Class MaxResultsView

main.py:306–433

        view.query_message_id = message.id 
      
class MaxResultsView(discord.ui.View):
    def __init__(self, query):
        super().__init__()
        self.query = query
        self.message_id = None  # Used for the MaxResultsView message
        self.query_message_id = None 

    async def handle_search(self, interaction: discord.Interaction, max_results: int):
        shodan_client = interaction.client.shodan
        user = interaction.user.name
        guild_name = interaction.guild.name if interaction.guild else "Direct Message"

        logger.info(f"{user} initiated a search with query '{self.query}' in {guild_name}")

        try:
            result = shodan_client.search(self.query)
            matches = result.get('matches', [])

            if matches:
                limited_matches = matches[:max_results]
                total = result.get('total', 0)
                info = f"{user} used /shodan to search for `{self.query}` from {guild_name} and found {total} results. Here are the top {max_results} results:\n\n"

                if self.query_message_id:
                    try:
                        message_channel = interaction.channel
                        query_message = await message_channel.fetch_message(self.query_message_id)
                        await query_message.edit(content=info, view=None)  # Edit the message to display results
                        # logger.info(f"Successfully edited QueryModal message with ID: {self.query_message_id} to display results.")
                    except Exception as e:
                        logger.error(f"Error editing QueryModal message with ID {self.query_message_id}: {e}")

                await interaction.followup.send(info)

                for match in limited_matches:
                    ip = match.get('ip_str', 'No IP available.')
                    port = match.get('port', 'No port available.')

                    # Extract geolocation details for easy mode
                    lat = match.get('location', {}).get('latitude')
                    long = match.get('location', {}).get('longitude')
                    google_maps_link = f"https://www.google.com/maps?q={lat},{long}" if lat and long else None
                    geolocation_text = f"([map](<{google_maps_link}>))" if google_maps_link else ""

                    # Check if IP is IPv6
                    if is_ipv6(ip):
                        clickable_link = f"{ip} (port: {port}) {geolocation_text}"
                    else:
                        clickable_link = f"[{ip}:{port}](http://{ip}:{port}) {geolocation_text}"
                    
                    await interaction.channel.send(clickable_link)

                    screenshot_data = match.get('screenshot', {}).get('data')
                    if screenshot_data:
                        try:
                            screenshot_bytes = base64.b64decode(screenshot_data)
                            screenshot_file = io.BytesIO(screenshot_bytes)
                            screenshot_filename = f"screenshot_{match.get('ip_str', 'unknown')}.jpg"