upstream-univer/packages/sheets-table-ui/src/controllers/sheet-table-component.controller.ts

/**
 * Copyright 2023-present DreamNum Co., Ltd.
 *
 * 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 { IDisposable, Nullable } from '@univerjs/core';
import { Disposable, IContextService, Inject } from '@univerjs/core';
import { SheetCanvasPopManagerService } from '@univerjs/sheets-ui';
import { IDialogService } from '@univerjs/ui';
import { distinctUntilChanged, startWith } from 'rxjs';
import { SHEETS_TABLE_FILTER_PANEL_OPENED_KEY, UNIVER_SHEET_TABLE_FILTER_PANEL_ID } from '../const';

export interface ITableFilterPanelInfo {
    unitId: string;
    subUnitId: string;
    tableId: string;
    column: number;
    row: number;
}
export class SheetsTableComponentController extends Disposable {
    private _popupDisposable?: Nullable<IDisposable>;
    private _currentTableFilterInfo: Nullable<ITableFilterPanelInfo> = null;
    constructor(
        @IContextService protected readonly _contextService: IContextService,
        @Inject(SheetCanvasPopManagerService) private _sheetCanvasPopupService: SheetCanvasPopManagerService,
        @Inject(IDialogService) protected readonly _dialogService: IDialogService

    ) {
        super();
        this._initUIPopup();
    }

    public setCurrentTableFilterInfo(info: ITableFilterPanelInfo): void {
        this._currentTableFilterInfo = info;
    }

    public openOrToggleFilterPanel(info: ITableFilterPanelInfo): void {
        const opened = this._contextService.getContextValue(SHEETS_TABLE_FILTER_PANEL_OPENED_KEY);

        if (opened && this._isSameFilterPanelInfo(this._currentTableFilterInfo, info)) {
            this.closeFilterPanel();
            return;
        }

        this.setCurrentTableFilterInfo(info);

        if (opened) {
            this._popupDisposable?.dispose();
            this._popupDisposable = null;
            this._openFilterPopup();
            return;
        }

        this._contextService.setContextValue(SHEETS_TABLE_FILTER_PANEL_OPENED_KEY, true);
    }

    public clearCurrentTableFilterInfo(): void {
        this._currentTableFilterInfo = null;
    }

    public getCurrentTableFilterInfo(): Nullable<ITableFilterPanelInfo> {
        return this._currentTableFilterInfo;
    }

    private _initUIPopup() {
        this.disposeWithMe(this._contextService.subscribeContextValue$(SHEETS_TABLE_FILTER_PANEL_OPENED_KEY)

            .pipe(startWith(undefined), distinctUntilChanged())
            .subscribe((open) => {
                if (open) {
                    this._openFilterPopup();
                } else if (open === false) {
                    this._closeFilterPopup();
                }
            }));
    }

    public closeFilterPanel(): void {
        this._contextService.setContextValue(SHEETS_TABLE_FILTER_PANEL_OPENED_KEY, false);
    }

    protected _openFilterPopup(): void {
        const currentFilterModel = this._currentTableFilterInfo;
        if (!currentFilterModel) {
            throw new Error('[SheetsFilterUIController]: no filter model when opening filter popup!');
        }

        const { row: startRow, column: col } = currentFilterModel;
        this._popupDisposable = this._sheetCanvasPopupService.attachPopupToCell(startRow, col, {
            componentKey: SHEETS_TABLE_FILTER_PANEL_OPENED_KEY,
            direction: 'horizontal',
            onClickOutside: () => {
                this._dialogService.close(UNIVER_SHEET_TABLE_FILTER_PANEL_ID);
                this._contextService.setContextValue(SHEETS_TABLE_FILTER_PANEL_OPENED_KEY, false);
            },
            offset: [5, 0],
            portal: true,
        });
    }

    private _closeFilterPopup(): void {
        this._dialogService.close(UNIVER_SHEET_TABLE_FILTER_PANEL_ID);
        this._popupDisposable?.dispose();
        this._popupDisposable = null;
        this.clearCurrentTableFilterInfo();
    }

    private _isSameFilterPanelInfo(a: Nullable<ITableFilterPanelInfo>, b: ITableFilterPanelInfo): boolean {
        return Boolean(a && a.unitId === b.unitId && a.subUnitId === b.subUnitId && a.tableId === b.tableId && a.column === b.column && a.row === b.row);
    }
}