From c759fe0cbe348d2d6e69df6e8995ae6bb1a9a5a5 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 8 Oct 2023 11:07:38 +0100
Subject: Support iteration through struct types

---
 src/main/java/org/shedlang/compiler/typechecker/TypeInfos.java | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/main/java')

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,
-- 
cgit v1.2.3