fix #1677

Author: sumneko

changelog.md

@@ -3,8 +3,10 @@
 ## 3.6.2
 * `FIX` incorrect type check for generic with nil
 * `FIX` [#1676]
+* `FIX` [#1677]
 
 [#1676]: https://github.com/sumneko/lua-language-server/issues/1676
+[#1677]: https://github.com/sumneko/lua-language-server/issues/1677
 
 ## 3.6.1
 `2022-11-8`

script/parser/guide.lua

@@ -281,17 +281,10 @@ function m.isLiteral(obj)
 end
 
 --- 获取字面量
----@param obj parser.object
+---@param obj table
 ---@return any
 function m.getLiteral(obj)
-    local tp = obj.type
-    if     tp == 'boolean' then
-        return obj[1]
-    elseif tp == 'string' then
-        return obj[1]
-    elseif tp == 'number' then
-        return obj[1]
-    elseif tp == 'integer' then
+    if m.isLiteral(obj) then
         return obj[1]
     end
     return nil

script/vm/compiler.lua

@@ -284,7 +284,7 @@ local searchFieldSwitch = util.switch()
 
 ---@param suri uri
 ---@param object vm.global
----@param key? string|vm.global
+---@param key? string|number|integer|boolean|vm.global
 ---@param ref boolean
 ---@param pushResult fun(field: vm.object, isMark?: boolean)
 function vm.getClassFields(suri, object, key, ref, pushResult)
@@ -341,22 +341,22 @@ function vm.getClassFields(suri, object, key, ref, pushResult)
                             end
                         end
                     else
-                        local typeName
+                        local keyObject
                         if keyType == 'number' then
                             if math.tointeger(key) then
-                                typeName = 'integer'
+                                keyObject = { type = 'integer', [1] = key }
                             else
-                                typeName = 'number'
+                                keyObject = { type = 'number', [1] = key }
                             end
                         elseif keyType == 'boolean'
                         or     keyType == 'string' then
-                            typeName = keyType
+                            keyObject = { type = keyType, [1] = key }
                         end
-                        if typeName and field.field.type ~= 'doc.field.name' then
+                        if keyObject and field.field.type ~= 'doc.field.name' then
                             -- ---@field [integer] boolean -> class[1]
                             local fieldNode = vm.compileNode(field.field)
-                            if vm.isSubType(suri, typeName, fieldNode) then
-                                local nkey = '|' .. typeName
+                            if vm.isSubType(suri, keyObject, fieldNode) then
+                                local nkey = '|' .. keyType
                                 if not searchedFields[nkey] then
                                     pushResult(field, true)
                                     hasFounded[nkey] = true

test/type_inference/init.lua

@@ -3947,3 +3947,21 @@ end
 
 local <?x?> = f(true)
 ]]
+
+TEST 'number' [[
+---@class A
+---@field [1] number
+---@field [2] boolean
+local t
+
+local <?n?> = t[1]
+]]
+
+TEST 'boolean' [[
+---@class A
+---@field [1] number
+---@field [2] boolean
+local t
+
+local <?n?> = t[2]
+]]