@dev-angsu/cli 1.0.4 → 1.0.6

package/bin/index.js

@@ -11,6 +11,7 @@ import ora from "ora";
 import fs from "fs";
 
 import { run as review } from "../src/utils/engine.js";
+import { copyCode } from "../src/commands/copy.js";
 
 const program = new Command();
 
@@ -84,5 +85,16 @@ program
     await review();
   });
 
+// 4. Copy Code Command
+program
+  .command("copycode")
+  .alias("cp")
+  .description("Copy codebase context to clipboard for LLMs")
+  .option("-o, --output <file>", "Output result to a file instead of clipboard")
+  .option("-d, --dry-run", "Dry run: only list files and stats")
+  .action(async (options) => {
+    await copyCode(process.cwd(), options);
+  });
+
 // 6. Parse Arguments
 program.parse(process.argv);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dev-angsu/cli",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "",
   "main": "index.js",
   "scripts": {
@@ -18,9 +18,13 @@
     "@actions/core": "^2.0.1",
     "@actions/github": "^6.0.1",
     "chalk": "^5.6.2",
+    "clipboardy": "^4.0.0",
     "commander": "^14.0.2",
     "dotenv": "^17.2.3",
+    "fast-glob": "^3.3.3",
+    "ignore": "^5.3.2",
     "inquirer": "^13.1.0",
+    "isbinaryfile": "^5.0.7",
     "openai": "^6.15.0",
     "ora": "^9.0.0",
     "simple-git": "^3.30.0"

package/src/commands/copy.js

@@ -0,0 +1,87 @@
+import { getValidFiles } from "../utils/file-reader.js";
+import clipboardy from "clipboardy";
+import fs from "fs";
+import path from "path";
+import ora from "ora";
+import chalk from "chalk";
+import inquirer from "inquirer";
+
+export async function copyCode(dir = process.cwd(), options = {}) {
+  const spinner = ora("Scanning directory...").start();
+
+  try {
+    const files = await getValidFiles(dir);
+
+    if (files.length === 0) {
+      spinner.fail("No valid files found to copy.");
+      return;
+    }
+
+    spinner.text = `Processing ${files.length} files...`;
+
+    let output = "";
+    let tokenCount = 0;
+
+    // Generate Tree Structure for context
+    const tree = files.join("\n");
+
+    for (const file of files) {
+      const content = fs.readFileSync(path.join(dir, file), "utf-8");
+      // Wrap in Markdown code blocks with filename
+      if (!options.dryRun) {
+        output += `\n\n# File: ${file}\n\`\`\`\n${content}\n\`\`\``;
+      }
+      tokenCount += content.length / 4; // Rough estimate (1 token ~= 4 chars)
+    }
+
+    if (options.dryRun) {
+      spinner.succeed(chalk.green("Dry run complete."));
+      console.log(chalk.bold("\nFile Structure:"));
+      console.log(tree);
+    } else {
+      // Final Payload
+      const finalOutput = `# Project Context\n\n## File Structure\n\`\`\`\n${tree}\n\`\`\`\n${output}`;
+
+      if (options.output) {
+        fs.writeFileSync(path.join(dir, options.output), finalOutput);
+        spinner.succeed(chalk.green(`Context saved to ${options.output} 📝`));
+      } else {
+        const LARGE_FILE_LIMIT = 500000; // ~500KB
+        if (finalOutput.length > LARGE_FILE_LIMIT) {
+          spinner.stop();
+          console.log(
+            chalk.yellow(
+              `\n⚠️  Warning: Output is large (${Math.round(
+                finalOutput.length / 1024
+              )}KB).`
+            )
+          );
+
+          const { proceed } = await inquirer.prompt([
+            {
+              type: "confirm",
+              name: "proceed",
+              message: "Copying this might freeze your clipboard. Proceed?",
+              default: false,
+            },
+          ]);
+
+          if (!proceed)
+            return console.log(
+              chalk.blue("Aborted. Try using --output <file> instead.")
+            );
+          spinner.start("Copying to clipboard...");
+        }
+
+        await clipboardy.write(finalOutput);
+        spinner.succeed(chalk.green("Context copied to clipboard! 📋"));
+      }
+    }
+    console.log(chalk.dim(`\nStats:`));
+    console.log(chalk.dim(`- Files: ${files.length}`));
+    console.log(chalk.dim(`- Est. Tokens: ~${Math.round(tokenCount)}`));
+  } catch (error) {
+    spinner.fail("Failed to copy context.");
+    console.error(error);
+  }
+}

package/src/utils/engine.js

@@ -30,25 +30,42 @@ async function getPRDiff() {
   const octokit = github.getOctokit(token);
 
   const context = github.context;
-  const prNumber = context.payload.pull_request?.number;
 
-  if (!prNumber) {
-    throw new Error(
-      "❌ No Pull Request found in context. Are you running this on push?"
-    );
-  }
+  if (context.payload.pull_request) {
+    const prNumber = context.payload.pull_request.number;
+    // Fetch the diff specifically for PR
+    const { data: diff } = await octokit.rest.pulls.get({
+      owner: context.repo.owner,
+      repo: context.repo.repo,
+      pull_number: prNumber,
+      mediaType: {
+        format: "diff", // Ask GitHub to return the raw diff text
+      },
+    });
+    return diff;
+  } else if (context.eventName === "push") {
+    const { before, after, repository } = context.payload;
+
+    let base = before;
+    // Handle new branch creation (before commit is zeroed out)
+    if (before === "0000000000000000000000000000000000000000") {
+      base = repository.default_branch;
+    }
 
-  // Fetch the diff specifically
-  const { data: diff } = await octokit.rest.pulls.get({
-    owner: context.repo.owner,
-    repo: context.repo.repo,
-    pull_number: prNumber,
-    mediaType: {
-      format: "diff", // Ask GitHub to return the raw diff text
-    },
-  });
+    // Compare head commit against previous commit
+    const { data: diff } = await octokit.rest.repos.compareCommits({
+      owner: context.repo.owner,
+      repo: context.repo.repo,
+      base,
+      head: after,
+      mediaType: { format: "diff" },
+    });
+    return diff;
+  }
 
-  return diff;
+  throw new Error(
+    "❌ Unsupported event type. Only PRs and Pushes are supported."
+  );
 }
 
 async function generateReview(diff) {
@@ -120,13 +137,23 @@ export async function run() {
       const octokit = github.getOctokit(token);
       const context = github.context;
 
-      await octokit.rest.issues.createComment({
-        owner: context.repo.owner,
-        repo: context.repo.repo,
-        issue_number: context.payload.pull_request.number,
-        body: review,
-      });
-      console.log("✅ Review posted to GitHub PR!");
+      if (context.payload.pull_request) {
+        await octokit.rest.issues.createComment({
+          owner: context.repo.owner,
+          repo: context.repo.repo,
+          issue_number: context.payload.pull_request.number,
+          body: review,
+        });
+        console.log("✅ Review posted to GitHub PR!");
+      } else if (context.eventName === "push") {
+        await octokit.rest.repos.createCommitComment({
+          owner: context.repo.owner,
+          repo: context.repo.repo,
+          commit_sha: context.payload.after,
+          body: review,
+        });
+        console.log("✅ Review posted to Commit!");
+      }
     } else {
       // Locally, we just print it
       console.log("\n================ 🤖 AI CODE REVIEW ================ \n");

package/src/utils/file-reader.js

@@ -0,0 +1,58 @@
+import glob from "fast-glob";
+import ignore from "ignore";
+import fs from "fs";
+import path from "path";
+import { isBinaryFile } from "isbinaryfile";
+
+export async function getValidFiles(dir) {
+  const ig = ignore();
+
+  // 1. Default Ignores (Standard noise)
+  ig.add([
+    ".git",
+    "node_modules",
+    "dist",
+    "build",
+    "coverage",
+    ".DS_Store",
+    "package-lock.json",
+    "yarn.lock",
+    "pnpm-lock.yaml",
+    ".env",
+    ".env.local",
+    "*.png",
+    "*.jpg",
+    "*.jpeg",
+    "*.gif",
+    "*.ico",
+    "*.svg", // Images
+  ]);
+
+  // 2. Load .gitignore if exists
+  const gitignorePath = path.join(dir, ".gitignore");
+  if (fs.existsSync(gitignorePath)) {
+    const gitignoreContent = fs.readFileSync(gitignorePath, "utf-8");
+    ig.add(gitignoreContent);
+  }
+
+  // 3. Scan Files (Fast Glob)
+  const allFiles = await glob("**/*", {
+    cwd: dir,
+    dot: true,
+    ignore: ["**/node_modules/**", "**/.git/**"], // Optimization: Hard ignore
+    onlyFiles: true,
+  });
+
+  // 4. Filter Results
+  const validFiles = [];
+  for (const file of allFiles) {
+    // Check .gitignore rules
+    if (ig.ignores(file)) continue;
+
+    // Check if Binary (prevents garbage text)
+    const isBinary = await isBinaryFile(path.join(dir, file));
+    if (!isBinary) validFiles.push(file);
+  }
+
+  return validFiles;
+}