fix close-non-object

Author: sumneko

script/core/diagnostics/close-non-object.lua

@@ -1,6 +1,7 @@
 local files    = require 'files'
 local guide    = require 'parser.guide'
 local lang     = require 'language'
+local vm       = require 'vm'
 
 return function (uri, callback)
     local state = files.getState(uri)
@@ -23,18 +24,15 @@ return function (uri, callback)
             }
             return
         end
-        if source.value.type == 'nil'
-        or source.value.type == 'number'
-        or source.value.type == 'integer'
-        or source.value.type == 'boolean'
-        or source.value.type == 'table'
-        or source.value.type == 'function' then
+        local infer = vm.getInfer(source.value)
+        if  not infer:hasClass()
+        and not infer:hasType 'nil'
+        and infer:view('any', uri) ~= 'any'  then
             callback {
                 start   = source.value.start,
                 finish  = source.value.finish,
                 message = lang.script.DIAG_COSE_NON_OBJECT,
             }
-            return
         end
     end)
 end

test/diagnostics/common.lua

@@ -333,13 +333,13 @@ TEST [[
 local _ <close> = <!1!>
 ]]
 
---TEST [[
---local _ <close> = <!''!>
---]]
---
---TEST [[
---local c <close> = <!(function () return 1 end)()!>
---]]
+TEST [[
+local _ <close> = <!''!>
+]]
+
+TEST [[
+local c <close> = <!(function () return 1 end)()!>
+]]
 
 config.get(nil, 'Lua.diagnostics.disable')['unused-local'] = true
 TEST [[