snapshot of https://github.com/dream-num/univer.git (dev, 2026-09-16; blobs > 131072B omitted)

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

@@ -0,0 +1,122 @@
+/**
+ * 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);
+    }
+}