remove-row-col.mutation.ts

/**
 * Copyright 2023-present DreamNum Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import type { IColumnData, IMutation, IObjectArrayPrimitiveType, IRowData, Worksheet } from '@univerjs/core';
import { CommandType, concatMatrixArray, IUniverInstanceService, sliceMatrixArray, spliceArray } from '@univerjs/core';
import type { IAccessor } from '@wendellhu/redi';

import type {
    IInsertColMutationParams,
    IInsertRowMutationParams,
    IRemoveColMutationParams,
    IRemoveRowsMutationParams,
} from '../../basics/interfaces/mutation-interface';

export const RemoveRowsUndoMutationFactory = (
    params: IRemoveRowsMutationParams,
    worksheet: Worksheet
): IInsertRowMutationParams => {
    const manager = worksheet.getRowManager();
    const rowPrimitive = manager.getRowData();
    const rowWrapper = rowPrimitive;
    const rowInfo: IObjectArrayPrimitiveType<IRowData> = {};

    const range = params.range;
    const slice = sliceMatrixArray(range.startRow, range.endRow, rowWrapper);
    const _rowInfo = concatMatrixArray(rowInfo, slice);

    return {
        unitId: params.unitId,
        subUnitId: params.subUnitId,
        range: params.range,
        rowInfo: _rowInfo as IObjectArrayPrimitiveType<IRowData>,
    };
};

export const RemoveRowMutation: IMutation<IRemoveRowsMutationParams> = {
    id: 'sheet.mutation.remove-rows',

    type: CommandType.MUTATION,

    handler: (accessor, params) => {
        const univerInstanceService = accessor.get(IUniverInstanceService);
        const universheet = univerInstanceService.getUniverSheetInstance(params.unitId);

        if (universheet == null) {
            throw new Error('universheet is null error!');
        }

        const worksheet = universheet.getSheetBySheetId(params.subUnitId);
        if (!worksheet) return false;
        const range = params.range;
        const manager = worksheet.getRowManager();
        const rowPrimitive = manager.getRowData();

        const filterOutRows = [];
        for (let i = range.startRow; i <= range.endRow; i++) {
            if (worksheet.getRowFiltered(i)) {
                filterOutRows.push(i);
            }
        }

        const rowCount = range.endRow - range.startRow + 1;
        spliceArray(range.startRow, rowCount, rowPrimitive);

        // remove cells contents by directly mutating worksheetCellMatrix
        const cellMatrix = worksheet.getCellMatrix();
        cellMatrix.removeRows(range.startRow, rowCount);

        worksheet.setRowCount(worksheet.getRowCount() - rowCount);

        return true;
    },
};

export const RemoveColMutationFactory = (
    accessor: IAccessor,
    params: IRemoveColMutationParams
): IInsertColMutationParams => {
    const univerInstanceService = accessor.get(IUniverInstanceService);
    const universheet = univerInstanceService.getUniverSheetInstance(params.unitId);

    if (universheet == null) {
        throw new Error('universheet is null error!');
    }

    const worksheet = universheet.getSheetBySheetId(params.subUnitId);

    if (worksheet == null) {
        throw new Error('worksheet is null error!');
    }

    const manager = worksheet.getColumnManager();
    const columnPrimitive = manager.getColumnData();
    const columnWrapper = columnPrimitive;
    const colInfo: IObjectArrayPrimitiveType<IColumnData> = {};

    const range = params.range;

    const slice = sliceMatrixArray(range.startColumn, range.endColumn, columnWrapper);

    const _colInfo = concatMatrixArray(colInfo, slice);

    return {
        unitId: params.unitId,
        subUnitId: params.subUnitId,
        range: params.range,
        colInfo: _colInfo as IObjectArrayPrimitiveType<IColumnData>,
    };
};

export const RemoveColMutation: IMutation<IRemoveColMutationParams> = {
    id: 'sheet.mutation.remove-col',
    type: CommandType.MUTATION,
    handler: (accessor, params) => {
        const univerInstanceService = accessor.get(IUniverInstanceService);
        const universheet = univerInstanceService.getUniverSheetInstance(params.unitId);

        if (universheet == null) {
            throw new Error('universheet is null error!');
        }

        const worksheet = universheet.getSheetBySheetId(params.subUnitId);
        if (!worksheet) return false;

        const range = params.range;
        const manager = worksheet.getColumnManager();
        const colPrimitive = manager.getColumnData();
        const colCount = range.endColumn - range.startColumn + 1;
        spliceArray(range.startColumn, colCount, colPrimitive);
        worksheet.setColumnCount(worksheet.getColumnCount() - colCount);

        // remove cells contents by directly mutating worksheetCellMatrix
        const cellMatrix = worksheet.getCellMatrix();
        cellMatrix.removeColumns(range.startColumn, colCount);

        return true;
    },
};