rubdottocom / chatgpt-python-to-javascript-chat Goto Github PK

require('dotenv').config();
const { Octokit } = require("@octokit/rest");

// Set your GitHub token here
const GITHUB_TOKEN = process.env.GITHUB_TOKEN || 'YOUR TOKEN HERE';

const octokit = new Octokit({ auth: GITHUB_TOKEN });

async function getReadmeContent(repoOwner, repoName) {
  try {
    const { data } = await octokit.repos.getContent({
      owner: repoOwner,
      repo: repoName,
      path: "README.md"
    });
    return Buffer.from(data.content, 'base64').toString('utf-8');
  } catch {
    return "README not found.";
  }
}

async function traverseRepoIteratively(repoOwner, repoName) {
  let structure = "";
  let dirsToVisit = [{ path: "", content: await octokit.repos.getContent({ owner: repoOwner, repo: repoName, path: "" }) }];
  let dirsVisited = new Set();

  while (dirsToVisit.length > 0) {
    const { path, content } = dirsToVisit.pop();
    dirsVisited.add(path);
    for (let item of content.data) {
      if (item.type === "dir") {
        if (!dirsVisited.has(item.path)) {
          structure += `${path}/${item.name}/\n`;
          const contents = await octokit.repos.getContent({ owner: repoOwner, repo: repoName, path: item.path });
          dirsToVisit.push({ path: `${path}/${item.name}`, content: contents });
        }
      } else {
        structure += `${path}/${item.name}\n`;
      }
    }
  }
  return structure;
}

async function getFileContentsIteratively(repoOwner, repoName) {
  let fileContents = "";
  // Este ejemplo no incluye la lista completa de extensiones binarias por brevedad.
  let binaryExtensions = ['.exe', '.dll', '.png', '.jpg'];
  let dirsToVisit = [{ path: "", content: await octokit.repos.getContent({ owner: repoOwner, repo: repoName, path: "" }) }];
  let dirsVisited = new Set();

  while (dirsToVisit.length > 0) {
    const { path, content } = dirsToVisit.pop();
    dirsVisited.add(path);
    for (let item of content.data) {
      if (item.type === "dir") {
        if (!dirsVisited.has(item.path)) {
          const contents = await octokit.repos.getContent({ owner: repoOwner, repo: repoName, path: item.path });
          dirsToVisit.push({ path: `${path}/${item.name}`, content: contents });
        }
      } else {
        const isBinary = binaryExtensions.some(ext => item.name.endsWith(ext));
        if (isBinary) {
          fileContents += `File: ${path}/${item.name}\nContent: Skipped binary file\n\n`;
        } else {
          try {
            const { data } = await octokit.repos.getContent({ owner: repoOwner, repo: repoName, path: item.path });
            const decodedContent = Buffer.from(data.content, 'base64').toString('utf-8');
            fileContents += `File: ${path}/${item.name}\nContent:\n${decodedContent}\n\n`;
          } catch {
            fileContents += `File: ${path}/${item.name}\nContent: Skipped due to error\n\n`;
          }
        }
      }
    }
  }
  return fileContents;
}

async function getRepoContents(repoUrl) {
  const repoName = repoUrl.split('/').pop();
  if (!GITHUB_TOKEN) {
    throw new Error("Please set the 'GITHUB_TOKEN' environment variable.");
  }

  const repoOwner = repoUrl.split('/')[repoUrl.split('/').length - 2];
  
  console.log(`Fetching README for: ${repoName}`);
  const readmeContent = await getReadmeContent(repoOwner, repoName);

  console.log(`\nFetching repository structure for: ${repoName}`);
  const repoStructure = `Repository Structure: ${repoName}\n` + await traverseRepoIteratively(repoOwner, repoName);

  console.log(`\nFetching file contents for: ${repoName}`);
  const fileContents = await getFileContentsIteratively(repoOwner, repoName);

  const instructions = `Prompt: Analyze the ${repoName} repository to understand its structure, purpose, and functionality. Follow these steps to study the codebase:\n\n`
    + "1. Read the README file to gain an overview of the project, its goals, and any setup instructions.\n\n"
    + "2. Examine the repository structure to understand how the files and directories are organized.\n\n"
    + "3. Identify the main entry point of the application (e.g., main.py, app.py, index.js) and start analyzing the code flow from there.\n\n"
    + "4. Study the dependencies and libraries used in the project to understand the external tools and frameworks being utilized.\n\n"
    + "5. Analyze the core functionality of the project by examining the key modules, classes, and functions.\n\n"
    + "6. Look for any configuration files (e.g., config.py, .env) to understand how the project is configured and what settings are available.\n\n"
    + "7. Investigate any tests or test directories to see how the project ensures code quality and handles different scenarios.\n\n"
    + "8. Review any documentation or inline comments to gather insights into the codebase and its intended behavior.\n\n"
    + "9. Identify any potential areas for improvement, optimization, or further exploration based on your analysis.\n\n"
    + "10. Provide a summary of your findings, including the project's purpose, key features, and any notable observations or recommendations.\n\n"
    + "Use the files and contents provided below to complete this analysis:\n\n";

  return { repoName, instructions, readmeContent, repoStructure, fileContents };
}

if (require.main === module) {
  const readline = require('readline').createInterface({
    input: process.stdin,
    output: process.stdout
  });

  readline.question("Please enter the GitHub repository URL: ", async (repoUrl) => {
    try {
      const { repoName, instructions, readmeContent, repoStructure, fileContents } = await getRepoContents(repoUrl);
      const outputFilename = `${repoName}_contents.txt`;
      const fs = require('fs');
      fs.writeFileSync(outputFilename, `${instructions}README:\n${readmeContent}\n\n${repoStructure}\n\n${fileContents}`, 'utf-8');
      console.log(`Repository contents saved to '${outputFilename}'.`);
    } catch (error) {
      console.error(`Error: ${error.message}`);
    } finally {
      readline.close();
    }
  });
}