feat: manually mirror opencoze's code from bytedance
Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
17
frontend/packages/workflow/nodes/src/service/index.ts
Normal file
17
frontend/packages/workflow/nodes/src/service/index.ts
Normal file
@@ -0,0 +1,17 @@
|
||||
/*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
export * from './workflow-nodes-service';
|
||||
@@ -0,0 +1,122 @@
|
||||
/*
|
||||
* 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 { customAlphabet } from 'nanoid';
|
||||
import { inject, injectable } from 'inversify';
|
||||
import { FlowNodeFormData } from '@flowgram-adapter/free-layout-editor';
|
||||
import { EntityManager } from '@flowgram-adapter/free-layout-editor';
|
||||
import { WorkflowNodeEntity } from '@flowgram-adapter/free-layout-editor';
|
||||
import { Emitter } from '@flowgram-adapter/common';
|
||||
|
||||
import { type FormNodeMeta } from '../typings';
|
||||
import { DEFAULT_NODE_META_PATH } from '../constants';
|
||||
|
||||
@injectable()
|
||||
export class WorkflowNodesService {
|
||||
protected onNodesTitleChangeEmitter = new Emitter<void>();
|
||||
@inject(EntityManager) protected readonly entityManager: EntityManager;
|
||||
private nanoid = customAlphabet('1234567890', 5);
|
||||
/**
|
||||
* 节点标题变化
|
||||
*/
|
||||
readonly onNodesTitleChange = this.onNodesTitleChangeEmitter.event;
|
||||
|
||||
/**
|
||||
* 节点标题更新
|
||||
* @param node
|
||||
*/
|
||||
getNodeTitle(node: WorkflowNodeEntity): string {
|
||||
const formData = node.getData<FlowNodeFormData>(FlowNodeFormData);
|
||||
const nodeMeta = formData.formModel.getFormItemValueByPath<FormNodeMeta>(
|
||||
DEFAULT_NODE_META_PATH,
|
||||
);
|
||||
|
||||
return nodeMeta?.title || '';
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有节点
|
||||
*/
|
||||
getAllNodes(ignoreNode?: WorkflowNodeEntity): WorkflowNodeEntity[] {
|
||||
return this.entityManager
|
||||
.getEntities<WorkflowNodeEntity>(WorkflowNodeEntity)
|
||||
.filter(n => n.id !== 'root' && n !== ignoreNode);
|
||||
}
|
||||
/**
|
||||
* 获取所有节点的标题
|
||||
*/
|
||||
getAllTitles(ignoreNode?: WorkflowNodeEntity): string[] {
|
||||
return this.getAllNodes(ignoreNode).map(node => this.getNodeTitle(node));
|
||||
}
|
||||
/**
|
||||
* 获取开始节点
|
||||
*/
|
||||
getStartNode(): WorkflowNodeEntity {
|
||||
return this.entityManager
|
||||
.getEntities<WorkflowNodeEntity>(WorkflowNodeEntity)
|
||||
.find(node => node.isStart) as WorkflowNodeEntity;
|
||||
}
|
||||
/**
|
||||
* 触发节点标题更新事件
|
||||
*/
|
||||
fireNodesTitleChange(): void {
|
||||
this.onNodesTitleChangeEmitter.fire();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建不会重复的title
|
||||
*
|
||||
* abc_1 -> abc_2
|
||||
*/
|
||||
|
||||
createUniqTitle(
|
||||
titlePrefix: string,
|
||||
ignoreNode?: WorkflowNodeEntity | undefined,
|
||||
ignoreTitles?: string[],
|
||||
): string {
|
||||
const allTitles = this.getAllTitles(ignoreNode);
|
||||
if (ignoreTitles) {
|
||||
allTitles.push(...ignoreTitles);
|
||||
}
|
||||
|
||||
const allTitlesSet = new Set(allTitles);
|
||||
|
||||
let startIndex = 0;
|
||||
let newTitle = `${titlePrefix}`;
|
||||
|
||||
const matched = titlePrefix.match(/_([0-9]+)$/);
|
||||
if (matched) {
|
||||
startIndex = Number(matched[1]);
|
||||
titlePrefix = titlePrefix.slice(0, matched.index);
|
||||
}
|
||||
|
||||
while (allTitlesSet.has(newTitle)) {
|
||||
startIndex += 1;
|
||||
newTitle = `${titlePrefix}_${startIndex}`;
|
||||
}
|
||||
return newTitle;
|
||||
}
|
||||
|
||||
/** 创建唯一ID */
|
||||
createUniqID() {
|
||||
let id: string;
|
||||
do {
|
||||
// 防止 id 以 0 开头,后端会转成 int64 导致 0 丢失,从而一些 id 匹配不上
|
||||
id = `1${this.nanoid()}`;
|
||||
} while (this.entityManager.getEntityById(id));
|
||||
return id;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user