support ---@type and --[[@as]] for return

resolve #2144

Author: sumneko

changelog.md

@@ -1,6 +1,7 @@
 # changelog
 
 ## 3.7.0
+* `NEW` support `---@type` and `--[[@as]]` for return statement
 * `FIX` wrong hover and signature for method with varargs and overloads
 * `FIX` [#2224]
 

script/vm/compiler.lua

@@ -1168,6 +1168,16 @@ local compilerSwitch = util.switch()
             vm.compileCallArg(source, call)
         end
 
+        if source.parent.type == 'return' then
+            local myIndex = util.arrayIndexOf(source.parent, source)
+            ---@cast myIndex -?
+            local parentNode = vm.selectNode(source.parent, myIndex)
+            if not parentNode:isEmpty() then
+                vm.setNode(source, parentNode)
+                return
+            end
+        end
+
         if source.parent.type == 'setglobal'
         or source.parent.type == 'local'
         or source.parent.type == 'setlocal'

test/diagnostics/missing-fields.lua

@@ -197,3 +197,11 @@ local b = {
     c = 3,
 }
 ]]
+
+TEST [[
+---@class A
+---@field x integer
+
+---@type A
+return <!{}!>
+]]