js/federated-core.js

/**
 * Federated AI Core - Architecture Distribuée Avancée
 * Système Neural Fedéré avec Byzantine Fault Tolerance
 * Firewall Connectif Intégré - Protocole SPECTRE-F
 */

class FederatedNode {
    constructor(nodeId, role = 'worker') {
        this.nodeId = nodeId;
        this.role = role; // 'master', 'worker', 'validator'
        this.status = 'initializing';
        this.latency = 0;
        this.encryptionKey = null;
        this.lastHeartbeat = Date.now();
        this.trustScore = 100;
        this.anomaliesDetected = 0;
        this.firewallRules = new Map();
    }

    async initializeSecureChannel() {
        // Simulation de chiffrement post-quantique
        this.encryptionKey = await this.generateQuantumKey();
        this.status = 'active';
        console.log(`🔐 Node ${this.nodeId}: Secure channel established with quantum-resistant encryption`);
    }

    async generateQuantumKey() {
        // Simulation de génération de clé CRYSTALS-Kyber
        return Array.from({length: 32}, () => Math.floor(Math.random() * 256));
    }

    async processFederatedUpdate(modelUpdate) {
        if (this.role === 'validator') {
            return this.validateUpdate(modelUpdate);
        }
        return this.aggregateUpdate(modelUpdate);
    }

    validateUpdate(update) {
        // Algorithme de consensus byzantin simplifié
        const checksum = this.calculateSecureChecksum(update);
        const isValid = checksum % 997 === 0; // Simulation de validation cryptographique
        
        if (!isValid) {
            this.trustScore -= 5;
            this.anomaliesDetected++;
            return { valid: false, penalty: 'trust_degradation' };
        }
        
        return { valid: true, signature: this.signUpdate(update) };
    }

    aggregateUpdate(update) {
        // Agrégation fédérée sécurisée
        return {
            nodeId: this.nodeId,
            weights: update.weights,
            timestamp: Date.now(),
            signature: this.signUpdate(update),
            differentialPrivacy: this.applyDPNoise(update)
        };
    }

    applyDPNoise(data) {
        // Bruit de Laplace pour differential privacy
        const epsilon = 0.1;
        return data.weights.map(w => w + (Math.random() - 0.5) * epsilon);
    }

    calculateSecureChecksum(data) {
        let hash = 0;
        const str = JSON.stringify(data);
        for (let i = 0; i < str.length; i++) {
            const char = str.charCodeAt(i);
            hash = ((hash << 5) - hash) + char;
            hash = hash & hash;
        }
        return Math.abs(hash);
    }

    signUpdate(update) {
        // Signature numérique simulée
        return `SIG_${this.nodeId}_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
    }
}

class FederatedAICore {
    constructor() {
        this.nodes = new Map();
        this.masterNode = null;
        this.globalModel = null;
        this.consensusThreshold = 0.67; // 2/3 pour tolérance byzantine
        this.firewall = new ConnectiveFirewall();
        this.encryptionLayer = new PostQuantumCrypto();
        this.initializeCluster();
    }

    initializeCluster() {
        // Initialisation du cluster fédéré
        this.masterNode = new FederatedNode('MASTER-01', 'master');
        this.masterNode.initializeSecureChannel();
        this.nodes.set('MASTER-01', this.masterNode);

        // Workers distribués
        for (let i = 1; i <= 22; i++) {
            const nodeId = `WORKER-${String(i).padStart(2, '0')}`;
            const node = new FederatedNode(nodeId, i % 3 === 0 ? 'validator' : 'worker');
            node.initializeSecureChannel();
            this.nodes.set(nodeId, node);
        }

        console.log(`🧠 Federated Cluster Initialized: 1 Master, ${this.nodes.size - 1} Nodes`);
        this.startConsensusProtocol();
    }

    async startConsensusProtocol() {
        setInterval(() => {
            this.runConsensusRound();
        }, 5000);
    }

    async runConsensusRound() {
        const activeNodes = Array.from(this.nodes.values()).filter(n => n.status === 'active');
        const updates = [];
        
        // Collecte fédérée
        for (const node of activeNodes) {
            if (node.role !== 'master') {
                const update = await this.generateModelUpdate(node);
                updates.push(update);
            }
        }

        // Agrégation sécurisée
        const aggregated = this.secureAggregation(updates);
        this.globalModel = aggregated;
        
        // Propagation avec firewall checking
        this.propagateModel(aggregated);
    }

    async generateModelUpdate(node) {
        // Simulation d'entraînement local
        return {
            nodeId: node.nodeId,
            weights: Array.from({length: 128}, () => Math.random()),
            metrics: {
                accuracy: 0.95 + Math.random() * 0.05,
                loss: Math.random() * 0.01,
                fraudDetected: Math.floor(Math.random() * 100)
            },
            timestamp: Date.now()
        };
    }

    secureAggregation(updates) {
        // Agrégation fédérée avec vérification Byzantine
        const validUpdates = updates.filter(u => {
            const node = this.nodes.get(u.nodeId);
            if (!node) return false;
            const validation = node.validateUpdate(u);
            return validation.valid;
        });

        if (validUpdates.length < this.nodes.size * this.consensusThreshold) {
            console.warn('⚠️ Consensus failed: Insufficient valid nodes');
            return null;
        }

        // Moyenne pondérée par le trust score
        const weightedSum = validUpdates.map(u => {
            const node = this.nodes.get(u.nodeId);
            return u.weights.map(w => w * (node.trustScore / 100));
        });

        return {
            weights: this.averageArrays(weightedSum),
            consensusNodes: validUpdates.length,
            timestamp: Date.now(),
            hash: this.generateModelHash(validUpdates)
        };
    }

    averageArrays(arrays) {
        const length = arrays[0].length;
        const result = new Array(length).fill(0);
        for (const arr of arrays) {
            for (let i = 0; i < length; i++) {
                result[i] += arr[i] / arrays.length;
            }
        }
        return result;
    }

    generateModelHash(updates) {
        return `MODEL_${Date.now()}_${updates.length}_NODES`;
    }

    propagateModel(model) {
        if (!model) return;
        
        console.log(`📡 Propagating model to ${this.nodes.size} nodes with firewall validation`);
        
        for (const [nodeId, node] of this.nodes) {
            if (nodeId !== 'MASTER-01') {
                // Vérification firewall avant propagation
                const securityCheck = this.firewall.inspectTraffic({
                    type: 'model_update',
                    payload: model,
                    destination: nodeId
                });

                if (securityCheck.allowed) {
                    node.lastHeartbeat = Date.now();
                } else {
                    node.status = 'quarantined';
                    console.warn(`🚫 Node ${nodeId} quarantined by firewall`);
                }
            }
        }
    }

    getClusterHealth() {
        const nodes = Array.from(this.nodes.values());
        return {
            total: nodes.length,
            active: nodes.filter(n => n.status === 'active').length,
            quarantined: nodes.filter(n => n.status === 'quarantined').length,
            averageTrust: nodes.reduce((sum, n) => sum + n.trustScore, 0) / nodes.length,
            globalModelVersion: this.globalModel?.hash || 'N/A',
            firewallStatus: this.firewall.status
        };
    }

    async detectAdvancedThreats(entityData) {
        // Détection distribuée avec consensus
        const threatSignals = [];
        
        for (const [nodeId, node] of this.nodes) {
            if (node.status === 'active' && node.role !== 'master') {
                const signal = await this.analyzeLocalThreat(node, entityData);
                threatSignals.push(signal);
            }
        }

        // Consensus sur la menace
        const threatScore = threatSignals.filter(s => s.threat).length / threatSignals.length;
        
        return {
            threatDetected: threatScore > 0.5,
            confidence: threatScore,
            distributedConsensus: threatSignals.length,
            recommendedAction: threatScore > 0.8 ? 'IMMEDIATE_ISOLATION' : 'MONITORING',
            signatures: threatSignals.map(s => s.signature)
        };
    }

    async analyzeLocalThreat(node, data) {
        // Analyse locale par nœud fédéré
        const riskFactors = [
            data.transactionVelocity > 1000 ? 0.3 : 0,
            data.jurisdictionRisk > 0.7 ? 0.25 : 0,
            data.shellIndicators > 3 ? 0.35 : 0,
            Math.random() * 0.1 // Bruit de détection
        ];
        
        const totalRisk = riskFactors.reduce((a, b) => a + b, 0);
        
        return {
            nodeId: node.nodeId,
            threat: totalRisk > 0.6,
            riskScore: totalRisk,
            signature: node.signUpdate(data),
            timestamp: Date.now()
        };
    }
}

class ConnectiveFirewall {
    constructor() {
        this.rules = new Map();
        this.threatDatabase = new Set();
        this.status = 'active';
        this.blockedAttempts = 0;
        this.initializeRules();
    }

    initializeRules() {
        // Règles de sécurité avancées
        this.rules.set('SQL_INJECTION', /(\b(union|select|insert|update|delete|drop|create|alter)\b.*\b(from|into|table|database)\b)/i);
        this.rules.set('XSS_ATTEMPT', /<script\b[^<]*(?:(?!<\/script>)<[^<]*)*<\/script>/gi);
        this.rules.set('INJECTION_PAYLOAD', /(\b(eval|exec|system|passthru|shell_exec|popen|proc_open)\b)/i);
        this.rules.set('BYZANTINE_ATTACK', this.detectByzantineBehavior.bind(this));
    }

    inspectTraffic(packet) {
        const checks = [];
        
        for (const [ruleName, rule] of this.rules) {
            if (typeof rule === 'function') {
                checks.push(rule(packet));
            } else if (rule.test && rule.test(JSON.stringify(packet))) {
                checks.push({ rule: ruleName, blocked: true, severity: 'HIGH' });
                this.blockedAttempts++;
            }
        }

        const blocked = checks.some(c => c.blocked);
        
        if (blocked) {
            this.triggerSecurityAlert(packet, checks);
        }

        return {
            allowed: !blocked,
            checks: checks,
            timestamp: Date.now(),
            signature: this.generatePacketSignature(packet)
        };
    }

    detectByzantineBehavior(packet) {
        // Détection d'attaque byzantine (nœud malveillant)
        if (packet.payload && packet.payload.weights) {
            const variance = this.calculateVariance(packet.payload.weights);
            if (variance > 0.5) {
                return { rule: 'BYZANTINE_HIGH_VARIANCE', blocked: true, severity: 'CRITICAL' };
            }
        }
        return { blocked: false };
    }

    calculateVariance(array) {
        const mean = array.reduce((a, b) => a + b) / array.length;
        return array.reduce((sum, val) => sum + Math.pow(val - mean, 2), 0) / array.length;
    }

    generatePacketSignature(packet) {
        return `SEC_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
    }

    triggerSecurityAlert(packet, violations) {
        console.error(`🚨 SECURITY ALERT: ${violations.length} violations detected`, {
            packet: packet.destination || 'unknown',
            violations: violations.map(v => v.rule),
            timestamp: new Date().toISOString()
        });
    }

    getSecurityReport() {
        return {
            status: this.status,
            blockedAttempts: this.blockedAttempts,
            threatDatabaseSize: this.threatDatabase.size,
            activeRules: this.rules.size,
            lastUpdate: new Date().toISOString()
        };
    }
}

class PostQuantumCrypto {
    constructor() {
        this.algorithm = 'CRYSTALS-Kyber-1024';
        this.mode = 'FIPS-203-DRAFT';
    }

    async encrypt(data, publicKey) {
        // Simulation de chiffrement post-quantique
        const encrypted = btoa(JSON.stringify(data));
        return {
            ciphertext: encrypted,
            algorithm: this.algorithm,
            nonce: Math.random().toString(36).substr(2, 16)
        };
    }

    async decrypt(ciphertext, privateKey) {
        try {
            return JSON.parse(atob(ciphertext));
        } catch {
            return null;
        }
    }
}

// Export pour utilisation globale
window.FederatedAICore = FederatedAICore;
window.ConnectiveFirewall = ConnectiveFirewall;
window.FederatedNode = FederatedNode;