shadowx-fca 8.0.0

package/e2eetest.js

@@ -0,0 +1,356 @@
+#!/usr/bin/env node
+"use strict";
+
+/**
+ * ╔══════════════════════════════════════════════════════════════╗
+ * ║           SHADOWX-FCA — E2EE Test Bot                       ║
+ * ║           Dev by Mueid Mursalin Rifat                       ║
+ * ╚══════════════════════════════════════════════════════════════╝
+ *
+ * Uses SHADOWX-FCA's own index.js login (no Koffi / lib/index.mjs).
+ * Auto-discovers all src/ modules and tests them via bot commands.
+ * Supports both E2EE encrypted and standard MQTT messages.
+ *
+ * Usage:
+ *   node e2eetest.js                     # reads ./appstate.json
+ *   node e2eetest.js --cookie cookie.txt # reads JSON cookie file
+ */
+
+const fs   = require('fs');
+const path = require('path');
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 1  AUTO-DISCOVER SRC MODULES
+// ─────────────────────────────────────────────────────────────────────────────
+function discoverModules() {
+    const srcDir = path.join(__dirname, 'src');
+    const SKIP_DIRS = new Set(['e2ee', 'vendor', 'data']);
+    return fs.readdirSync(srcDir)
+        .filter(f => f.endsWith('.js') && !SKIP_DIRS.has(f.replace('.js', '')))
+        .map(f => f.replace('.js', ''))
+        .sort();
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 2  MODULE HEALTH CHECK
+// ─────────────────────────────────────────────────────────────────────────────
+function checkAllModules() {
+    const srcDir = path.join(__dirname, 'src');
+    const files  = fs.readdirSync(srcDir).filter(f => f.endsWith('.js'));
+    const ok = [], failed = [];
+
+    for (const f of files) {
+        try {
+            const mod = require(path.join(srcDir, f));
+            if (typeof mod === 'function' || typeof mod === 'object')
+                ok.push(f.replace('.js', ''));
+            else
+                failed.push({ name: f, reason: 'unexpected export type' });
+        } catch (e) {
+            failed.push({ name: f, reason: e.message });
+        }
+    }
+    return { ok, failed };
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 3  APPSTATE / COOKIE LOADER
+//   Supports:  appstate.json  |  cookie.txt  |  --cookie <file>  CLI arg
+// ─────────────────────────────────────────────────────────────────────────────
+function loadAppState() {
+    // CLI: --cookie <path>
+    const idx = process.argv.indexOf('--cookie');
+    if (idx !== -1 && process.argv[idx + 1]) {
+        return readJsonFile(path.resolve(process.argv[idx + 1]), 'cookie');
+    }
+    // default: appstate.json beside the script
+    const candidates = [
+        path.join(process.cwd(), 'appstate.json'),
+        path.join(process.cwd(), 'cookie.txt'),
+        path.join(__dirname, 'appstate.json'),
+    ];
+    for (const p of candidates) {
+        if (fs.existsSync(p)) {
+            console.log(`[AUTH] Using: ${p}`);
+            return readJsonFile(p, path.basename(p));
+        }
+    }
+    throw new Error(
+        'No appstate found.\n' +
+        '  Put appstate.json in the working directory, or run:\n' +
+        '  node e2eetest.js --cookie /path/to/cookie.json'
+    );
+}
+
+function readJsonFile(filePath, label) {
+    const raw = fs.readFileSync(filePath, 'utf8').trim();
+    try {
+        const parsed = JSON.parse(raw);
+        if (Array.isArray(parsed) && parsed.length > 0) {
+            console.log(`[AUTH] ✓ Loaded ${parsed.length} cookies from ${label}`);
+            return parsed;
+        }
+        throw new Error('Expected a JSON array of cookies');
+    } catch (e) {
+        throw new Error(`Cannot parse ${filePath}: ${e.message}`);
+    }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 4  COMMAND HANDLER
+// ─────────────────────────────────────────────────────────────────────────────
+async function handleMessage(api, event, allModules) {
+    const PREFIX = '!';
+    const body   = (event.body || '').trim();
+    if (!body.startsWith(PREFIX)) return;
+
+    const parts    = body.slice(PREFIX.length).trim().split(/\s+/);
+    const cmd      = (parts.shift() || '').toLowerCase();
+    const args     = parts;
+    const threadID = event.threadID;
+
+    const tag = event.isE2EE ? '🔐' : '📨';
+    console.log(`\n${tag} Command: !${cmd}  args: ${JSON.stringify(args)}`);
+
+    const send = (text) => api.sendMessage(text, threadID);
+
+    try {
+        switch (cmd) {
+
+            // ── Basic ──────────────────────────────────────────────────────
+            case 'ping':
+                await send('🏓 Pong! SHADOWX-FCA is alive!');
+                break;
+
+            case 'info':
+                await send(
+                    `ℹ️ SHADOWX-FCA Info\n` +
+                    `├─ MsgID  : ${event.messageID || 'N/A'}\n` +
+                    `├─ Sender : ${event.senderID}\n` +
+                    `├─ Thread : ${threadID}\n` +
+                    `├─ E2EE   : ${event.isE2EE ? '🔐 Yes' : '📨 No'}\n` +
+                    `├─ Type   : ${event.isGroup ? '👥 GROUP' : '👤 DM'}\n` +
+                    `└─ Time   : ${new Date(event.timestamp || Date.now()).toLocaleString()}`
+                );
+                break;
+
+            case 'echo':
+                await send(`🔊 ${args.join(' ') || '(nothing to echo)'}`);
+                break;
+
+            // ── Module listing & checking ──────────────────────────────────
+            case 'modules': {
+                const page  = parseInt(args[0]) || 1;
+                const size  = 40;
+                const start = (page - 1) * size;
+                const slice = allModules.slice(start, start + size);
+                const total = allModules.length;
+                const pages = Math.ceil(total / size);
+                await send(
+                    `📦 Modules (${total} total) — page ${page}/${pages}:\n` +
+                    slice.join(', ') +
+                    (page < pages ? `\n\nSend !modules ${page + 1} for next page` : '')
+                );
+                break;
+            }
+
+            case 'check': {
+                await send('🔍 Checking all src/ modules, please wait...');
+                const results  = checkAllModules();
+                const failText = results.failed.length
+                    ? results.failed.map(f => `  ✗ ${f.name}: ${f.reason.slice(0, 80)}`).join('\n')
+                    : '  (none)';
+                await send(
+                    `📊 Module Check — ${results.ok.length + results.failed.length} files\n` +
+                    `✅ OK     : ${results.ok.length}\n` +
+                    `❌ Failed : ${results.failed.length}\n\n` +
+                    (results.failed.length
+                        ? `Failed modules:\n${failText}`
+                        : `🎉 All modules loaded successfully!`)
+                );
+                break;
+            }
+
+            case 'find': {
+                const kw = (args[0] || '').toLowerCase();
+                if (!kw) { await send('❓ Usage: !find <keyword>'); break; }
+                const matches = allModules.filter(m => m.toLowerCase().includes(kw));
+                await send(
+                    matches.length
+                        ? `🔎 "${kw}" — ${matches.length} match(es):\n${matches.join(', ')}`
+                        : `❌ No modules matching "${kw}"`
+                );
+                break;
+            }
+
+            case 'has': {
+                const mod = args[0] || '';
+                await send(
+                    allModules.includes(mod)
+                        ? `✅ Module "${mod}" exists in SHADOWX-FCA`
+                        : `❌ Module "${mod}" not found — try !find ${mod}`
+                );
+                break;
+            }
+
+            case 'categories': {
+                const groups = {};
+                for (const m of allModules) {
+                    const prefix = m.replace(/([A-Z])/g, ' $1').trim().split(' ')[0].toLowerCase();
+                    (groups[prefix] = groups[prefix] || []).push(m);
+                }
+                const lines = Object.entries(groups)
+                    .sort((a, b) => b[1].length - a[1].length)
+                    .map(([k, v]) => `  ${k.padEnd(12)}: ${v.length}`);
+                await send(`📂 Module Categories (${allModules.length} total):\n${lines.join('\n')}`);
+                break;
+            }
+
+            // ── E2EE status ────────────────────────────────────────────────
+            case 'e2ee': {
+                const e2eeDir   = path.join(__dirname, 'src', 'e2ee');
+                const e2eeFiles = fs.readdirSync(e2eeDir)
+                    .filter(f => !fs.statSync(path.join(e2eeDir, f)).isDirectory());
+                const connected = !!(api.e2ee && api.e2ee.connected);
+                await send(
+                    `🔐 E2EE Engine Status\n` +
+                    `├─ Connected    : ${connected ? '✅ Yes' : '❌ No'}\n` +
+                    `├─ Engine files : ${e2eeFiles.join(', ')}\n` +
+                    `├─ Store path   : .shadowx-fca/e2ee_device.json\n` +
+                    `└─ Protocol     : Signal (Double Ratchet + Noise_XX)`
+                );
+                break;
+            }
+
+            // ── Help ───────────────────────────────────────────────────────
+            case 'help':
+            default:
+                await send(
+                    `📋 SHADOWX-FCA E2EE Test Bot\n` +
+                    `Dev: Mueid Mursalin Rifat\n\n` +
+                    `!ping              — Bot alive check\n` +
+                    `!info              — Message & thread info\n` +
+                    `!echo <text>       — Echo text back\n` +
+                    `!modules [page]    — List all src/ modules (paged)\n` +
+                    `!check             — Health-check all src/ files\n` +
+                    `!find <keyword>    — Search modules by name\n` +
+                    `!has <name>        — Check if module exists\n` +
+                    `!categories        — Modules grouped by type\n` +
+                    `!e2ee              — E2EE engine status\n` +
+                    `!help              — Show this help`
+                );
+                break;
+        }
+    } catch (err) {
+        console.error(`✗ Error in !${cmd}: ${err.message}`);
+        try { await send(`❌ Error in !${cmd}: ${err.message}`); } catch (_) {}
+    }
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 5  MAIN — login via SHADOWX-FCA index.js (no Koffi / lib/index.mjs)
+// ─────────────────────────────────────────────────────────────────────────────
+async function main() {
+    console.log('\n╔════════════════════════════════════════════════════════════╗');
+    console.log('║         SHADOWX-FCA — E2EE Test Bot                       ║');
+    console.log('║         Dev by Mueid Mursalin Rifat                       ║');
+    console.log('╚════════════════════════════════════════════════════════════╝\n');
+
+    // ── Module discovery & health check ───────────────────────────────────
+    const allModules = discoverModules();
+    console.log(`[MODULES] ✓ Discovered ${allModules.length} src modules`);
+
+    const check = checkAllModules();
+    if (check.failed.length) {
+        console.warn(`[MODULES] ⚠  ${check.failed.length} module(s) failed to load:`);
+        check.failed.forEach(f => console.warn(`  ✗ ${f.name}: ${f.reason}`));
+    } else {
+        console.log(`[MODULES] ✓ ${check.ok.length} / ${check.ok.length} modules OK`);
+    }
+
+    // ── Load appstate ──────────────────────────────────────────────────────
+    const appState = loadAppState();
+
+    // ── Login via SHADOWX-FCA's own index.js ──────────────────────────────
+    console.log('[LOGIN] Logging in via SHADOWX-FCA...');
+    const login = require('./index.js');
+
+    const api = await new Promise((resolve, reject) => {
+        login({ appState }, { listenEvents: true }, (err, api) => {
+            if (err) return reject(err);
+            resolve(api);
+        });
+    });
+
+    console.log(`[LOGIN] ✓ Logged in as UID: ${api.getCurrentUserID()}\n`);
+
+    // ── Try to connect E2EE (gracefully falls back if engine not available) ─
+    let e2eeActive = false;
+    try {
+        console.log('[E2EE] Connecting E2EE encryption bridge...');
+        await api.connectE2EE();
+        e2eeActive = true;
+        console.log('[E2EE] ✓ E2EE bridge connected\n');
+    } catch (err) {
+        console.warn(`[E2EE] ⚠  E2EE not available: ${err.message}`);
+        console.warn('[E2EE]    Falling back to standard MQTT only.\n');
+    }
+
+    // ── Choose listener: E2EE combined or standard MQTT ───────────────────
+    const listenFn = (e2eeActive && typeof api.listenE2EE === 'function')
+        ? api.listenE2EE.bind(api)
+        : api.listen.bind(api);
+
+    const mode = e2eeActive ? '🔐 E2EE + MQTT' : '📨 MQTT only';
+
+    console.log('─'.repeat(62));
+    console.log(`🚀 SHADOWX-FCA E2EE Test Bot is ready!`);
+    console.log(`   UID            : ${api.getCurrentUserID()}`);
+    console.log(`   Modules loaded : ${allModules.length}`);
+    console.log(`   Listen mode    : ${mode}`);
+    console.log('   Send !help in any chat to see all commands.');
+    console.log('   Press Ctrl+C to stop');
+    console.log('─'.repeat(62) + '\n');
+
+    // ── Start listening ────────────────────────────────────────────────────
+    listenFn((err, event) => {
+        if (err) {
+            console.error('❌ [listen error]', err);
+            return;
+        }
+        if (!event) return;
+
+        const tag = event.isE2EE ? '🔐' : '📨';
+
+        // Log every event
+        console.log(`${tag} [${event.type}] thread:${event.threadID} sender:${event.senderID}`);
+        if (event.body) console.log(`   body: ${event.body.slice(0, 120)}`);
+
+        // Handle bot commands
+        if (event.type === 'message' || event.type === 'message_reply') {
+            handleMessage(api, event, allModules).catch(e =>
+                console.error('handleMessage error:', e.message)
+            );
+        }
+    });
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// § 6  SHUTDOWN
+// ─────────────────────────────────────────────────────────────────────────────
+process.on('SIGINT',  () => { console.log('\n\n👋 Shutting down...'); process.exit(0); });
+process.on('SIGTERM', () => { console.log('\n💥 Force shutdown');    process.exit(1); });
+process.on('uncaughtException', err => {
+    console.error('\n💥 Uncaught Exception:', err.message);
+    console.error(err.stack);
+    process.exit(1);
+});
+
+main().catch(err => {
+    console.error('\n❌ FATAL:', err.message);
+    console.error('\nTroubleshooting:');
+    console.error('  1. Make sure appstate.json is in the current directory');
+    console.error('  2. Or run: node e2eetest.js --cookie /path/to/cookie.json');
+    console.error('  3. Run: npm install (if you see missing module errors)');
+    process.exit(1);
+});