cheatchat 1.0.0 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cheatchat",
-  "version": "1.0.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/connect.js

@@ -4,6 +4,7 @@ import { connectWebSocket } from "../websocket/connect.js";
 import { disconnectWebSocket } from "../websocket/disconnect.js";
 import { handleIncomingMessage } from "../websocket/receiver.js";
 import { sendMessage } from "../websocket/sender.js";
+import { requestHistory, requestOnline, requestUsers } from "../services/socket.js";
 import { showChatHeader } from "../ui/banner.js";
 import {
   printChatHelp,
@@ -109,8 +110,13 @@ export async function connectCommand() {
   showChatHeader(username, chatMode, dmTarget);
   printBlank();
   printSystem("You are in Global Chat. Type a message or /help for commands.");
+  printSystem("Loading chat history...");
   printBlank();
 
+  // ─── Auto-load global history on connect ─────────────────────────────
+
+  requestHistory(ws, "global");
+
   // ─── Start Interactive Readline ──────────────────────────────────────
 
   rl = readline.createInterface({
@@ -141,6 +147,8 @@ export async function connectCommand() {
           chatMode = CHAT_MODE.GLOBAL;
           dmTarget = null;
           printModeSwitch("global");
+          // Auto-load global history on mode switch
+          requestHistory(ws, "global");
           refreshPrompt();
           return;
 
@@ -159,10 +167,31 @@ export async function connectCommand() {
           chatMode = CHAT_MODE.PRIVATE;
           dmTarget = target;
           printModeSwitch("private", target);
+          // Auto-load private history when switching to DM
+          requestHistory(ws, "private", target);
           refreshPrompt();
           return;
         }
 
+        case "/history":
+          if (chatMode === CHAT_MODE.GLOBAL) {
+            requestHistory(ws, "global");
+          } else if (chatMode === CHAT_MODE.PRIVATE && dmTarget) {
+            requestHistory(ws, "private", dmTarget);
+          }
+          rl.prompt();
+          return;
+
+        case "/online":
+          requestOnline(ws);
+          rl.prompt();
+          return;
+
+        case "/users":
+          requestUsers(ws);
+          rl.prompt();
+          return;
+
         case "/help":
           printChatHelp();
           rl.prompt();

package/src/commands/help.js

@@ -27,6 +27,9 @@ export function helpCommand() {
   const chatCommands = [
     ["/global", "Switch to global chat"],
     ["/dm <username>", "Switch to private messaging"],
+    ["/history", "Load message history"],
+    ["/online", "Show online users"],
+    ["/users", "Show all registered users"],
     ["/help", "Show chat commands"],
     ["/exit", "Disconnect and exit"],
   ];
@@ -35,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) {

package/src/services/socket.js

@@ -69,6 +69,37 @@ export function sendPrivateMessage(ws, to, text) {
   return true;
 }
 
+/**
+ * Request chat history (global or private).
+ */
+export function requestHistory(ws, type, withUser, limit = 50) {
+  if (ws.readyState !== WebSocket.OPEN) return false;
+  const payload = { action: "history", type, limit };
+  if (type === "private" && withUser) {
+    payload.with = withUser;
+  }
+  ws.send(JSON.stringify(payload));
+  return true;
+}
+
+/**
+ * Request the list of currently online users.
+ */
+export function requestOnline(ws) {
+  if (ws.readyState !== WebSocket.OPEN) return false;
+  ws.send(JSON.stringify({ action: "online" }));
+  return true;
+}
+
+/**
+ * Request the list of all registered users.
+ */
+export function requestUsers(ws) {
+  if (ws.readyState !== WebSocket.OPEN) return false;
+  ws.send(JSON.stringify({ action: "users" }));
+  return true;
+}
+
 /**
  * Close the WebSocket connection gracefully.
  */

package/src/ui/printer.js

@@ -13,6 +13,7 @@ import {
   warning,
   accent,
   brand,
+  white,
   whiteBold,
   dim,
 } from "./color.js";
@@ -93,6 +94,9 @@ export function printChatHelp() {
   console.log(`  ${whiteBold("Chat Commands:")}`);
   console.log(`  ${brand("/global")}          ${muted("Switch to global chat")}`);
   console.log(`  ${brand("/dm <username>")}   ${muted("Switch to private messaging")}`);
+  console.log(`  ${brand("/history")}         ${muted("Load message history for current chat")}`);
+  console.log(`  ${brand("/online")}          ${muted("Show currently connected users")}`);
+  console.log(`  ${brand("/users")}           ${muted("Show all registered users")}`);
   console.log(`  ${brand("/help")}            ${muted("Show this help menu")}`);
   console.log(`  ${brand("/exit")}            ${muted("Disconnect and exit")}`);
   console.log();
@@ -111,6 +115,66 @@ export function printModeSwitch(mode, target) {
   console.log();
 }
 
+/**
+ * Print a history header.
+ */
+export function printHistoryHeader(type, target) {
+  console.log();
+  const sep = muted("─".repeat(50));
+  console.log(`  ${sep}`);
+  if (type === "global") {
+    console.log(`  ${systemTag(" HISTORY ")} ${accent("🌍 Global Chat History")}`);
+  } else {
+    console.log(`  ${systemTag(" HISTORY ")} ${accent(`🔒 DM History with ${whiteBold(target)}`)}`);
+  }
+  console.log(`  ${sep}`);
+}
+
+/**
+ * Print a history footer.
+ */
+export function printHistoryFooter(count) {
+  const sep = muted("─".repeat(50));
+  console.log(`  ${sep}`);
+  console.log(`  ${muted(`${count} message${count !== 1 ? "s" : ""} loaded`)}`);
+  console.log(`  ${sep}`);
+  console.log();
+}
+
+/**
+ * Print online users list.
+ */
+export function printOnlineUsers(users) {
+  console.log();
+  console.log(`  ${whiteBold("Online Users")} ${muted(`(${users.length})`)}`);
+  console.log(`  ${muted("─".repeat(30))}`);
+  if (users.length === 0) {
+    console.log(`  ${muted("No users online")}`);
+  } else {
+    for (const u of users) {
+      console.log(`  ${success("●")} ${accent(u)}`);
+    }
+  }
+  console.log();
+}
+
+/**
+ * Print all registered users list.
+ */
+export function printRegisteredUsers(users) {
+  console.log();
+  console.log(`  ${whiteBold("Registered Users")} ${muted(`(${users.length})`)}`);
+  console.log(`  ${muted("─".repeat(30))}`);
+  if (users.length === 0) {
+    console.log(`  ${muted("No users found")}`);
+  } else {
+    for (const u of users) {
+      console.log(`  ${brand("•")} ${white(u)}`);
+    }
+  }
+  console.log();
+}
+
 /**
  * Print a blank line.
  */

package/src/websocket/receiver.js

@@ -1,4 +1,13 @@
-import { printGlobalMessage, printPrivateMessage, printError, printSystem } from "../ui/printer.js";
+import {
+  printGlobalMessage,
+  printPrivateMessage,
+  printError,
+  printSystem,
+  printHistoryHeader,
+  printHistoryFooter,
+  printOnlineUsers,
+  printRegisteredUsers,
+} from "../ui/printer.js";
 
 /**
  * Handle incoming WebSocket messages and dispatch to the appropriate printer.
@@ -15,6 +24,18 @@ export function handleIncomingMessage(data, currentUser) {
       printPrivateMessage(data.sender, data.text, data.timestamp, currentUser);
       break;
 
+    case "history":
+      renderHistory(data, currentUser);
+      break;
+
+    case "online":
+      printOnlineUsers(data.users || []);
+      break;
+
+    case "users":
+      printRegisteredUsers(data.users || []);
+      break;
+
     case "error":
       printError(data.message || "Unknown server error");
       break;
@@ -24,3 +45,28 @@ export function handleIncomingMessage(data, currentUser) {
       break;
   }
 }
+
+/**
+ * Render a history response (global or private).
+ */
+function renderHistory(data, currentUser) {
+  const messages = data.messages || [];
+  const type = data.type;
+  const target = data.with;
+
+  printHistoryHeader(type, target);
+
+  if (messages.length === 0) {
+    printSystem("No messages yet. Start the conversation!");
+  } else {
+    for (const msg of messages) {
+      if (type === "global") {
+        printGlobalMessage(msg.sender, msg.text, msg.timestamp, currentUser);
+      } else {
+        printPrivateMessage(msg.sender, msg.text, msg.timestamp, currentUser);
+      }
+    }
+  }
+
+  printHistoryFooter(messages.length);
+}