IT WORKS HELL YEAH.

assembler/src/resolver.rs

@@ -1,9 +1,11 @@
 use std::{collections::HashMap, path::PathBuf};
 
+use common::prelude::Register;
+
 use crate::{
     AssembleError,
     model::{Module, Node, Opcode, Symbol, Token},
-    quick_hash,
+    node, quick_hash,
 };
 
 pub fn resolve_symbols(nodes: &mut Vec<Node>) -> Result<(), AssembleError> {
@@ -119,3 +121,36 @@ pub fn resolve_dependencies(mut nodes: Vec<Node>) -> Result<Vec<Node>, AssembleE
 
     Ok(nodes)
 }
+
+pub fn create_sections(nodes: &mut Vec<Node>) -> Result<(), AssembleError> {
+    let mut res = Vec::<Node>::with_capacity(nodes.len());
+
+    res.push(node!(None, Opcode::Segment, Token::Immediate(0)));
+
+    for n in nodes.iter() {
+        if n.opcode() == Opcode::Data {
+            res.push(n.clone());
+        }
+    }
+
+    res.push(node!(None, Opcode::Segment, Token::Immediate(1)));
+    let start = res.len() + 1;
+    res.insert(
+        0,
+        node!(
+            None,
+            Opcode::Jmp,
+            Token::Immediate(start as u32 * 4),
+            Token::Register(Register::Zero)
+        ),
+    );
+    for n in nodes.iter() {
+        if !matches!(n.opcode(), Opcode::Data | Opcode::Include) {
+            res.push(n.clone());
+        }
+    }
+
+    *nodes = res;
+
+    Ok(())
+}