#!/usr/bin/env node

const fs = require("fs");
const path = require("path");

function extractTitle(filePath) {
  try {
    const content = fs.readFileSync(filePath, "utf8");

    // For prompt files, look for the main heading after frontmatter
    if (filePath.includes(".prompt.md") || filePath.includes(".chatmode.md")) {
      const lines = content.split("\n");
      let inFrontmatter = false;
      let frontmatterEnded = false;

      for (const line of lines) {
        if (line.trim() === "---") {
          if (!inFrontmatter) {
            inFrontmatter = true;
          } else if (inFrontmatter && !frontmatterEnded) {
            frontmatterEnded = true;
          }
          continue;
        }

        if (frontmatterEnded && line.startsWith("# ")) {
          return line.substring(2).trim();
        }
      }

      // For prompt files without heading, clean up filename
      const basename = path.basename(
        filePath,
        filePath.includes(".prompt.md") ? ".prompt.md" : ".chatmode.md"
      );
      return basename
        .replace(/[-_]/g, " ")
        .replace(/\b\w/g, (l) => l.toUpperCase());
    }

    // For instruction files, look for the first heading
    const lines = content.split("\n");
    for (const line of lines) {
      if (line.startsWith("# ")) {
        return line.substring(2).trim();
      }
    }

    // Fallback to filename
    const basename = path.basename(filePath, path.extname(filePath));
    return basename
      .replace(/[-_]/g, " ")
      .replace(/\b\w/g, (l) => l.toUpperCase());
  } catch (error) {
    // Fallback to filename
    const basename = path.basename(filePath, path.extname(filePath));
    return basename
      .replace(/[-_]/g, " ")
      .replace(/\b\w/g, (l) => l.toUpperCase());
  }
}

function extractDescription(filePath) {
  try {
    const content = fs.readFileSync(filePath, "utf8");

    // Parse frontmatter for description (for both prompts and instructions)
    const lines = content.split("\n");
    let inFrontmatter = false;
    let frontmatterEnded = false;

    // For multi-line descriptions
    let isMultilineDescription = false;
    let multilineDescription = [];

    for (let i = 0; i < lines.length; i++) {
      const line = lines[i];

      if (line.trim() === "---") {
        if (!inFrontmatter) {
          inFrontmatter = true;
        } else if (inFrontmatter && !frontmatterEnded) {
          frontmatterEnded = true;
          break;
        }
        continue;
      }

      if (inFrontmatter && !frontmatterEnded) {
        // Check for multi-line description with pipe syntax (|)
        const multilineMatch = line.match(/^description:\s*\|(\s*)$/);
        if (multilineMatch) {
          isMultilineDescription = true;
          // Continue to next line to start collecting the multi-line content
          continue;
        }

        // If we're collecting a multi-line description
        if (isMultilineDescription) {
          // If the line has no indentation or has another frontmatter key, stop collecting
          if (!line.startsWith("  ") || line.match(/^[a-zA-Z0-9_-]+:/)) {
            isMultilineDescription = false;
            // Join the collected lines and return
            return multilineDescription.join(" ").trim();
          }

          // Add the line to our multi-line collection (removing the 2-space indentation)
          multilineDescription.push(line.substring(2));
        } else {
          // Look for single-line description field in frontmatter
          const descriptionMatch = line.match(
            /^description:\s*['"]?(.+?)['"]?$/
          );
          if (descriptionMatch) {
            return descriptionMatch[1];
          }
        }
      }
    }

    // If we've collected multi-line description but the frontmatter ended
    if (multilineDescription.length > 0) {
      return multilineDescription.join(" ").trim();
    }

    return null;
  } catch (error) {
    return null;
  }
}

function generateReadme() {
  const instructionsDir = path.join(__dirname, "instructions");
  const promptsDir = path.join(__dirname, "prompts");
  const chatmodesDir = path.join(__dirname, "chatmodes");
  const readmePath = path.join(__dirname, "README.md");

  // Check if README file exists
  if (!fs.existsSync(readmePath)) {
    console.error(
      "README.md not found! Please create a base README.md file first."
    );
    process.exit(1);
  }

  // Read the current README content
  let currentReadme = fs.readFileSync(readmePath, "utf8");

  // Get all instruction files
  const instructionFiles = fs
    .readdirSync(instructionsDir)
    .filter((file) => file.endsWith(".md"))
    .sort();

  // Get all prompt files - we'll use this to find new prompts
  const promptFiles = fs
    .readdirSync(promptsDir)
    .filter((file) => file.endsWith(".prompt.md"))
    .sort();

  // Get all chat mode files - we'll use this to update the chat modes section
  const chatmodeFiles = fs.existsSync(chatmodesDir)
    ? fs
        .readdirSync(chatmodesDir)
        .filter((file) => file.endsWith(".chatmode.md"))
        .sort()
    : [];

  // Update instructions section - rebuild the whole list
  const instructionsSection = currentReadme.match(
    /## 📋 Custom Instructions\n\nTeam and project-specific instructions.+?(?=\n\n>)/s
  );
  if (instructionsSection) {
    let instructionsListContent = "\n\n";

    // Generate alphabetically sorted list of instruction links
    for (const file of instructionFiles) {
      const filePath = path.join(instructionsDir, file);
      const title = extractTitle(filePath);
      const link = encodeURI(`instructions/${file}`);

      // Check if there's a description in the frontmatter
      const customDescription = extractDescription(filePath);

      if (customDescription) {
        // Use the description from frontmatter
        instructionsListContent += `- [${title}](${link}) - ${customDescription}\n`;
      } else {
        // Fallback to the default approach - use last word of title for description, removing trailing 's' if present
        const topic = title.split(" ").pop().replace(/s$/, "");
        instructionsListContent += `- [${title}](${link}) - ${topic} specific coding standards and best practices\n`;
      }
    }

    // Replace the current instructions section with the updated one
    const newInstructionsSection =
      "## 📋 Custom Instructions\n\nTeam and project-specific instructions to enhance GitHub Copilot's behavior for specific technologies and coding practices:" +
      instructionsListContent;
    currentReadme = currentReadme.replace(
      instructionsSection[0],
      newInstructionsSection
    );
  }

  // Extract existing prompt links from README
  const existingPromptLinks = [];
  const promptLinkRegex = /\[.*?\]\(prompts\/(.+?)\)/g;
  let match;

  while ((match = promptLinkRegex.exec(currentReadme)) !== null) {
    existingPromptLinks.push(match[1]);
  }

  // Find new prompts that aren't already in the README
  const newPromptFiles = promptFiles.filter(
    (file) => !existingPromptLinks.includes(file)
  );

  if (newPromptFiles.length > 0) {
    console.log(`Found ${newPromptFiles.length} new prompts to add.`);

    // Create content for new prompts (in Uncategorised section)
    let newPromptsContent = "";

    // Check if we already have an Uncategorised section
    const uncategorisedSectionRegex = /### Uncategorised\n/;
    const hasUncategorisedSection =
      uncategorisedSectionRegex.test(currentReadme);

    // If we need to add the section header
    if (!hasUncategorisedSection) {
      newPromptsContent += "### Uncategorised\n";
    }

    // Add each new prompt
    for (const file of newPromptFiles) {
      const filePath = path.join(promptsDir, file);
      const title = extractTitle(filePath);
      const description = extractDescription(filePath);
      const link = encodeURI(`prompts/${file}`);

      if (description) {
        newPromptsContent += `- [${title}](${link}) - ${description}\n`;
      } else {
        newPromptsContent += `- [${title}](${link})\n`;
      }
    }

    // Add a newline if we created a new section
    if (!hasUncategorisedSection) {
      newPromptsContent += "\n";
    }

    // Update the README content - insert new content in the right place
    if (hasUncategorisedSection) {
      // Add to existing Uncategorised section
      const uncategorisedSectionPos = currentReadme.match(
        uncategorisedSectionRegex
      ).index;
      const sectionEndRegex = /\n\n/;
      let sectionEndMatch = sectionEndRegex.exec(
        currentReadme.slice(uncategorisedSectionPos + 16)
      ); // 16 is length of "### Uncategorised\n"

      let insertPos;
      if (sectionEndMatch) {
        insertPos = uncategorisedSectionPos + 16 + sectionEndMatch.index;
      } else {
        // If we can't find the end of the section, just insert at the end of the section header
        insertPos = uncategorisedSectionPos + 16;
      }

      currentReadme =
        currentReadme.slice(0, insertPos) +
        newPromptsContent +
        currentReadme.slice(insertPos);
    } else {
      // No Uncategorised section exists yet - find where to add it
      // Look for the "Ready-to-use prompt templates" section and the next section after it
      const promptSectionRegex =
        /## 🎯 Reusable Prompts\n\nReady-to-use prompt templates/;
      const promptSectionMatch = currentReadme.match(promptSectionRegex);

      if (promptSectionMatch) {
        // Find where to insert the new section - after any existing categories
        let insertPos;
        // First check if there are any existing categories
        const existingCategoriesRegex = /### [^\n]+\n/g;
        let lastCategoryMatch = null;
        while ((match = existingCategoriesRegex.exec(currentReadme)) !== null) {
          lastCategoryMatch = match;
        }

        if (lastCategoryMatch) {
          // Find the end of the last category section
          const afterLastCategory = currentReadme.slice(
            lastCategoryMatch.index + lastCategoryMatch[0].length
          );
          const nextSectionRegex = /\n\n>/;
          const nextSectionMatch = afterLastCategory.match(nextSectionRegex);

          if (nextSectionMatch) {
            insertPos =
              lastCategoryMatch.index +
              lastCategoryMatch[0].length +
              nextSectionMatch.index;
          } else {
            // If we can't find the next section, add at the end of the prompt section
            insertPos = currentReadme.indexOf(
              "> 💡 **Usage**: Use `/prompt-name`"
            );
            if (insertPos === -1) {
              // Fallback position - before Additional Resources
              insertPos = currentReadme.indexOf("## 📚 Additional Resources");
            }
          }
        } else {
          // No categories yet, add right after the intro text
          const afterIntroRegex = /prompt\` command\.\n\n/;
          const afterIntroMatch = currentReadme.match(afterIntroRegex);

          if (afterIntroMatch) {
            insertPos = afterIntroMatch.index + afterIntroMatch[0].length;
          } else {
            // Fallback position - before Additional Resources
            insertPos = currentReadme.indexOf("## 📚 Additional Resources");
          }
        }

        if (insertPos !== -1) {
          currentReadme =
            currentReadme.slice(0, insertPos) +
            newPromptsContent +
            currentReadme.slice(insertPos);
        } else {
          console.error(
            "Could not find a suitable place to insert new prompts."
          );
        }
      } else {
        console.error(
          "Could not find the Reusable Prompts section in the README."
        );
      }
    }
  } else {
    console.log("No new prompts to add.");
  }

  // Update chat modes section
  const chatmodesSection = currentReadme.match(
    /## 🎭 Custom Chat Modes\n\nCustom chat modes define.+?(?=\n\n>)/s
  );

  if (chatmodesSection && chatmodeFiles.length > 0) {
    let chatmodesListContent = "\n\n";

    // Generate list of chat mode links
    for (const file of chatmodeFiles) {
      const filePath = path.join(chatmodesDir, file);
      const title = extractTitle(filePath);
      const link = encodeURI(`chatmodes/${file}`);

      // Check if there's a description in the frontmatter
      const customDescription = extractDescription(filePath);

      if (customDescription) {
        // Use the description from frontmatter
        chatmodesListContent += `- [${title}](${link}) - ${customDescription}\n`;
      } else {
        // Just add a link without description
        chatmodesListContent += `- [${title}](${link})\n`;
      }
    }

    // Replace the current chat modes section with the updated one
    const newChatmodesSection =
      '## 🎭 Custom Chat Modes\n\nCustom chat modes define specific behaviors and tools for GitHub Copilot Chat, enabling enhanced context-aware assistance for particular tasks or workflows. These `.chatmode.md` files can be loaded by selecting "Chat with Custom Mode" from the Copilot menu.' +
      chatmodesListContent;

    currentReadme = currentReadme.replace(
      chatmodesSection[0],
      newChatmodesSection
    );
  } else if (!chatmodesSection && chatmodeFiles.length > 0) {
    // Chat modes section doesn't exist yet but we have chat mode files
    const chatmodesListContent = chatmodeFiles
      .map((file) => {
        const filePath = path.join(chatmodesDir, file);
        const title = extractTitle(filePath);
        const link = `chatmodes/${file}`;
        const customDescription = extractDescription(filePath);

        if (customDescription) {
          return `- [${title}](${link}) - ${customDescription}`;
        } else {
          return `- [${title}](${link})`;
        }
      })
      .join("\n");

    const newChatmodesSection =
      "## 🎭 Custom Chat Modes\n\n" +
      'Custom chat modes define specific behaviors and tools for GitHub Copilot Chat, enabling enhanced context-aware assistance for particular tasks or workflows. These `.chatmode.md` files can be loaded by selecting "Chat with Custom Mode" from the Copilot menu.\n\n' +
      chatmodesListContent +
      '\n\n> 💡 **Usage**: Create a `.chatmode.md` file in your workspace or repository, then select "Chat with Custom Mode" from the Copilot Chat menu and select your chat mode file.\n';

    // Insert before Additional Resources section
    const additionalResourcesPos = currentReadme.indexOf(
      "## 📚 Additional Resources"
    );
    if (additionalResourcesPos !== -1) {
      currentReadme =
        currentReadme.slice(0, additionalResourcesPos) +
        newChatmodesSection +
        "\n" +
        currentReadme.slice(additionalResourcesPos);
    }
  }

  return currentReadme;
}

// Generate and write the README
const updatedReadme = generateReadme();

// Only write file if we have content to write
if (updatedReadme) {
  fs.writeFileSync(path.join(__dirname, "README.md"), updatedReadme);
  console.log("README.md updated successfully!");
}