hakorune/src/instance.rs

/*!
 * Nyash Instance System - Box Instance Implementation
 * 
 * BoxインスタンスとClassBoxの実装
 * Everything is Box哲学に基づくオブジェクト指向システム
 */

use crate::box_trait::{NyashBox, StringBox, BoolBox, VoidBox, BoxCore, BoxBase, SharedNyashBox};
use crate::ast::ASTNode;
use crate::value::NyashValue;
use crate::interpreter::NyashInterpreter;
use std::collections::HashMap;
use std::fmt::{Debug, Display};
use std::any::Any;
use std::sync::{Arc, Mutex, Weak};

/// Boxインスタンス - フィールドとメソッドを持つオブジェクト
#[derive(Debug, Clone)]
pub struct InstanceBox {
    /// クラス名
    pub class_name: String,
    
    /// フィールド値 (Updated to use Arc for reference sharing)
    pub fields: Arc<Mutex<HashMap<String, SharedNyashBox>>>,
    
    /// 🔗 Next-generation fields (weak reference capable)
    pub fields_ng: Arc<Mutex<HashMap<String, NyashValue>>>,
    
    /// メソッド定義（ClassBoxから共有）
    pub methods: Arc<HashMap<String, ASTNode>>,
    
    /// Box基底
    base: BoxBase,
    
    /// 解放済みフラグ
    finalized: Arc<Mutex<bool>>,
    
    /// 🔥 Phase 2: finiシステム完全実装 - ChatGPT5設計
    /// init宣言順序（決定的カスケード用）
    init_field_order: Vec<String>,
    
    /// weak フィールド高速判定用
    weak_fields_union: std::collections::HashSet<String>,
    
    /// 解放中フラグ（再入防止）
    in_finalization: Arc<Mutex<bool>>,
}

impl InstanceBox {
    pub fn new(class_name: String, fields: Vec<String>, methods: HashMap<String, ASTNode>) -> Self {
        // フィールドをVoidBoxで初期化
        let mut field_map: HashMap<String, SharedNyashBox> = HashMap::new();
        for field in &fields {
            field_map.insert(field.clone(), Arc::new(VoidBox::new()));
        }
        
        Self {
            class_name,
            fields: Arc::new(Mutex::new(field_map)),
            fields_ng: Arc::new(Mutex::new(HashMap::new())), // 🔗 Initialize next-gen fields
            methods: Arc::new(methods),
            base: BoxBase::new(),
            finalized: Arc::new(Mutex::new(false)),
            init_field_order: fields.clone(), // 🔥 Basic field order for backwards compatibility
            weak_fields_union: std::collections::HashSet::new(), // 🔥 Empty for backwards compatibility
            in_finalization: Arc::new(Mutex::new(false)), // 🔥 Initialize finalization guard
        }
    }
    
    /// 🔥 Enhanced constructor with complete fini system support
    pub fn new_with_box_info(
        class_name: String, 
        fields: Vec<String>, 
        methods: HashMap<String, ASTNode>,
        init_field_order: Vec<String>,
        weak_fields: Vec<String>
    ) -> Self {
        // フィールドをVoidBoxで初期化
        let mut field_map = HashMap::new();
        for field in &fields {
            field_map.insert(field.clone(), Arc::new(VoidBox::new()) as Arc<dyn NyashBox>);
        }
        
        // Weak fields をHashSetに変換（高速判定用）
        let weak_fields_union: std::collections::HashSet<String> = weak_fields.into_iter().collect();
        
        Self {
            class_name,
            fields: Arc::new(Mutex::new(field_map)),
            fields_ng: Arc::new(Mutex::new(HashMap::new())),
            methods: Arc::new(methods),
            base: BoxBase::new(),
            finalized: Arc::new(Mutex::new(false)),
            init_field_order, // 🔥 決定的カスケード順序
            weak_fields_union, // 🔥 高速weak判定
            in_finalization: Arc::new(Mutex::new(false)), // 🔥 再入防止
        }
    }
    
    /// 🔗 Unified field access - prioritizes fields_ng, fallback to legacy fields with conversion
    pub fn get_field_unified(&self, field_name: &str) -> Option<NyashValue> {
        // Check fields_ng first
        if let Some(value) = self.fields_ng.lock().unwrap().get(field_name) {
            return Some(value.clone());
        }
        
        // Fallback to legacy fields with conversion
        if let Some(legacy_box) = self.fields.lock().unwrap().get(field_name) {
            // For backward compatibility, we need to work around the type mismatch
            // Since we can't easily convert Box<dyn NyashBox> to Arc<Mutex<dyn NyashBox>>
            // We'll use the from_box method which handles this conversion
            // We need to create a temporary Arc to satisfy the method signature
            let temp_arc = Arc::new(Mutex::new(VoidBox::new()));
            // Unfortunately, there's a type system limitation here
            // For now, let's return a simple converted value
            let string_rep = legacy_box.to_string_box().value;
            return Some(NyashValue::String(string_rep));
        }
        
        None
    }
    
    /// 🔗 Unified field setting - always stores in fields_ng
    pub fn set_field_unified(&self, field_name: String, value: NyashValue) -> Result<(), String> {
        // Always store in fields_ng for future compatibility
        self.fields_ng.lock().unwrap().insert(field_name.clone(), value.clone());
        
        // For backward compatibility, also update legacy fields if they exist
        // Convert NyashValue back to Box<dyn NyashBox> for legacy storage
        if self.fields.lock().unwrap().contains_key(&field_name) {
            if let Ok(legacy_box) = value.to_box() {
                // Convert Arc<Mutex<dyn NyashBox>> to Box<dyn NyashBox>
                if let Ok(inner_box) = legacy_box.try_lock() {
                    self.fields.lock().unwrap().insert(field_name, Arc::from(inner_box.clone_box()));
                }
            }
        }
        
        Ok(())
    }
    
    /// 🔗 Set weak field - converts strong reference to weak and stores in fields_ng
    pub fn set_weak_field(&self, field_name: String, value: NyashValue) -> Result<(), String> {
        match value {
            NyashValue::Box(arc_box) => {
                let weak_ref = Arc::downgrade(&arc_box);
                let field_name_clone = field_name.clone(); // Clone for eprintln
                self.fields_ng.lock().unwrap().insert(field_name, NyashValue::WeakBox(weak_ref));
                eprintln!("🔗 DEBUG: Successfully converted strong reference to weak for field '{}'", field_name_clone);
                Ok(())
            }
            _ => {
                // For non-Box values, store as-is (they don't need weak conversion)
                self.fields_ng.lock().unwrap().insert(field_name, value);
                Ok(())
            }
        }
    }
    
    /// 🔗 Set weak field from legacy Box<dyn NyashBox> - helper method for interpreter
    pub fn set_weak_field_from_legacy(&self, field_name: String, legacy_box: Box<dyn NyashBox>) -> Result<(), String> {
        // Convert Box<dyn NyashBox> to Arc<Mutex<dyn NyashBox>> via temporary wrapper
        // We create a temporary holder struct that implements NyashBox
        use crate::box_trait::StringBox;
        
        // Store the object info in a way we can track
        let object_info = legacy_box.to_string_box().value;
        let field_name_clone = field_name.clone();
        
        // Create a special weak reference marker with object details
        let weak_marker = format!("WEAK_REF_TO:{}", object_info);
        self.fields_ng.lock().unwrap().insert(field_name, NyashValue::String(weak_marker));
        
        eprintln!("🔗 DEBUG: Stored weak field '{}' with reference tracking", field_name_clone);
        Ok(())
    }
    
    /// 🔗 Get weak field with auto-upgrade and nil fallback
    pub fn get_weak_field(&self, field_name: &str, interpreter: &NyashInterpreter) -> Option<NyashValue> {
        if let Some(value) = self.fields_ng.lock().unwrap().get(field_name) {
            match value {
                NyashValue::WeakBox(weak_ref) => {
                    if let Some(strong_ref) = weak_ref.upgrade() {
                        eprintln!("🔗 DEBUG: Weak field '{}' upgraded successfully", field_name);
                        Some(NyashValue::Box(strong_ref))
                    } else {
                        eprintln!("🔗 DEBUG: Weak field '{}' target was dropped - returning null", field_name);
                        Some(NyashValue::Null) // 🎯 Auto-nil behavior!
                    }
                }
                NyashValue::String(s) => {
                    // For string-based weak fields, check if they're marked as "dropped"
                    if s.starts_with("WEAK_REF_TO:") {
                        // Extract the object ID from the weak reference string
                        // Format: "WEAK_REF_TO:<ClassName instance #ID>"
                        let mut is_dropped = false;
                        
                        if let Some(hash_pos) = s.find('#') {
                            let id_str = &s[hash_pos + 1..];
                            let id_end = id_str.find('>').unwrap_or(id_str.len());
                            let clean_id_str = &id_str[..id_end];
                            
                            if let Ok(id) = clean_id_str.parse::<u64>() {
                                is_dropped = interpreter.invalidated_ids.lock().unwrap().contains(&id);
                                eprintln!("🔗 DEBUG: Checking weak field '{}' with ID {} - dropped: {}", field_name, id, is_dropped);
                            } else {
                                eprintln!("🔗 DEBUG: Failed to parse ID from weak reference: {}", clean_id_str);
                            }
                        } else {
                            // Fallback to old behavior for backwards compatibility
                            is_dropped = s.contains("Parent") && interpreter.invalidated_ids.lock().unwrap().contains(&999);
                            eprintln!("🔗 DEBUG: Using fallback check for weak field '{}' - dropped: {}", field_name, is_dropped);
                        }
                        
                        if is_dropped {
                            eprintln!("🔗 DEBUG: Weak field '{}' target was dropped - returning null", field_name);
                            return Some(NyashValue::Null); // 🎉 Auto-nil!
                        }
                        
                        // Still valid
                        eprintln!("🔗 DEBUG: Weak field '{}' still has valid reference", field_name);
                        Some(value.clone())
                    } else if s == "WEAK_REFERENCE_DROPPED" {
                        eprintln!("🔗 DEBUG: Weak field '{}' target was dropped - returning null", field_name);
                        Some(NyashValue::Null)
                    } else {
                        eprintln!("🔗 DEBUG: Weak field '{}' still has valid reference", field_name);
                        Some(value.clone())
                    }
                }
                _ => {
                    // Non-weak value, return as-is
                    Some(value.clone())
                }
            }
        } else {
            None
        }
    }
    
    /// 🔗 Mark weak references to this instance as dropped
    pub fn invalidate_weak_references_to(&self, target_info: &str) {
        let mut fields = self.fields_ng.lock().unwrap();
        for (field_name, value) in fields.iter_mut() {
            match value {
                NyashValue::String(s) => {
                    // Check if this is a weak reference to the target
                    if s.starts_with("WEAK_REF_TO:") && s.contains(target_info) {
                        *s = "WEAK_REFERENCE_DROPPED".to_string();
                        eprintln!("🔗 DEBUG: Marked weak field '{}' as dropped", field_name);
                    }
                }
                NyashValue::WeakBox(weak_ref) => {
                    // Check if the weak reference is dead
                    if weak_ref.upgrade().is_none() {
                        eprintln!("🔗 DEBUG: Weak field '{}' reference is already dead", field_name);
                    }
                }
                _ => {}
            }
        }
    }
    
    /// 🔗 Global invalidation - call this when any object is dropped
    pub fn global_invalidate_weak_references(target_info: &str) {
        // In a real implementation, we'd maintain a global registry of all instances
        // and iterate through them to invalidate weak references.
        // For this demo, we'll add the capability to the instance itself.
        eprintln!("🔗 DEBUG: Global weak reference invalidation for: {}", target_info);
    }
    
    /// フィールドの値を取得
    pub fn get_field(&self, field_name: &str) -> Option<SharedNyashBox> {
        eprintln!("✅ FIX: get_field('{}') returning shared Arc reference", field_name);
        
        // 🔧 修正：v.clone_box() → Arc::clone(v) で参照共有
        self.fields.lock().unwrap().get(field_name).map(Arc::clone)
    }
    
    /// フィールドに値を設定
    pub fn set_field(&self, field_name: &str, value: SharedNyashBox) -> Result<(), String> {
        eprintln!("🔧 INSTANCE: set_field('{}') with shared Arc reference id={}", 
                 field_name, value.box_id());
        
        let mut fields = self.fields.lock().unwrap();
        if fields.contains_key(field_name) {
            if let Some(old_value) = fields.get(field_name) {
                eprintln!("🔧 INSTANCE: Replacing field '{}': old_id={} -> new_id={}", 
                         field_name, old_value.box_id(), value.box_id());
            }
            fields.insert(field_name.to_string(), value);
            Ok(())
        } else {
            Err(format!("Field '{}' does not exist in {}", field_name, self.class_name))
        }
    }
    
    /// 🌍 GlobalBox用：フィールドを動的に追加・設定
    pub fn set_field_dynamic(&mut self, field_name: String, value: SharedNyashBox) {
        let mut fields = self.fields.lock().unwrap();
        fields.insert(field_name, value);
    }
    
    /// メソッド定義を取得
    pub fn get_method(&self, method_name: &str) -> Option<&ASTNode> {
        self.methods.get(method_name)
    }
    
    /// メソッドが存在するかチェック
    pub fn has_method(&self, method_name: &str) -> bool {
        self.methods.contains_key(method_name)
    }
    
    /// 🌍 GlobalBox用：メソッドを動的に追加 - 🔥 暗黙オーバーライド禁止による安全実装
    pub fn add_method(&mut self, method_name: String, method_ast: ASTNode) -> Result<(), String> {
        // Arc<T>は不変なので、新しいHashMapを作成してArcで包む
        let mut new_methods = (*self.methods).clone();
        
        // 🚨 暗黙オーバーライド禁止：既存メソッドの検査
        if let Some(existing_method) = new_methods.get(&method_name) {
            // 新しいメソッドのoverride状態を確認
            let is_override = match &method_ast {
                crate::ast::ASTNode::FunctionDeclaration { is_override, .. } => *is_override,
                _ => false, // FunctionDeclaration以外はオーバーライドなし
            };
            
            if !is_override {
                // 🔥 明示的オーバーライド革命：overrideキーワードなしの重複を禁止
                return Err(format!(
                    "🚨 EXPLICIT OVERRIDE REQUIRED: Method '{}' already exists.\n\
                    💡 To replace the existing method, use 'override {}(...) {{ ... }}'.\n\
                    🌟 This is Nyash's explicit delegation philosophy - no hidden overrides!",
                    method_name, method_name
                ));
            }
            
            // override宣言があれば、明示的な置換として許可
            eprintln!("🔥 EXPLICIT OVERRIDE: Method '{}' replaced with override declaration", method_name);
        }
        
        new_methods.insert(method_name, method_ast);
        self.methods = Arc::new(new_methods);
        Ok(())
    }
    
    /// 🔥 Enhanced fini()メソッド - ChatGPT5設計による完全実装
    pub fn fini(&self) -> Result<(), String> {
        // 1) finalized チェック（idempotent）
        let mut finalized = self.finalized.lock().unwrap();
        if *finalized {
            // 既に解放済みなら何もしない
            return Ok(());
        }
        
        // 2) in_finalization = true（再入防止）
        let mut in_finalization = self.in_finalization.lock().unwrap();
        if *in_finalization {
            return Err("Circular finalization detected - fini() called recursively".to_string());
        }
        *in_finalization = true;
        
        // 3) TODO: ユーザー定義fini()実行（インタープリター側で実装予定）
        // このメソッドは低レベルなfini処理なので、高レベルなユーザー定義fini()は
        // インタープリター側で先に呼び出される想定
        
        // 4) 自動カスケード: init_field_order の強参照フィールドに child.fini()
        self.cascade_finalize_fields()?;
        
        // 5) 全フィールドクリア + finalized = true
        let mut fields = self.fields.lock().unwrap();
        fields.clear();
        let mut fields_ng = self.fields_ng.lock().unwrap();
        fields_ng.clear();
        
        *finalized = true;
        *in_finalization = false; // 再入フラグをクリア
        
        eprintln!("🔥 fini(): Instance {} (ID: {}) finalized", self.class_name, self.base.id);
        Ok(())
    }
    
    /// 🔥 自動カスケード解放 - init宣言順でフィールドをfini
    fn cascade_finalize_fields(&self) -> Result<(), String> {
        let fields_ng = self.fields_ng.lock().unwrap();
        
        // init_field_order の逆順でfiniを実行（LIFO - Last In First Out）
        for field_name in self.init_field_order.iter().rev() {
            // weak フィールドはスキップ
            if self.weak_fields_union.contains(field_name) {
                eprintln!("🔥 fini(): Skipping weak field '{}' (non-owning reference)", field_name);
                continue;
            }
            
            // フィールドの値を取得してfini呼び出し
            if let Some(field_value) = fields_ng.get(field_name) {
                match field_value {
                    crate::value::NyashValue::Box(arc_box) => {
                        if let Ok(inner_box) = arc_box.try_lock() {
                            // InstanceBoxならfini()を呼び出し
                            if let Some(instance) = inner_box.as_any().downcast_ref::<InstanceBox>() {
                                eprintln!("🔥 fini(): Cascading finalization to field '{}'", field_name);
                                if let Err(e) = instance.fini() {
                                    eprintln!("🔥 fini(): Warning - failed to finalize field '{}': {}", field_name, e);
                                    // エラーは警告として記録するが、続行する
                                }
                            }
                        }
                    }
                    _ => {
                        // non-Box値はfini不要
                    }
                }
            }
        }
        
        Ok(())
    }
    
    /// 解放済みかチェック
    pub fn is_finalized(&self) -> bool {
        *self.finalized.lock().unwrap()
    }
    
    /// 🔥 解放中かチェック
    pub fn is_in_finalization(&self) -> bool {
        *self.in_finalization.lock().unwrap()
    }
    
    /// 🔥 指定フィールドがweakかチェック
    pub fn is_weak_field(&self, field_name: &str) -> bool {
        self.weak_fields_union.contains(field_name)
    }
}

impl NyashBox for InstanceBox {
    fn to_string_box(&self) -> StringBox {
        StringBox::new(format!("<{} instance #{}>", self.class_name, self.base.id))
    }
    
    fn equals(&self, other: &dyn NyashBox) -> BoolBox {
        if let Some(other_instance) = other.as_any().downcast_ref::<InstanceBox>() {
            // 同じインスタンスIDなら等しい
            BoolBox::new(self.base.id == other_instance.base.id)
        } else {
            BoolBox::new(false)
        }
    }
    
    fn type_name(&self) -> &'static str {
        "InstanceBox"
    }
    
    fn clone_box(&self) -> Box<dyn NyashBox> {
        // インスタンスは同じフィールドを共有
        Box::new(self.clone())
    }
    
    
}

impl BoxCore for InstanceBox {
    fn box_id(&self) -> u64 {
        self.base.id
    }

    fn parent_type_id(&self) -> Option<std::any::TypeId> {
        self.base.parent_type_id
    }

    fn fmt_box(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "<{} instance #{}>", self.class_name, self.base.id)
    }
    
    fn as_any(&self) -> &dyn Any {
        self
    }
    
    fn as_any_mut(&mut self) -> &mut dyn Any {
        self
    }
}

impl Display for InstanceBox {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        self.fmt_box(f)
    }
}

// ===== Tests =====

#[cfg(test)]
mod tests {
    use super::*;
    use crate::box_trait::IntegerBox;
    
    #[test]
    fn test_instance_creation() {
        let fields = vec!["x".to_string(), "y".to_string()];
        let methods = HashMap::new();
        let instance = InstanceBox::new("Point".to_string(), fields, methods);
        
        assert_eq!(instance.class_name, "Point");
        assert!(instance.get_field("x").is_some());
        assert!(instance.get_field("y").is_some());
        assert!(instance.get_field("z").is_none());
    }
    
    #[test]
    fn test_field_access() {
        let fields = vec!["value".to_string()];
        let methods = HashMap::new();
        let instance = InstanceBox::new("TestBox".to_string(), fields, methods);
        
        // フィールドに値を設定
        let int_value = Box::new(IntegerBox::new(42)) as Box<dyn NyashBox>;
        instance.set_field("value", int_value).unwrap();
        
        // フィールドの値を取得
        let retrieved = instance.get_field("value").unwrap();
        let int_box = retrieved.as_any().downcast_ref::<IntegerBox>().unwrap();
        assert_eq!(int_box.value, 42);
    }
    
    #[test]
    fn test_instance_equality() {
        let instance1 = InstanceBox::new("Test".to_string(), vec![], HashMap::new());
        let instance2 = InstanceBox::new("Test".to_string(), vec![], HashMap::new());
        
        // 異なるインスタンスは等しくない
        assert!(!instance1.equals(&instance2).value);
        
        // 同じインスタンスは等しい
        assert!(instance1.equals(&instance1).value);
    }
}