can now apply '#' on enum values

teal-language · Jul 26, 2024 · cd378f9 · cd378f9
1 parent f97625d
commit cd378f9
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/spec/operator/len_spec.lua b/spec/operator/len_spec.lua
@@ -5,6 +5,20 @@ describe("#", function()
       local x: integer = #({1, 2, 3})
    ]]))
 
+   it("can run on a string", util.check([[
+      local s = "hello"
+      local len = #s
+   ]]))
+
+   it("can run on an enum value", util.check([[
+      local enum Enum
+         "hello"
+      end
+
+      local s: Enum = "hello"
+      local len = #s
+   ]]))
+
    it("returns an integer when used on tuple", util.check([[
       local x: integer = #({1, "hi"})
    ]]))

diff --git a/tl.lua b/tl.lua
@@ -6293,6 +6293,7 @@ local equality_binop = {
 
 local unop_types = {
    ["#"] = {
+      ["enum"] = "integer",
       ["string"] = "integer",
       ["array"] = "integer",
       ["tupletable"] = "integer",

diff --git a/tl.tl b/tl.tl
@@ -6293,6 +6293,7 @@ local equality_binop = {
 
 local unop_types: {string:{TypeName:TypeName}} = {
    ["#"] = {
+      ["enum"] = "integer",
       ["string"] = "integer",
       ["array"] = "integer",
       ["tupletable"] = "integer",