import { RowModifyInfo } from '@/lib/types/api';
import { RowSelectionOperator } from '@/lib/operator/RowSelectionOperator';
import { StructureCalculator } from '@/lib/calculate/StructureCalculator';

export const useDataOperator = (
    rowSelectionOperator: RowSelectionOperator,
    structureCalculator: StructureCalculator,
) => {
    return {
        modifyRows: (modifyRows: RowModifyInfo[]) => structureCalculator.modifyRows(modifyRows),
        removeRows: (rowIds: string[]) => structureCalculator.removeRows(rowIds),
        getSelectionRows: () => rowSelectionOperator.getSelectedRowData(),
        getTreeSelectionRows: () => rowSelectionOperator.getTreeSelectedRowData(),
        getData: () => structureCalculator.getData()
    };
};