upstream-univer/packages/docs-drawing-ui/src/plugin.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 { Dependency } from '@univerjs/core';
import type { IUniverDocsDrawingUIConfig } from './config/config';
import { DependentOn, IConfigService, Inject, Injector, merge, Plugin, UniverInstanceType } from '@univerjs/core';
import { UniverDocsPlugin } from '@univerjs/docs';
import { UniverDocsDrawingPlugin } from '@univerjs/docs-drawing';
import { UniverDocsUIPlugin } from '@univerjs/docs-ui';
import { UniverDrawingPlugin } from '@univerjs/drawing';
import { UniverDrawingUIPlugin } from '@univerjs/drawing-ui';
import { IRenderManagerService, UniverRenderEnginePlugin } from '@univerjs/engine-render';
import pkg from '../package.json';
import { defaultPluginConfig, DOCS_DRAWING_UI_PLUGIN_CONFIG_KEY, DOCS_DRAWING_UI_PLUGIN_NAME } from './config/config';
import { ComponentsController } from './controllers/components.controller';
import { DocDrawingAddRemoveController } from './controllers/doc-drawing-notification.controller';
import { DocDrawingPrintingController } from './controllers/doc-drawing-printing.controller';
import { DocDrawingTransformerController } from './controllers/doc-drawing-transformer-update.controller';
import { DocFloatDomController } from './controllers/doc-float-dom.controller';
import {
    DocDrawingTransformUpdateController,
} from './controllers/render-controllers/doc-drawing-transform-update.controller';
import {
    DocDrawingUpdateRenderController,
} from './controllers/render-controllers/doc-drawing-update.render-controller';
import { DocDrawingUIController } from './controllers/ui.controller';
import { DocDrawingPopupMenuController } from './menu/drawing-popup-menu.controller';
import { DocDrawingFloatingToolbarAdapterService } from './services/doc-drawing-floating-toolbar-adapter.service';
import { DocRefreshDrawingsService } from './services/doc-refresh-drawings.service';

@DependentOn(
    UniverDocsPlugin,
    UniverDrawingPlugin,
    UniverRenderEnginePlugin,
    UniverDocsDrawingPlugin,
    UniverDocsUIPlugin,
    UniverDrawingUIPlugin
)
export class UniverDocsDrawingUIPlugin extends Plugin {
    static override type = UniverInstanceType.UNIVER_DOC;
    static override pluginName = DOCS_DRAWING_UI_PLUGIN_NAME;
    static override packageName = pkg.name;
    static override version = pkg.version;

    constructor(
        private readonly _config: Partial<IUniverDocsDrawingUIConfig> = defaultPluginConfig,
        @Inject(Injector) protected _injector: Injector,
        @IRenderManagerService private readonly _renderManagerSrv: IRenderManagerService,
        @IConfigService private readonly _configService: IConfigService
    ) {
        super();

        // Manage the plugin configuration.
        const { ...rest } = merge(
            {},
            defaultPluginConfig,
            this._config
        );
        this._configService.setConfig(DOCS_DRAWING_UI_PLUGIN_CONFIG_KEY, rest);
    }

    override onStarting(): void {
        const dependencies: Dependency[] = [
            [ComponentsController],
            [DocDrawingUIController],
            [DocDrawingPopupMenuController],
            [DocDrawingTransformerController],
            [DocDrawingAddRemoveController],
            [DocRefreshDrawingsService],
            [DocDrawingFloatingToolbarAdapterService],
            [DocFloatDomController],
            [DocDrawingPrintingController],
        ];

        dependencies.forEach((dependency) => this._injector.add(dependency));
        this._injector.get(ComponentsController);
    }

    override onReady(): void {
        ([
            [DocDrawingUpdateRenderController],
            [DocDrawingTransformUpdateController],
        ] as Dependency[]).forEach((m) => this._renderManagerSrv.registerRenderModule(UniverInstanceType.UNIVER_DOC, m));

        this._injector.get(DocDrawingAddRemoveController);
        this._injector.get(DocDrawingUIController);
        this._injector.get(DocDrawingTransformerController);
        this._injector.get(DocDrawingPrintingController);
    }

    override onRendered(): void {
        this._injector.get(DocDrawingPopupMenuController);
        this._injector.get(DocFloatDomController);
    }
}