From bc1c77d8a89b36d7b1c2ec4240655fdcc0daae69 Mon Sep 17 00:00:00 2001
From: jesopo <github@lolnerd.net>
Date: Sat, 21 Dec 2019 21:16:50 +0000
Subject: [PATCH] implement standardised $token replacement in utils.parse

---
 src/utils/parse.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/src/utils/parse.py b/src/utils/parse.py
index ce2ee793..5beb4637 100644
--- a/src/utils/parse.py
+++ b/src/utils/parse.py
@@ -120,3 +120,34 @@ def timed_args(args, min_args):
         return time, args[1:]
     return None, args
 
+def format_tokens(s: str, names: typing.List[str], sigil: str="$"
+        ) -> typing.List[typing.Tuple[int, str]]:
+    i = 0
+    max = len(s)-1
+    sigil_found = False
+    tokens: typing.List[typing.Tuple[int, str]] = []
+
+    while i < max:
+        if s[i] == sigil:
+            if s[i+1] == sigil:
+                i += 1
+            else:
+                sigil_found = True
+        elif sigil_found:
+            sigil_found = False
+            for name in names:
+                if len(name) <= (len(s)-i) and s[i:i+len(name)] == name:
+                    tokens.append((i-1, "%s%s" % (sigil, name)))
+                    i += len(name)
+                    break
+        i += 1
+    return tokens
+
+def format_token_replace(s: str, vars: typing.Dict[str, str],
+        sigil: str="$") -> str:
+    tokens = format_tokens(s, list(vars.keys()), sigil)
+    tokens.sort(key=lambda x: x[0])
+    tokens.reverse()
+    for i, token in tokens:
+        s = s[:i] + vars[token.replace(sigil, "", 1)] + s[i+len(token):]
+    return s