support protected check for completion

Author: CppCXY

crates/code_analysis/src/semantic/mod.rs

@@ -23,6 +23,7 @@ use semantic_info::{
     infer_node_property_owner, infer_node_semantic_info, infer_token_property_owner,
     infer_token_semantic_info,
 };
+use type_compact::is_sub_type_of;
 use visibility::check_visibility;
 
 use crate::LuaFunctionType;
@@ -154,6 +155,14 @@ impl<'a> SemanticModel<'a> {
         .unwrap_or(true)
     }
 
+    pub fn is_sub_type_of(
+        &mut self,
+        sub_type_ref_id: &LuaTypeDeclId,
+        super_type_ref_id: &LuaTypeDeclId,
+    ) -> bool {
+        is_sub_type_of(self.db, sub_type_ref_id, super_type_ref_id)
+    }
+
     pub fn get_emmyrc(&self) -> &Emmyrc {
         &self.emmyrc
     }

crates/code_analysis/src/semantic/type_compact/mod.rs

@@ -1,4 +1,5 @@
 mod func_type;
+mod sub_type;
 
 use func_type::infer_doc_func_type_compact;
 
@@ -8,6 +9,7 @@ use crate::db_index::{
 };
 
 use super::{InferGuard, LuaInferConfig};
+pub use sub_type::is_sub_type_of;
 
 #[allow(unused)]
 pub fn check_type_compact(

crates/code_analysis/src/semantic/type_compact/sub_type.rs

@@ -0,0 +1,34 @@
+use crate::{DbIndex, InferGuard, LuaType, LuaTypeDeclId};
+
+pub fn is_sub_type_of(
+    db: &DbIndex,
+    sub_type_ref_id: &LuaTypeDeclId,
+    super_type_ref_id: &LuaTypeDeclId,
+) -> bool {
+    let mut infer_guard = InferGuard::new();
+    check_sub_type_of(db, sub_type_ref_id, super_type_ref_id, &mut infer_guard).unwrap_or(false)
+}
+
+fn check_sub_type_of(
+    db: &DbIndex,
+    sub_type_ref_id: &LuaTypeDeclId,
+    super_type_ref_id: &LuaTypeDeclId,
+    infer_guard: &mut InferGuard,
+) -> Option<bool> {
+    infer_guard.check(super_type_ref_id)?;
+
+    let supers = db.get_type_index().get_super_types(sub_type_ref_id)?;
+    for super_type in supers {
+        if let LuaType::Ref(super_id) = super_type {
+            if super_id == *super_type_ref_id {
+                return Some(true);
+            }
+
+            if check_sub_type_of(db, sub_type_ref_id, &super_id, infer_guard).unwrap_or(false) {
+                return Some(true);
+            }
+        }
+    }
+
+    Some(false)
+}

crates/code_analysis/src/semantic/visibility/mod.rs

@@ -5,7 +5,7 @@ use emmylua_parser::{
 
 use crate::{DbIndex, Emmyrc, FileId, LuaMemberOwner, LuaPropertyOwnerId, LuaType};
 
-use super::{infer_expr, LuaInferConfig};
+use super::{infer_expr, type_compact::is_sub_type_of, LuaInferConfig};
 
 pub fn check_visibility(
     db: &DbIndex,
@@ -94,7 +94,9 @@ fn check_block_visibility(
                         return Some(true);
                     }
 
-                    // todo is subclass
+                    if is_sub_type_of(db, &left, &right) {
+                        return Some(true);
+                    }
                 }
                 _ => {}
             }