meshcore-bot/modules/commands/satpass_command.py

#!/usr/bin/env python3
"""
Satellite Pass Command - Provides satellite pass information
"""

from .base_command import BaseCommand
from ..solar_conditions import get_next_satellite_pass
from ..models import MeshMessage


class SatpassCommand(BaseCommand):
    """Command to get satellite pass information"""
    
    # Plugin metadata
    name = "satpass"
    keywords = ['satpass']
    description = "Get satellite pass info: satpass <NORAD_number_or_shortcut> [visual]"
    category = "solar"
    requires_internet = True  # Requires internet access for N2YO API
    
    # Common satellite shortcuts
    SATELLITE_SHORTCUTS = {
    'iss': '25544',
    'hst': '20580',  # Hubble Space Telescope
    'hubble': '20580',
    'starlink': '44294',  # Example Starlink satellite
    'tiangong': '48274',  # Tiangong space station
    'goes18': '51850',  # GOES-18 weather satellite
    }
    
    def __init__(self, bot):
        """Initialize the satpass command.
        
        Args:
            bot: The bot instance.
        """
        super().__init__(bot)
        self.satpass_enabled = self.get_config_value('Satpass_Command', 'enabled', fallback=True, value_type='bool')
    
    def can_execute(self, message: MeshMessage) -> bool:
        """Check if this command can be executed with the given message.
        
        Args:
            message: The message triggering the command.
            
        Returns:
            bool: True if command is enabled and checks pass, False otherwise.
        """
        if not self.satpass_enabled:
            return False
        return super().can_execute(message)
    
    async def execute(self, message: MeshMessage) -> bool:
        """Execute the satpass command.
        
        Args:
            message: The message triggering the command.
            
        Returns:
            bool: True if executed successfully, False otherwise.
        """
        try:
            # Check if user provided a satellite number
            content = message.content.strip()
            if content == 'satpass':
                # No satellite specified, show help
                help_text = self._get_help_text()
                await self.send_response(message, help_text)
                return True
            
            # Extract satellite identifier from command
            parts = content.split()
            if len(parts) < 2:
                error_msg = self.translate('commands.satpass.no_satellite')
                await self.send_response(message, error_msg)
                return True
            
            satellite_input = parts[1].lower()
            
            # Check for "visual" or "vis" option
            use_visual = False
            if len(parts) >= 3:
                option = parts[2].lower()
                if option in ['visual', 'vis']:
                    use_visual = True
            
            # Check if it's a shortcut first
            if satellite_input in self.SATELLITE_SHORTCUTS:
                satellite = self.SATELLITE_SHORTCUTS[satellite_input]
            else:
                # Assume it's a NORAD number
                satellite = satellite_input
            
            # Get satellite pass information
            pass_info = get_next_satellite_pass(satellite, use_visual=use_visual)
            
            # Send response
            response = self.translate('commands.satpass.header', pass_info=pass_info)
            await self.send_response(message, response)
            return True
            
        except Exception as e:
            error_msg = self.translate('commands.satpass.error', error=str(e))
            await self.send_response(message, error_msg)
            return False
    
    def _get_help_text(self) -> str:
        """Get detailed help text with shortcuts.
        
        Returns:
            str: Detailed help text including shortcuts.
        """
        shortcuts_text = self.translate('commands.satpass.help_header')
        
        # Group shortcuts by category for better organization
        weather_sats = ['noaa15', 'noaa18', 'noaa19', 'metop-a', 'metop-b', 'metop-c', 'goes16', 'goes17', 'goes18']
        space_stations = ['iss', 'tiangong', 'tiangong1', 'tiangong2']
        telescopes = ['hst', 'hubble']
        other = ['starlink']
        
        # Add weather satellites
        shortcuts_text += self.translate('commands.satpass.category_weather')
        weather_list = [f"{name} ({self.SATELLITE_SHORTCUTS[name]})" for name in weather_sats if name in self.SATELLITE_SHORTCUTS]
        shortcuts_text += ", ".join(weather_list) + "\n"
        
        # Add space stations
        shortcuts_text += self.translate('commands.satpass.category_stations')
        station_list = [f"{name} ({self.SATELLITE_SHORTCUTS[name]})" for name in space_stations if name in self.SATELLITE_SHORTCUTS]
        shortcuts_text += ", ".join(station_list) + "\n"
        
        # Add telescopes
        shortcuts_text += self.translate('commands.satpass.category_telescopes')
        telescope_list = [f"{name} ({self.SATELLITE_SHORTCUTS[name]})" for name in telescopes if name in self.SATELLITE_SHORTCUTS]
        shortcuts_text += ", ".join(telescope_list) + "\n"
        
        # Add other satellites
        shortcuts_text += self.translate('commands.satpass.category_other')
        other_list = [f"{name} ({self.SATELLITE_SHORTCUTS[name]})" for name in other if name in self.SATELLITE_SHORTCUTS]
        shortcuts_text += ", ".join(other_list) + "\n"
        
        shortcuts_text += self.translate('commands.satpass.examples')
        
        return shortcuts_text
    
    def get_help_text(self) -> str:
        """Get help text for this command.
        
        Returns:
            str: The help text for this command.
        """
        return self.translate('commands.satpass.description')