loren-code 0.2.2 → 0.2.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "loren-code",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Ollama Cloud Model Manager - Dynamic model switching, API key rotation, and real-time configuration updates",
   "author": "lorenzune",
   "license": "MIT",

package/scripts/loren.js

@@ -19,15 +19,34 @@ const logFilePath = path.join(runtimeDir, "bridge.log");
 const errorLogFilePath = path.join(runtimeDir, "bridge.err.log");
 const userHome = process.env.USERPROFILE || process.env.HOME || projectRoot;
 const claudeSettingsPath = path.join(userHome, ".claude", "settings.json");
+const displayName = getDisplayName();
 
 process.chdir(projectRoot);
 ensureRuntimeDir();
 const envStatus = ensureEnvLocal(projectRoot, { logger: { warn() {} } });
 
-const BANNER = `
-LOREN CODE
-Smarter bridge, fewer rituals.
-`;
+const ASCII_BANNER_LINES = [
+  "██╗      ██████╗ ██████╗ ███████╗███╗   ██╗     ██████╗ ██████╗ ██████╗ ███████╗",
+  "██║     ██╔═══██╗██╔══██╗██╔════╝████╗  ██║    ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
+  "██║     ██║   ██║██████╔╝█████╗  ██╔██╗ ██║    ██║     ██║   ██║██║  ██║█████╗",
+  "██║     ██║   ██║██╔══██╗██╔══╝  ██║╚██╗██║    ██║     ██║   ██║██║  ██║██╔══╝",
+  "███████╗╚██████╔╝██║  ██║███████╗██║ ╚████║    ╚██████╗╚██████╔╝██████╔╝███████╗",
+  "╚══════╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝╚═╝  ╚═══╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
+];
+const GREEN = "\x1b[32m";
+const YELLOW = "\x1b[33m";
+const RED = "\x1b[31m";
+const CYAN = "\x1b[36m";
+const MAGENTA = "\x1b[35m";
+const RESET = "\x1b[0m";
+const BANNER_COLORS = [
+  "\x1b[38;2;198;218;255m",
+  "\x1b[38;2;190;212;255m",
+  "\x1b[38;2;181;206;255m",
+  "\x1b[38;2;188;201;255m",
+  "\x1b[38;2;196;197;255m",
+  "\x1b[38;2;205;193;255m",
+];
 
 const COMMANDS = {
   model: {
@@ -101,25 +120,8 @@ async function main() {
 }
 
 async function listModels() {
-  const config = loadConfig();
-
   try {
-    const response = await fetch(`${config.upstreamBaseUrl}/api/tags`, {
-      headers: { accept: "application/json" },
-    });
-
-    if (!response.ok) {
-      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-    }
-
-    const data = await response.json();
-    let models = Array.isArray(data.models) ? data.models : [];
-
-    models = models.sort((a, b) => {
-      const dateA = a.modified_at ? new Date(a.modified_at).getTime() : 0;
-      const dateB = b.modified_at ? new Date(b.modified_at).getTime() : 0;
-      return dateB - dateA;
-    });
+    const { config, models } = await fetchAvailableModels();
 
     console.log("\nAvailable models from Ollama Cloud:");
     console.log("─".repeat(70));
@@ -202,12 +204,7 @@ function setModel(args) {
   saveEnvFile(envFilePath, envVars);
   syncClaudeSelectedModel(requestedModel);
 
-  console.log(`\nDefault model set to ${requestedModel}.`);
-  console.log("Fresh requests will use it right away.");
-  if (fs.existsSync(claudeSettingsPath)) {
-    console.log("Claude Code settings were updated too.");
-  }
-  console.log("");
+  return requestedModel;
 }
 
 function showCurrentModel() {
@@ -342,13 +339,16 @@ function showPaths() {
   console.log("");
 }
 
-function startServer() {
+function startServer(options = {}) {
+  const quiet = options.quiet === true;
   const existingPid = readPidFile();
   if (existingPid && isProcessRunning(existingPid)) {
     const config = loadConfig();
-    console.log("\nLoren is already running.");
-    console.log(`URL: ${getBridgeBaseUrl(config)}`);
-    console.log("");
+    if (!quiet) {
+      console.log("\nLoren is already running.");
+      console.log(`URL: ${getBridgeBaseUrl(config)}`);
+      console.log("");
+    }
     return;
   }
 
@@ -364,10 +364,12 @@ function startServer() {
   child.unref();
   fs.writeFileSync(pidFilePath, `${child.pid}\n`, "utf8");
 
-  const config = loadConfig();
-  console.log("\nLoren is up and listening.");
-  console.log(`URL: ${getBridgeBaseUrl(config)}`);
-  console.log("");
+  if (!quiet) {
+    const config = loadConfig();
+    console.log("\nLoren is up and listening.");
+    console.log(`URL: ${getBridgeBaseUrl(config)}`);
+    console.log("");
+  }
 }
 
 function stopServer() {
@@ -495,33 +497,34 @@ async function runSetupWizard(config) {
   });
 
   try {
-    const rawKeys = (await rl.question("Paste your Ollama API key(s), separated by commas: ")).trim();
-
-    if (rawKeys) {
-      const keys = splitKeyList(rawKeys);
-      const envVars = loadEnvFile(envFilePath);
-      envVars.OLLAMA_API_KEYS = keys.join(",");
-      saveEnvFile(envFilePath, envVars);
-      console.log(`\nNice. Loren is holding ${keys.length} key(s) and feeling organized.`);
-    } else {
-      console.log("\nNo keys yet. Loren will wait here and act casual about it.");
-    }
-
-    const startNow = (await rl.question("Start the bridge now? [Y/n] ")).trim().toLowerCase();
-    if (startNow === "" || startNow === "y" || startNow === "yes") {
-      startServer();
-    }
+    const keys = await promptForApiKeys(rl);
+    const envVars = loadEnvFile(envFilePath);
+    envVars.OLLAMA_API_KEYS = keys.join(",");
+    saveEnvFile(envFilePath, envVars);
+    console.log(`${GREEN}✓ Saved ${keys.length} API key(s).${RESET}`);
+    console.log("");
 
     if (process.platform === "win32") {
-      const installClaude = (await rl.question("Install Claude Code integration too? [y/N] ")).trim().toLowerCase();
-      if (installClaude === "y" || installClaude === "yes") {
+      const installClaude = (await askQuestion(rl, "Install Claude Code integration too? [Y/n] ")).trim().toLowerCase();
+      if (installClaude === "" || installClaude === "y" || installClaude === "yes") {
         installClaudeIntegration();
+        console.log(`${GREEN}✓ Claude Code integration installed.${RESET}`);
+        console.log("");
       } else {
         console.log("\nNo problem. You can wire Claude in later.");
       }
     }
 
-    console.log("Setup complete. Fewer steps, fewer goblins.");
+    await promptForModelSelection(rl);
+
+    const startNow = (await askQuestion(rl, "Start the bridge now? [Y/n] ")).trim().toLowerCase();
+    if (startNow === "" || startNow === "y" || startNow === "yes") {
+      startServer({ quiet: true });
+      console.log(`${GREEN}✓ Bridge started.${RESET}`);
+      console.log("");
+    }
+
+    console.log(`${GREEN}Setup complete. Fewer steps, fewer goblins.${RESET}`);
     console.log("");
   } finally {
     rl.close();
@@ -529,35 +532,39 @@ async function runSetupWizard(config) {
 }
 
 function printWizardIntro() {
-  console.log(BANNER);
+  printBanner();
   if (envStatus.migrated) {
-    console.log("Your previous settings were imported automatically.");
+    console.log(`${MAGENTA}Your previous settings were imported automatically.${RESET}`);
   } else if (envStatus.created) {
-    console.log("A fresh config is ready.");
+    console.log(`${GREEN}A fresh config is ready.${RESET}`);
   }
-  console.log("Let's get Loren ready in one quick pass.");
+  console.log(`${CYAN}Welcome${displayName ? `, ${displayName}` : ""}.${RESET}`);
+  console.log(`${YELLOW}Run \`loren\` in an interactive terminal to finish setup.${RESET}`);
+  console.log(`${GREEN}Let's get Loren ready in one quick pass.${RESET}`);
   console.log("");
 }
 
 function printWelcomeBack(config) {
-  console.log(BANNER);
-  console.log(`Welcome back. ${config.apiKeys.length} key(s) loaded.`);
+  printBanner();
+  console.log(`Welcome back${displayName ? `, ${displayName}` : ""}.`);
+  console.log(`${config.apiKeys.length} key(s) loaded.`);
   console.log(`Current default model: ${config.defaultModel}`);
+  console.log(`${GREEN}Run \`loren start\` to launch the bridge.${RESET}`);
   console.log("");
-  console.log("Useful commands:");
-  console.log("  loren start");
-  console.log("  loren model:list");
-  console.log("  loren config:show");
-  console.log("");
+  printCommandSummary();
 }
 
 function printQuickSetup(config) {
   if (config.apiKeys.length > 0) {
-    console.log("Run `loren start` to launch the bridge.");
-    console.log("");
+    printWelcomeBack(config);
     return;
   }
 
+  printBanner();
+  console.log(`Welcome${displayName ? `, ${displayName}` : ""}.`);
+  console.log(`${YELLOW}Run \`loren\` in an interactive terminal to finish setup.${RESET}`);
+  console.log("");
+  printCommandSummary();
   console.log("Quick start:");
   console.log("  1. Run `loren` in an interactive terminal");
   console.log("  2. Add your Ollama API key(s)");
@@ -577,8 +584,94 @@ function installClaudeIntegration() {
   }
 }
 
+async function promptForApiKeys(rl) {
+  while (true) {
+    const rawKeys = (await askQuestion(rl, "Paste your Ollama API key(s), separated by commas: ")).trim();
+    const keys = splitKeyList(rawKeys);
+
+    if (keys.length > 0) {
+      return keys;
+    }
+
+    console.log(`${RED}At least one API key is required to continue.${RESET}`);
+    console.log("");
+  }
+}
+
+async function promptForModelSelection(rl) {
+  try {
+    const { models } = await fetchAvailableModels();
+
+    console.log("Available models:");
+    console.log("─".repeat(70));
+    for (const model of models) {
+      const modelId = model.model || model.name;
+      const size = formatSize(model.size);
+      const modified = model.modified_at ? new Date(model.modified_at).toLocaleDateString() : "unknown";
+      console.log(`${modelId.padEnd(30)}${size.padStart(12)}${modified.padStart(12)}`);
+    }
+    console.log("");
+
+    while (true) {
+      const requestedModel = (await askQuestion(rl, "Choose the default model: ")).trim();
+      const match = models.find((model) => (model.model || model.name) === requestedModel);
+
+      if (!requestedModel) {
+        console.log(`${RED}Please choose a model from the list above.${RESET}`);
+        console.log("");
+        continue;
+      }
+
+      if (!match) {
+        console.log(`${RED}That model is not in the current list.${RESET}`);
+        console.log("");
+        continue;
+      }
+
+      const model = setModel([requestedModel]);
+      console.log(`${GREEN}✓ Default model set to ${model}.${RESET}`);
+      console.log("");
+      return;
+    }
+  } catch (error) {
+    console.log(`${RED}Couldn't load models right now: ${error.message}${RESET}`);
+    console.log(`${RED}Please fix your keys and run \`loren model:list\` after setup.${RESET}`);
+    console.log("");
+    throw error;
+  }
+}
+
+async function fetchAvailableModels() {
+  const config = loadConfig();
+  const response = await fetch(`${config.upstreamBaseUrl}/api/tags`, {
+    headers: { accept: "application/json" },
+  });
+
+  if (!response.ok) {
+    throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+  }
+
+  const data = await response.json();
+  let models = Array.isArray(data.models) ? data.models : [];
+  models = models.sort((a, b) => {
+    const dateA = a.modified_at ? new Date(a.modified_at).getTime() : 0;
+    const dateB = b.modified_at ? new Date(b.modified_at).getTime() : 0;
+    return dateB - dateA;
+  });
+
+  return { config, models };
+}
+
+function askQuestion(rl, prompt) {
+  return rl.question(`${CYAN}${prompt}${RESET}`);
+}
+
 function printHelp() {
-  console.log(BANNER);
+  printBanner();
+  printCommandSummary();
+}
+
+function printCommandSummary() {
   console.log("Commands:");
   console.log("  loren setup                 Run the setup wizard");
   console.log("  loren start                 Start the bridge");
@@ -602,6 +695,28 @@ function printHelp() {
   console.log("");
 }
 
+function getDisplayName() {
+  const explicit = process.env.USERNAME || process.env.USER;
+  if (explicit && explicit.trim()) {
+    return explicit.trim();
+  }
+
+  const baseName = path.basename(userHome || "").trim();
+  return baseName || "";
+}
+
+function printBanner() {
+  const coloredBanner = ASCII_BANNER_LINES
+    .map((line, index) => `${BANNER_COLORS[index] || ""}${line}${RESET}`)
+    .join("\n");
+
+  console.log(coloredBanner);
+  console.log("");
+  console.log(`${CYAN}LOREN CODE${RESET}`);
+  console.log("Smarter bridge, fewer rituals.");
+  console.log("");
+}
+
 main().catch((error) => {
   console.error(error instanceof Error ? error.message : String(error));
   process.exit(1);