setlocal can be marked by ---@type

Author: sumneko

script/parser/luadoc.lua

@@ -1390,6 +1390,7 @@ local function bindDocsBetween(sources, binded, bindSources, start, finish)
             if src.start >= start then
                 if src.type == 'local'
                 or src.type == 'self'
+                or src.type == 'setlocal'
                 or src.type == 'setglobal'
                 or src.type == 'tablefield'
                 or src.type == 'tableindex'

script/vm/compiler.lua

@@ -976,6 +976,14 @@ local compilerSwitch = util.switch()
         local runner = vm.createRunner(source)
         runner:launch(function (src, node)
             if src.type == 'setlocal' then
+                if src.bindDocs then
+                    for _, doc in ipairs(src.bindDocs) do
+                        if doc.type == 'doc.type' then
+                            vm.setNode(src, vm.compileNode(doc), true)
+                            return
+                        end
+                    end
+                end
                 if src.value and guide.isLiteral(src.value) then
                     if src.value.type == 'table' then
                         vm.setNode(src, vm.createNode(src.value), true)
@@ -1260,6 +1268,9 @@ local compilerSwitch = util.switch()
         local uri = guide.getUri(source)
         vm.setNode(source, source)
         local global = globalMgr.getGlobal('type', source.node[1])
+        if not global then
+            return
+        end
         for _, set in ipairs(global:getSets(uri)) do
             if set.type == 'doc.class' then
                 if set.extends then

test/type_inference/init.lua

@@ -2056,3 +2056,10 @@ local iter
 for <?x?> in iter do
 end
 ]]
+
+TEST 'integer' [[
+local x
+
+---@type integer
+<?x?> = XXX
+]]