Replacing newlines

examples/basic.🍺

@@ -3,3 +3,9 @@ y = 5
 // This will generate a single - because it's faster
 z = 255
 array = [1 2 3 4 5]
+c = 'c'
+basm(".")
+newline = '\n'
+basm(".")
+x = "\nHello, World!"
+basm("<<<<<<<<<<<<.>.>.>.>.>.>.>.>.>.>.>.>.")

src/ast.rs

@@ -103,3 +103,12 @@ pub enum Iterator {
     },
     Path(Box<Expression>),
 }
+
+impl Expression {
+    pub fn as_number(&self) -> Option<u8> {
+        match self {
+            Expression::Number(n) => Some(*n),
+            _ => None,
+        }
+    }
+}

src/codegen.rs

@@ -1,4 +1,5 @@
 use std::collections::HashMap;
+use std::str::from_utf8;
 use crate::ast::Expression;
 
 pub struct Codegen {
@@ -149,6 +150,25 @@ impl Codegen {
             Expression::Number(x) => {
                 self.set_new_value(*x);
             }
+            Expression::Path(x) => {
+                self.move_to(*self.variables.get(x).unwrap());
+            }
+            Expression::Call { name, args } => {
+                match name.as_str() {
+                    "basm" => {
+                        assert_eq!(args.len(), 1);
+                        match args.get(0).unwrap() {
+                            Expression::Array(x) => {
+                                for value in x.iter() {
+                                    self.code += from_utf8(&[value.as_number().unwrap()]).unwrap()
+                                }
+                            }
+                            _ => { todo!() }
+                        }
+                    }
+                    _ => { todo!("Implement other functions"); }
+                }
+            }
             Expression::Assignment { name, value } => {
                 if self.variables.get(name).is_none() {
                     match *value.clone() {
@@ -176,7 +196,9 @@ impl Codegen {
                     self.add_value(3);
                 }
             }
-            _ => {}
+            _ => {
+                todo!("Implement other expressions");
+            }
         }
     }
 }

src/lexer.rs

@@ -32,7 +32,7 @@ pub fn tokenize_indexed(input: &str) -> Vec<IndexedToken> {
             }
         })
         .collect();
-    output
+    postprocess(output)
 }
 
 #[allow(dead_code)]
@@ -42,3 +42,16 @@ pub fn tokenize(input: &str) -> Vec<Token> {
         .map(|x| x.token)
         .collect()
 }
+
+fn postprocess(mut input: Vec<IndexedToken>) -> Vec<IndexedToken> {
+    for token in input.iter_mut() {
+        match token.token {
+            Token::String(ref mut x) => {
+                *x = x.replace("\\n", "\n");
+            }
+            _ => {}
+        }
+    }
+
+    input
+}

src/tokens.rs

@@ -58,7 +58,8 @@ pub enum Token {
     #[regex(r"[0-9]+", |lex| lex.slice().parse().ok())]
     Integer(u8),
 
-    #[regex(r"'(.|\\n|\\d)'", |lex| lex.slice().chars().nth(1).map(|c| c as u8))]
+    #[regex(r"'(.)'", |lex| lex.slice().chars().nth(1).map(|c| c as u8))]
+    #[regex(r"'\\n'", |_| Some(10))]
     Char(u8),
 
     #[regex(r#""[^"]*""#, |lex| {