@ibiz-template/runtime 0.7.41-alpha.128 → 0.7.41-alpha.130

package/src/register/helper/tree-node-register.ts

@@ -0,0 +1,78 @@
+import { IDETree, IDETreeNode } from '@ibiz/model-core';
+import { ITreeNodeProvider } from '../../interface';
+import { CustomRegister } from '../custom-register';
+import { getPluginRegisterKey } from './common-register';
+
+/** 树节点适配器前缀 */
+export const TREE_NODE_PROVIDER_PREFIX = 'TREE_NODE';
+
+/**
+ * @description 注册树节点适配器
+ * @export
+ * @param {string} key
+ * @param {() => ITreeNodeProvider} callback
+ */
+export function registerTreeNodeProvider(
+  key: string,
+  callback: () => ITreeNodeProvider,
+): void {
+  ibiz.register.register(`${TREE_NODE_PROVIDER_PREFIX}_${key}`, callback);
+}
+
+function getProvider(key: string): ITreeNodeProvider | undefined {
+  return ibiz.register.get(
+    `${TREE_NODE_PROVIDER_PREFIX}_${key}`,
+  ) as ITreeNodeProvider;
+}
+
+/**
+ * @description 获取树节点适配器
+ * @export
+ * @param {IDETreeNode} model
+ * @param {IDETree} tree
+ * @returns {*}  {(Promise<ITreeNodeProvider | undefined>)}
+ */
+export async function getTreeNodeProvider(
+  model: IDETreeNode,
+  tree: IDETree,
+): Promise<ITreeNodeProvider | undefined> {
+  let provider: ITreeNodeProvider | undefined;
+  const { sysPFPluginId, appId, treeNodeType } = model as Required<IDETreeNode>;
+
+  // 找自定义注册的适配器
+  const registerKey = CustomRegister.getRegisterKey(TREE_NODE_PROVIDER_PREFIX, {
+    mainModel: model,
+    control: tree,
+  });
+  provider = getProvider(registerKey);
+  if (!provider) {
+    ibiz.log.debug(
+      ibiz.i18n.t('runtime.register.helper.treeNodeCustomRegistration', {
+        registerKey,
+      }),
+    );
+  } else {
+    return provider;
+  }
+
+  // 找插件适配器
+  if (sysPFPluginId) {
+    const pluginKey = await getPluginRegisterKey(sysPFPluginId, appId);
+    if (pluginKey) {
+      provider = getProvider(pluginKey);
+    }
+    if (!provider) {
+      ibiz.log.warn(
+        ibiz.i18n.t('runtime.register.helper.treeNodePlugin', {
+          pluginKey,
+        }),
+      );
+    } else {
+      return provider;
+    }
+  }
+
+  // 找树节点类型适配器
+  provider = getProvider(treeNodeType);
+  return provider;
+}