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

packages/sheets-drawing-ui/src/controllers/sheet-cell-image-copy-paste.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 { DocumentDataModel, IDocDrawingBase } from '@univerjs/core';
+import type { IInnerPasteCommandParams } from '@univerjs/docs-ui';
+import type { LocaleKey } from '../locale/types';
+import {
+    BuildTextUtils,
+    createDocumentModelWithStyle,
+    Disposable,
+    DOCS_FORMULA_BAR_EDITOR_UNIT_ID_KEY,
+    DOCS_NORMAL_EDITOR_UNIT_ID_KEY,
+    ICommandService,
+    Inject,
+    IUniverInstanceService,
+    LocaleService,
+    UniverInstanceType,
+} from '@univerjs/core';
+import { InnerPasteCommand } from '@univerjs/docs-ui';
+import { getCurrentTypeOfRenderer, IRenderManagerService } from '@univerjs/engine-render';
+import { EditingRenderController, SetCellEditVisibleOperation } from '@univerjs/sheets-ui';
+import { IDialogService } from '@univerjs/ui';
+
+const DISABLE_UNITS = [
+    DOCS_NORMAL_EDITOR_UNIT_ID_KEY,
+    DOCS_FORMULA_BAR_EDITOR_UNIT_ID_KEY,
+];
+export class SheetCellImageCopyPasteController extends Disposable {
+    constructor(
+        @ICommandService private readonly _commandService: ICommandService,
+        @IUniverInstanceService private readonly _univerInstanceService: IUniverInstanceService,
+        @IDialogService private readonly _dialogService: IDialogService,
+        @IRenderManagerService private readonly _renderManagerService: IRenderManagerService,
+        @Inject(LocaleService) private readonly _localeService: LocaleService
+    ) {
+        super();
+        this._initDocImageCopyPasteHooks();
+    }
+
+    private _setCellImage(drwaing: IDocDrawingBase) {
+        const docDataModel = createDocumentModelWithStyle('', {});
+        const editingRenderController = getCurrentTypeOfRenderer(UniverInstanceType.UNIVER_SHEET, this._univerInstanceService, this._renderManagerService)?.with(EditingRenderController);
+        const jsonXActions = BuildTextUtils.drawing.add({
+            documentDataModel: docDataModel,
+            drawings: [drwaing],
+            selection: {
+                collapsed: true,
+                startOffset: 0,
+                endOffset: 0,
+            },
+        });
+        if (jsonXActions) {
+            docDataModel.apply(jsonXActions);
+            if (editingRenderController) {
+                editingRenderController.submitCellData(docDataModel);
+            }
+        }
+    }
+
+    private _initDocImageCopyPasteHooks() {
+        this.disposeWithMe(
+            this._commandService.beforeCommandExecuted((commandInfo) => {
+                if (commandInfo.id === InnerPasteCommand.id) {
+                    const params = commandInfo.params as IInnerPasteCommandParams;
+                    const { doc } = params;
+                    const currentDoc = this._univerInstanceService.getCurrentUnitOfType<DocumentDataModel>(UniverInstanceType.UNIVER_DOC);
+                    if (currentDoc == null || !Object.keys(doc.drawings ?? {}).length) {
+                        return;
+                    }
+                    const docUnitId = currentDoc.getUnitId();
+                    if (DISABLE_UNITS.includes(docUnitId)) {
+                        const handleCloseDialog = () => {
+                            this._dialogService.close('sheet-cell-image-copy-paste');
+                            this._commandService.syncExecuteCommand(SetCellEditVisibleOperation.id, {
+                                visible: false,
+                            });
+                        };
+                        // empty
+                        if (currentDoc.getBody()?.dataStream === '\r\n') {
+                            this._commandService.syncExecuteCommand(SetCellEditVisibleOperation.id, {
+                                visible: false,
+                            });
+                            this._setCellImage(Object.values(doc.drawings!)[0]);
+                        } else {
+                            this._dialogService.open({
+                                id: 'sheet-cell-image-copy-paste',
+                                title: {
+                                    label: this._localeService.t<LocaleKey>('sheets-drawing-ui.cell-image.pasteTitle'),
+                                },
+                                children: {
+                                    label: this._localeService.t<LocaleKey>('sheets-drawing-ui.cell-image.pasteContent'),
+                                },
+                                width: 320,
+                                onClose: handleCloseDialog,
+                                showOk: true,
+                                showCancel: true,
+                                onOk: () => {
+                                    handleCloseDialog();
+                                    this._setCellImage(Object.values(doc.drawings!)[0]);
+                                },
+                                onCancel: handleCloseDialog,
+                            });
+                        }
+                    }
+                }
+            })
+        );
+    }
+}