Implement semantics analyzer

4 files changed, 75 insertions, 3 deletions

diff --git a/src/main/java/no/eliashaugsbakk/kompilator/Main.java b/src/main/java/no/eliashaugsbakk/kompilator/Main.java
index fa7f79c..b1ad70a 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/Main.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/Main.java
@@ -9,6 +9,8 @@ import no.eliashaugsbakk.kompilator.assembleAndLink.AssemblerAndLinker;
 import no.eliashaugsbakk.kompilator.parsing.AST;
 import no.eliashaugsbakk.kompilator.parsing.Parser;
 import no.eliashaugsbakk.kompilator.parsing.ParserException;
+import no.eliashaugsbakk.kompilator.semanticAnalysis.Analyzer;
+import no.eliashaugsbakk.kompilator.semanticAnalysis.SemanticException;
 import no.eliashaugsbakk.kompilator.tokenization.Lexer;
 import no.eliashaugsbakk.kompilator.tokenization.Token;
 
@@ -44,10 +46,17 @@ public class Main {
       ast = new Parser(tokens).parse();
     } catch (ParserException e) {
       IO.println("Error while parsing: " + e.getMessage());
+      System.exit(1);
+    }
+
+    try {
+      new Analyzer(ast).analyze();
+    } catch (SemanticException e) {
+      IO.println("Semantic error: " + e.getMessage());
+      System.exit(1);
     }
 
     // TODO:
-    // new Analyzer(ast).analyze();
     // List<String> IR = new IRGenerator(ast).generate();
 
     List<String> IR = List.of();
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/parsing/AST.java b/src/main/java/no/eliashaugsbakk/kompilator/parsing/AST.java
index 95ec77c..ad8a3a2 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/parsing/AST.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/parsing/AST.java
@@ -3,9 +3,13 @@ package no.eliashaugsbakk.kompilator.parsing;
 import no.eliashaugsbakk.kompilator.parsing.node.ASTNode;
 
 public class AST {
-  ASTNode root;
+  private final ASTNode root;
 
-  AST(ASTNode root) {
+  public AST(ASTNode root) {
     this.root = root;
   }
+
+  public ASTNode getRoot() {
+    return root;
+  }
 }
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/Analyzer.java b/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/Analyzer.java
index 891018c..0b01b14 100644
--- a/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/Analyzer.java
+++ b/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/Analyzer.java
@@ -1,4 +1,56 @@
 package no.eliashaugsbakk.kompilator.semanticAnalysis;
 
+import no.eliashaugsbakk.kompilator.parsing.AST;
+import no.eliashaugsbakk.kompilator.parsing.node.Program;
+import no.eliashaugsbakk.kompilator.parsing.node.expression.Expression;
+import no.eliashaugsbakk.kompilator.parsing.node.expression.FunctionCall;
+import no.eliashaugsbakk.kompilator.parsing.node.expression.StringLiteral;
+import no.eliashaugsbakk.kompilator.parsing.node.statement.ExpressionStatement;
+import no.eliashaugsbakk.kompilator.parsing.node.statement.Statement;
+
 public class Analyzer {
+  private final AST ast;
+
+  public Analyzer(AST ast) {
+    this.ast = ast;
+  }
+
+  public void analyze() throws SemanticException {
+    Program program = (Program) ast.getRoot();
+
+    for (Statement stmt : program.statements) {
+      analyzeStatement(stmt);
+    }
+  }
+
+  private void analyzeStatement(Statement stmt) throws SemanticException {
+    if (stmt instanceof ExpressionStatement exprStmt) {
+      analyzeExpression(exprStmt.expression);
+    } else {
+      // Analyze the statement
+      // NO other statements implemented
+    }
+  }
+
+  private void analyzeExpression(Expression expr) throws SemanticException {
+    if (expr instanceof FunctionCall call) {
+      checkFunctionCall(call);
+    } else {
+      // analyze the expression
+    }
+  }
+
+  private void checkFunctionCall(FunctionCall call) throws SemanticException {
+    if (!call.functionName.equals("print")) {
+      throw new SemanticException("unknown function: " + call.functionName);
+    }
+
+    if (call.arguments.size() != 1) {
+      throw new SemanticException("print expects 1 argument, got " + call.arguments.size());
+    }
+
+    if (!(call.arguments.getFirst() instanceof StringLiteral)) {
+      throw new SemanticException("print expects String argument only");
+    }
+  }
 }
diff --git a/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/SemanticException.java b/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/SemanticException.java
new file mode 100644
index 0000000..402cfe7
--- /dev/null
+++ b/src/main/java/no/eliashaugsbakk/kompilator/semanticAnalysis/SemanticException.java
@@ -0,0 +1,7 @@
+package no.eliashaugsbakk.kompilator.semanticAnalysis;
+
+public class SemanticException extends Exception {
+  public SemanticException(String message) {
+    super(message);
+  }
+}