aboutsummaryrefslogtreecommitdiffstats
path: root/libdevcore/Whiskers.cpp
blob: 4bad8476c4a8a8f1f789964eb364ab914e8a13bd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
    This file is part of solidity.

    solidity is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    solidity is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with solidity.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file Whiskers.cpp
 * @author Chris <chis@ethereum.org>
 * @date 2017
 *
 * Moustache-like templates.
 */

#include <libdevcore/Whiskers.h>

#include <libdevcore/Assertions.h>

#include <boost/regex.hpp>

using namespace std;
using namespace dev;

Whiskers::Whiskers(string const& _template):
m_template(_template)
{
}

Whiskers& Whiskers::operator ()(string const& _parameter, string const& _value)
{
    assertThrow(
        m_parameters.count(_parameter) == 0,
        WhiskersError,
        _parameter + " already set."
    );
    assertThrow(
        m_listParameters.count(_parameter) == 0,
        WhiskersError,
        _parameter + " already set as list parameter."
    );
    m_parameters[_parameter] = _value;

    return *this;
}

Whiskers& Whiskers::operator ()(
    string const& _listParameter,
    vector<map<string, string>> const& _values
)
{
    assertThrow(
        m_listParameters.count(_listParameter) == 0,
        WhiskersError,
        _listParameter + " already set."
    );
    assertThrow(
        m_parameters.count(_listParameter) == 0,
        WhiskersError,
        _listParameter + " already set as value parameter."
    );
    m_listParameters[_listParameter] = _values;

    return *this;
}

string Whiskers::render() const
{
    return replace(m_template, m_parameters, m_listParameters);
}

string Whiskers::replace(
    string const& _template,
    StringMap const& _parameters,
    map<string, vector<StringMap>> const& _listParameters
)
{
    using namespace boost;
    static regex listOrTag("<([^#/>]+)>|<#([^>]+)>(.*?)</\\2>");
    return regex_replace(_template, listOrTag, [&](match_results<string::const_iterator> _match) -> string
    {
        string tagName(_match[1]);
        if (!tagName.empty())
        {
            assertThrow(_parameters.count(tagName), WhiskersError, "Tag " + tagName + " not found.");
            return _parameters.at(tagName);
        }
        else
        {
            string listName(_match[2]);
            string templ(_match[3]);
            assertThrow(!listName.empty(), WhiskersError, "");
            assertThrow(
                _listParameters.count(listName),
                WhiskersError, "List parameter " + listName + " not set."
            );
            string replacement;
            for (auto const& parameters: _listParameters.at(listName))
                replacement += replace(templ, joinMaps(_parameters, parameters));
            return replacement;
        }
    });
}

Whiskers::StringMap Whiskers::joinMaps(
    Whiskers::StringMap const& _a,
    Whiskers::StringMap const& _b
)
{
    Whiskers::StringMap ret = _a;
    for (auto const& x: _b)
        assertThrow(
            ret.insert(x).second,
            WhiskersError,
            "Parameter collision"
        );
    return ret;
}