1 files changed, 39 insertions, 0 deletions
diff --git a/src/asm_pass2.py b/src/asm_pass2.py
new file mode 100644
index 0000000..77b8fa6
--- /dev/null
+++ b/src/asm_pass2.py
@@ -0,0 +1,39 @@
+from dataclasses import dataclass
+from assembler_ops import *
+from assembler_eval import *
+from asm_pass1 import *
+
+@dataclass
+class Result2Line:
+  line_num: int
+  parts: list[str]
+
+@dataclass
+class Result2:
+  lines: list[Result2Line]
+  options: Options
+  operations: OpExpansionDict
+
+def pass_2(result1: Result1) -> Result2:
+  options = Options()
+  result_lines: list[Result2Line] = []
+
+  for line in result1.lines:
+    keyword, *args = line.parts
+    if not keyword.startswith("@"):
+      fn = result1.operations[keyword]
+      output: list[list[str]] = fn(*args)
+    else:
+      output = [line.parts]
+
+    for parts in output:
+      result_lines.append(Result2Line(
+        line_num=line.line_num,
+        parts=parts
+      ))
+
+  return Result2(
+    operations=result1.operations,
+    options=options,
+    lines=result_lines
+  )