letfixit 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "letfixit",
-  "version": "0.1.2",
+  "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",
@@ -59,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

@@ -642,6 +642,8 @@ function printHelp() {
     ${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
 
@@ -665,6 +667,12 @@ if (cmd === '--version' || cmd === '-v') {
   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
 }