fix build_func_stat_override_hint

Author: xuhuanzy

crates/emmylua_ls/src/handlers/inlay_hint/build_inlay_hint.rs

@@ -435,18 +435,20 @@ fn get_super_member_id(
     member_key: &LuaMemberKey,
     infer_guard: &mut InferGuard,
 ) -> Option<LuaMemberId> {
-    if let LuaType::Ref(super_type_id) = &super_type {
-        infer_guard.check(super_type_id).ok()?;
-        let member_map = semantic_model.get_member_info_map(&super_type)?;
-
-        if let Some(member_infos) = member_map.get(&member_key) {
-            let first_property = member_infos.first()?.property_owner_id.clone()?;
-            if let LuaSemanticDeclId::Member(member_id) = first_property {
-                return Some(member_id);
-            }
+    let super_type_id = match &super_type {
+        LuaType::Ref(id) => id,
+        LuaType::Generic(generic) => generic.get_base_type_id_ref(),
+        _ => return None,
+    };
+    infer_guard.check(super_type_id).ok()?;
+    let member_map = semantic_model.get_member_info_map(&super_type)?;
+
+    if let Some(member_infos) = member_map.get(&member_key) {
+        let first_property = member_infos.first()?.property_owner_id.clone()?;
+        if let LuaSemanticDeclId::Member(member_id) = first_property {
+            return Some(member_id);
         }
     }
-
     None
 }
 

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

@@ -319,4 +319,28 @@ mod tests {
             .unwrap();
         assert_eq!(result.len(), 1);
     }
+
+    #[test]
+    fn test_generic_type_override() {
+        let mut ws = ProviderVirtualWorkspace::new();
+        ws.def(
+            r#"
+            ---@class A<T>
+            ---@field aaa fun(a: integer): integer
+        "#,
+        );
+        let result = ws
+            .check_inlay_hint(
+                r#"
+                ---@class B<T>: A<T>
+                local B
+
+                function B:aaa(a)
+                    return a
+                end
+            "#,
+            )
+            .unwrap();
+        assert_eq!(result.len(), 1);
+    }
 }