fix error when handling index with a non-literal key. eg: t[#t+1] = true

xebecnan · Nov 4, 2021 · 601d799 · 601d799
1 parent 14ffdab
commit 601d799
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/src/symbols.lua b/src/symbols.lua
@@ -115,20 +115,18 @@ local function find_symbol(namespace, ast, narrow_func)
             return { tag='Id', 'Any' }
         end
 
-        local Types = require('types')
-        local field_type = Types.get_node_type(ast[2])
-        if field_type.tag == 'Id' and field_type[1] == 'Any' then
-            return { tag='Id', 'Any' }
-        elseif field_type.tag == 'Id' and field_type[1] == 'Str' then
-            -- literal string as index
-            assert(ast[2].tag == 'Str')
+        if ast[2].tag == 'Str' then
             local key = ast[2][1]
             return get_table_field(si1, key)
-        elseif field_type.tag == 'Id' and field_type[1] == 'Integer' then
-            -- literal integer as index
-            assert(ast[2].tag == 'Integer')
+        elseif ast[2].tag == 'Integer' then
             local key = ast[2][1]
             return get_table_field(si1, key)
+        end
+
+        local Types = require('types')
+        local field_type = Types.get_node_type(ast[2])
+        if field_type.tag == 'Id' and field_type[1] == 'Any' then
+            return { tag='Id', 'Any' }
         else
             -- TODO: 其他类型作为 key
             ast_error(ast, "find_symbol not support '%s' yet: TODO", Types.get_full_type_name(field_type, false))