@sanjay5114/cdx 1.0.0

package/index.js

@@ -0,0 +1,530 @@
+#!/usr/bin/env node
+"use strict";
+
+/**
+ * CDX v3.0.0 Enterprise
+ * Always checks auth on startup. Gates all features behind login.
+ * Dex (animated mascot) greets and guides the user throughout.
+ */
+
+const { Command } = require("commander");
+const chalk       = require("chalk");
+
+const UI = require("./lib/ui");
+const {
+  T, W, rule, center, printBanner, section, ok, warn, err, info, note,
+  spinner, table, badge, panel, ICONS, footer, typewrite,
+  dex, dexAnimate, dexGreet,
+} = UI;
+
+const store = require("./lib/store");
+const auth  = require("./lib/auth");
+
+const sleep = ms => new Promise(r => setTimeout(r, ms));
+
+// ─────────────────────────────────────────────────────────────────────────────
+//  AUTH GATE — runs before any command action
+//  Returns the logged-in user, or null if login was cancelled/failed
+// ─────────────────────────────────────────────────────────────────────────────
+async function requireAuth(inquirer) {
+  // 1. Try to restore existing session silently
+  const spin = spinner("Checking session…").start();
+  await sleep(300);
+  const user = await auth.restoreSession().catch(() => null);
+
+  if (user) {
+    spin.ok(`Session active — ${user.displayName || user.email}`);
+    await dexGreet(user.displayName || user.email.split("@")[0]);
+    return user;
+  }
+
+  spin.warn("No active session — authentication required");
+  console.log();
+
+  // 2. Show Dex prompting login
+  dex("Hey there! I'm Dex, your documentation assistant. You'll need to sign in before we can get started.", "idle");
+
+  // 3. Login / signup choice
+  const { authChoice } = await inquirer.prompt([{
+    type   : "list",
+    name   : "authChoice",
+    message: T.brand("What would you like to do?"),
+    choices: [
+      { name: `  ${T.brand("◉")}  ${T.whiteBold("Sign In")}    ${T.dim("I already have an account")}`, value: "login" },
+      { name: `  ${T.success("◉")}  ${T.whiteBold("Sign Up")}    ${T.dim("Create a new CDX account")}`,  value: "signup" },
+      { name: `  ${T.dim("◉")}  ${T.dim("Exit")}`,                                                       value: "exit" },
+    ],
+    prefix: `   ${T.accent(ICONS.arrow)}`,
+  }]);
+
+  if (authChoice === "exit") {
+    await typewrite(`\n   ${T.dim("Goodbye! Come back when you're ready.")}`, 18);
+    footer();
+    process.exit(0);
+  }
+
+  const apiKey = store.get("firebaseApiKey");  // always returns bundled key
+
+  if (authChoice === "login") {
+    return await doLogin(inquirer, apiKey);
+  } else {
+    return await doSignup(inquirer, apiKey);
+  }
+}
+
+async function doLogin(inquirer, apiKey) {
+  section("Sign In", "Firebase Authentication");
+  dex("Enter your credentials and I'll get you right in!", "think");
+
+  let attempts = 0;
+  while (attempts < 3) {
+    const { email } = await inquirer.prompt([{
+      type    : "input",
+      name    : "email",
+      message : T.brand("Email address:"),
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: v => v.includes("@") || "Please enter a valid email address.",
+    }]);
+    const { password } = await inquirer.prompt([{
+      type    : "password",
+      name    : "password",
+      message : T.brand("Password:"),
+      mask    : "•",
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: v => v.length >= 6 || "Password must be at least 6 characters.",
+    }]);
+
+    console.log();
+    const anim = dexAnimate("Authenticating with Firebase…", "think");
+    await sleep(300);
+
+    try {
+      const session = await auth.signIn(apiKey, email, password);
+      auth.persistSession(session);
+      anim.stop("Authentication successful!", "happy");
+      console.log();
+      await dexGreet(session.displayName || session.email.split("@")[0]);
+      return session;
+    } catch (e) {
+      anim.stop("Authentication failed — " + e.message, "error");
+      attempts++;
+      if (attempts < 3) {
+        dex("No worries! Double-check your credentials and let's try again.", "idle");
+        const { retry } = await inquirer.prompt([{
+          type   : "confirm",
+          name   : "retry",
+          message: T.brand("Try again?"),
+          default: true,
+          prefix : `   ${T.accent(ICONS.arrow)}`,
+        }]);
+        if (!retry) break;
+      }
+    }
+  }
+
+  err("Could not authenticate. Run  cdx auth signup  to create a new account.");
+  footer();
+  process.exit(1);
+}
+
+async function doSignup(inquirer, apiKey) {
+  section("Create Account", "Firebase Authentication");
+  dex("Awesome! Let me set up your CDX account. This only takes a moment.", "happy");
+
+  const answers = await inquirer.prompt([
+    {
+      type    : "input",
+      name    : "displayName",
+      message : T.brand("Display name:"),
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: v => v.trim().length >= 2 || "Display name must be at least 2 characters.",
+    },
+    {
+      type    : "input",
+      name    : "email",
+      message : T.brand("Email address:"),
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: v => v.includes("@") || "Please enter a valid email address.",
+    },
+    {
+      type    : "password",
+      name    : "password",
+      message : T.brand("Password (min. 8 characters):"),
+      mask    : "•",
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: v => v.length >= 8 || "Password must be at least 8 characters.",
+    },
+    {
+      type    : "password",
+      name    : "confirm",
+      message : T.brand("Confirm password:"),
+      mask    : "•",
+      prefix  : `   ${T.accent(ICONS.arrow)}`,
+      validate: (v, a) => v === a.password || "Passwords do not match.",
+    },
+  ]);
+
+  console.log();
+  const anim = dexAnimate("Creating your account…", "think");
+  await sleep(400);
+
+  try {
+    const session = await auth.signUp(apiKey, answers.email, answers.password, answers.displayName);
+    auth.persistSession(session);
+    anim.stop("Account created successfully!", "happy");
+    console.log();
+    ok(`Welcome to CDX, ${session.displayName}!`);
+    note("Your session is stored securely at ~/.cdx/.session");
+    console.log();
+    await dexGreet(session.displayName);
+    return session;
+  } catch (e) {
+    anim.stop("Registration failed — " + e.message, "error");
+    err(e.message);
+    footer();
+    process.exit(1);
+  }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+//  COMMANDER
+// ─────────────────────────────────────────────────────────────────────────────
+const program = new Command();
+
+program
+  .name("cdx")
+  .description("CDX Enterprise — SaaS-grade AI documentation generation")
+  .version("3.0.0", "-v, --version")
+  .addHelpText("beforeAll", () => { printBanner(); return ""; })
+  .configureHelp({ helpWidth: W, sortSubcommands: true });
+
+// ── create ─────────────────────────────────────────────────────────────────────
+program
+  .command("create <filename>")
+  .description("Generate comprehensive documentation for the current project")
+  .option("-a, --all",        "Include hidden files and directories")
+  .option("--push",           "Push generated docs to GitHub immediately")
+  .option("--no-interactive", "Skip post-generation improvement loop")
+  .option("--model <model>",  "Override Gemini model for this run")
+  .action(async (filename, opts) => {
+    printBanner();
+    const inquirer = (await import("inquirer").catch(() => null))?.default;
+    if (!inquirer) { err("'inquirer' not installed."); process.exit(1); }
+
+    const user = await requireAuth(inquirer);
+
+    section("Create Documentation", filename);
+    dex("Let me scan this project and generate enterprise-grade docs for you!", "think");
+
+    table([
+      ["Target file",    chalk.cyan(filename)],
+      ["Signed in as",   user.displayName || user.email],
+      ["Include hidden", opts.all  ? badge("YES", "info")    : badge("NO", "default")],
+      ["GitHub push",    opts.push ? badge("YES", "success") : badge("deferred", "default")],
+      ["Interactive",    opts.interactive !== false ? badge("YES", "success") : badge("NO", "default")],
+    ]);
+    console.log();
+
+    if (opts.model) store.set("geminiModel", opts.model);
+
+    try {
+      await require("./commands/create")(filename, {
+        all: opts.all, push: opts.push,
+        noInteractive: opts.interactive === false,
+        onProgress: msg => process.stdout.write(`\r   ${T.dim(msg)}                                   `),
+      });
+    } catch (e) {
+      dex("Hmm, something went wrong. Check the error below.", "error");
+      err("Generation failed: " + e.message);
+      if (process.env.CDX_DEBUG) console.error(e.stack);
+      process.exit(1);
+    }
+  });
+
+// ── config ─────────────────────────────────────────────────────────────────────
+program
+  .command("config [category]")
+  .description("Configure CDX settings — categories: ai, firebase, github, output, show, reset")
+  .action(async (category) => {
+    printBanner();
+    const inquirer = (await import("inquirer").catch(() => null))?.default;
+    if (!inquirer) { err("'inquirer' not installed."); process.exit(1); }
+
+    await requireAuth(inquirer);
+    try { await require("./commands/config")(category); }
+    catch (e) { err("Config error: " + e.message); process.exit(1); }
+  });
+
+// ── auth (direct sub-commands, no gate — needed to log in) ────────────────────
+program
+  .command("auth <action>")
+  .description("Firebase auth — actions: login, signup, logout, whoami, reset-password")
+  .action(async (action) => {
+    printBanner();
+    try { await require("./commands/auth")(action); }
+    catch (e) { err("Auth error: " + e.message); process.exit(1); }
+  });
+
+// ── start ──────────────────────────────────────────────────────────────────────
+program
+  .command("start")
+  .description("Launch interactive CDX terminal UI")
+  .action(async () => {
+    printBanner();
+
+    const inquirer = (await import("inquirer").catch(() => null))?.default;
+    if (!inquirer) { err("'inquirer' not installed."); return; }
+
+    // ── ALWAYS authenticate first ─────────────────────────────────────────────
+    const user = await requireAuth(inquirer);
+
+    // ── System info ───────────────────────────────────────────────────────────
+    const bootSpin = spinner("Initialising CDX Enterprise shell…").start();
+    await sleep(350);
+    bootSpin.ok("Shell ready");
+    console.log();
+
+    section("System Information");
+    const cfg = store.load();
+    table([
+      ["CDX Version",  "3.0.0 Enterprise"],
+      ["Signed In As", user.displayName || user.email],
+      ["Node.js",      process.version],
+      ["Platform",     process.platform + " / " + process.arch],
+      ["Working Dir",  process.cwd().length > 42 ? "…" + process.cwd().slice(-40) : process.cwd()],
+      ["Gemini Key",   cfg.geminiApiKey   ? badge("Configured", "success") : badge("Not set — run cdx config ai", "warn")],
+      ["GitHub Token", cfg.githubToken    ? badge("Configured", "success") : badge("Not set", "default")],
+      ["Firebase",     badge("Auto-configured (mkchit)", "info")],
+    ]);
+
+    if (!cfg.geminiApiKey) {
+      console.log();
+      dex("Quick heads up — you haven't configured a Gemini API key yet. Without it, I can only generate basic local docs. Run config → AI Settings to add your key!", "idle");
+    }
+
+    // ── Main menu loop ─────────────────────────────────────────────────────────
+    while (true) {
+      const freshCfg  = store.load();
+      const freshUser = await auth.restoreSession().catch(() => null);
+
+      // Re-auth if session expired mid-session
+      if (!freshUser) {
+        warn("Session expired. Please sign in again.");
+        const reUser = await requireAuth(inquirer);
+        if (!reUser) { process.exit(0); }
+        continue;
+      }
+
+      const MENU = [
+        {
+          name : `  ${T.success("◆")}  ${T.whiteBold("Create Docs")} ${T.success("(local)")}        ${T.dim("Analyze current directory & generate")}`,
+          value: "create-local",
+        },
+        {
+          name : `  ${T.brand("◆")}  ${T.whiteBold("Create Docs")} ${T.brand("(GitHub)")}       ${T.dim("Remote repo → analyze → push")}` +
+                 (freshCfg.githubToken ? "" : T.warn("  ⚠ token needed")),
+          value: "create-github",
+        },
+        {
+          name : `  ${T.violet("◆")}  ${T.whiteBold("Configuration")}               ${T.dim("Manage AI, GitHub & output settings")}`,
+          value: "config",
+        },
+        {
+          name : `  ${T.warn("◆")}  ${T.whiteBold("Account")}                    ${T.dim("Session info / sign out  ")}` +
+                 badge("● " + (freshUser.email || "").slice(0, 20), "success"),
+          value: "account",
+        },
+        {
+          name : `  ${T.muted("◆")}  ${T.muted("Exit")}`,
+          value: "exit",
+        },
+      ];
+
+      console.log();
+      console.log(rule("─", T.dim));
+      console.log();
+
+      const { action } = await inquirer.prompt([{
+        type    : "list",
+        name    : "action",
+        message : T.brand("What would you like to do?"),
+        choices : MENU,
+        prefix  : `   ${T.accent(ICONS.arrow)}`,
+        pageSize: 10,
+      }]);
+      console.log();
+
+      // ── CREATE LOCAL ──────────────────────────────────────────────────────────
+      if (action === "create-local") {
+        section("Create Documentation", "Local Project");
+        dex("I'll scan your project, detect the stack, and produce detailed docs across multiple passes. Let's go!", "think");
+
+        const answers = await inquirer.prompt([
+          {
+            type    : "input",
+            name    : "filename",
+            message : T.brand("Output filename:"),
+            default : freshCfg.defaultOutput || "README.md",
+            prefix  : `   ${T.accent(ICONS.arrow)}`,
+          },
+          {
+            type    : "confirm",
+            name    : "all",
+            message : T.brand("Include hidden files/directories?"),
+            default : false,
+            prefix  : `   ${T.accent(ICONS.arrow)}`,
+          },
+          {
+            type    : "confirm",
+            name    : "push",
+            message : T.brand("Push to GitHub after generation?"),
+            default : freshCfg.autoPush || false,
+            prefix  : `   ${T.accent(ICONS.arrow)}`,
+            when    : () => !!(freshCfg.githubToken && freshCfg.githubRepo),
+          },
+        ]);
+        console.log();
+
+        try {
+          await require("./commands/create")(answers.filename, {
+            all        : answers.all,
+            push       : answers.push,
+            onProgress : msg => process.stdout.write(`\r   ${T.dim(msg)}                              `),
+          });
+          dex("Docs generated and saved! Check the improvement menu that just appeared to refine them further.", "happy");
+        } catch (e) {
+          dex("Something went wrong during generation. Check the error below.", "error");
+          err("Generation failed: " + e.message);
+        }
+
+      // ── CREATE GITHUB ──────────────────────────────────────────────────────────
+      } else if (action === "create-github") {
+        if (!freshCfg.githubToken) {
+          dex("You'll need to add a GitHub token first. Head to Configuration → GitHub Integration.", "idle");
+          warn("GitHub token not configured.");
+        } else {
+          section("Create Documentation", "GitHub Repository");
+
+          const fetchSpin = spinner("Fetching your GitHub repositories…").start();
+          let repos = [];
+          try {
+            const { fetchGitHubRepos } = require("./lib/scanner");
+            repos = await fetchGitHubRepos(freshCfg.githubToken);
+            fetchSpin.ok(`Fetched ${repos.length} repositories`);
+          } catch (e) {
+            fetchSpin.fail("Failed: " + e.message);
+            dex("Couldn't reach GitHub. Check your token and internet connection.", "error");
+            continue;
+          }
+
+          const repoChoices = [
+            ...repos.map(r => ({
+              name : `  ${r.private ? T.dim("🔒") : T.accent("◆")}  ${T.white(r.full_name)}  ${T.dim((r.description || "").slice(0, 48))}`,
+              value: r.full_name,
+            })),
+            new inquirer.Separator(),
+            { name: `  ${T.dim("↩  Cancel")}`, value: null },
+          ];
+
+          const { selectedRepo } = await inquirer.prompt([{
+            type    : "list",
+            name    : "selectedRepo",
+            message : T.brand("Select a repository:"),
+            choices : repoChoices,
+            prefix  : `   ${T.accent(ICONS.arrow)}`,
+            pageSize: 16,
+          }]);
+
+          if (!selectedRepo) { info("Cancelled."); continue; }
+
+          const [owner, repo] = selectedRepo.split("/");
+          store.set("githubRepo", selectedRepo);
+          ok(`Target → ${chalk.cyan(selectedRepo)}`);
+          console.log();
+
+          dex("Great choice! I'll fetch the source tree, analyze every module, and push the docs back automatically.", "think");
+
+          const { confirmGen } = await inquirer.prompt([{
+            type    : "confirm",
+            name    : "confirmGen",
+            message : T.brand(`Generate & push docs for ${selectedRepo}?`),
+            default : true,
+            prefix  : `   ${T.accent(ICONS.arrow)}`,
+          }]);
+
+          if (!confirmGen) { info("Cancelled."); continue; }
+
+          try {
+            await require("./commands/create")("README.md", {
+              remoteGithub: { owner, repo, token: freshCfg.githubToken },
+              push        : true,
+              onProgress  : msg => process.stdout.write(`\r   ${T.dim(msg)}                              `),
+            });
+          } catch (e) {
+            dex("Hmm, the remote generation hit an issue.", "error");
+            err("Generation failed: " + e.message);
+          }
+        }
+
+      // ── CONFIG ─────────────────────────────────────────────────────────────────
+      } else if (action === "config") {
+        await require("./commands/config")();
+
+      // ── ACCOUNT ────────────────────────────────────────────────────────────────
+      } else if (action === "account") {
+        section("Account", freshUser.email);
+
+        table([
+          ["Display Name", freshUser.displayName || T.dim("—")],
+          ["Email",        freshUser.email],
+          ["User ID",      freshUser.uid || T.dim("—")],
+          ["Session",      badge("Active", "success")],
+          ["Firebase",     badge("mkchit project", "info")],
+        ]);
+        console.log();
+
+        const { accountAction } = await inquirer.prompt([{
+          type   : "list",
+          name   : "accountAction",
+          message: T.brand("Account options:"),
+          choices: [
+            { name: `  ${T.warn("◆")}  ${T.white("Sign Out")}  ${T.dim("Clear session and return to login")}`, value: "logout" },
+            { name: `  ${T.dim("◆")}  ${T.dim("Back")}`,                                                        value: "back" },
+          ],
+          prefix : `   ${T.accent(ICONS.arrow)}`,
+        }]);
+
+        if (accountAction === "logout") {
+          auth.signOut();
+          dex("Signed out successfully. Come back soon!", "happy");
+          ok("Session cleared.");
+          console.log();
+          // Force re-auth on next loop iteration
+          const reUser = await requireAuth(inquirer);
+          if (!reUser) process.exit(0);
+        }
+
+      // ── EXIT ───────────────────────────────────────────────────────────────────
+      } else if (action === "exit") {
+        dex("Great work today! Your docs are in safe hands. See you next time!", "happy");
+        await sleep(400);
+        await typewrite(`   ${T.dim("Goodbye, " + (freshUser.displayName || "friend") + "! Happy shipping! 🚀")}`, 18);
+        console.log();
+        footer();
+        process.exit(0);
+      }
+    }
+  });
+
+// ── unknown commands ──────────────────────────────────────────────────────────
+program.on("command:*", cmds => {
+  printBanner();
+  err(`Unknown command: ${chalk.bold(cmds[0])}`);
+  console.log();
+  info(`Valid commands: ${["create","config","auth","start"].map(c => T.accent(c)).join("  ")}`);
+  note(`Run ${T.accent("cdx --help")} for full usage.`);
+  console.log();
+  process.exit(1);
+});
+
+program.parse(process.argv);