/**
 * tools/parsePRD.js
 * Tool to parse PRD document and generate tasks
 */

import { z } from "zod";
import {
  handleApiResult,
  createErrorResponse,
  getProjectRootFromSession
} from "./utils.js";
import { parsePRDDirect } from "../core/task-master-core.js";

/**
 * Register the parsePRD tool with the MCP server
 * @param {Object} server - FastMCP server instance
 */
export function registerParsePRDTool(server) {
  server.addTool({
    name: "parse_prd",
    description: "Parse a Product Requirements Document (PRD) or text file to automatically generate initial tasks.",
    parameters: z.object({
      input: z.string().default("tasks/tasks.json").describe("Path to the PRD document file (relative to project root or absolute)"),
      numTasks: z.union([z.number(), z.string()]).optional().describe("Approximate number of top-level tasks to generate (default: 10)"),
      output: z.string().optional().describe("Output path for tasks.json file (relative to project root or absolute, default: tasks/tasks.json)"),
      force: z.boolean().optional().describe("Allow overwriting an existing tasks.json file."),
      projectRoot: z
        .string()
        .optional()
        .describe(
          "Root directory of the project (default: automatically detected from session or CWD)"
        ),
    }),
    execute: async (args, { log, session, reportProgress }) => {
      try {
        log.info(`Parsing PRD document with args: ${JSON.stringify(args)}`);
        
        let rootFolder = getProjectRootFromSession(session, log);
        
        if (!rootFolder && args.projectRoot) {
          rootFolder = args.projectRoot;
          log.info(`Using project root from args as fallback: ${rootFolder}`);
        }
        
        const result = await parsePRDDirect({
          projectRoot: rootFolder,
          ...args
        }, log, { reportProgress, mcpLog: log, session});
        
        await reportProgress({ progress: 100 });
        
        if (result.success) {
          log.info(`Successfully generated ${result.data?.taskCount || 0} tasks from PRD at ${result.data?.outputPath}`);
        } else {
          log.error(`Failed to parse PRD: ${result.error?.message || 'Unknown error'}`);
        }
        
        return handleApiResult(result, log, 'Error parsing PRD document');
      } catch (error) {
        log.error(`Error in parse_prd tool: ${error.message}`);
        return createErrorResponse(error.message);
      }
    },
  });
}