import aiohttp
import asyncio
from modules.blockchain_logger import BlockchainLogger

class ThreatIntelligence:
    def __init__(self):
        self.sources = [
            "https://api.threatsource1.com/v1/threats",
            "https://api.threatsource2.com/v1/threats",
            "https://api.threatsource3.com/v1/threats"
        ]
        self.blockchain_logger = BlockchainLogger()

    async def fetch_data(self, url):
        async with aiohttp.ClientSession() as session:
            async with session.get(url) as response:
                data = await response.json()
                self.blockchain_logger.log_event(f"Fetched data from {url}")
                return data

    async def fetch_all_data(self):
        tasks = [self.fetch_data(url) for url in self.sources]
        return await asyncio.gather(*tasks)

    def process_data(self, data):
        processed_data = []
        for source_data in data:
            for threat in source_data:
                processed_data.append({
                    "threat_id": threat["id"],
                    "description": threat["description"],
                    "severity": threat["severity"],
                    "timestamp": threat["timestamp"]
                })
        self.blockchain_logger.log_event("Processed threat data")
        return processed_data

    async def get_threat_intelligence(self):
        raw_data = await self.fetch_all_data()
        return self.process_data(raw_data)

    async def integrate_with_new_components(self, new_component_data):
        latest_threats = await self.get_threat_intelligence()
        analyzed_threats = self.process_data(latest_threats)
        updated_simulations = self.generate_attack_simulations(analyzed_threats)
        return updated_simulations

    def ensure_compatibility(self, existing_data, new_component_data):
        compatible_data = {
            "existing_data": existing_data,
            "new_component_data": new_component_data
        }
        return compatible_data

    def generate_attack_simulations(self, threats):
        simulations = []
        for threat in threats:
            if threat["severity"] > 0.9:
                simulations.append("High-Risk Attack Simulation")
            elif threat["severity"] > 0.7:
                simulations.append("Medium-Risk Attack Simulation")
            else:
                simulations.append("Low-Risk Attack Simulation")
        self.blockchain_logger.log_event("Generated attack simulations")
        return simulations