aitbc/apps/blockchain-node/src/aitbc_chain/network/bridge_manager.py

"""
Bridge Manager
Manages island bridging with manual approval for federated mesh
"""

import asyncio
import logging
import uuid
from typing import Dict, List, Optional, Set
from dataclasses import dataclass
from enum import Enum
import time

logger = logging.getLogger(__name__)


class BridgeState(Enum):
    """Bridge connection state"""
    PENDING = "pending"
    APPROVED = "approved"
    ACTIVE = "active"
    REJECTED = "rejected"
    TERMINATED = "terminated"


@dataclass
class BridgeConnection:
    """Represents a bridge connection between islands"""
    bridge_id: str
    source_island_id: str
    target_island_id: str
    source_node_id: str
    target_node_id: Optional[str]  # Node on target island that approved
    state: BridgeState
    requested_at: float
    approved_at: Optional[float] = None
    activated_at: Optional[float] = None
    terminated_at: Optional[float] = None
    rejection_reason: Optional[str] = None


class BridgeManager:
    """Manages island bridging with manual approval"""
    
    def __init__(self, local_node_id: str, local_island_id: str):
        self.local_node_id = local_node_id
        self.local_island_id = local_island_id
        
        # Bridge connections
        self.bridges: Dict[str, BridgeConnection] = {}  # bridge_id -> BridgeConnection
        
        # Active bridges (island_id -> bridge_id)
        self.active_bridges: Dict[str, str] = {}  # target_island_id -> bridge_id
        
        # Pending bridge requests (island_id -> bridge_id)
        self.pending_requests: Dict[str, str] = {}  # target_island_id -> bridge_id
        
        self.running = False
    
    def request_bridge(self, target_island_id: str) -> str:
        """
        Request a bridge to another island
        
        Returns:
            Bridge request ID
        """
        if target_island_id == self.local_island_id:
            logger.warning("Cannot bridge to own island")
            return ""
        
        if target_island_id in self.active_bridges:
            logger.warning(f"Already have active bridge to {target_island_id}")
            return self.active_bridges[target_island_id]
        
        if target_island_id in self.pending_requests:
            logger.warning(f"Already have pending bridge request to {target_island_id}")
            return self.pending_requests[target_island_id]
        
        bridge_id = str(uuid.uuid4())
        
        bridge = BridgeConnection(
            bridge_id=bridge_id,
            source_island_id=self.local_island_id,
            target_island_id=target_island_id,
            source_node_id=self.local_node_id,
            target_node_id=None,
            state=BridgeState.PENDING,
            requested_at=time.time()
        )
        
        self.bridges[bridge_id] = bridge
        self.pending_requests[target_island_id] = bridge_id
        
        logger.info(f"Requested bridge to island {target_island_id} (bridge_id: {bridge_id})")
        return bridge_id
    
    def approve_bridge_request(self, bridge_id: str, approving_node_id: str) -> bool:
        """
        Approve a bridge request
        
        Args:
            bridge_id: Bridge request ID
            approving_node_id: Node ID approving the bridge
            
        Returns:
            True if successful, False otherwise
        """
        if bridge_id not in self.bridges:
            logger.warning(f"Unknown bridge request {bridge_id}")
            return False
        
        bridge = self.bridges[bridge_id]
        
        if bridge.state != BridgeState.PENDING:
            logger.warning(f"Bridge {bridge_id} not in pending state")
            return False
        
        bridge.state = BridgeState.APPROVED
        bridge.target_node_id = approving_node_id
        bridge.approved_at = time.time()
        
        # Move from pending to active
        if bridge.target_island_id in self.pending_requests:
            del self.pending_requests[bridge.target_island_id]
        
        self.active_bridges[bridge.target_island_id] = bridge_id
        
        logger.info(f"Approved bridge request {bridge_id} to island {bridge.target_island_id}")
        return True
    
    def reject_bridge_request(self, bridge_id: str, reason: str = "") -> bool:
        """
        Reject a bridge request
        
        Args:
            bridge_id: Bridge request ID
            reason: Rejection reason
            
        Returns:
            True if successful, False otherwise
        """
        if bridge_id not in self.bridges:
            logger.warning(f"Unknown bridge request {bridge_id}")
            return False
        
        bridge = self.bridges[bridge_id]
        
        if bridge.state != BridgeState.PENDING:
            logger.warning(f"Bridge {bridge_id} not in pending state")
            return False
        
        bridge.state = BridgeState.REJECTED
        bridge.rejection_reason = reason
        
        # Remove from pending
        if bridge.target_island_id in self.pending_requests:
            del self.pending_requests[bridge.target_island_id]
        
        logger.info(f"Rejected bridge request {bridge_id} (reason: {reason})")
        return True
    
    def establish_bridge(self, bridge_id: str) -> bool:
        """
        Establish an active bridge connection
        
        Args:
            bridge_id: Bridge ID to establish
            
        Returns:
            True if successful, False otherwise
        """
        if bridge_id not in self.bridges:
            logger.warning(f"Unknown bridge {bridge_id}")
            return False
        
        bridge = self.bridges[bridge_id]
        
        if bridge.state != BridgeState.APPROVED:
            logger.warning(f"Bridge {bridge_id} not approved")
            return False
        
        bridge.state = BridgeState.ACTIVE
        bridge.activated_at = time.time()
        
        logger.info(f"Established active bridge {bridge_id} to island {bridge.target_island_id}")
        return True
    
    def terminate_bridge(self, island_id: str) -> bool:
        """
        Terminate a bridge to an island
        
        Args:
            island_id: Target island ID
            
        Returns:
            True if successful, False otherwise
        """
        if island_id not in self.active_bridges:
            logger.warning(f"No active bridge to island {island_id}")
            return False
        
        bridge_id = self.active_bridges[island_id]
        bridge = self.bridges[bridge_id]
        
        bridge.state = BridgeState.TERMINATED
        bridge.terminated_at = time.time()
        
        del self.active_bridges[island_id]
        
        logger.info(f"Terminated bridge to island {island_id}")
        return True
    
    def get_bridge_status(self, island_id: str) -> Optional[BridgeConnection]:
        """
        Get bridge status for a specific island
        
        Args:
            island_id: Target island ID
            
        Returns:
            Bridge connection if exists, None otherwise
        """
        # Check active bridges
        if island_id in self.active_bridges:
            bridge_id = self.active_bridges[island_id]
            return self.bridges[bridge_id]
        
        # Check pending requests
        if island_id in self.pending_requests:
            bridge_id = self.pending_requests[island_id]
            return self.bridges[bridge_id]
        
        return None
    
    def get_all_bridges(self) -> List[BridgeConnection]:
        """Get all bridge connections"""
        return list(self.bridges.values())
    
    def get_active_bridges(self) -> List[BridgeConnection]:
        """Get all active bridge connections"""
        return [
            self.bridges[bridge_id]
            for bridge_id in self.active_bridges.values()
        ]
    
    def get_pending_requests(self) -> List[BridgeConnection]:
        """Get all pending bridge requests"""
        return [
            self.bridges[bridge_id]
            for bridge_id in self.pending_requests.values()
        ]
    
    def is_bridged_to_island(self, island_id: str) -> bool:
        """Check if node has active bridge to an island"""
        return island_id in self.active_bridges
    
    def has_pending_request(self, island_id: str) -> bool:
        """Check if node has pending bridge request to an island"""
        return island_id in self.pending_requests
    
    async def start(self):
        """Start bridge manager"""
        self.running = True
        logger.info("Starting bridge manager")
        
        # Start background tasks
        tasks = [
            asyncio.create_task(self._request_timeout_monitor())
        ]
        
        try:
            await asyncio.gather(*tasks)
        except Exception as e:
            logger.error(f"Bridge manager error: {e}")
        finally:
            self.running = False
    
    async def stop(self):
        """Stop bridge manager"""
        self.running = False
        logger.info("Stopping bridge manager")
    
    async def _request_timeout_monitor(self):
        """Monitor bridge requests and handle timeouts"""
        while self.running:
            try:
                current_time = time.time()
                
                # Remove expired pending requests (older than 1 hour)
                expired_requests = []
                for island_id, bridge_id in list(self.pending_requests.items()):
                    bridge = self.bridges[bridge_id]
                    if current_time - bridge.requested_at > 3600:
                        expired_requests.append((island_id, bridge_id))
                
                for island_id, bridge_id in expired_requests:
                    bridge = self.bridges[bridge_id]
                    bridge.state = BridgeState.REJECTED
                    bridge.rejection_reason = "Request timeout"
                    
                    del self.pending_requests[island_id]
                    logger.info(f"Removed expired bridge request {bridge_id} to island {island_id}")
                
                await asyncio.sleep(60)  # Check every minute
                
            except Exception as e:
                logger.error(f"Bridge request timeout monitor error: {e}")
                await asyncio.sleep(10)


# Global bridge manager instance
bridge_manager_instance: Optional[BridgeManager] = None


def get_bridge_manager() -> Optional[BridgeManager]:
    """Get global bridge manager instance"""
    return bridge_manager_instance


def create_bridge_manager(node_id: str, island_id: str) -> BridgeManager:
    """Create and set global bridge manager instance"""
    global bridge_manager_instance
    bridge_manager_instance = BridgeManager(node_id, island_id)
    return bridge_manager_instance