cortxt/src/commands/tree.js

AI-friendly CLI to share project context or file code easily. The fastest way to provide project context to AI intelligence like ChatGPT, Claude, and other AI assistants.

import fs from "fs";
import path from "path";
import { shouldIgnore } from "../utils/ignore.js";
import { copyToClipboard } from "../utils/clipboard.js";
import { colors } from "../utils/colors.js";

export async function runTree(options = {}) {
  const maxDepth = parseInt(options.depth) || 3;
  const rootDir = process.cwd();

  console.log(
    `${colors.folder} Project Structure ${colors.path(
      "(depth:"
    )} ${colors.number(maxDepth)}${colors.path(")")}\n`
  );

  const tree = buildTree(rootDir, "", maxDepth, 0);
  const projectName = path.basename(rootDir);
  const output = `📁 ${projectName}/\n${tree}`;

  console.log(output);
  copyToClipboard(output);
  console.log(
    `\n${colors.success("✅ Project tree copied to clipboard!")} ${colors.brand(
      "Paste to any AI"
    )} ✨`
  );
}

function buildTree(dir, prefix = "", maxDepth = 3, currentDepth = 0) {
  if (currentDepth >= maxDepth) return "";

  try {
    const files = fs.readdirSync(dir);
    let result = "";

    // Filter and sort files
    const validFiles = files
      .filter((file) => {
        const fullPath = path.join(dir, file);
        return !shouldIgnore(fullPath, file, dir);
      })
      .sort();

    validFiles.forEach((file, index) => {
      const fullPath = path.join(dir, file);
      const isLast = index === validFiles.length - 1;
      const connector = isLast ? "└── " : "├── ";
      const nextPrefix = prefix + (isLast ? "    " : "│   ");

      try {
        const stat = fs.statSync(fullPath);
        const icon = stat.isDirectory() ? "📁 " : "📄 ";
        result += `${prefix}${connector}${icon}${file}\n`;

        if (stat.isDirectory()) {
          result += buildTree(fullPath, nextPrefix, maxDepth, currentDepth + 1);
        }
      } catch (err) {
        // Skip files we can't read
      }
    });

    return result;
  } catch (err) {
    return "";
  }
}