Add manual type guard for __GET_ITEMS__

es-meta · Sep 19, 2024 · 723064f · 723064f
1 parent 48c9c0f
commit 723064f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/scala/esmeta/analyzer/TypeAnalyzer.scala b/src/main/scala/esmeta/analyzer/TypeAnalyzer.scala
@@ -214,6 +214,12 @@ class TypeAnalyzer(
         "__FLAT_LIST__" -> { (xs, vs, retTy) =>
           AbsValue(vs(0).ty.list.elem, Map())
         },
+        "__GET_ITEMS__" -> { (xs, vs, retTy) =>
+          val ast = vs(1).ty.toValue.grammarSymbol match
+            case Fin(set) => AstT(set.map(_.name))
+            case Inf      => AstT
+          AbsValue(ListT(ast), Map())
+        },
         "Completion" -> { (xs, vs, retTy) =>
           AbsValue(vs(0).ty && CompT, Map())
         },