76 lines
2.6 KiB
TypeScript
76 lines
2.6 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* 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 {
|
|
definePluginCreator,
|
|
type PluginCreator,
|
|
} from '@flowgram-adapter/free-layout-editor';
|
|
|
|
import { WorkflowEncapsulateContainerModule } from './workflow-encapsulate-container-module';
|
|
import { EncapsulateValidatorsContainerModule } from './validators';
|
|
import { EncapsulateValidateManager } from './validate';
|
|
import {
|
|
type GetGlobalStateOption,
|
|
type GetNodeTemplateOption,
|
|
type OnEncapsulateOption,
|
|
} from './types';
|
|
import { EncapsulateRenderContainerModule } from './render';
|
|
import { EncapsulateContext } from './encapsulate-context';
|
|
import { EncapsulateManager, EncapsulateService } from './encapsulate';
|
|
|
|
interface EncapsulatePluginOptions {
|
|
getNodeTemplate?: GetNodeTemplateOption;
|
|
getGlobalState?: GetGlobalStateOption;
|
|
onEncapsulate?: OnEncapsulateOption;
|
|
}
|
|
|
|
export const createWorkflowEncapsulatePlugin: PluginCreator<EncapsulatePluginOptions> =
|
|
definePluginCreator<EncapsulatePluginOptions>({
|
|
onInit(ctx, options) {
|
|
ctx.get<EncapsulateManager>(EncapsulateManager).init();
|
|
|
|
ctx.get<EncapsulateContext>(EncapsulateContext).setPluginContext(ctx);
|
|
|
|
if (options.getNodeTemplate) {
|
|
ctx
|
|
.get<EncapsulateContext>(EncapsulateContext)
|
|
.setGetNodeTemplate(options.getNodeTemplate);
|
|
}
|
|
|
|
if (options.getGlobalState) {
|
|
ctx
|
|
.get<EncapsulateContext>(EncapsulateContext)
|
|
.setGetGlobalState(options.getGlobalState);
|
|
}
|
|
|
|
if (options.onEncapsulate) {
|
|
ctx.get<EncapsulateService>(EncapsulateService).onEncapsulate(res => {
|
|
options?.onEncapsulate?.(res, ctx);
|
|
});
|
|
}
|
|
},
|
|
onDispose(ctx) {
|
|
ctx.get<EncapsulateValidateManager>(EncapsulateValidateManager).dispose();
|
|
ctx.get<EncapsulateService>(EncapsulateService).dispose();
|
|
ctx.get<EncapsulateManager>(EncapsulateManager).dispose();
|
|
},
|
|
containerModules: [
|
|
WorkflowEncapsulateContainerModule,
|
|
EncapsulateRenderContainerModule,
|
|
EncapsulateValidatorsContainerModule,
|
|
],
|
|
});
|