idea-manager 0.2.0 → 0.3.1

package/src/lib/mcp/tools.ts

@@ -1,82 +1,84 @@
-import type { IItemTree, IItem, IPrompt } from '@/types';
-
-// Re-export query functions for MCP tools
-// These are imported from DB queries when the MCP server initializes
+import type { ITask, ITaskPrompt, ISubProject } from '@/types';
 
 export interface McpToolContext {
   listProjects: () => { id: string; name: string; description: string; created_at: string; updated_at: string }[];
   getProject: (id: string) => { id: string; name: string; description: string } | undefined;
-  getItemTree: (projectId: string) => IItemTree[];
-  getItems: (projectId: string) => IItem[];
-  getPrompt: (itemId: string) => IPrompt | undefined;
-  updateItem: (id: string, data: Record<string, unknown>) => IItem | undefined;
+  getSubProjects: (projectId: string) => ISubProject[];
+  getTasksByProject: (projectId: string) => ITask[];
+  getTaskPrompt: (taskId: string) => ITaskPrompt | undefined;
+  updateTask: (id: string, data: Record<string, unknown>) => ITask | undefined;
 }
 
 export function getNextTask(ctx: McpToolContext, projectId: string): {
-  item: IItem;
-  prompt?: IPrompt;
+  task: ITask;
+  prompt?: ITaskPrompt;
 } | null {
-  const items = ctx.getItems(projectId);
-
-  // Find unlocked + pending items (ready to execute)
-  const ready = items.filter(i => !i.is_locked && i.status === 'pending');
+  const tasks = ctx.getTasksByProject(projectId);
 
+  // Find submitted tasks with prompts (ready to execute)
+  const ready = tasks.filter(t => t.status === 'submitted');
   if (ready.length === 0) return null;
 
-  // Sort by sort_order
   ready.sort((a, b) => a.sort_order - b.sort_order);
-  const item = ready[0];
-  const prompt = ctx.getPrompt(item.id);
+  const task = ready[0];
+  const prompt = ctx.getTaskPrompt(task.id);
 
-  return { item, prompt };
+  return { task, prompt };
 }
 
 export function getProjectContext(ctx: McpToolContext, projectId: string): {
   project: { id: string; name: string; description: string } | undefined;
-  tree: IItemTree[];
-  stats: { total: number; pending: number; inProgress: number; done: number; unlocked: number };
+  subProjects: ISubProject[];
+  tasks: ITask[];
+  stats: { total: number; idea: number; writing: number; submitted: number; testing: number; done: number; problem: number };
 } {
   const project = ctx.getProject(projectId);
-  const tree = ctx.getItemTree(projectId);
-  const items = ctx.getItems(projectId);
+  const subProjects = ctx.getSubProjects(projectId);
+  const tasks = ctx.getTasksByProject(projectId);
 
   const stats = {
-    total: items.length,
-    pending: items.filter(i => i.status === 'pending').length,
-    inProgress: items.filter(i => i.status === 'in_progress').length,
-    done: items.filter(i => i.status === 'done').length,
-    unlocked: items.filter(i => !i.is_locked).length,
+    total: tasks.length,
+    idea: tasks.filter(t => t.status === 'idea').length,
+    writing: tasks.filter(t => t.status === 'writing').length,
+    submitted: tasks.filter(t => t.status === 'submitted').length,
+    testing: tasks.filter(t => t.status === 'testing').length,
+    done: tasks.filter(t => t.status === 'done').length,
+    problem: tasks.filter(t => t.status === 'problem').length,
   };
 
-  return { project, tree, stats };
+  return { project, subProjects, tasks, stats };
 }
 
-export function formatItemForMcp(item: IItem, prompt?: IPrompt): string {
+export function formatTaskForMcp(task: ITask, prompt?: ITaskPrompt): string {
   const lines = [
-    `제목: ${item.title}`,
-    `설명: ${item.description}`,
-    `유형: ${item.item_type}`,
-    `우선순위: ${item.priority}`,
-    `상태: ${item.status}`,
-    `잠금: ${item.is_locked ? '잠금' : '해제'}`,
+    `Title: ${task.title}`,
+    `Description: ${task.description}`,
+    `Status: ${task.status}`,
+    `Priority: ${task.priority}`,
   ];
 
-  if (prompt) {
-    lines.push('', '--- 프롬프트 ---', prompt.content);
+  if (prompt?.content) {
+    lines.push('', '--- Prompt ---', prompt.content);
   }
 
   return lines.join('\n');
 }
 
-export function formatTreeForMcp(tree: IItemTree[], indent = 0): string {
+export function formatProjectForMcp(
+  subProjects: ISubProject[],
+  tasks: ITask[],
+): string {
   const lines: string[] = [];
-  for (const item of tree) {
-    const prefix = '  '.repeat(indent);
-    const lock = item.is_locked ? '🔐' : '🔓';
-    const status = item.status === 'done' ? '✅' : item.status === 'in_progress' ? '🔄' : '⏳';
-    lines.push(`${prefix}${lock} ${item.title} ${status}`);
-    if (item.children.length > 0) {
-      lines.push(formatTreeForMcp(item.children, indent + 1));
+  for (const sp of subProjects) {
+    lines.push(`[${sp.name}]`);
+    const spTasks = tasks.filter(t => t.sub_project_id === sp.id);
+    if (spTasks.length === 0) {
+      lines.push('  (no tasks)');
+    } else {
+      for (const t of spTasks) {
+        const icon = t.status === 'done' ? 'v' : t.status === 'submitted' ? '>' : t.status === 'testing' ? '~' : t.status === 'problem' ? '!' : '-';
+        lines.push(`  ${icon} ${t.title} [${t.status}]`);
+      }
     }
   }
   return lines.join('\n');