Parse native function declarations

7 files changed, 152 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNamespaceStatementNode.java b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNamespaceStatementNode.java
index 8d94d5b8..c4981b68 100644
--- a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNamespaceStatementNode.java
+++ b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNamespaceStatementNode.java
@@ -1,4 +1,4 @@
 package org.shedlang.compiler.ast.untyped;
 
-public sealed interface UntypedNamespaceStatementNode extends UntypedNode permits UntypedAssignNode, UntypedEffectDefinitionNode, UntypedFunctionDefinitionNode, UntypedStructDefinitionNode, UntypedSumTypeDefinitionNode {
+public sealed interface UntypedNamespaceStatementNode extends UntypedNode permits UntypedAssignNode, UntypedEffectDefinitionNode, UntypedFunctionDefinitionNode, UntypedNativeFunctionDeclarationNode, UntypedStructDefinitionNode, UntypedSumTypeDefinitionNode {
 }
diff --git a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNativeFunctionDeclarationNode.java b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNativeFunctionDeclarationNode.java
new file mode 100644
index 00000000..989d6044
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedNativeFunctionDeclarationNode.java
@@ -0,0 +1,104 @@
+package org.shedlang.compiler.ast.untyped;
+
+import org.shedlang.compiler.sources.Source;
+import org.shedlang.compiler.sources.Sources;
+import org.shedlang.compiler.types.Visibility;
+
+import java.util.Optional;
+
+public record UntypedNativeFunctionDeclarationNode(
+    String name,
+    Visibility visibility,
+    UntypedFunctionSignatureNode signature,
+    Source source
+) implements UntypedNamespaceStatementNode, UntypedVariableDeclarationNode {
+    public static Builder builder() {
+        return new Builder(
+            "f",
+            Visibility.PRIVATE,
+            UntypedFunctionSignatureNode.builder()
+        );
+    }
+
+    public record Builder(
+        String name,
+        Visibility visibility,
+        UntypedFunctionSignatureNode.Builder signature
+    ) {
+        public UntypedNativeFunctionDeclarationNode build() {
+            return new UntypedNativeFunctionDeclarationNode(
+                name,
+                visibility,
+                signature.build(),
+                Sources.NULL
+            );
+        }
+
+        public Builder withName(String name) {
+            return new Builder(
+                name,
+                this.visibility,
+                this.signature
+            );
+        }
+
+        public Builder addPositionalParam(String name, UntypedTypeLevelExpressionNode type) {
+            return addPositionalParam(name, Optional.of(type));
+        }
+
+        public Builder addPositionalParam(
+            String name,
+            Optional<UntypedTypeLevelExpressionNode> type
+        ) {
+            return new Builder(
+                this.name,
+                this.visibility,
+                this.signature.addPositionalParam(name, type)
+            );
+        }
+
+        public Builder addNamedParam(String name, UntypedTypeLevelExpressionNode type) {
+            return new Builder(
+                this.name,
+                this.visibility,
+                this.signature.addNamedParam(name, type)
+            );
+        }
+
+        public Builder withEffect(UntypedTypeLevelExpressionNode effect) {
+            return new Builder(
+                this.name,
+                this.visibility,
+                this.signature.withEffect(effect)
+            );
+        }
+
+        public Builder withReturnTypeUnit() {
+            return withReturnType(new UntypedTypeLevelReferenceNode("Unit", Sources.NULL));
+        }
+
+        public Builder withReturnType(UntypedTypeLevelExpressionNode returnType) {
+            return new Builder(
+                this.name,
+                this.visibility,
+                this.signature.withReturnType(returnType)
+            );
+        }
+
+        public Builder withoutReturnType() {
+            return new Builder(
+                this.name,
+                this.visibility,
+                this.signature.withoutReturnType()
+            );
+        }
+
+        public Builder withVisibility(Visibility visibility) {
+            return new Builder(
+                this.name,
+                visibility,
+                this.signature
+            );
+        }
+    }
+}
diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index cda360e3..45533683 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -776,6 +776,9 @@ public class Parser {
             case KEYWORD_FUN ->
                 parseFunctionDefinition(tokens);
 
+            case KEYWORD_NATIVE ->
+                parseNativeFunctionDeclaration(tokens);
+
             case KEYWORD_PUB ->
                 switch (tokens.peek(1).tokenType()) {
                     case KEYWORD_EFFECT ->
@@ -784,6 +787,9 @@ public class Parser {
                     case KEYWORD_FUN ->
                         parseFunctionDefinition(tokens);
 
+                    case KEYWORD_NATIVE ->
+                        parseNativeFunctionDeclaration(tokens);
+
                     case KEYWORD_STRUCT ->
                         parseStructDefinition(tokens);
 
@@ -811,6 +817,32 @@ public class Parser {
         };
     }
 
+    private static UntypedNativeFunctionDeclarationNode parseNativeFunctionDeclaration(
+        TokenIterator<TokenType> tokens
+    ) {
+        var start = tokens.startPosition();
+
+        var visibility = parseVisibility(tokens);
+
+        tokens.skip(TokenType.KEYWORD_NATIVE);
+        tokens.skip(TokenType.KEYWORD_FUN);
+        var name = tokens.nextValue(TokenType.IDENTIFIER);
+
+        var signature = parseFunctionSignature(tokens);
+
+        tokens.skip(TokenType.SYMBOL_SEMICOLON);
+
+        var end = tokens.endPosition();
+        var source = new FileFragmentSource(start.to(end));
+
+        return new UntypedNativeFunctionDeclarationNode(
+            name,
+            visibility,
+            signature,
+            source
+        );
+    }
+
     private static UntypedParamsNode parseParams(TokenIterator<TokenType> tokens) {
         var paramsStart = tokens.startPosition();
 
diff --git a/src/main/java/org/shedlang/compiler/parser/TokenType.java b/src/main/java/org/shedlang/compiler/parser/TokenType.java
index dbe32b0d..5eee208c 100644
--- a/src/main/java/org/shedlang/compiler/parser/TokenType.java
+++ b/src/main/java/org/shedlang/compiler/parser/TokenType.java
@@ -21,6 +21,7 @@ public enum TokenType {
     KEYWORD_IF,
     KEYWORD_IMPORT,
     KEYWORD_IS,
+    KEYWORD_NATIVE,
     KEYWORD_PUB,
     KEYWORD_RESUME,
     KEYWORD_RETURN,
diff --git a/src/main/java/org/shedlang/compiler/parser/Tokeniser.java b/src/main/java/org/shedlang/compiler/parser/Tokeniser.java
index dfac78a0..6e5bed3f 100644
--- a/src/main/java/org/shedlang/compiler/parser/Tokeniser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Tokeniser.java
@@ -25,6 +25,7 @@ public class Tokeniser {
         Map.entry("if", TokenType.KEYWORD_IF),
         Map.entry("import", TokenType.KEYWORD_IMPORT),
         Map.entry("is", TokenType.KEYWORD_IS),
+        Map.entry("native", TokenType.KEYWORD_NATIVE),
         Map.entry("pub", TokenType.KEYWORD_PUB),
         Map.entry("resume", TokenType.KEYWORD_RESUME),
         Map.entry("return", TokenType.KEYWORD_RETURN),
diff --git a/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java b/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
index 2a842d93..a2a8b253 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/DeclaredVariables.java
@@ -104,6 +104,12 @@ class DeclaredVariables {
                     functionDefinition
                 ));
 
+            case UntypedNativeFunctionDeclarationNode nativeFunctionDeclaration ->
+                variables.plus(DeclaredVariable.fromNode(
+                    VariableType.CONSTANT,
+                    nativeFunctionDeclaration
+                ));
+
             case UntypedStructDefinitionNode struct ->
                 variables.plus(DeclaredVariable.fromNode(
                     VariableType.CONSTANT,
diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index daee11c4..4486e457 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -1484,6 +1484,10 @@ public class TypeChecker {
                     );
                 }
 
+                case UntypedNativeFunctionDeclarationNode untypedNativeFunctionDeclaration -> {
+                    throw new UnsupportedOperationException("TODO");
+                }
+
                 case UntypedStructDefinitionNode untypedStruct -> {
                     dependencyChecker = dependencyChecker.addDeferredDependencies(
                         untypedStruct.name(),
@@ -1556,6 +1560,9 @@ public class TypeChecker {
             case UntypedFunctionDefinitionNode functionDefinition ->
                 typeCheckStepsFunctionDefinition(functionDefinition);
 
+            case UntypedNativeFunctionDeclarationNode nativeFunctionDeclaration ->
+                throw new UnsupportedOperationException("TODO");
+
             case UntypedStructDefinitionNode structDefinition ->
                 typeCheckStepsStructDefinition(structDefinition);