upstream-univer/packages/slides-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 { SlideDataModel } from '@univerjs/slides';
import type { IUniverSlidesUIConfig } from './config/config';
import {
    DependentOn,
    IConfigService,
    Inject,
    Injector,
    IUniverInstanceService,
    merge,
    mergeOverrideWithDependencies,
    Plugin,
    UniverInstanceType,
} from '@univerjs/core';
import { UniverDocsPlugin } from '@univerjs/docs';
import { UniverDocsUIPlugin } from '@univerjs/docs-ui';
import { UniverDrawingPlugin } from '@univerjs/drawing';
import { IRenderManagerService, UniverRenderEnginePlugin } from '@univerjs/engine-render';
import { UniverSlidesPlugin } from '@univerjs/slides';
import pkg from '../package.json';
import { defaultPluginConfig, SLIDES_UI_PLUGIN_CONFIG_KEY } from './config/config';
import { CanvasView } from './controllers/canvas-view';
import { ComponentsController } from './controllers/components.controller';
import { SlideEditingRenderController } from './controllers/slide-editing.render-controller';
import { SlideEditorBridgeRenderController } from './controllers/slide-editor-bridge.render-controller';
import { SlideRenderController } from './controllers/slide.render-controller';
import { SlidesUIController } from './controllers/ui.controller';
import { SlidePopupMenuController } from './menu/popup-menu.controller';
import { ISlideEditorBridgeService, SlideEditorBridgeService } from './services/slide-editor-bridge.service';
import { ISlideEditorManagerService, SlideEditorManagerService } from './services/slide-editor-manager.service';
import { SlideCanvasPopMangerService } from './services/slide-popup-manager.service';
import { SlideRenderService } from './services/slide-render.service';

@DependentOn(
    UniverDocsPlugin,
    UniverDrawingPlugin,
    UniverRenderEnginePlugin,
    UniverSlidesPlugin,
    UniverDocsUIPlugin
)
export class UniverSlidesUIPlugin extends Plugin {
    static override pluginName = 'UNIVER_SLIDES_UI_PLUGIN';
    static override packageName = pkg.name;
    static override version = pkg.version;
    static override type = UniverInstanceType.UNIVER_SLIDE;

    constructor(
        private readonly _config: Partial<IUniverSlidesUIConfig> = defaultPluginConfig,
        @Inject(Injector) override readonly _injector: Injector,
        @IRenderManagerService private readonly _renderManagerService: IRenderManagerService,
        @IUniverInstanceService private readonly _univerInstanceService: IUniverInstanceService,
        @IConfigService private readonly _configService: IConfigService
    ) {
        super();

        // Manage the plugin configuration.
        const { menu, ...rest } = merge(
            {},
            defaultPluginConfig,
            this._config
        );
        if (menu) {
            this._configService.setConfig('menu', menu, { merge: true });
        }
        this._configService.setConfig(SLIDES_UI_PLUGIN_CONFIG_KEY, rest);
    }

    override onStarting(): void {
        this._injector.add([ComponentsController]);
        this._injector.get(ComponentsController);
        mergeOverrideWithDependencies([
            [SlideRenderService],
            [ISlideEditorBridgeService, { useClass: SlideEditorBridgeService }],
            // used by SlideUIController --> EditorContainer
            [ISlideEditorManagerService, { useClass: SlideEditorManagerService }],
            [SlideCanvasPopMangerService],
        ] as Dependency[], this._config.override)
            .forEach((d) => this._injector.add(d));
    }

    override onReady(): void {
        ([
            // SlideRenderService will be init in ready stage, and then calling RenderManagerService@createRender --> init all deps in this rendering register block.

            [SlideRenderController],
        ] as Dependency[]).forEach((m) => {
            this.disposeWithMe(this._renderManagerService.registerRenderModule(UniverInstanceType.UNIVER_SLIDE, m));
        });

        mergeOverrideWithDependencies([
            [CanvasView],
            // cannot register in _renderManagerService now.
            // [ISlideEditorBridgeService, { useClass: SlideEditorBridgeService }],
            // // used by SlideUIController --> EditorContainer
            // [ISlideEditorManagerService, { useClass: SlideEditorManagerService }],

            // SlidesUIController controller should be registered in Ready stage.
            // SlidesUIController controller would add a new RenderUnit (__INTERNAL_EDITOR__DOCS_NORMAL)
            [SlidesUIController], // editor service was create in renderManagerService
            [SlideRenderController],
            [SlidePopupMenuController],
        ] as Dependency[], this._config.override).forEach((m) => {
            this._injector.add(m);
        });

        this._injector.get(CanvasView);
        this._injector.get(SlideRenderService);
    }

    override onRendered(): void {
        ([
            // need slideEditorBridgeService
            // need TextSelectionRenderService which init by EditorContainer
            [SlideEditorBridgeRenderController],
            [SlideEditingRenderController],
        ] as Dependency[]).forEach((m) => {
            // find all renderMap and register module to each item in renderMap
            this.disposeWithMe(this._renderManagerService.registerRenderModule(UniverInstanceType.UNIVER_SLIDE, m));
        });

        this._markSlideAsFocused();

        this._injector.get(SlidesUIController);
    }

    override onSteady(): void {
        this._injector.get(SlidePopupMenuController);
    }

    private _markSlideAsFocused() {
        const currentService = this._univerInstanceService;
        try {
            const slideDataModel = currentService.getCurrentUnitOfType<SlideDataModel>(UniverInstanceType.UNIVER_SLIDE)!;
            currentService.focusUnit(slideDataModel.getUnitId());
        } catch (e) {
        }
    }
}