Support access methods on structs

2 files changed, 17 insertions, 3 deletions

diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index ed522427..16728f8c 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -1358,8 +1358,9 @@ public class TypeChecker {
             }
 
             case SimpleStructType receiverTypeStruct -> {
+                // TODO: handle ambiguity
                 var structTypeInfo = context.structTypeInfo(receiverTypeStruct, source);
-                return structTypeInfo.fieldType(memberName);
+                return structTypeInfo.memberType(memberName);
             }
 
             case NamespaceType receiverTypeNamespace -> {
diff --git a/src/main/java/org/shedlang/compiler/types/StructTypeInfo.java b/src/main/java/org/shedlang/compiler/types/StructTypeInfo.java
index b8a7df38..d90a5f6b 100644
--- a/src/main/java/org/shedlang/compiler/types/StructTypeInfo.java
+++ b/src/main/java/org/shedlang/compiler/types/StructTypeInfo.java
@@ -15,8 +15,13 @@ public record StructTypeInfo(
         return fields().fieldCount();
     }
 
-    public Optional<Type> fieldType(String fieldName) {
-        return fields.fieldType(fieldName);
+    public Optional<Type> memberType(String memberName) {
+        return fields.fieldType(memberName).or(() ->
+            methods.stream()
+                .filter(method -> method.name().equals(memberName))
+                .findFirst()
+                .map(method -> method.type())
+        );
     }
 
     public int fieldIndex(String fieldName) {
@@ -51,5 +56,13 @@ public record StructTypeInfo(
                 this.methods
             );
         }
+
+        public Builder addMethod(String name, FunctionType type) {
+            return new Builder(
+                this.fields,
+                fieldsArePositional,
+                this.methods.plus(new MethodInfo(name, type))
+            );
+        }
     }
 }