Support iteration through struct types

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeInfos.java b/src/main/java/org/shedlang/compiler/typechecker/TypeInfos.java
index 5dc01ae7..f18b8836 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeInfos.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeInfos.java
@@ -7,6 +7,7 @@ import org.shedlang.compiler.types.*;
 import org.shedlang.compiler.util.P;
 
 import java.util.Optional;
+import java.util.Set;
 
 public class TypeInfos {
     public static final TypeInfos EMPTY = new TypeInfos(Empty.map(), P.orderedPMap(), Empty.map());
@@ -55,6 +56,10 @@ public class TypeInfos {
         return Optional.ofNullable(structTypeInfos.get(type));
     }
 
+    public Set<StructType> structTypes() {
+        return structTypeInfos.keySet();
+    }
+
     public TypeInfos addSumTypeInfo(SumType sumType, SumTypeInfo sumTypeInfo) {
         return new TypeInfos(
             this.namespaceTypeInfos,