letfixit 0.1.1 → 0.1.2

package/package.json

@@ -1,8 +1,18 @@
 {
   "name": "letfixit",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "LetFixIt — AI-based realtime performance analysis & suggestions for Flutter, Android & iOS, powered by the local Ultron n0-beta model.",
-  "keywords": ["flutter", "android", "ios", "performance", "profiler", "devtools", "observability", "llm", "local-ai"],
+  "keywords": [
+    "flutter",
+    "android",
+    "ios",
+    "performance",
+    "profiler",
+    "devtools",
+    "observability",
+    "llm",
+    "local-ai"
+  ],
   "license": "MIT",
   "author": "JahidDev24",
   "main": "server/index.js",

package/server/index.js

@@ -29,7 +29,7 @@ const ENABLE_LOCAL_AI = process.env.ENABLE_LOCAL_AI !== 'false';
 const MODEL_PATH = model.MODEL_PATH;
 
 // ── Terminal styling + branded startup ─────────────────────────────────────────
-const C = { reset: '\x1b[0m', bold: '\x1b[1m', dim: '\x1b[2m', mag: '\x1b[95m', grn: '\x1b[92m', yel: '\x1b[93m', cyan: '\x1b[96m' };
+const C = { reset: '\x1b[0m', bold: '\x1b[1m', dim: '\x1b[2m', mag: '\x1b[95m', grn: '\x1b[92m', yel: '\x1b[93m', red: '\x1b[91m', cyan: '\x1b[96m' };
 const col = (x, s) => `${x}${s}${C.reset}`;
 const stripAnsi = (s) => s.replace(/\x1b\[[0-9;]*m/g, '');
 const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
@@ -42,10 +42,11 @@ function bannerBox(lines) {
 }
 
 function printWelcome() {
+  const v = require('../package.json').version;
   console.log('');
   bannerBox([
     '',
-    col(C.bold + C.mag, '✦  P A R I T Y F I X') + col(C.dim, '                      v0.1.0'),
+    col(C.bold + C.mag, '✦  P A R I T Y F I X') + col(C.dim, `                      v${v}`),
     col(C.reset, 'AI-based realtime performance analysis & suggestions'),
     col(C.dim, 'Flutter  ·  Android  ·  iOS      powered by Ultron n0-beta'),
     '',
@@ -53,32 +54,40 @@ function printWelcome() {
   console.log('');
 }
 
-// "Ultron n0-beta initializing…" — starts the local model and waits briefly for
-// it to answer, without blocking the whole boot on weight loading.
+// Boot the local model. On first run (model not installed) this RUNS the setup
+// — download + key prompt + decrypt + merge — then starts the model. Returns
+// true if AI is available (local model ready, or a cloud provider configured).
 async function initUltron() {
   if (!ENABLE_LOCAL_AI) {
     console.log('  ' + col(C.dim, 'Ultron n0-beta is disabled (ENABLE_LOCAL_AI=false) — using cloud fallback if configured.\n'));
-    return;
+    return aiStatus().anyAvailable;
   }
+
+  // First-time setup: install the model (prompts for the decryption key).
   if (!model.isReady()) {
-    console.log('  ' + col(C.yel, '⚠  Ultron n0-beta not installed.') + ' Run ' + col(C.cyan, 'npm run setup') + ' to download it.');
-    const s = aiStatus();
-    console.log('  ' + col(C.dim, s.anyAvailable ? 'A cloud fallback key is set — analysis will use it for now.\n' : 'No AI provider yet — findings will show without suggestions until setup.\n'));
-    return;
+    console.log('  ' + col(C.mag, '✦ First-time setup') + col(C.dim, ` — installing ${model.MODEL_NAME}. This runs once.\n`));
+    try {
+      await require('../scripts/setup-model').setup();
+    } catch (e) {
+      const msg = String(e.message || e);
+      if (msg.startsWith('WRONG_KEY:')) console.log('\n  ' + col(C.red, '❌ ' + msg.replace('WRONG_KEY: ', '')));
+      else console.log('\n  ' + col(C.red, '❌ Setup failed: ' + msg));
+    }
+    console.log('');
   }
 
-  process.stdout.write('  ' + col(C.mag, '✦ Ultron n0-beta') + col(C.dim, ' initializing'));
-  startLlamafile();
-  // Poll health for a few seconds; don't hang the boot if weights are still loading.
-  let ok = false;
-  for (let i = 0; i < 8 && !ok; i++) { process.stdout.write(col(C.dim, '.')); ok = await isLlamaHealthy(); if (!ok) await sleep(500); }
-  process.stdout.write('\n');
-  if (ok) {
-    console.log('  ' + col(C.grn, '✓ ready') + col(C.dim, `  — ${model.MODEL_NAME} running locally on :${LOCAL_AI_PORT} (${model.sizeGB()} GB, private)`));
-  } else {
-    console.log('  ' + col(C.yel, '… still loading') + col(C.dim, '  — weights load in the background (first start can take ~20s).'));
+  if (model.isReady()) {
+    process.stdout.write('  ' + col(C.mag, '✦ Ultron n0-beta') + col(C.dim, ' initializing'));
+    startLlamafile();
+    let ok = false;
+    for (let i = 0; i < 8 && !ok; i++) { process.stdout.write(col(C.dim, '.')); ok = await isLlamaHealthy(); if (!ok) await sleep(500); }
+    process.stdout.write('\n');
+    if (ok) console.log('  ' + col(C.grn, '✓ ready') + col(C.dim, `  — ${model.MODEL_NAME} running locally on :${LOCAL_AI_PORT} (${model.sizeGB()} GB, private)`));
+    else    console.log('  ' + col(C.yel, '… still loading') + col(C.dim, '  — weights load in the background (first start can take ~20s).'));
+    console.log('  ' + col(C.dim, model.DISCLAIMER.split(' — ')[0] + ' — treat suggestions as hints.\n'));
   }
-  console.log('  ' + col(C.dim, model.DISCLAIMER.split(' — ')[0] + ' — treat suggestions as hints.\n'));
+
+  return aiStatus().anyAvailable; // local ready OR cloud key present
 }
 
 // ── Interactive startup ───────────────────────────────────────────────────────
@@ -569,9 +578,16 @@ server.on('error', (err) => {
 // ── Startup ───────────────────────────────────────────────────────────────────
 async function main() {
   printWelcome();
-  // "Ultron n0-beta initializing…" — warms the local model up front (idempotent)
-  // so it's ready by the time analysis needs it, and owns the AI startup output.
-  await initUltron();
+  // First-run: install + start the model. Returns true only if AI is available.
+  const aiReady = await initUltron();
+
+  // AI is required to analyze — don't show the menu without it.
+  if (!aiReady) {
+    console.log('  ' + col(C.red, '⚠  AI is required to run analysis, and it isn\'t set up yet.'));
+    console.log('  ' + col(C.dim, `Re-run ${col(C.cyan, 'letfixit')} and paste your ${model.MODEL_NAME} key when prompted,`));
+    console.log('  ' + col(C.dim, `or add a cloud key (GROQ/GEMINI/CLAUDE) to .env, then start again.\n`));
+    process.exit(0);
+  }
 
   const setup = await interactiveSetup();
 
@@ -616,4 +632,39 @@ function shutdown() {
 process.on('SIGINT',  shutdown);
 process.on('SIGTERM', shutdown);
 
-main();
+// ── CLI dispatch ────────────────────────────────────────────────────────────
+function printHelp() {
+  const v = require('../package.json').version;
+  console.log(`
+  ${col(C.bold + C.mag, 'LetFixIt')} v${v} ${col(C.dim, '— AI-based realtime performance analysis for Flutter, Android & iOS')}
+
+  ${col(C.bold, 'Usage')}
+    ${col(C.cyan, 'letfixit')}              Launch the analyzer (first run installs ${model.MODEL_NAME})
+    ${col(C.cyan, 'letfixit setup')}        Download & set up the model (prompts for the decryption key)
+    ${col(C.cyan, 'letfixit setup --force')} Re-download / reinstall the model
+    ${col(C.cyan, 'letfixit version')}      Print the installed version (or --version / -v)
+    ${col(C.cyan, 'letfixit --help')}       Show this help
+
+  ${col(C.bold, 'Environment')}
+    ${col(C.dim, 'PARITYFIX_MODEL_KEY')}   decryption key for non-interactive / CI installs
+    ${col(C.dim, 'DASHBOARD_PORT')}        dashboard port (default 8081)
+    ${col(C.dim, 'LOCAL_AI_PORT')}         local model port (default 8080)
+    ${col(C.dim, 'ENABLE_LOCAL_AI')}       set to "false" to use cloud fallback only
+`);
+}
+
+const cmd = process.argv[2];
+if (cmd === '--version' || cmd === '-v') {
+  console.log(require('../package.json').version);            // bare number (scripts)
+} else if (cmd === 'version') {
+  console.log(`letfixit v${require('../package.json').version}`); // friendly line
+} else if (cmd === '--help' || cmd === '-h' || cmd === 'help') {
+  printHelp();
+} else if (cmd === 'setup') {
+  // Explicit setup: run the installer directly and exit (like `npm run setup`).
+  require('../scripts/setup-model').setup()
+    .then(() => process.exit(0))
+    .catch((e) => { console.error('  ❌ setup failed:', e.message); process.exit(1); });
+} else {
+  main(); // default: launch the analyzer
+}