@mybricks/plugin-ai 0.0.1 → 0.0.2

package/src/agents/workspace-by-knowledges/providers/focus-info-provider.ts

@@ -0,0 +1,131 @@
+/**
+ * 聚焦信息 Provider
+ * 提供当前聚焦组件的树结构和描述信息
+ */
+
+import { DirectoryProvider, IKnowledgeNode, KnowledgeNodeType } from '../../knowledges/types';
+import { context } from '../../../context';
+import { FocusOutlineInfoManager } from '../utils';
+
+/**
+ * 聚焦信息
+ */
+interface FocusInfo {
+  pageId?: string;
+  comId?: string;
+  title?: string;
+  type?: 'page' | 'uiCom' | 'section' | 'logicCom';
+  focusArea?: {
+    selector: string;
+    title: string;
+  };
+  diagramId?: string;
+}
+
+/**
+ * 获取聚焦信息
+ */
+function getFocusInfo(): FocusInfo {
+  return (context.currentFocus || {}) as FocusInfo;
+}
+
+/**
+ * 创建聚焦信息 Provider
+ * 将当前聚焦的页面和组件信息作为单个文件呈现
+ */
+export function createFocusInfoProvider(): DirectoryProvider {
+  const providerId = 'focus-info';
+  const providerName = '聚焦信息';
+  let outlineInfoManager: FocusOutlineInfoManager | null = null;
+
+  /**
+   * 初始化 outline info manager
+   */
+  function initializeManager(): void {
+    if (outlineInfoManager) {
+      return;
+    }
+
+    const focusInfo = getFocusInfo();
+    if (focusInfo?.pageId) {
+      outlineInfoManager = new FocusOutlineInfoManager({ 
+        api: context.api, 
+        focusInfo 
+      });
+    }
+  }
+
+  /**
+   * 获取聚焦描述
+   */
+  function getFocusDescription(): string {
+    initializeManager();
+    
+    if (!outlineInfoManager) {
+      return '当前没有聚焦到任何页面或组件。';
+    }
+    
+    return outlineInfoManager.generateFocusDescription();
+  }
+
+  /**
+   * 获取聚焦层级结构
+   */
+  function getFocusHierarchy(): string {
+    initializeManager();
+    
+    if (!outlineInfoManager) {
+      return '';
+    }
+    
+    return outlineInfoManager.generateFocusHierarchy();
+  }
+
+  return new DirectoryProvider({
+    id: providerId,
+    name: providerName,
+    description: '当前聚焦组件的树结构和描述',
+    weight: 90, // 高权重，排在项目信息之后
+    
+    // 自定义根节点为文件而不是目录
+    customRootNode: {
+      id: providerId,
+      name: providerName,
+      type: KnowledgeNodeType.FILE,
+      extname: '.md',
+      description: '当前聚焦组件的树结构和描述',
+      metadata: { providerId }
+    },
+    
+    // 因为根节点是文件，所以没有子节点
+    getChildren: async (parentId: string) => {
+      return [];
+    },
+    
+    // 读取文件内容
+    readFile: async (fileId: string) => {
+      if (fileId !== providerId) {
+        throw new Error(`Invalid file ID: ${fileId}`);
+      }
+
+      const focusInfo = getFocusInfo();
+      
+      // 如果没有聚焦信息
+      if (!focusInfo.pageId && !focusInfo.comId) {
+        return '当前未聚焦任何页面或组件';
+      }
+
+      // 获取聚焦的层级结构和描述
+      const contentHierarchy = getFocusHierarchy();
+      const focusDescription = getFocusDescription();
+
+      return `## 聚焦信息
+以下是当前聚焦组件的简略树结构，展示了聚焦元素的父级、兄弟、子级元素的关系。
+注意：此树结构并不完整，折叠了无关元素信息，如需详细信息请打开DSL文档获取。
+
+${contentHierarchy}
+
+${focusDescription}`;
+    }
+  });
+}

package/src/agents/workspace-by-knowledges/providers/index.ts

@@ -0,0 +1,8 @@
+/**
+ * Workspace Providers
+ * 导出所有 provider 创建函数
+ */
+
+export { createProjectInfoProvider } from './project-info-provider';
+export { createFocusInfoProvider } from './focus-info-provider';
+export { createComponentDocsProvider } from './component-docs-provider';

package/src/agents/workspace-by-knowledges/providers/project-info-provider.ts

@@ -0,0 +1,151 @@
+/**
+ * 项目信息 Provider
+ * 提供项目的画布索引信息
+ */
+
+import { DirectoryProvider, IKnowledgeNode, KnowledgeNodeType } from '../../knowledges/types';
+import { context } from '../../../context';
+
+/**
+ * 页面信息
+ */
+interface PageInfo {
+  id: string;
+  title: string;
+  type: string;
+  componentType?: string;
+  children?: PageInfo[];
+}
+
+/**
+ * 画布数据
+ */
+interface PagesData {
+  pageAry: PageInfo[];
+}
+
+/**
+ * 获取所有页面信息
+ */
+function getAllPages(): PagesData | PagesData[] {
+  const result = context.api?.global?.api?.getAllPageInfo?.();
+  return result || [];
+}
+
+/**
+ * 获取当前聚焦的页面ID
+ */
+function getFocusedPageId(): string | undefined {
+  return context.currentFocus?.pageId;
+}
+
+/**
+ * 处理原始数据
+ */
+function processRawData(rawData: PagesData | PagesData[]): PageInfo[] {
+  // 如果 rawData 是数组
+  if (Array.isArray(rawData)) {
+    const allPages: PageInfo[] = [];
+    rawData.forEach((canvas: any) => {
+      if (canvas.pageAry && Array.isArray(canvas.pageAry)) { // 多画布
+        allPages.push(...canvas.pageAry.map((page: any) => ({
+          id: page.id,
+          title: page.title,
+          type: page.type,
+          componentType: page.componentType || undefined,
+          children: page.children || []
+        })));
+      } else if (canvas.id) {
+        allPages.push({ ...canvas })
+      }
+    });
+    return allPages;
+  }
+
+  // 如果不是数组
+  if (!rawData?.pageAry) {
+    return [];
+  }
+
+  return rawData.pageAry.map((page: PageInfo) => ({
+    id: page.id,
+    title: page.title,
+    type: page.type,
+    componentType: page.componentType || undefined,
+    children: page.children || []
+  }));
+}
+
+/**
+ * 生成树形文本
+ */
+function generateTreeText(pages: PageInfo[], focusedPageId?: string, level = 0): string {
+  let result = '';
+  const indent = '  '.repeat(level);
+
+  pages.forEach(page => {
+    let line = `${indent}- ${page.title}[id=${page.id}]`;
+
+    if (page.componentType) {
+      line += `(${page.componentType})`;
+    }
+
+    if (focusedPageId && page.id === focusedPageId) {
+      line += ' 【当前聚焦】';
+    }
+
+    result += line + '\n';
+
+    if (page.children && page.children.length > 0) {
+      result += generateTreeText(page.children, focusedPageId, level + 1);
+    }
+  });
+
+  return result;
+}
+
+/**
+ * 创建项目信息 Provider
+ * 作为单个文件呈现，包含画布索引
+ */
+export function createProjectInfoProvider(): DirectoryProvider {
+  const providerId = 'project-info';
+  const providerName = '项目信息';
+  
+  return new DirectoryProvider({
+    id: providerId,
+    name: providerName,
+    description: '包含项目的画布索引',
+    weight: 100, // 高权重，排在最前面
+    
+    // 自定义根节点为文件而不是目录
+    customRootNode: {
+      id: providerId,
+      name: providerName,
+      type: KnowledgeNodeType.FILE,
+      extname: '.md',
+      description: '包含项目的画布索引',
+      metadata: { providerId }
+    },
+    
+    // 因为根节点是文件，所以没有子节点
+    getChildren: async (parentId: string) => {
+      return [];
+    },
+    
+    // 读取文件内容
+    readFile: async (fileId: string) => {
+      if (fileId !== providerId) {
+        throw new Error(`Invalid file ID: ${fileId}`);
+      }
+
+      const pagesInfo = getAllPages();
+      const focusedPageId = getFocusedPageId();
+      const processedPages = processRawData(pagesInfo);
+      const pageTree = generateTreeText(processedPages, focusedPageId);
+
+      return `## 画布索引
+${pageTree}`;
+    }
+  });
+}

package/src/agents/workspace-by-knowledges/test.ts

@@ -0,0 +1,240 @@
+/**
+ * Workspace 测试文件
+ * 使用真实的 context API，测试函数注入到 window 供浏览器控制台使用
+ */
+
+import { Workspace } from './workspace';
+import { context } from '../../context';
+
+// 测试函数
+async function runWorkspaceTests() {
+  console.log('='.repeat(60));
+  console.log('开始测试 Workspace');
+  console.log('='.repeat(60));
+  
+  try {
+    // 检查 context 是否有必要的 API
+    if (!context.api) {
+      throw new Error('context.api 未初始化，请确保在设计器环境中运行此测试');
+    }
+
+    // 1. 创建 Workspace
+    console.log('\n[测试 1] 创建 Workspace');
+    const workspace = new Workspace({
+      name: '测试工作空间',
+      description: '这是一个测试工作空间'
+    });
+    console.log('[SUCCESS] Workspace 创建成功');
+
+    // 2. 获取目录树
+    console.log('\n[测试 2] 获取目录树');
+    const tree = await workspace.getDirectoryTree({ showDescription: true });
+    console.log('目录树：\n', tree);
+
+    // 3. 检查默认打开的文件
+    console.log('\n[测试 3] 检查默认打开的文件');
+    const defaultFiles = workspace.getOpenedFilePaths();
+    console.log('默认打开的文件：', defaultFiles);
+    if (defaultFiles.includes('项目信息.md') && defaultFiles.includes('聚焦信息.md')) {
+      console.log('[SUCCESS] 项目信息和聚焦信息已默认打开');
+    } else {
+      console.log('[WARN] 默认文件未完全打开');
+    }
+
+    // 4. 测试打开组件文档（如果有组件的话）
+    console.log('\n[测试 4] 测试打开组件文档');
+    try {
+      // 尝试打开一个常见的组件文档
+      await workspace.openComponentDoc('组件配置文档/mybricks.normal-pc.antd5.icon.md');
+      console.log('[SUCCESS] 组件文档已打开');
+    } catch (error) {
+      console.log('[INFO] 无法打开组件文档（可能组件不存在）:', (error as Error).message);
+    }
+
+    // 5. 测试动态文档功能
+    console.log('\n[测试 5] 测试动态文档功能');
+    
+    // 注册一个自定义动态文档目录
+    workspace.registerDynamicDirectory({
+      id: 'custom-docs',
+      name: '自定义文档',
+      description: '用户自定义的文档',
+      weight: 50,
+      hidden: false // 不隐藏，在目录树中显示
+    });
+    
+    // 添加动态文档到自定义目录
+    await workspace.openDocument('doc1', {
+      type: '组件',
+      content: '这是第一个动态文档的内容\n包含一些测试数据',
+      title: '动态文档1',
+      description: '这是一个测试用的动态文档',
+      directoryId: 'custom-docs'
+    });
+    
+    // 添加另一个动态文档到默认目录
+    await workspace.openDocument('doc2', {
+      type: '画布',
+      content: '这是第二个动态文档的内容',
+      title: '动态文档2',
+      description: '使用默认动态文档目录'
+    });
+    
+    console.log('[SUCCESS] 动态文档已添加');
+
+    // 6. 获取已打开的文件列表
+    console.log('\n[测试 6] 获取已打开的文件列表');
+    const openedFiles = workspace.getOpenedFilePaths();
+    console.log('已打开的文件：', openedFiles);
+
+    // 7. 再次获取目录树（应该包含自定义文档目录）
+    console.log('\n[测试 7] 获取包含动态文档目录的目录树');
+    const treeWithDynamic = await workspace.getDirectoryTree({ showDescription: true });
+    console.log('目录树（包含动态文档）：\n', treeWithDynamic);
+
+    // 8. 导出工作空间内容
+    console.log('\n[测试 8] 导出工作空间内容');
+    const exportedContent = await workspace.export({ includeTree: true });
+    console.log('导出的内容长度：', exportedContent.length);
+    console.log('导出的内容（前500字符）：\n', exportedContent.substring(0, 500));
+
+    console.log('\n' + '='.repeat(60));
+    console.log('[SUCCESS] 所有测试通过！');
+    console.log('='.repeat(60));
+
+    return workspace;
+
+  } catch (error) {
+    console.error('\n[ERROR] 测试失败：', error);
+    console.error('错误堆栈：', (error as Error).stack);
+    throw error;
+  }
+}
+
+// 快捷测试函数
+async function quickTest() {
+  const workspace = new Workspace({
+    name: '快速测试工作空间'
+  });
+  
+  console.log('[INIT] 快速测试工作空间已创建');
+  console.log('使用方法：');
+  console.log('  - workspace.getDirectoryTree() - 获取目录树');
+  console.log('  - workspace.openComponentDoc(path) - 打开组件文档');
+  console.log('  - workspace.openComponentDocByNamespace(namespace) - 根据 namespace 打开组件文档');
+  console.log('  - workspace.openDocument(id, options) - 打开动态文档');
+  console.log('  - workspace.registerDynamicDirectory(config) - 注册动态文档目录');
+  console.log('  - workspace.export() - 导出工作空间内容');
+  console.log('  - workspace.refresh() - 刷新工作空间');
+
+  try {
+    // 先看看目录树
+    console.log('\n[步骤1] 获取目录树');
+    const tree = await workspace.getDirectoryTree({ showDescription: true });
+    console.log(tree);
+    
+    console.log('\n[步骤2] 检查默认打开的文件');
+    const openedFiles = workspace.getOpenedFilePaths();
+    console.log('已打开的文件列表：', openedFiles);
+    console.log('项目信息和聚焦信息应该已默认打开');
+    
+    console.log('\n[步骤3] 开始导出...');
+    const result = await workspace.export();
+    console.log('导出完成！');
+    console.log('导出内容长度：', result.length);
+    console.log('\n导出内容（前500字符）：\n', result.substring(0, 500));
+    return workspace;
+  } catch (error) {
+    console.error('[ERROR] 测试失败：', error);
+    console.error('错误堆栈：', (error as Error).stack);
+    throw error;
+  }
+}
+
+// 动态文档功能演示
+async function testDynamicDocs() {
+  console.log('='.repeat(60));
+  console.log('测试动态文档功能');
+  console.log('='.repeat(60));
+  
+  const workspace = new Workspace({
+    name: '动态文档测试工作空间'
+  });
+
+  // 1. 注册多个动态文档目录
+  console.log('\n[步骤1] 注册多个动态文档目录');
+  
+  workspace.registerDynamicDirectory({
+    id: 'opened-pages',
+    name: '已打开的页面',
+    description: '用户打开的页面文档',
+    weight: 80,
+    hidden: false
+  });
+
+  workspace.registerDynamicDirectory({
+    id: 'opened-components',
+    name: '已打开的组件',
+    description: '用户打开的组件文档',
+    weight: 70,
+    hidden: false
+  });
+
+  console.log('[SUCCESS] 已注册 2 个动态文档目录');
+
+  // 2. 添加文档到不同的目录
+  console.log('\n[步骤2] 添加文档到不同的目录');
+  
+  await workspace.openDocument('page_u_abc123', {
+    type: '画布',
+    content: '<页面1[id=u_abc123]>\n  <容器[id=u_def456]>\n    <按钮[id=u_ghi789]>',
+    title: '页面1',
+    description: '这是第一个页面',
+    directoryId: 'opened-pages'
+  });
+
+  await workspace.openDocument('component_u_xyz999', {
+    type: '组件',
+    content: '<按钮组件[id=u_xyz999]>\n  属性: { text: "点击我" }',
+    title: '按钮组件',
+    description: '一个普通按钮',
+    directoryId: 'opened-components'
+  });
+
+  console.log('[SUCCESS] 已添加文档到不同目录');
+
+  // 3. 查看目录树
+  console.log('\n[步骤3] 查看目录树');
+  const tree = await workspace.getDirectoryTree({ showDescription: true });
+  console.log(tree);
+
+  // 4. 导出内容
+  console.log('\n[步骤4] 导出内容');
+  const exported = await workspace.export({ includeTree: true });
+  console.log('导出内容长度：', exported.length);
+  console.log('\n完整导出内容：\n', exported);
+
+  console.log('\n' + '='.repeat(60));
+  console.log('[SUCCESS] 动态文档功能测试完成！');
+  console.log('='.repeat(60));
+
+  return workspace;
+}
+
+// 注入到 window 对象供浏览器控制台使用
+if (typeof window !== 'undefined') {
+  (window as any).workspaceTest = {
+    runTests: runWorkspaceTests,
+    quickTest,
+    testDynamicDocs,
+    Workspace,
+    context
+  };
+  
+  console.log('[READY] 测试函数已注入到 window.workspaceTest');
+  console.log('使用方法：');
+  console.log('  - window.workspaceTest.runTests() - 运行完整测试');
+  console.log('  - window.workspaceTest.quickTest() - 快速创建测试实例');
+  console.log('  - window.workspaceTest.testDynamicDocs() - 测试动态文档功能');
+  console.log('  - window.workspaceTest.context - 查看 context 对象');
+}

package/src/agents/workspace-by-knowledges/types.ts

@@ -0,0 +1,56 @@
+/**
+ * Workspace 类型定义
+ */
+
+import { KnowledgeBaseConfig } from '../knowledges/types';
+
+/**
+ * 页面信息
+ */
+export interface PageInfo {
+  id: string;
+  title: string;
+  type: string;
+  componentType?: string;
+  children?: PageInfo[];
+}
+
+/**
+ * 画布数据
+ */
+export interface PagesData {
+  pageAry: PageInfo[];
+}
+
+/**
+ * 聚焦信息
+ */
+export interface FocusInfo {
+  pageId?: string;
+  comId?: string;
+  title?: string;
+  type?: 'page' | 'uiCom' | 'section' | 'logicCom';
+  focusArea?: {
+    selector: string;
+    title: string;
+  };
+  diagramId?: string;
+}
+
+/**
+ * 组件信息
+ */
+export interface ComponentInfo {
+  namespace: string;
+  name: string;
+  abbreviation: string;
+  description?: string;
+}
+
+/**
+ * Workspace 配置（简化版，从 context 获取所有依赖）
+ */
+export interface WorkspaceConfig extends KnowledgeBaseConfig {
+  // 所有依赖从 context 获取，无需传入
+}
+