[IR] adding linkage for consts

Cr0a3 · Sep 8, 2024 · d624c9e · d624c9e
1 parent ccc5384
commit d624c9e
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 7 deletions.
diff --git a/src/IR/parser/lexer.rs b/src/IR/parser/lexer.rs
@@ -4,7 +4,7 @@ use super::IrError;
 
 /// A location reference.
 /// Is recommended to be used for giving tokens locations
-#[derive(Debug, Clone, PartialEq, Eq)]
+#[derive(Debug, Clone, PartialEq, Eq, Default)]
 pub struct Loc {
     /// The line number
     pub line: u64,

diff --git a/src/IR/parser/parser.rs b/src/IR/parser/parser.rs
@@ -274,20 +274,47 @@ impl IrParser {
         PARSE NAME
         */
 
-        let name;
+        let mut scope = Linkage::External;
+        let mut parsed_scope = false;
+
+        let mut name = "unreachable".into();
 
-        let mut location;
+        let mut location = Loc::default();
 
         self.expect( TokenType::Ident(String::new()) )?;
 
         let tok = self.current_token()?;
         if let TokenType::Ident(ident) = &tok.typ {
-            name = ident.to_string();
-            location = tok.loc.clone();
+            match ident.as_str() {
+                "local" | "internal" | "private" => {
+                    parsed_scope = true;
+                    scope = Linkage::Internal
+                },
+                "public" | "external" => {
+                    parsed_scope = true;
+                    scope = Linkage::External
+                },
+                _ => {
+                    name = ident.to_string();
+                    location = tok.loc.clone();
+                }
+            }
         } else { unreachable!() }
 
         self.input.pop_front();
 
+        if parsed_scope {
+            self.expect(TokenType::Ident(String::new()))?;
+            let tok = self.current_token()?;
+
+            if let TokenType::Ident(ident) = &tok.typ {
+                name = ident.to_owned();
+                location = tok.loc.clone();
+            } else { unreachable!() }
+
+            self.input.pop_front();
+        }
+
         self.expect(TokenType::Equal)?;
         self.input.pop_front();
 
@@ -320,7 +347,7 @@ impl IrParser {
             name: name, 
             data: data,
             location: location,
-            scope: Linkage::Extern,
+            scope: scope,
         })
     }
 

diff --git a/tools/ytest/example.yl b/tools/ytest/example.yl
@@ -1,4 +1,4 @@
-# RUN: cargo run -p ylc -- -in=%s -o=out.o -asm-clr && gcc out.o -o test.exe && ./a.exe 
+# RUN: cargo run -p ylc -- -in=%s -o=out.o -asm-clr && gcc out.o -o a.exe && ./a.exe 
 
 const const0 = "Hello World!\n"