tango/packages/context/src/context.ts

import type { Engine } from '@music163/tango-core';
import { createContext } from '@music163/tango-helpers';

export interface ITangoEngineContext {
  /**
   * 低代码引擎
   */
  engine: Engine;
}

const [TangoEngineProvider, useTangoEngine] = createContext<ITangoEngineContext>({
  name: 'TangoEngineContext',
});

export { TangoEngineProvider };

export const useWorkspace = () => {
  return useTangoEngine()?.engine.workspace;
};

export const useDesigner = () => {
  return useTangoEngine()?.engine.designer;
};

const builtinHelpers = [
  {
    title: 'setStoreValue',
    key: '() => tango.setStoreValue("variableName", "variableValue")',
    type: 'function',
  },
  {
    title: 'getStoreValue',
    key: '() => tango.getStoreValue("variableName")',
    type: 'function',
  },
  { title: 'openModal', key: '() => tango.openModal("")', type: 'function' },
  { title: 'closeModal', key: '() => tango.closeModal("")', type: 'function' },
  { title: 'navigateTo', key: '() => tango.navigateTo("/")', type: 'function' },
  { title: 'showToast', key: '() => tango.showToast("hello")', type: 'function' },
  { title: 'formatDate', key: '() => tango.formatDate("2022-12-12")', type: 'function' },
  { title: 'formatNumber', key: '() => tango.formatDate(9999)', type: 'function' },
  {
    title: 'copyToClipboard',
    key: '() => tango.copyToClipboard("hello")',
    type: 'function',
  },
];

export const useWorkspaceData = () => {
  const workspace = useWorkspace();
  const modelVariables: any[] = []; // 绑定变量列表
  const storeActionVariables: any[] = []; // 模型中的所有 actions
  const storeVariables: any[] = []; // 模型中的所有变量
  const serviceVariables: any[] = []; // 服务中的所有变量

  Object.values(workspace.storeModules).forEach((file) => {
    const prefix = `stores.${file.name}`;
    const states = file.states.map((item) => ({
      title: item.name,
      key: `${prefix}.${item.name}`,
      raw: item.code,
    }));
    const actions = file.actions.map((item) => ({
      title: item.name,
      key: `${prefix}.${item.name}`,
      type: 'function',
      raw: item.code,
    }));

    modelVariables.push({
      title: file.name,
      key: prefix,
      selectable: false,
      children: states,
    });

    storeActionVariables.push({
      title: file.name,
      key: prefix,
      selectable: false,
      children: actions,
    });

    storeVariables.push({
      title: file.name,
      key: prefix,
      selectable: false,
      children: [...states, ...actions],
      showAddChildIcon: true,
      showRemoveIcon: true,
    });
  });

  Object.values(workspace.serviceModules).forEach((file) => {
    const prefix = file.name !== 'index' ? `services.${file.name}` : 'services';
    serviceVariables.push({
      title: file.name,
      key: prefix,
      selectable: false,
      children: Object.keys(file.serviceFunctions || {}).map((key) => ({
        title: key,
        key: [prefix, key].join('.'),
        type: 'function',
      })),
    });
  });

  // 路由选项列表
  const routeOptions = workspace.pages?.map((item) => ({
    label: `${item.name}  (${item.path})`,
    value: item.path,
  }));

  return {
    modelVariables: [buildVariableOptions('数据模型', 'stores', modelVariables)],
    actionVariables: [
      buildVariableOptions('数据模型', 'stores', storeActionVariables),
      buildVariableOptions('服务函数', 'services', serviceVariables),
      buildVariableOptions('工具函数', 'helpers', builtinHelpers),
    ],
    storeVariables,
    serviceVariables,
    expressionVariables: [
      buildVariableOptions('数据模型', 'stores', storeVariables),
      buildVariableOptions('服务函数', 'services', serviceVariables),
      buildVariableOptions('工具函数', 'helpers', builtinHelpers),
    ],
    routeOptions,
  };
};

function buildVariableOptions(title: string, key: string, children: any[]) {
  return {
    key,
    title,
    children,
  };
}