simplify code

This commit is contained in:
Daniel Sipka 2015-04-27 14:14:21 +02:00
parent 16783f76e3
commit 4d6744dc63
8 changed files with 48 additions and 27 deletions

View File

@ -13,5 +13,6 @@ std::string mstch::render(
std::map<std::string, template_type> partial_templates; std::map<std::string, template_type> partial_templates;
for (auto& partial: partials) for (auto& partial: partials)
partial_templates.insert({partial.first, {partial.second}}); partial_templates.insert({partial.first, {partial.second}});
return render_context(root, partial_templates).render(tmplt); return render_context(root, partial_templates).render(tmplt);
} }

View File

@ -15,18 +15,20 @@ std::string in_section::render(render_context& ctx, const token& token) {
if (token.name() == section_name && skipped_openings == 0) { if (token.name() == section_name && skipped_openings == 0) {
auto& node = ctx.get_node(section_name); auto& node = ctx.get_node(section_name);
std::string out; std::string out;
if (m_type == type::normal && !visit(is_node_empty(), node)) if (m_type == type::normal && !visit(is_node_empty(), node))
out = visit(render_section(ctx, section), node); out = visit(render_section(ctx, section), node);
else if (m_type == type::inverted && visit(is_node_empty(), node)) else if (m_type == type::inverted && visit(is_node_empty(), node))
out = render_context::push(ctx).render(section); out = render_context::push(ctx).render(section);
ctx.set_state<outside_section>(); ctx.set_state<outside_section>();
return out; return out;
} else { } else
skipped_openings--; skipped_openings--;
}
else if (token.token_type() == token::type::inverted_section_open || else if (token.token_type() == token::type::inverted_section_open ||
token.token_type() == token::type::section_open) token.token_type() == token::type::section_open)
skipped_openings++; skipped_openings++;
section << token; section << token;
return ""; return "";
} }

View File

@ -18,32 +18,43 @@ void template_type::process_text(citer begin, citer end) {
} }
} }
void template_type::tokenize(const std::string& tmplt) { void template_type::tokenize(const std::string& tmp) {
std::string o{"{{"}, c{"}}"}; std::string open{"{{"}, close{"}}"};
citer beg = tmplt.begin(); citer beg = tmp.begin();
for (unsigned long pos = 0; pos < tmplt.size();) { auto npos = std::string::npos;
auto to = tmplt.find(o, pos);
auto tc = tmplt.find(c, (to == std::string::npos)?to:(to + 1)); for (unsigned long cur_pos = 0; cur_pos < tmp.size();) {
if (tc != std::string::npos && to != std::string::npos) { auto open_pos = tmp.find(open, cur_pos);
if (*(beg + to + o.size()) == '{' && *(beg + tc + c.size()) == '}') auto close_pos = tmp.find(
++tc; close, (open_pos == npos) ? open_pos : (open_pos + 1));
process_text(beg + pos, beg + to);
pos = tc + c.size(); if (close_pos != npos && open_pos != npos) {
tokens.push_back({{beg + to, beg + tc + c.size()}, o.size(), c.size()}); if (*(beg + open_pos + open.size()) == '{' &&
if (*(beg + to + o.size()) == '=' && *(beg + tc - 1) == '=') { *(beg + close_pos + close.size()) == '}')
o = {beg + to + o.size() + 1, beg + tmplt.find(' ', to)}; ++close_pos;
c = {beg + tmplt.find(' ', to) + 1, beg + tc - 1};
process_text(beg + cur_pos, beg + open_pos);
cur_pos = close_pos + close.size();
tokens.push_back({{beg + open_pos, beg + close_pos + close.size()},
open.size(), close.size()});
if (*(beg + open_pos + open.size()) == '=' &&
*(beg + close_pos - 1) == '=')
{
open = {beg + open_pos + open.size() + 1, beg + tmp.find(' ',open_pos)};
close = {beg + tmp.find(' ', open_pos) + 1, beg + close_pos - 1};
} }
} else { } else {
process_text(beg + pos, tmplt.end()); process_text(beg + cur_pos, tmp.end());
pos = tc; cur_pos = close_pos;
} }
} }
} }
void template_type::strip_whitespace() { void template_type::strip_whitespace() {
auto lbegin = tokens.begin(); auto line_begin = tokens.begin();
bool has_tag = false, non_space = false; bool has_tag = false, non_space = false;
for (auto it = tokens.begin(); it != tokens.end(); ++it) { for (auto it = tokens.begin(); it != tokens.end(); ++it) {
auto type = (*it).token_type(); auto type = (*it).token_type();
if (type != token::type::text && type != token::type::variable && if (type != token::type::text && type != token::type::variable &&
@ -51,12 +62,15 @@ void template_type::strip_whitespace() {
has_tag = true; has_tag = true;
else if (!(*it).ws_only()) else if (!(*it).ws_only())
non_space = true; non_space = true;
if ((*it).eol()) { if ((*it).eol()) {
if (has_tag && !non_space) if (has_tag && !non_space)
for (auto c = lbegin; it != c-1; c = (*c).ws_only()?tokens.erase(c):++c) for (auto cur = line_begin; it != cur - 1;
it = (*c).eol()?c-1:it; cur = (*cur).ws_only() ? tokens.erase(cur) : ++cur)
it = (*cur).eol() ? cur - 1 : it;
non_space = has_tag = false; non_space = has_tag = false;
lbegin = it + 1; line_begin = it + 1;
} }
} }
} }

View File

@ -20,7 +20,7 @@ class template_type {
std::vector<token> tokens; std::vector<token> tokens;
void strip_whitespace(); void strip_whitespace();
void process_text(citer beg, citer end); void process_text(citer beg, citer end);
void tokenize(const std::string& tmplt); void tokenize(const std::string& tmp);
}; };
} }

View File

@ -15,10 +15,12 @@ mstch::citer mstch::first_not_ws(mstch::criter begin, mstch::criter end) {
std::string mstch::html_escape(const std::string& str) { std::string mstch::html_escape(const std::string& str) {
std::string out; std::string out;
citer start = str.begin(); citer start = str.begin();
auto add_escape = [&out, &start](const std::string& escaped, citer& it) { auto add_escape = [&out, &start](const std::string& escaped, citer& it) {
out += std::string{start, it} + escaped; out += std::string{start, it} + escaped;
start = it + 1; start = it + 1;
}; };
for (auto it = str.begin(); it != str.end(); ++it) for (auto it = str.begin(); it != str.end(); ++it)
switch (*it) { switch (*it) {
case '&': add_escape("&amp;", it); break; case '&': add_escape("&amp;", it); break;
@ -29,5 +31,6 @@ std::string mstch::html_escape(const std::string& str) {
case '/': add_escape("&#x2F;", it); break; case '/': add_escape("&#x2F;", it); break;
default: break; default: break;
} }
return out + std::string{start, str.end()}; return out + std::string{start, str.end()};
} }

View File

@ -35,6 +35,7 @@ inline std::string render_section::operator()(const lambda& fun) const {
std::string section_str; std::string section_str;
for(auto& token: section) for(auto& token: section)
section_str += token.raw(); section_str += token.raw();
return fun(section_str, [this](const std::string& str) { return fun(section_str, [this](const std::string& str) {
return render_context::push(ctx).render(template_type{str}); return render_context::push(ctx).render(template_type{str});
}); });