esengine/packages/editor-app/src/presentation/hooks/useConnectionOperations.ts

import { useCallback, useMemo } from 'react';
import { ConnectionType } from '../../domain/models/Connection';
import { IValidator } from '../../domain/interfaces/IValidator';
import { CommandManager } from '../../application/commands/CommandManager';
import { TreeStateAdapter } from '../../application/state/BehaviorTreeDataStore';
import { AddConnectionUseCase } from '../../application/use-cases/AddConnectionUseCase';
import { RemoveConnectionUseCase } from '../../application/use-cases/RemoveConnectionUseCase';

/**
 * 连接操作 Hook
 */
export function useConnectionOperations(
    validator: IValidator,
    commandManager: CommandManager
) {
    const treeState = useMemo(() => new TreeStateAdapter(), []);

    const addConnectionUseCase = useMemo(
        () => new AddConnectionUseCase(commandManager, treeState, validator),
        [commandManager, treeState, validator]
    );

    const removeConnectionUseCase = useMemo(
        () => new RemoveConnectionUseCase(commandManager, treeState),
        [commandManager, treeState]
    );

    const addConnection = useCallback((
        from: string,
        to: string,
        connectionType: ConnectionType = 'node',
        fromProperty?: string,
        toProperty?: string
    ) => {
        try {
            return addConnectionUseCase.execute(from, to, connectionType, fromProperty, toProperty);
        } catch (error) {
            console.error('添加连接失败:', error);
            throw error;
        }
    }, [addConnectionUseCase]);

    const removeConnection = useCallback((
        from: string,
        to: string,
        fromProperty?: string,
        toProperty?: string
    ) => {
        try {
            removeConnectionUseCase.execute(from, to, fromProperty, toProperty);
        } catch (error) {
            console.error('移除连接失败:', error);
            throw error;
        }
    }, [removeConnectionUseCase]);

    return useMemo(() => ({
        addConnection,
        removeConnection
    }), [addConnection, removeConnection]);
}