letfixit 0.1.1 → 0.1.3

package/package.json

@@ -1,8 +1,18 @@
 {
   "name": "letfixit",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "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",
@@ -49,7 +59,10 @@
     "setup:force": "node scripts/setup-model.js --force",
     "encrypt-model": "node scripts/encrypt-model.js",
     "fill-manifest": "node scripts/fill-manifest.js",
+    "verify": "node scripts/verify-chunk.js",
+    "clean": "node scripts/clean-model.js",
     "postinstall": "node scripts/setup-model.js",
+    "preuninstall": "node scripts/clean-model.js",
     "prepublishOnly": "npm run build"
   },
   "dependencies": {

package/scripts/clean-model.js

@@ -0,0 +1,27 @@
+#!/usr/bin/env node
+// ── Cleanup ───────────────────────────────────────────────────────────────────
+// Removes the downloaded model and any temp/partial files so nothing is left on
+// disk. Runs as a `preuninstall` hook (best-effort — npm's uninstall hooks are
+// unreliable across versions) and is also exposed as `letfixit clean` /
+// `npm run clean` for a guaranteed manual wipe.
+const fs = require('fs');
+const mc = require('../server/model_config');
+
+function clean() {
+  let removed = false;
+  try {
+    if (fs.existsSync(mc.MODEL_DIR)) {
+      const gb = mc.sizeGB();
+      fs.rmSync(mc.MODEL_DIR, { recursive: true, force: true }); // model + .tmp partials
+      console.log(`  ✓ Removed ${mc.MODEL_NAME}${gb ? ` (${gb} GB)` : ''} — ${mc.MODEL_DIR}`);
+      removed = true;
+    }
+  } catch (e) {
+    // Never fail an uninstall on cleanup trouble.
+    console.log(`  ⚠ cleanup note: ${e.message}`);
+  }
+  if (!removed) console.log('  ✓ No model files to remove.');
+}
+
+if (require.main === module) clean();
+module.exports = { clean };

package/scripts/setup-model.js

@@ -142,7 +142,7 @@ async function fetchChunk(ch, encPath, baseLabel) {
 async function resolveKey() {
   if (process.env.PARITYFIX_MODEL_KEY) return process.env.PARITYFIX_MODEL_KEY.trim();
   if (!process.stdin.isTTY) return null; // can't prompt (postinstall in CI, piped stdin)
-  const k = await promptHidden(`  🔑 Enter the ${mc.MODEL_NAME} decryption key: `);
+  const k = await promptHidden('  🔑 Enter acceptance key: ');
   return k || null;
 }
 

package/scripts/verify-chunk.js

@@ -0,0 +1,73 @@
+#!/usr/bin/env node
+// ── Quick verify ──────────────────────────────────────────────────────────────
+// Downloads ONLY the first chunk from the manifest and decrypts it, to confirm
+// the acceptance key + hosted chunks + manifest all line up — without pulling the
+// full ~2.6 GB model.
+//
+//   PARITYFIX_MODEL_KEY=<key> npm run verify       (or: node scripts/verify-chunk.js)
+//   node scripts/verify-chunk.js <key>             (pass the key as an argument)
+// If no key is given it prompts (hidden input).
+require('./load-env')();
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const readline = require('readline');
+const mc = require('../server/model_config');
+const { deriveKey } = require('./model_crypto');
+const sm = require('./setup-model'); // fetchChunk, decryptChunk, sha256File
+
+const C = { reset: '\x1b[0m', mag: '\x1b[95m', grn: '\x1b[92m', red: '\x1b[91m', dim: '\x1b[2m', cyan: '\x1b[96m' };
+const c = (x, s) => `${x}${s}${C.reset}`;
+
+function promptHidden(q) {
+  return new Promise((resolve) => {
+    process.stdout.write(c(C.mag, q));
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl._writeToOutput = () => {};
+    rl.question('', (a) => { rl.close(); process.stdout.write('\n'); resolve(a.trim()); });
+  });
+}
+
+async function run(keyArg) {
+  const manifest = mc.loadManifest();
+  if (!manifest || !manifest.chunks || !manifest.chunks.length) {
+    console.error(c(C.red, '  ❌ No manifest found (config/model.manifest.json).'));
+    process.exit(1);
+  }
+
+  let key = process.env.PARITYFIX_MODEL_KEY || keyArg;
+  if (!key && process.stdin.isTTY) key = await promptHidden('  🔑 Enter acceptance key: ');
+  if (!key) {
+    console.error(c(C.red, '  ❌ No key. Set PARITYFIX_MODEL_KEY, pass it as an argument, or run in a terminal.'));
+    process.exit(1);
+  }
+
+  const ch = [...manifest.chunks].sort((a, b) => a.index - b.index)[0];
+  const tmp = path.join(os.tmpdir(), 'letfixit-verify');
+  fs.mkdirSync(tmp, { recursive: true });
+  const enc = path.join(tmp, 'chunk0.enc');
+  const dec = path.join(tmp, 'chunk0.dec');
+
+  console.log(c(C.dim, `  Verifying chunk 0 of ${manifest.chunks.length} — ${ch.url}\n`));
+  let ok = false;
+  try {
+    await sm.fetchChunk(ch, enc, c(C.cyan, 'chunk 0')); // downloads + verifies sha256
+    await sm.decryptChunk(enc, dec, deriveKey(key.trim(), manifest.salt), ch.iv, ch.authTag);
+    const size = fs.statSync(dec).size;
+    ok = size === (ch.bytes || size);
+    console.log('\n  ' + c(C.grn, '✅ VERIFIED') + c(C.dim, ` — download + checksum + decrypt all passed (${(size / 1048576).toFixed(1)} MB plaintext).`));
+    console.log('  ' + c(C.dim, 'Your acceptance key and hosted chunks are correct; the full setup will work.'));
+  } catch (e) {
+    if (/authenticate|unable to authenticate|bad decrypt/i.test(e.message)) {
+      console.log('\n  ' + c(C.red, '❌ Decryption failed — the acceptance key is incorrect for these chunks.'));
+    } else {
+      console.log('\n  ' + c(C.red, `❌ Verify failed: ${e.message}`));
+    }
+  } finally {
+    fs.rmSync(tmp, { recursive: true, force: true });
+  }
+  process.exit(ok ? 0 : 1);
+}
+
+if (require.main === module) run(process.argv[2]);   // node scripts/verify-chunk.js [key]
+module.exports = { run };

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,47 @@ 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 verify')}       Quick test: download + decrypt only chunk 0
+    ${col(C.cyan, 'letfixit clean')}        Remove the downloaded model (free disk / no footprint)
+    ${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 if (cmd === 'verify') {
+  // Quick test: download + decrypt only the first chunk to validate the key/host.
+  require('../scripts/verify-chunk').run(process.argv[3]);
+} else if (cmd === 'clean') {
+  // Remove the downloaded model + temp files (no footprint).
+  require('../scripts/clean-model').clean();
+} else {
+  main(); // default: launch the analyzer
+}