""" System service for monitoring and controlling the fishbowl system """ import logging from datetime import datetime, timedelta, timezone from typing import Dict, List, Any, Optional import psutil import json from admin.models import SystemStatus, SystemStatusEnum, SystemConfiguration, LogEntry logger = logging.getLogger(__name__) class SystemService: """Service for system monitoring and control""" def __init__(self): self.system_state = SystemStatusEnum.RUNNING self.start_time = datetime.now(timezone.utc) self.error_count = 0 self.warnings_count = 0 self.log_buffer = [] @classmethod async def initialize(cls): """Initialize system service""" logger.info("System service initialized") async def get_status(self) -> SystemStatus: """Get current system status""" try: uptime_seconds = (datetime.now(timezone.utc) - self.start_time).total_seconds() uptime_str = self._format_uptime(uptime_seconds) # Get resource usage memory = psutil.virtual_memory() cpu_percent = psutil.cpu_percent(interval=1) resource_usage = { "cpu_percent": cpu_percent, "memory_total_mb": memory.total // (1024 * 1024), "memory_used_mb": memory.used // (1024 * 1024), "memory_percent": memory.percent } # Performance metrics performance_metrics = { "avg_response_time": 2.5, # Would track actual response times "requests_per_minute": 30, # Would track actual request rate "database_query_time": 0.05 # Would track actual DB performance } return SystemStatus( status=self.system_state, uptime=uptime_str, version="1.0.0", database_status="healthy", redis_status="healthy", llm_service_status="healthy", discord_bot_status="connected", active_processes=["main", "conversation_engine", "scheduler", "admin_interface"], error_count=self.error_count, warnings_count=self.warnings_count, performance_metrics=performance_metrics, resource_usage=resource_usage ) except Exception as e: logger.error(f"Error getting system status: {e}") return SystemStatus( status=SystemStatusEnum.ERROR, uptime="unknown", version="1.0.0", database_status="error", redis_status="unknown", llm_service_status="unknown", discord_bot_status="unknown", active_processes=[], error_count=self.error_count + 1, warnings_count=self.warnings_count, performance_metrics={}, resource_usage={} ) async def pause_system(self): """Pause the entire system""" try: logger.info("Pausing system operations") self.system_state = SystemStatusEnum.PAUSED # Would integrate with main application to pause operations except Exception as e: logger.error(f"Error pausing system: {e}") raise async def resume_system(self): """Resume system operations""" try: logger.info("Resuming system operations") self.system_state = SystemStatusEnum.RUNNING # Would integrate with main application to resume operations except Exception as e: logger.error(f"Error resuming system: {e}") raise async def get_configuration(self) -> SystemConfiguration: """Get system configuration""" # Default configuration values return SystemConfiguration( conversation_frequency=0.5, response_delay_min=1.0, response_delay_max=5.0, personality_change_rate=0.1, memory_retention_days=90, max_conversation_length=50, creativity_boost=True, conflict_resolution_enabled=True, safety_monitoring=True, auto_moderation=False, backup_frequency_hours=24 ) async def update_configuration(self, config: Dict[str, Any]): """Update system configuration""" try: logger.info(f"Updating system configuration: {config}") # Would integrate with main application to update configuration except Exception as e: logger.error(f"Error updating configuration: {e}") raise async def get_logs(self, limit: int = 100, level: Optional[str] = None) -> List[LogEntry]: """Get system logs""" try: # In production, this would read from actual log files sample_logs = [ LogEntry( timestamp=datetime.now(timezone.utc) - timedelta(minutes=i), level="INFO" if i % 3 != 0 else "DEBUG", component="conversation_engine", message=f"Sample log message {i}", metadata={"log_id": i} ) for i in range(min(limit, 50)) ] if level: sample_logs = [log for log in sample_logs if log.level == level.upper()] return sample_logs except Exception as e: logger.error(f"Error getting logs: {e}") return [] def _format_uptime(self, seconds: float) -> str: """Format uptime in human-readable format""" days, remainder = divmod(int(seconds), 86400) hours, remainder = divmod(remainder, 3600) minutes, seconds = divmod(remainder, 60) if days > 0: return f"{days}d {hours}h {minutes}m" elif hours > 0: return f"{hours}h {minutes}m" else: return f"{minutes}m {seconds}s"