aifastdb 2.2.2 → 2.2.6

package/dist/dev-plan-store.js

@@ -138,6 +138,7 @@ class DevPlanStore {
         this.projectName = projectName;
         this.docStore = new native_1.EnhancedDocumentStore(config.documentPath, (0, document_store_1.documentStoreProductionConfig)());
         this.taskStore = new native_1.EnhancedDocumentStore(config.taskPath, (0, document_store_1.documentStoreProductionConfig)());
+        this.moduleStore = new native_1.EnhancedDocumentStore(config.modulePath, (0, document_store_1.documentStoreProductionConfig)());
     }
     // ==========================================================================
     // Document Section Operations
@@ -155,15 +156,20 @@ class DevPlanStore {
         }
         const version = input.version || '1.0.0';
         const now = Date.now();
+        const finalModuleId = input.moduleId || existing?.moduleId;
+        const tags = [
+            `plan:${this.projectName}`,
+            `section:${input.section}`,
+            ...(input.subSection ? [`sub:${input.subSection}`] : []),
+            `ver:${version}`,
+        ];
+        if (finalModuleId) {
+            tags.push(`module:${finalModuleId}`);
+        }
         const docInput = {
             content: input.content,
             contentType: native_2.ContentType.Text,
-            tags: [
-                `plan:${this.projectName}`,
-                `section:${input.section}`,
-                ...(input.subSection ? [`sub:${input.subSection}`] : []),
-                `ver:${version}`,
-            ],
+            tags,
             metadata: {
                 projectName: this.projectName,
                 section: input.section,
@@ -171,6 +177,7 @@ class DevPlanStore {
                 version,
                 subSection: input.subSection || null,
                 relatedSections: input.relatedSections || [],
+                moduleId: finalModuleId || null,
                 createdAt: existing?.createdAt || now,
                 updatedAt: now,
             },
@@ -302,20 +309,25 @@ class DevPlanStore {
             totalSubtasks: 0,
             completedSubtasks: 0,
         };
+        const tags = [
+            `plan:${this.projectName}`,
+            'type:main-task',
+            `mtask:${input.taskId}`,
+            `priority:${input.priority}`,
+            'status:pending',
+        ];
+        if (input.moduleId) {
+            tags.push(`module:${input.moduleId}`);
+        }
         const docInput = {
             content: JSON.stringify(taskData),
             contentType: native_2.ContentType.Text,
-            tags: [
-                `plan:${this.projectName}`,
-                'type:main-task',
-                `mtask:${input.taskId}`,
-                `priority:${input.priority}`,
-                'status:pending',
-            ],
+            tags,
             metadata: {
                 projectName: this.projectName,
                 taskId: input.taskId,
                 status: 'pending',
+                moduleId: input.moduleId || null,
                 createdAt: now,
                 updatedAt: now,
                 completedAt: null,
@@ -328,6 +340,7 @@ class DevPlanStore {
             id,
             projectName: this.projectName,
             ...taskData,
+            moduleId: input.moduleId,
             status: 'pending',
             createdAt: now,
             updatedAt: now,
@@ -378,6 +391,7 @@ class DevPlanStore {
         this.deleteAndEnsureTimestampAdvance(this.taskStore, existing.id);
         const now = Date.now();
         const completedAt = finalStatus === 'completed' ? (existing.completedAt || now) : null;
+        const finalModuleId = input.moduleId || existing.moduleId;
         const taskData = {
             taskId: input.taskId,
             title: input.title,
@@ -388,20 +402,25 @@ class DevPlanStore {
             totalSubtasks: existing.totalSubtasks,
             completedSubtasks: existing.completedSubtasks,
         };
+        const tags = [
+            `plan:${this.projectName}`,
+            'type:main-task',
+            `mtask:${input.taskId}`,
+            `priority:${input.priority}`,
+            `status:${finalStatus}`,
+        ];
+        if (finalModuleId) {
+            tags.push(`module:${finalModuleId}`);
+        }
         const docInput = {
             content: JSON.stringify(taskData),
             contentType: native_2.ContentType.Text,
-            tags: [
-                `plan:${this.projectName}`,
-                'type:main-task',
-                `mtask:${input.taskId}`,
-                `priority:${input.priority}`,
-                `status:${finalStatus}`,
-            ],
+            tags,
             metadata: {
                 projectName: this.projectName,
                 taskId: input.taskId,
                 status: finalStatus,
+                moduleId: finalModuleId || null,
                 createdAt: existing.createdAt,
                 updatedAt: now,
                 completedAt,
@@ -414,6 +433,7 @@ class DevPlanStore {
             ...taskData,
             id,
             projectName: this.projectName,
+            moduleId: finalModuleId,
             status: finalStatus,
             createdAt: existing.createdAt,
             updatedAt: now,
@@ -454,6 +474,10 @@ class DevPlanStore {
             const priorityTag = `priority:${filter.priority}`;
             docs = docs.filter((doc) => doc.tags.includes(priorityTag));
         }
+        if (filter?.moduleId) {
+            const moduleTag = `module:${filter.moduleId}`;
+            docs = docs.filter((doc) => doc.tags.includes(moduleTag));
+        }
         return docs.map((doc) => this.docToMainTask(doc));
     }
     /**
@@ -476,20 +500,25 @@ class DevPlanStore {
             totalSubtasks: mainTask.totalSubtasks,
             completedSubtasks: mainTask.completedSubtasks,
         };
+        const tags = [
+            `plan:${this.projectName}`,
+            'type:main-task',
+            `mtask:${mainTask.taskId}`,
+            `priority:${mainTask.priority}`,
+            `status:${status}`,
+        ];
+        if (mainTask.moduleId) {
+            tags.push(`module:${mainTask.moduleId}`);
+        }
         const docInput = {
             content: JSON.stringify(taskData),
             contentType: native_2.ContentType.Text,
-            tags: [
-                `plan:${this.projectName}`,
-                'type:main-task',
-                `mtask:${mainTask.taskId}`,
-                `priority:${mainTask.priority}`,
-                `status:${status}`,
-            ],
+            tags,
             metadata: {
                 projectName: this.projectName,
                 taskId: mainTask.taskId,
                 status,
+                moduleId: mainTask.moduleId || null,
                 createdAt: mainTask.createdAt,
                 updatedAt: now,
                 completedAt,
@@ -968,6 +997,196 @@ class DevPlanStore {
         return md;
     }
     // ==========================================================================
+    // Module Operations
+    // ==========================================================================
+    /**
+     * 创建功能模块
+     */
+    createModule(input) {
+        const existing = this.getModule(input.moduleId);
+        if (existing) {
+            throw new Error(`Module "${input.moduleId}" already exists for project "${this.projectName}"`);
+        }
+        const now = Date.now();
+        const status = input.status || 'active';
+        const moduleData = {
+            moduleId: input.moduleId,
+            name: input.name,
+            description: input.description || '',
+        };
+        const docInput = {
+            content: JSON.stringify(moduleData),
+            contentType: native_2.ContentType.Text,
+            tags: [
+                `plan:${this.projectName}`,
+                'type:module',
+                `module:${input.moduleId}`,
+                `status:${status}`,
+            ],
+            metadata: {
+                projectName: this.projectName,
+                moduleId: input.moduleId,
+                status,
+                createdAt: now,
+                updatedAt: now,
+            },
+            importance: 0.85,
+        };
+        const id = this.moduleStore.put(docInput);
+        this.moduleStore.flush();
+        return {
+            id,
+            projectName: this.projectName,
+            moduleId: input.moduleId,
+            name: input.name,
+            description: input.description,
+            status,
+            mainTaskCount: 0,
+            subTaskCount: 0,
+            completedSubTaskCount: 0,
+            docCount: 0,
+            createdAt: now,
+            updatedAt: now,
+        };
+    }
+    /**
+     * 获取功能模块（含自动计算的 taskCount/docCount）
+     */
+    getModule(moduleId) {
+        const tag = `module:${moduleId}`;
+        const docs = this.moduleStore.findByTag(tag)
+            .filter((doc) => doc.tags.includes(`plan:${this.projectName}`));
+        if (docs.length === 0)
+            return null;
+        const latest = docs.sort((a, b) => this.getDocUpdatedAt(b) - this.getDocUpdatedAt(a))[0];
+        return this.docToModule(latest);
+    }
+    /**
+     * 列出所有功能模块
+     */
+    listModules(filter) {
+        let docs = this.moduleStore.findByTag(`plan:${this.projectName}`)
+            .filter((doc) => doc.tags.includes('type:module'));
+        // 按 moduleId 去重
+        const latestMap = new Map();
+        for (const doc of docs) {
+            const data = JSON.parse(doc.content);
+            const moduleId = data.moduleId;
+            if (!moduleId)
+                continue;
+            const existing = latestMap.get(moduleId);
+            if (!existing || this.getDocUpdatedAt(doc) > this.getDocUpdatedAt(existing)) {
+                latestMap.set(moduleId, doc);
+            }
+        }
+        docs = Array.from(latestMap.values());
+        if (filter?.status) {
+            const statusTag = `status:${filter.status}`;
+            docs = docs.filter((doc) => doc.tags.includes(statusTag));
+        }
+        return docs.map((doc) => this.docToModule(doc));
+    }
+    /**
+     * 更新功能模块
+     */
+    updateModule(moduleId, updates) {
+        const existing = this.getModule(moduleId);
+        if (!existing)
+            return null;
+        this.deleteAndEnsureTimestampAdvance(this.moduleStore, existing.id);
+        const now = Date.now();
+        const newName = updates.name || existing.name;
+        const newDescription = updates.description !== undefined ? updates.description : existing.description;
+        const newStatus = updates.status || existing.status;
+        const moduleData = {
+            moduleId,
+            name: newName,
+            description: newDescription || '',
+        };
+        const docInput = {
+            content: JSON.stringify(moduleData),
+            contentType: native_2.ContentType.Text,
+            tags: [
+                `plan:${this.projectName}`,
+                'type:module',
+                `module:${moduleId}`,
+                `status:${newStatus}`,
+            ],
+            metadata: {
+                projectName: this.projectName,
+                moduleId,
+                status: newStatus,
+                createdAt: existing.createdAt,
+                updatedAt: now,
+            },
+            importance: 0.85,
+        };
+        const id = this.moduleStore.put(docInput);
+        this.moduleStore.flush();
+        return {
+            id,
+            projectName: this.projectName,
+            moduleId,
+            name: newName,
+            description: newDescription,
+            status: newStatus,
+            mainTaskCount: existing.mainTaskCount,
+            subTaskCount: existing.subTaskCount,
+            completedSubTaskCount: existing.completedSubTaskCount,
+            docCount: existing.docCount,
+            createdAt: existing.createdAt,
+            updatedAt: now,
+        };
+    }
+    /**
+     * 删除功能模块
+     */
+    deleteModule(moduleId) {
+        const existing = this.getModule(moduleId);
+        if (!existing)
+            return false;
+        this.moduleStore.delete(existing.id);
+        this.moduleStore.flush();
+        return true;
+    }
+    /**
+     * 获取模块详情 — 包含关联的任务和文档
+     */
+    getModuleDetail(moduleId) {
+        const mod = this.getModule(moduleId);
+        if (!mod)
+            return null;
+        // 获取关联的主任务
+        const moduleTag = `module:${moduleId}`;
+        let taskDocs = this.taskStore.findByTag(moduleTag)
+            .filter((doc) => doc.tags.includes(`plan:${this.projectName}`) &&
+            doc.tags.includes('type:main-task'));
+        taskDocs = this.deduplicateByTaskId(taskDocs);
+        const mainTasks = taskDocs.map((doc) => this.docToMainTask(doc));
+        // 获取关联的所有子任务（通过主任务间接关联）
+        const subTasks = [];
+        for (const mt of mainTasks) {
+            const subs = this.listSubTasks(mt.taskId);
+            subTasks.push(...subs);
+        }
+        // 获取关联的文档
+        let docDocs = this.docStore.findByTag(moduleTag)
+            .filter((doc) => doc.tags.includes(`plan:${this.projectName}`));
+        // 按 section+subSection 去重
+        const latestDocMap = new Map();
+        for (const doc of docDocs) {
+            const sectionTag = doc.tags.find((t) => t.startsWith('section:'));
+            const subTag = doc.tags.find((t) => t.startsWith('sub:'));
+            const key = `${sectionTag || 'unknown'}|${subTag || ''}`;
+            const ex = latestDocMap.get(key);
+            if (!ex || this.getDocUpdatedAt(doc) > this.getDocUpdatedAt(ex)) {
+                latestDocMap.set(key, doc);
+            }
+        }
+        const documents = Array.from(latestDocMap.values()).map((doc) => this.docToDevPlanDoc(doc));
+        return { module: mod, mainTasks, subTasks, documents };
+    }
+    // ==========================================================================
     // Utility
     // ==========================================================================
     /**
@@ -976,6 +1195,7 @@ class DevPlanStore {
     sync() {
         this.docStore.flush();
         this.taskStore.flush();
+        this.moduleStore.flush();
     }
     /**
      * 获取项目名称
@@ -1058,6 +1278,8 @@ class DevPlanStore {
         const section = (sectionTag?.replace('section:', '') || 'custom');
         const subTag = doc.tags.find((t) => t.startsWith('sub:'));
         const subSection = subTag?.replace('sub:', '');
+        const moduleTag = doc.tags.find((t) => t.startsWith('module:'));
+        const moduleId = moduleTag?.replace('module:', '') || undefined;
         return {
             id: doc.id,
             projectName: this.projectName,
@@ -1067,6 +1289,7 @@ class DevPlanStore {
             version: doc.metadata?.version || '1.0.0',
             subSection,
             relatedSections: doc.metadata?.relatedSections || [],
+            moduleId,
             createdAt: doc.metadata?.createdAt || doc.createdAt,
             updatedAt: doc.metadata?.updatedAt || doc.createdAt,
         };
@@ -1075,6 +1298,8 @@ class DevPlanStore {
         const data = JSON.parse(doc.content);
         const statusTag = doc.tags.find((t) => t.startsWith('status:'));
         const status = (statusTag?.replace('status:', '') || 'pending');
+        const moduleTag = doc.tags.find((t) => t.startsWith('module:'));
+        const moduleId = moduleTag?.replace('module:', '') || undefined;
         return {
             id: doc.id,
             projectName: this.projectName,
@@ -1084,6 +1309,7 @@ class DevPlanStore {
             description: data.description,
             estimatedHours: data.estimatedHours,
             relatedSections: data.relatedSections || [],
+            moduleId,
             totalSubtasks: data.totalSubtasks || 0,
             completedSubtasks: data.completedSubtasks || 0,
             status,
@@ -1092,6 +1318,71 @@ class DevPlanStore {
             completedAt: doc.metadata?.completedAt || null,
         };
     }
+    docToModule(doc) {
+        const data = JSON.parse(doc.content);
+        const statusTag = doc.tags.find((t) => t.startsWith('status:'));
+        const status = (statusTag?.replace('status:', '') || 'active');
+        const moduleId = data.moduleId;
+        // 计算关联的主任务数（去重）
+        const moduleTag = `module:${moduleId}`;
+        const taskDocs = this.taskStore.findByTag(moduleTag)
+            .filter((d) => d.tags.includes(`plan:${this.projectName}`) &&
+            d.tags.includes('type:main-task'));
+        const uniqueTaskIds = new Set();
+        for (const td of taskDocs) {
+            try {
+                uniqueTaskIds.add(JSON.parse(td.content).taskId);
+            }
+            catch { }
+        }
+        // 计算关联的子任务数（遍历关联主任务下的所有子任务）
+        let subTaskCount = 0;
+        let completedSubTaskCount = 0;
+        for (const mainTaskId of uniqueTaskIds) {
+            const subDocs = this.taskStore.findByTag(`parent:${mainTaskId}`)
+                .filter((d) => d.tags.includes(`plan:${this.projectName}`) &&
+                d.tags.includes('type:sub-task'));
+            // 按 taskId 去重
+            const seenSubIds = new Set();
+            for (const sd of subDocs) {
+                try {
+                    const subData = JSON.parse(sd.content);
+                    if (seenSubIds.has(subData.taskId))
+                        continue;
+                    seenSubIds.add(subData.taskId);
+                    subTaskCount++;
+                    const subStatusTag = sd.tags.find((t) => t.startsWith('status:'));
+                    if (subStatusTag === 'status:completed') {
+                        completedSubTaskCount++;
+                    }
+                }
+                catch { }
+            }
+        }
+        // 计算关联的文档数（按 section+subSection 去重）
+        const docDocs = this.docStore.findByTag(moduleTag)
+            .filter((d) => d.tags.includes(`plan:${this.projectName}`));
+        const uniqueDocKeys = new Set();
+        for (const dd of docDocs) {
+            const st = dd.tags.find((t) => t.startsWith('section:'));
+            const sub = dd.tags.find((t) => t.startsWith('sub:'));
+            uniqueDocKeys.add(`${st || ''}|${sub || ''}`);
+        }
+        return {
+            id: doc.id,
+            projectName: this.projectName,
+            moduleId,
+            name: data.name,
+            description: data.description || undefined,
+            status,
+            mainTaskCount: uniqueTaskIds.size,
+            subTaskCount,
+            completedSubTaskCount,
+            docCount: uniqueDocKeys.size,
+            createdAt: doc.metadata?.createdAt || doc.createdAt,
+            updatedAt: doc.metadata?.updatedAt || doc.createdAt,
+        };
+    }
     docToSubTask(doc) {
         const data = JSON.parse(doc.content);
         const statusTag = doc.tags.find((t) => t.startsWith('status:'));
@@ -1140,20 +1431,25 @@ class DevPlanStore {
             totalSubtasks: subs.length,
             completedSubtasks: completedCount,
         };
+        const tags = [
+            `plan:${this.projectName}`,
+            'type:main-task',
+            `mtask:${mainTask.taskId}`,
+            `priority:${mainTask.priority}`,
+            `status:${mainTask.status}`,
+        ];
+        if (mainTask.moduleId) {
+            tags.push(`module:${mainTask.moduleId}`);
+        }
         const docInput = {
             content: JSON.stringify(taskData),
             contentType: native_2.ContentType.Text,
-            tags: [
-                `plan:${this.projectName}`,
-                'type:main-task',
-                `mtask:${mainTask.taskId}`,
-                `priority:${mainTask.priority}`,
-                `status:${mainTask.status}`,
-            ],
+            tags,
             metadata: {
                 projectName: this.projectName,
                 taskId: mainTask.taskId,
                 status: mainTask.status,
+                moduleId: mainTask.moduleId || null,
                 createdAt: mainTask.createdAt,
                 updatedAt: now,
                 completedAt: mainTask.completedAt,
@@ -1335,6 +1631,7 @@ function createDevPlan(projectName, basePath) {
     return new DevPlanStore(projectName, {
         documentPath: path.join(base, projectName, 'documents.jsonl'),
         taskPath: path.join(base, projectName, 'tasks.jsonl'),
+        modulePath: path.join(base, projectName, 'modules.jsonl'),
     });
 }
 /**