fix hover union function

Author: xuhuanzy

crates/emmylua_code_analysis/src/semantic/infer/infer_table.rs

@@ -1,4 +1,4 @@
-use std::ops::Deref;
+use std::{ops::Deref, sync::Arc};
 
 use emmylua_parser::{
     LuaAssignStat, LuaAst, LuaAstNode, LuaCallArgList, LuaCallExpr, LuaExpr, LuaIndexMemberExpr,
@@ -8,7 +8,7 @@ use emmylua_parser::{
 use crate::{
     db_index::{DbIndex, LuaType},
     infer_call_expr_func, infer_expr, InferGuard, LuaArrayType, LuaDeclId, LuaInferCache,
-    LuaMemberId, LuaTupleStatus, LuaTupleType, VariadicType,
+    LuaMemberId, LuaTupleStatus, LuaTupleType, LuaUnionType, TypeOps, VariadicType,
 };
 
 use super::{
@@ -221,12 +221,37 @@ fn infer_table_type_by_calleee(
             call_arg_number -= 1;
         }
     }
-    Ok(param_types
+    let typ = param_types
         .get(call_arg_number)
         .ok_or(InferFailReason::None)?
         .1
         .clone()
-        .unwrap_or(LuaType::Any))
+        .unwrap_or(LuaType::Any);
+    match &typ {
+        LuaType::TableConst(_) => {}
+        LuaType::Union(union) => {
+            // TODO: 假设存在多个匹配项, 我们需要根据字段的匹配情况来确定最终的类型
+            return Ok(union_remove_non_table_type(db, union));
+        }
+        _ => {}
+    }
+
+    Ok(typ)
+}
+
+/// 移除掉一些非`table`类型
+fn union_remove_non_table_type(db: &DbIndex, union: &Arc<LuaUnionType>) -> LuaType {
+    let mut result = LuaType::Unknown;
+    for typ in union.into_set().into_iter() {
+        match typ {
+            LuaType::Signature(_) | LuaType::DocFunction(_) => {}
+            _ if typ.is_string() || typ.is_number() || typ.is_boolean() => {}
+            _ => {
+                result = TypeOps::Union.apply(db, &result, &typ);
+            }
+        }
+    }
+    result
 }
 
 fn infer_table_field_type_by_parent(

crates/emmylua_ls/src/handlers/hover/build_hover.rs

@@ -217,6 +217,7 @@ fn build_member_hover(
         true,
     )
     .get_types(&builder.semantic_model);
+
     replace_semantic_type(&mut semantic_decls, &typ);
     let member_name = match member.get_key() {
         LuaMemberKey::Name(name) => name.to_string(),

crates/emmylua_ls/src/handlers/hover/find_origin.rs

@@ -200,10 +200,7 @@ fn resolve_member_owner(
                 // 非类, 那么通过右值推断
                 let value_expr = table_field.get_value_expr()?;
                 let value_node = value_expr.get_syntax_id().to_node_from_root(&root)?;
-                semantic_model.find_decl(
-                    value_node.into(),
-                    emmylua_code_analysis::SemanticDeclLevel::default(),
-                )
+                semantic_model.find_decl(value_node.into(), SemanticDeclLevel::default())
             } else {
                 None
             }
@@ -217,10 +214,8 @@ fn resolve_member_owner(
             for (var, expr) in vars.iter().zip(exprs.iter()) {
                 if var.syntax().text_range() == current_node.text_range() {
                     let expr_node = expr.get_syntax_id().to_node_from_root(&root)?;
-                    result = semantic_model.find_decl(
-                        expr_node.into(),
-                        emmylua_code_analysis::SemanticDeclLevel::default(),
-                    );
+                    result =
+                        semantic_model.find_decl(expr_node.into(), SemanticDeclLevel::default());
                     break;
                 }
             }
@@ -255,7 +250,7 @@ fn resolve_table_field_through_type_inference(
 
     if !matches!(
         table_type,
-        emmylua_code_analysis::LuaType::Ref(_) | emmylua_code_analysis::LuaType::Def(_)
+        LuaType::Ref(_) | LuaType::Def(_) | LuaType::Generic(_)
     ) {
         return None;
     }

crates/emmylua_ls/src/handlers/hover/function_humanize.rs

@@ -53,7 +53,7 @@ pub fn hover_function_type(
     // 已处理过的 semantic_decl_id, 用于解决`test_issue_499_3`
     let mut handled_semantic_decl_ids = HashSet::new();
     let mut type_descs: Vec<HoverFunctionInfo> = Vec::with_capacity(semantic_decls.len());
-    // 记录已处理过的类型，用于在 Union 中跳过重复类型.
+    // 记录已处理过的类型, 用于在 Union 中跳过重复类型.
     // 这是为了解决最后一个类型可能是前面所有类型的联合类型的情况
     let mut processed_types = HashSet::new();
 
@@ -122,7 +122,7 @@ pub fn hover_function_type(
             }
         }
 
-        // 如果当前类型是 Union，传入已处理的类型集合
+        // 如果当前类型是 Union, 传入已处理的类型集合
         let result = match typ {
             LuaType::Union(_) => process_single_function_type_with_exclusions(
                 builder,
@@ -158,10 +158,10 @@ pub fn hover_function_type(
                 function_info = info;
             }
             ProcessFunctionTypeResult::Multiple(infos) => {
-                // 对于 Union 类型，将每个子类型的结果都添加到 type_descs 中
+                // 对于 Union 类型, 将每个子类型的结果都添加到 type_descs 中
                 let infos_len = infos.len();
                 for (index, mut info) in infos.into_iter().enumerate() {
-                    // 合并描述信息，只有最后一个才设置描述
+                    // 合并描述信息, 只有最后一个才设置描述
                     if function_info.description.is_some()
                         && info.description.is_none()
                         && index == infos_len - 1
@@ -754,7 +754,7 @@ fn process_single_function_type_with_exclusions(
             }
         }
         _ => {
-            // 对于非 Union 类型，直接调用原函数
+            // 对于非 Union 类型, 直接调用原函数
             process_single_function_type(
                 builder,
                 db,

crates/emmylua_ls/src/handlers/test/hover_test.rs

@@ -245,4 +245,57 @@ mod tests {
             },
         ));
     }
+
+    #[test]
+    fn test_field_key() {
+        let mut ws = ProviderVirtualWorkspace::new();
+        ws.def(
+            r#"
+            ---@class ObserverParams
+            ---@field next fun() # 测试
+
+            ---@param params fun() | ObserverParams
+            function test(params)
+            end
+            "#,
+        );
+        assert!(ws.check_hover(
+            r#"
+                test({
+                    <??>next = function()
+                    end
+                })
+            "#,
+            VirtualHoverResult {
+                value: "```lua\n(field) ObserverParams.next()\n```\n\n---\n\n测试".to_string(),
+            },
+        ));
+    }
+
+    #[test]
+    fn test_field_key_for_generic() {
+        let mut ws = ProviderVirtualWorkspace::new();
+        ws.def(
+            r#"
+            ---@class ObserverParams<T>
+            ---@field next fun() # 测试
+
+            ---@generic T
+            ---@param params fun() | ObserverParams<T>
+            function test(params)
+            end
+            "#,
+        );
+        assert!(ws.check_hover(
+            r#"
+                test({
+                    <??>next = function()
+                    end
+                })
+            "#,
+            VirtualHoverResult {
+                value: "```lua\n(field) ObserverParams.next()\n```\n\n---\n\n测试".to_string(),
+            },
+        ));
+    }
 }