mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow contexted strings in replaceStrings

Browse source
This commit is contained in:
Nikolay Amiantov 2016-08-14 04:54:48 +03:00
parent eff80419c7
commit f1b8dd43be
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/libexpr/primops.cc
View file

@ -1620,13 +1620,18 @@ static void prim_replaceStrings(EvalState & state, const Pos & pos, Value * * ar
if (args[0]->listSize() != args[1]->listSize()) if (args[0]->listSize() != args[1]->listSize())
throw EvalError(format("‘from’ and ‘to’ arguments to ‘replaceStrings’ have different lengths, at %1%") % pos); throw EvalError(format("‘from’ and ‘to’ arguments to ‘replaceStrings’ have different lengths, at %1%") % pos);
Strings from; vector<string> from;
from.reserve(args[0]->listSize());
for (unsigned int n = 0; n < args[0]->listSize(); ++n) for (unsigned int n = 0; n < args[0]->listSize(); ++n)
from.push_back(state.forceStringNoCtx(*args[0]->listElems()[n], pos)); from.push_back(state.forceString(*args[0]->listElems()[n], pos));
Strings to; vector<std::pair<string, PathSet>> to;
for (unsigned int n = 0; n < args[1]->listSize(); ++n) to.reserve(args[1]->listSize());
to.push_back(state.forceStringNoCtx(*args[1]->listElems()[n], pos)); for (unsigned int n = 0; n < args[1]->listSize(); ++n) {
PathSet ctx;
auto s = state.forceString(*args[1]->listElems()[n], ctx, pos);
to.push_back(std::make_pair(std::move(s), std::move(ctx)));
}
PathSet context; PathSet context;
auto s = state.forceString(*args[2], context, pos); auto s = state.forceString(*args[2], context, pos);
@ -1634,11 +1639,16 @@ static void prim_replaceStrings(EvalState & state, const Pos & pos, Value * * ar
string res; string res;
for (size_t p = 0; p < s.size(); ) { for (size_t p = 0; p < s.size(); ) {
bool found = false; bool found = false;
for (auto i = from.begin(), j = to.begin(); i != from.end(); ++i, ++j) auto i = from.begin();
auto j = to.begin();
for (; i != from.end(); ++i, ++j)
if (s.compare(p, i->size(), *i) == 0) { if (s.compare(p, i->size(), *i) == 0) {
found = true; found = true;
p += i->size(); p += i->size();
res += *j; res += j->first;
for (auto& path : j->second)
context.insert(path);
j->second.clear();
break; break;
} }
if (!found) res += s[p++]; if (!found) res += s[p++];