tanqiu/assets/scripts/CombatSystem/WeaponBullet.ts

import { _decorator, Component, Node, Vec2, Vec3, RigidBody2D, Collider2D, Contact2DType, IPhysics2DContact, find, Prefab, instantiate, UITransform, resources, sp, JsonAsset } from 'cc';
import { BulletCount, BulletCountConfig, BulletSpawnInfo } from './BulletEffects/BulletCount';
import { BulletTrajectory, BulletTrajectoryConfig } from './BulletEffects/BulletTrajectory';
import { BulletHitEffect, HitEffectConfig, HitResult } from './BulletEffects/BulletHitEffect';
import { BulletLifecycle, BulletLifecycleConfig } from './BulletEffects/BulletLifecycle';

const { ccclass, property } = _decorator;

/**
 * WeaponBullet - 重构后的统一子弹控制器
 * 
 * 整合四个子模块：
 * 1. BulletCount - 子弹数量控制
 * 2. BulletTrajectory - 弹道控制
 * 3. BulletHitEffect - 命中效果（可叠加）
 * 4. BulletLifecycle - 生命周期控制
 * 
 * 从weapons.json读取配置并分发给各模块
 */

export interface WeaponConfig {
    id: string;
    name: string;
    bulletConfig: {
        count: BulletCountConfig;
        trajectory: BulletTrajectoryConfig;
        hitEffects: HitEffectConfig[];
        lifecycle: BulletLifecycleConfig;
        visual: {
            bulletPrefab: string;
            hitEffect: string;
            trailEffect?: string;
            muzzleFlash: string;
        };
    };
    stats: {
        damage: number;
        fireRate: number;
        range: number;
        bulletSpeed: number;
        accuracy: number;
    };
}

export interface BulletInitData {
    weaponId: string;           // 武器ID，用于查找配置
    firePosition: Vec3;         // 发射位置
    direction?: Vec3;           // 发射方向（可选，自动瞄准时会自动计算）
    autoTarget?: boolean;       // 是否自动瞄准
    weaponConfig?: WeaponConfig; // 直接传入的武器配置（优先级更高）
}

@ccclass('WeaponBullet')
export class WeaponBullet extends Component {
    // === 子模块组件 ===
    private bulletTrajectory: BulletTrajectory = null;
    private bulletHitEffect: BulletHitEffect = null;
    private bulletLifecycle: BulletLifecycle = null;
    
    // === 配置数据 ===
    private weaponConfig: WeaponConfig = null;
    private isInitialized: boolean = false;
    
    // === 静态武器配置缓存 ===
    private static weaponsData: any = null;
    
    /**
     * 加载武器配置数据
     */
    public static loadWeaponsData(): Promise<void> {
        return new Promise((resolve, reject) => {
            if (WeaponBullet.weaponsData) {
                resolve();
                return;
            }
            
            resources.load('data/weapons', JsonAsset, (err, jsonAsset: JsonAsset) => {
                if (err) {
                    console.error('❌ 加载武器配置失败:', err);
                    reject(err);
                    return;
                }
                
                WeaponBullet.weaponsData = jsonAsset.json;
                console.log('✅ 武器配置加载成功');
                resolve();
            });
        });
    }
    
    /**
     * 根据武器ID获取配置
     */
    public static getWeaponConfig(weaponId: string): WeaponConfig | null {
        if (!WeaponBullet.weaponsData) {
            console.error('❌ 武器配置未加载');
            return null;
        }
        
        const weapons = WeaponBullet.weaponsData.weapons;
        const weapon = weapons.find((w: any) => w.id === weaponId);
        
        if (!weapon) {
            console.error(`❌ 未找到武器配置: ${weaponId}`);
            return null;
        }
        
        return weapon as WeaponConfig;
    }
    
    /**
     * 创建多发子弹
     */
    public static createBullets(initData: BulletInitData, bulletPrefab: Node): Node[] {
        // 获取武器配置
        const config = initData.weaponConfig || WeaponBullet.getWeaponConfig(initData.weaponId);
        if (!config) {
            console.error('❌ 无法获取武器配置');
            return [];
        }
        
        // 计算子弹生成信息
        const direction = initData.direction || new Vec3(1, 0, 0);
        const spawnInfos = BulletCount.calculateBulletSpawns(
            config.bulletConfig.count,
            initData.firePosition,
            direction
        );
        
        const bullets: Node[] = [];
        
        // 为每个子弹创建实例
        for (const spawnInfo of spawnInfos) {
            const createBullet = () => {
                const bullet = instantiate(bulletPrefab);
                const weaponBullet = bullet.getComponent(WeaponBullet) || bullet.addComponent(WeaponBullet);
                
                // 初始化子弹
                weaponBullet.init({
                    ...initData,
                    firePosition: spawnInfo.position,
                    direction: spawnInfo.direction,
                    weaponConfig: config
                });
                
                bullets.push(bullet);
            };
            
            // 处理延迟发射
            if (spawnInfo.delay > 0) {
                // 这里需要一个全局的调度器来处理延迟，暂时直接创建
                setTimeout(createBullet, spawnInfo.delay * 1000);
            } else {
                createBullet();
            }
        }
        
        return bullets;
    }
    
    /**
     * 初始化子弹
     */
    public init(initData: BulletInitData) {
        console.log('🚀 WeaponBullet 初始化开始...');
        
        // 获取武器配置
        this.weaponConfig = initData.weaponConfig || WeaponBullet.getWeaponConfig(initData.weaponId);
        if (!this.weaponConfig) {
            console.error('❌ 无法获取武器配置，初始化失败');
            return;
        }
        
        console.log(`🔫 初始化武器: ${this.weaponConfig.name}`);
        
        // 设置位置
        this.setPositionInGameArea(initData.firePosition);
        
        // 延迟初始化物理组件，确保位置设置完成
        this.scheduleOnce(() => {
            this.initializeComponents(initData);
        }, 0.05);
    }
    
    /**
     * 在GameArea中设置位置
     */
    private setPositionInGameArea(worldPos: Vec3) {
        const gameArea = find('Canvas/GameLevelUI/GameArea');
        if (gameArea) {
            const gameAreaTransform = gameArea.getComponent(UITransform);
            if (gameAreaTransform) {
                const localPos = gameAreaTransform.convertToNodeSpaceAR(worldPos);
                this.node.position = localPos;
                console.log('✅ 子弹位置已设置:', localPos);
            }
        }
    }
    
    /**
     * 初始化各个组件
     */
    private initializeComponents(initData: BulletInitData) {
        const config = this.weaponConfig.bulletConfig;
        
        // 确保物理组件存在
        this.setupPhysics();
        
        // 初始化弹道组件
        this.bulletTrajectory = this.getComponent(BulletTrajectory) || this.addComponent(BulletTrajectory);
        const trajCfg: BulletTrajectoryConfig = { ...config.trajectory, speed: this.weaponConfig.stats.bulletSpeed };
        this.bulletTrajectory.init(
            trajCfg,
            initData.direction || this.calculateDirection(initData.autoTarget),
            initData.firePosition
        );
        
        // 初始化命中效果组件
        this.bulletHitEffect = this.getComponent(BulletHitEffect) || this.addComponent(BulletHitEffect);
        this.bulletHitEffect.init(config.hitEffects);
        
        // 传递默认特效路径
        if (this.bulletHitEffect && config.visual) {
            (this.bulletHitEffect as any).setDefaultEffects?.(config.visual.hitEffect, config.visual.trailEffect, (config.visual as any).burnEffect);
        }
        
        // 初始化生命周期组件
        this.bulletLifecycle = this.getComponent(BulletLifecycle) || this.addComponent(BulletLifecycle);
        this.bulletLifecycle.init(config.lifecycle, initData.firePosition);
        
        // 设置碰撞监听
        this.setupCollisionListener();
        
        this.isInitialized = true;
        console.log('✅ WeaponBullet 初始化完成');
    }
    
    /**
     * 设置物理组件
     */
    private setupPhysics() {
        const rigidBody = this.getComponent(RigidBody2D);
        if (rigidBody) {
            rigidBody.enabledContactListener = true;
            rigidBody.gravityScale = 0; // 由弹道组件控制重力
            rigidBody.linearDamping = 0;
            rigidBody.angularDamping = 0;
            rigidBody.allowSleep = false;
        }
        
        const collider = this.getComponent(Collider2D);
        if (collider) {
            collider.sensor = false;
            collider.on(Contact2DType.BEGIN_CONTACT, this.onHit, this);
        }
        
        console.log('✅ 物理组件设置完成');
    }
    
    /**
     * 计算子弹方向
     */
    private calculateDirection(autoTarget: boolean = false): Vec3 {
        if (!autoTarget) {
            // 随机方向
            const angle = Math.random() * Math.PI * 2;
            return new Vec3(Math.cos(angle), Math.sin(angle), 0);
        }
        
        // 寻找最近敌人
        const nearestEnemy = this.findNearestEnemy();
        if (nearestEnemy) {
            const direction = nearestEnemy.worldPosition.clone()
                .subtract(this.node.worldPosition)
                .normalize();
            console.log('🎯 自动瞄准最近敌人');
            return direction;
        }
        
        // 默认向右
        return new Vec3(1, 0, 0);
    }
    
    /**
     * 寻找最近敌人
     */
    private findNearestEnemy(): Node | null {
        const enemyContainer = find('Canvas/GameLevelUI/enemyContainer');
        if (!enemyContainer) return null;
        
        const enemies = enemyContainer.children.filter(child => 
            child.active && this.isEnemyNode(child)
        );
        
        if (enemies.length === 0) return null;
        
        let nearestEnemy: Node = null;
        let nearestDistance = Infinity;
        const bulletPos = this.node.worldPosition;
        
        for (const enemy of enemies) {
            const distance = Vec3.distance(bulletPos, enemy.worldPosition);
            if (distance < nearestDistance) {
                nearestDistance = distance;
                nearestEnemy = enemy;
            }
        }
        
        return nearestEnemy;
    }
    
    /**
     * 判断是否为敌人节点
     */
    private isEnemyNode(node: Node): boolean {
        const name = node.name.toLowerCase();
        return name.includes('enemy') || 
               name.includes('敌人') ||
               node.getComponent('EnemyInstance') !== null;
    }
    
    /**
     * 设置碰撞监听
     */
    private setupCollisionListener() {
        // 碰撞监听已在setupPhysics中设置
    }
    
    /**
     * 碰撞处理
     */
    private onHit(selfCollider: Collider2D, otherCollider: Collider2D, contact: IPhysics2DContact | null) {
        if (!this.isInitialized) return;
        
        const otherNode = otherCollider.node;
        console.log(`💥 子弹碰撞: ${otherNode.name}`);
        
        // 获取碰撞世界坐标
        let contactWorldPos: Vec3 = null;
        if (contact && (contact as any).getWorldManifold) {
            const wm = (contact as any).getWorldManifold();
            if (wm && wm.points && wm.points.length > 0) {
                contactWorldPos = new Vec3(wm.points[0].x, wm.points[0].y, 0);
            }
        }
        if (!contactWorldPos) {
            contactWorldPos = otherNode.worldPosition.clone();
        }
        
        // 处理命中效果
        const hitResult: HitResult = this.bulletHitEffect.processHit(otherNode, contactWorldPos);
        
        // 通知生命周期组件发生命中（可能影响内部计数或阶段）
        if (this.bulletLifecycle) {
            this.bulletLifecycle.onHit(otherNode);
        }
        // 不直接销毁，交由 BulletLifecycle 自行处理
        // 生命周期和命中效果自行决定是否销毁；此处不再直接销毁，由 BulletLifecycle.update() 完成
        if (hitResult.shouldRicochet) {
            console.log('🎾 子弹发生弹射');
        } else if (hitResult.shouldContinue) {
            console.log('🏹 子弹继续飞行');
        }
    }
    
    update(dt: number) {
        // 生命周期检查由BulletLifecycle组件处理
        // 弹道更新由BulletTrajectory组件处理
        // 不需要在这里做额外处理
    }
    
    /**
     * 获取武器配置
     */
    public getWeaponConfig(): WeaponConfig {
        return this.weaponConfig;
    }
    
    /**
     * 获取子弹状态信息
     */
    public getBulletStatus() {
        return {
            isInitialized: this.isInitialized,
            weaponName: this.weaponConfig?.name,
            trajectoryState: this.bulletTrajectory?.getState(),
            lifecycleState: this.bulletLifecycle?.getState(),
            hitStats: this.bulletHitEffect?.getHitStats()
        };
    }
    
    /**
     * 强制销毁子弹
     */
    public forceDestroy() {
        if (this.bulletLifecycle) {
            this.bulletLifecycle.forceDestroy();
        }
    }
    
    /**
     * 验证初始化数据
     */
    public static validateInitData(initData: BulletInitData): boolean {
        if (!initData) return false;
        if (!initData.weaponId && !initData.weaponConfig) return false;
        if (!initData.firePosition) return false;
        
        return true;
    }
    
    onDestroy() {
        console.log('💀 WeaponBullet 被销毁');
        
        // 清理事件监听
        const collider = this.getComponent(Collider2D);
        if (collider) {
            collider.off(Contact2DType.BEGIN_CONTACT, this.onHit, this);
        }
    }
}