update keyof completion

Author: xuhuanzy

crates/emmylua_code_analysis/src/semantic/generic/instantiate_type/instantiate_special_generic.rs

@@ -42,7 +42,6 @@ pub fn instantiate_alias_call(
             if operands.len() != 1 {
                 return LuaType::Unknown;
             }
-            // let is_tuple = operands.len() == 1 && operands[0].is_tuple();
 
             let members = get_keyof_members(db, &operands[0]).unwrap_or_default();
             let member_key_types = members
@@ -54,13 +53,6 @@ pub fn instantiate_alias_call(
                 })
                 .collect::<Vec<_>>();
             LuaType::Tuple(LuaTupleType::new(member_key_types, LuaTupleStatus::InferResolve).into())
-            // if is_tuple {
-            //     LuaType::Tuple(
-            //         LuaTupleType::new(member_key_types, LuaTupleStatus::InferResolve).into(),
-            //     )
-            // } else {
-            //     LuaType::from_vec(member_key_types)
-            // }
         }
         // 条件类型不在此处理
         LuaAliasCallKind::Extends => {
@@ -285,7 +277,7 @@ fn instantiate_index_call(db: &DbIndex, owner: &LuaType, key: &LuaType) -> LuaTy
     }
 }
 
-fn get_keyof_members(db: &DbIndex, prefix_type: &LuaType) -> Option<Vec<LuaMemberInfo>> {
+pub fn get_keyof_members(db: &DbIndex, prefix_type: &LuaType) -> Option<Vec<LuaMemberInfo>> {
     match prefix_type {
         LuaType::Variadic(variadic) => match variadic.deref() {
             VariadicType::Base(base) => Some(vec![LuaMemberInfo {

crates/emmylua_code_analysis/src/semantic/generic/instantiate_type/mod.rs

@@ -20,6 +20,7 @@ use crate::{
 use super::type_substitutor::{SubstitutorValue, TypeSubstitutor};
 use crate::TypeVisitTrait;
 pub use instantiate_func_generic::{build_self_type, infer_self_type, instantiate_func_generic};
+pub use instantiate_special_generic::get_keyof_members;
 pub use instantiate_special_generic::instantiate_alias_call;
 
 pub fn instantiate_type_generic(

crates/emmylua_code_analysis/src/semantic/generic/mod.rs

@@ -23,6 +23,7 @@ use crate::SemanticDeclLevel;
 use crate::TypeOps;
 use crate::infer_node_semantic_decl;
 use crate::semantic::semantic_info::infer_token_semantic_decl;
+pub use instantiate_type::get_keyof_members;
 
 pub fn get_tpl_ref_extend_type(
     db: &DbIndex,

crates/emmylua_code_analysis/src/semantic/mod.rs

@@ -58,6 +58,7 @@ use overload_resolve::resolve_signature;
 pub use semantic_info::SemanticDeclLevel;
 pub use type_check::{TypeCheckFailReason, TypeCheckResult};
 
+pub use generic::get_keyof_members;
 pub use infer::{DocTypeInferContext, infer_doc_type};
 
 #[derive(Debug)]

crates/emmylua_ls/src/handlers/completion/add_completions/add_member_completion.rs

@@ -1,6 +1,6 @@
 use emmylua_code_analysis::{
-    DbIndex, LuaMemberInfo, LuaMemberKey, LuaSemanticDeclId, LuaType, SemanticModel,
-    try_extract_signature_id_from_field,
+    DbIndex, LuaAliasCallKind, LuaMemberInfo, LuaMemberKey, LuaSemanticDeclId, LuaType,
+    SemanticModel, get_keyof_members, try_extract_signature_id_from_field,
 };
 use emmylua_parser::{
     LuaAssignStat, LuaAstNode, LuaAstToken, LuaFuncStat, LuaGeneralToken, LuaIndexExpr,
@@ -44,6 +44,26 @@ pub fn add_member_completion(
         CompletionTriggerStatus::Dot => match member_key {
             LuaMemberKey::Name(name) => name.to_string(),
             LuaMemberKey::Integer(index) => format!("[{}]", index),
+            LuaMemberKey::ExprType(typ) => {
+                if let LuaType::Call(alias_call) = typ {
+                    if alias_call.get_call_kind() == LuaAliasCallKind::KeyOf
+                        && alias_call.get_operands().len() == 1
+                    {
+                        let members = get_keyof_members(
+                            builder.semantic_model.get_db(),
+                            &alias_call.get_operands()[0],
+                        )
+                        .unwrap_or_default();
+                        let member_keys = members.iter().map(|m| m.key.clone()).collect::<Vec<_>>();
+                        for key in member_keys {
+                            let mut member_info = member_info.clone();
+                            member_info.key = key;
+                            add_member_completion(builder, member_info, status, None);
+                        }
+                    }
+                }
+                return None;
+            }
             _ => return None,
         },
         CompletionTriggerStatus::Colon => match member_key {

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

@@ -2243,4 +2243,27 @@ mod tests {
         ));
         Ok(())
     }
+
+    #[gtest]
+    fn test_keyof_enum() -> Result<()> {
+        let mut ws = ProviderVirtualWorkspace::new();
+        check!(ws.check_completion(
+            r#"
+            ---@enum A
+            local styles = {
+                reset = 1
+            }
+
+            ---@type table<keyof A, string>
+            local t
+            t.<??>
+            "#,
+            vec![VirtualCompletionItem {
+                label: "reset".to_string(),
+                kind: CompletionItemKind::VARIABLE,
+                ..Default::default()
+            },],
+        ));
+        Ok(())
+    }
 }