Refactor dir structure

1 files changed, 47 insertions, 0 deletions

diff --git a/atk16-asm/tokenizer.py b/atk16-asm/tokenizer.py
new file mode 100644
index 0000000..cf31b56
--- /dev/null
+++ b/atk16-asm/tokenizer.py
@@ -0,0 +1,47 @@
+def tokenize(line: str, retain_curlies = False) -> list[str]:
+  cur: str = ""
+  result: list[str] = []
+  is_py_expr = False
+  is_string = False
+  idx = 0
+  while idx < len(line):
+    c = line[idx]
+    if not is_py_expr and c == "$":
+      is_py_expr = True
+      if retain_curlies:
+        cur += "${"
+      idx += 1
+    elif is_py_expr and c == "$":
+      raise Exception("Unexpected start of python expr while already parsing a python expression:\n" + line)
+    elif not is_py_expr and c == "}":
+      raise Exception("Unexpected end of python expr while not parsing a python expression:\n" + line)
+    elif is_py_expr and c == "}":
+      is_py_expr = False
+      if retain_curlies:
+        cur += c
+      result.append(cur)
+      cur = ""
+    elif is_py_expr:
+      cur += c
+    elif not is_string and c == "\"":
+      cur += "\""
+      is_string = True
+    elif is_string and c == "\"":
+      cur += "\""
+      is_string = False
+      result.append(cur)
+      cur = ""
+    elif is_string:
+      cur += c
+    elif c == " " or c == "\t":
+      result.append(cur)
+      cur = ""
+    else:
+      cur += c
+
+    idx += 1
+
+  if len(cur) > 0:
+    result.append(cur)
+
+  return [r for r in result if r != ""]
\ No newline at end of file