From 927c018f12e73c254cec32c0f60e1c01b9fc0319 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Wed, 21 Feb 2024 12:07:07 +0200
Subject: Refactor names

---
 atk16_asm/tokenizer.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 atk16_asm/tokenizer.py

(limited to 'atk16_asm/tokenizer.py')

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
-- 
cgit v1.3