support async infer for closure

Author: CppCXY

crates/code_analysis/src/compilation/analyzer/unresolve/resolve_closure_param.rs

@@ -1,5 +1,6 @@
 use crate::{
-    infer_call_expr_func, infer_expr, DbIndex, InferGuard, LuaDocParamInfo, LuaInferConfig, LuaType,
+    infer_call_expr_func, infer_expr, DbIndex, InferGuard, LuaDocParamInfo, LuaInferConfig,
+    LuaPropertyOwnerId, LuaType,
 };
 
 use super::UnResolveClosureParams;
@@ -19,16 +20,19 @@ pub fn try_resolve_closure_params(
         call_expr,
         call_expr_type,
         &mut InferGuard::new(),
-        None
+        None,
     )?;
 
-    let signature = db
-        .get_signature_index_mut()
-        .get_mut(&closure_params.signature_id)?;
-
     let expr_closure_params =
         if let Some(param_type) = call_doc_func.get_params().get(closure_params.param_idx) {
             if let Some(LuaType::DocFunction(func)) = &param_type.1 {
+                if func.is_async() {
+                    let file_id = closure_params.file_id;
+                    let property_owner = LuaPropertyOwnerId::Signature(closure_params.signature_id);
+                    db.get_property_index_mut()
+                        .add_async(file_id, property_owner);
+                }
+
                 func.get_params()
             } else {
                 return Some(true);
@@ -37,6 +41,10 @@ pub fn try_resolve_closure_params(
             return Some(true);
         };
 
+    let signature = db
+        .get_signature_index_mut()
+        .get_mut(&closure_params.signature_id)?;
+
     let signature_params = &mut signature.param_docs;
     for (idx, (name, type_ref)) in expr_closure_params.iter().enumerate() {
         if signature_params.contains_key(&idx) {

resources/std/coroutine.lua

@@ -18,7 +18,7 @@ coroutine = {}
 ---
 --- Creates a new coroutine, with body `f`. `f` must be a Lua function. Returns
 --- this new coroutine, an object with type `"thread"`.
----@param f fun():thread
+---@param f async fun(...)
 ---@return thread
 function coroutine.create(f) end
 
@@ -62,7 +62,7 @@ function coroutine.resume(co, ...) end
 ---
 --- Returns the running coroutine plus a boolean, true when the running
 --- coroutine is the main one.
----@return thread|boolean, string
+---@return thread, string
 function coroutine.running() end
 
 ---
@@ -86,7 +86,7 @@ function coroutine.status(co) end
 --- passed to the function behave as the extra arguments to `resume`. Returns
 --- the same values returned by `resume`, except the first
 --- boolean. In case of error, propagates the error.
----@param f fun():thread
+---@param f async fun()
 ---@return fun():any
 function coroutine.wrap(f) end