docs: better type annotations typescript/lua codegen

Author: TheLeoP

lua/refactoring/code_generation/langs/lua.lua

@@ -1,5 +1,7 @@
 local code_utils = require("refactoring.code_generation.utils")
 
+---@param opts refactor.FuncParams
+---@return string
 local function lua_function(opts)
     return ([[
 local function %s(%s)

lua/refactoring/code_generation/langs/typescriptreact.lua

@@ -1,8 +1,11 @@
 local code_utils = require("refactoring.code_generation.utils")
 local ts = require("refactoring.code_generation.langs.typescript")
 
+---@param args string[]
+---@param arg_types table<string, string>
+---@return string[]
 local function build_args(args, arg_types)
-    local final_args = {}
+    local final_args = {} ---@type string[]
     for i, arg in pairs(args) do
         if arg_types[arg] ~= code_utils.default_func_param_type() then
             final_args[i] = arg .. ": " .. arg_types[arg]
@@ -15,15 +18,19 @@ end
 
 ---@param opts refactor.code_gen.function.Opts
 local function tsx_function(opts)
-    if opts.region_type == "jsx_element" then
-        local args
-        if opts.args_types ~= nil then
-            args = build_args(opts.args, opts.args_types)
-        else
-            args = opts.args
-        end
+    if opts.region_type ~= "jsx_element" then
+        return ts["function"](opts)
+    end
+
+    local args ---@type string[]
+    if opts.args_types ~= nil then
+        args = build_args(opts.args, opts.args_types)
+    else
+        args = opts.args
+    end
+    assert(args)
 
-        return ([[
+    return ([[
 %sfunction %s({%s}) {
 return (
 <>
@@ -33,15 +40,12 @@ return (
 %s}
 
 ]]):format(
-            opts.func_header,
-            opts.name,
-            table.concat(args, ", "),
-            code_utils.stringify_code(opts.body),
-            opts.func_header
-        )
-    else
-        return ts["function"](opts)
-    end
+        opts.func_header,
+        opts.name,
+        table.concat(args, ", "),
+        code_utils.stringify_code(opts.body),
+        opts.func_header
+    )
 end
 
 ---@param opts refactor.code_gen.call_function.Opts

lua/refactoring/refactor/106.lua

@@ -81,7 +81,7 @@ end
 ---@param args string[]
 ---@return table<string, string|nil>
 local function get_function_param_types(refactor, args)
-    local args_types = {}
+    local args_types = {} ---@type table<string, string>
 
     local local_types = refactor.ts:get_local_types(refactor.scope)
 
@@ -166,6 +166,7 @@ end
 
 ---@class refactor.FuncParams
 ---@field func_header? string
+---@field args_types? table<string, string>
 ---@field contains_jsx? boolean
 ---@field class_name? string
 ---@field visibility? string