cheatchat 1.1.0 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cheatchat",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "",
   "bin": {
     "chat": "./bin/chat.js"
@@ -27,6 +27,7 @@
     "cli-table3": "^0.6.5",
     "commander": "^15.0.0",
     "dotenv": "^17.4.2",
+    "fs-extra": "^11.3.6",
     "inquirer": "^14.0.2",
     "ora": "^9.4.1",
     "ws": "^8.21.0"

package/src/commands/ai.js

@@ -0,0 +1,79 @@
+import readline from "node:readline";
+import { aiChat } from "../services/api.js";
+import { showBanner } from "../ui/banner.js";
+import { createSpinner } from "../ui/spinner.js";
+import {
+  printBlank,
+  printSystem,
+  printError,
+} from "../ui/printer.js";
+import { brand, accent, muted, whiteBold, success as successColor } from "../ui/color.js";
+
+/**
+ * Interactive AI chat mode.
+ * Completely separate from the main chat application.
+ */
+export async function aiCommand() {
+  showBanner();
+  printBlank();
+  console.log(`  ${whiteBold("🤖 AI Chat Mode")}`);
+  console.log(`  ${muted("Ask anything. Type")} ${brand("exit")} ${muted("to quit.")}`);
+  printBlank();
+
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: `  ${accent("You")} ${brand("›")} `,
+    terminal: true,
+  });
+
+  rl.prompt();
+
+  rl.on("line", async (input) => {
+    const line = input.trim();
+
+    if (!line) {
+      rl.prompt();
+      return;
+    }
+
+    if (line.toLowerCase() === "exit") {
+      printBlank();
+      printSystem("Exiting AI chat. Goodbye! 👋");
+      printBlank();
+      rl.close();
+      process.exit(0);
+      return;
+    }
+
+    const spinner = createSpinner("Thinking...");
+    spinner.start();
+
+    const result = await aiChat(line);
+
+    if (result.reply) {
+      spinner.stop();
+      console.log();
+      console.log(`  ${successColor("AI")} ${brand("›")} ${result.reply}`);
+      console.log();
+    } else {
+      spinner.fail("AI error");
+      printError(result.error || "Failed to get a response");
+      printBlank();
+    }
+
+    rl.prompt();
+  });
+
+  rl.on("close", () => {
+    process.exit(0);
+  });
+
+  process.on("SIGINT", () => {
+    printBlank();
+    printSystem("Exiting AI chat. Goodbye! 👋");
+    printBlank();
+    rl.close();
+    process.exit(0);
+  });
+}

package/src/commands/help.js

@@ -38,5 +38,18 @@ export function helpCommand() {
     console.log(`  ${accent(cmd.padEnd(20))} ${muted(desc)}`);
   }
 
+  printBlank();
+  console.log(`  ${whiteBold("AI Tools:")}`);
+  printBlank();
+
+  const aiCommands = [
+    ["chat ai", "Interactive AI chat"],
+    ["chat solve <file>", "Solve/improve a file with AI"],
+  ];
+
+  for (const [cmd, desc] of aiCommands) {
+    console.log(`  ${brand(cmd.padEnd(20))} ${muted(desc)}`);
+  }
+
   printBlank();
 }

package/src/commands/solve.js

@@ -0,0 +1,87 @@
+import path from "node:path";
+import fs from "fs-extra";
+import { aiFile } from "../services/api.js";
+import { showBanner } from "../ui/banner.js";
+import { createSpinner } from "../ui/spinner.js";
+import {
+  printBlank,
+  printSuccess,
+  printError,
+} from "../ui/printer.js";
+import { muted } from "../ui/color.js";
+
+/**
+ * Strip markdown code fences from AI response (client-side safety net).
+ */
+function stripCodeFences(text) {
+  const trimmed = text.trim();
+  const fenceRegex = /^```[\w]*\n?([\s\S]*?)\n?```$/;
+  const match = trimmed.match(fenceRegex);
+  if (match) {
+    return match[1];
+  }
+  return trimmed;
+}
+
+/**
+ * Solve/improve a file using AI.
+ * Reads the file, sends it to POST /ai/file, overwrites with the result.
+ */
+export async function solveCommand(filePath) {
+  showBanner();
+  printBlank();
+
+  // Resolve the file path relative to cwd
+  const resolvedPath = path.resolve(process.cwd(), filePath);
+  const filename = path.basename(resolvedPath);
+
+  // Check if file exists
+  if (!(await fs.pathExists(resolvedPath))) {
+    printError(`File not found: ${resolvedPath}`);
+    printBlank();
+    return;
+  }
+
+  // Check it's a file, not a directory
+  const stat = await fs.stat(resolvedPath);
+  if (!stat.isFile()) {
+    printError(`Not a file: ${resolvedPath}`);
+    printBlank();
+    return;
+  }
+
+  // Read file content
+  const content = await fs.readFile(resolvedPath, "utf-8");
+
+  if (content.trim().length === 0) {
+    printError("File is empty. Nothing to solve.");
+    printBlank();
+    return;
+  }
+
+  console.log(`  ${muted("File:")} ${filename}`);
+  console.log(`  ${muted("Size:")} ${content.length} characters`);
+  printBlank();
+
+  const spinner = createSpinner("🤖 Solving...");
+  spinner.start();
+
+  const result = await aiFile(filename, content);
+
+  if (result.content) {
+    // Strip code fences if present (client-side safety)
+    const cleanContent = stripCodeFences(result.content);
+
+    // Overwrite the original file
+    await fs.writeFile(resolvedPath, cleanContent, "utf-8");
+
+    spinner.succeed("🤖 Solving complete!");
+    printBlank();
+    printSuccess(`File updated: ${resolvedPath}`);
+  } else {
+    spinner.fail("Solving failed");
+    printError(result.error || "Failed to process the file");
+  }
+
+  printBlank();
+}

package/src/index.js

@@ -5,6 +5,8 @@ import { loginCommand } from "./commands/login.js";
 import { logoutCommand } from "./commands/logout.js";
 import { helpCommand } from "./commands/help.js";
 import { connectCommand } from "./commands/connect.js";
+import { aiCommand } from "./commands/ai.js";
+import { solveCommand } from "./commands/solve.js";
 import { whoamiFlow } from "./auth/whoami.js";
 import { APP_NAME, APP_VERSION, APP_DESCRIPTION } from "./utils/constants.js";
 
@@ -67,6 +69,20 @@ program
     helpCommand();
   });
 
+program
+  .command("ai")
+  .description("Enter interactive AI chat mode")
+  .action(async () => {
+    await aiCommand();
+  });
+
+program
+  .command("solve <file>")
+  .description("Solve/improve a source file using AI")
+  .action(async (file) => {
+    await solveCommand(file);
+  });
+
 // ─── Default (no command) ────────────────────────────────────────────────────
 
 if (process.argv.length <= 2) {

package/src/services/api.js

@@ -53,6 +53,34 @@ export async function getMe(token) {
   }
 }
 
+// ─── AI Endpoints (Public — No Auth Required) ───────────────────────────────
+
+/**
+ * Send a prompt to the AI chat endpoint.
+ * Returns { reply } or { error }
+ */
+export async function aiChat(prompt) {
+  try {
+    const { data } = await api.post("/ai/chat", { prompt }, { timeout: 60000 });
+    return data;
+  } catch (err) {
+    return extractError(err);
+  }
+}
+
+/**
+ * Send a file to the AI file solver endpoint.
+ * Returns { content } or { error }
+ */
+export async function aiFile(filename, content) {
+  try {
+    const { data } = await api.post("/ai/file", { filename, content }, { timeout: 60000 });
+    return data;
+  } catch (err) {
+    return extractError(err);
+  }
+}
+
 // ─── Error Extraction ────────────────────────────────────────────────────────
 
 function extractError(err) {