feat(assembler): Implement beq and mov instructions

1 files changed, 38 insertions, 0 deletions

diff --git a/assembler/src/parser/instruction.rs b/assembler/src/parser/instruction.rs
index 18d67b7..cc0523f 100644
--- a/assembler/src/parser/instruction.rs
+++ b/assembler/src/parser/instruction.rs
@@ -52,6 +52,20 @@ impl Instruction {
             "add" => {
                 self.check_args_len(2)?;
             },
+            "mov" => {
+                self.check_args_len(2)?;
+
+                if ! self.args[1].is_location() {
+                    return Err(ArgumentError::NotLocation);
+                }
+            },
+            "beq" => {
+                self.check_args_len(2)?;
+
+                if ! self.args[1].is_location() {
+                    return Err(ArgumentError::NotLocation);
+                }
+            },
             _ => return Err(ArgumentError::UnknownInstruction),
         }
         Ok(())
@@ -71,6 +85,30 @@ impl Instruction {
                 Instruction::new_subleq(&Z,            &self.args[1], &Address::Instruction(next + 2)),
                 Instruction::new_subleq(&Z,            &Z,            &Address::Instruction(next + 3)),
             ],
+            "mov" => vec![
+                Instruction::new_subleq(&self.args[1], &self.args[1], &Address::Instruction(next + 1)),
+                Instruction::new_subleq(&self.args[0], &Z,            &Address::Instruction(next + 2)),
+                Instruction::new_subleq(&Z,            &self.args[1], &Address::Instruction(next + 3)),
+                Instruction::new_subleq(&Z,            &Z,            &Address::Instruction(next + 4)),
+            ],
+            "beq" => vec![
+                Instruction::new_subleq(&self.args[0], &Z,            &Address::Instruction(next + 2)),
+                Instruction::new_subleq(&Z,            &Z,            &Address::Instruction(next + 4)),
+                Instruction::new_subleq(&Z,            &Z,            &Address::Instruction(next + 3)),
+                Instruction::new_subleq(&Z,            &self.args[0], &self.args[1]                  ),
+            ],
+            _ => panic!("Unknown instruction {}!", self.name),
+        }
+    }
+
+    pub(crate) fn collapse_len(&self) -> usize {
+        match &*self.name {
+            "subleq" | "jmp"
+                => 1,
+            "add"
+                => 3,
+            "mov" | "beq"
+                => 4,
             _ => panic!("Unknown instruction {}!", self.name),
         }
     }