Raise a compile error when trying to assign comptime records

edubart · Oct 31, 2023 · 3343daa · 3343daa
1 parent 596fcca
commit 3343daa
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/lualib/nelua/analyzer.lua b/lualib/nelua/analyzer.lua
@@ -2320,9 +2320,14 @@ function visitors.VarDecl(context, node)
           valtype = primtypes.niltype
         end
       end
-      if varnode.attr.comptime and not (valnode.attr.comptime and valtype) then
-        varnode:raisef("compile time variables can only assign to compile time expressions")
-      elseif vartype and vartype.is_auto then
+      if varnode.attr.comptime then
+        if not (valnode.attr.comptime and valtype) then
+          varnode:raisef("compile time variables can only assign to compile time expressions")
+        elseif (valnode.attr.value == nil and valnode.attr.type ~= primtypes.niltype) then
+          varnode:raisef("compile time variables cannot be of type '%s'", vartype)
+        end
+      end
+      if vartype and vartype.is_auto then
         if not valtype then
           varnode:raisef("auto variables must be assigned to expressions where type is known ahead")
         elseif valtype.is_nolvalue then

diff --git a/spec/analyzer_spec.lua b/spec/analyzer_spec.lua
@@ -1258,6 +1258,10 @@ it("records", function()
     local Record
     local a <comptime> = Record{}
   ]], "can only assign to compile time expressions")
+  expect.analyze_error([[
+    local Box = @record{n: number}
+    local b: Box <comptime> = {n=3}
+  ]], "compile time variables cannot be of type")
   expect.ast_type_equals(
     "local a: record{x: boolean}; local b = a.x",
     "local a: record{x: boolean}; local b: boolean = a.x")