import sys

from src.channel_manager import ChannelManager
from src.plugin_base import PluginBase


class Plugin(PluginBase):
    def handle_message(self, source_nick, channel, message):
        message_parts = message.split()
        self.channel_manager = ChannelManager()
        if message_parts[0] == '!hello':
            self.bot.ircsend(f'PRIVMSG {channel} :Hello, {source_nick}!')

        elif message_parts[0] == '!join':
            if len(message_parts) == 0:
                self.bot.ircsend(f'PRIVMSG {channel} :Please specify a channel to join')
                return
            else:
                self.channel_manager.save_channel(message_parts[1])
                self.bot.ircsend(f'JOIN {message_parts[1]}')

        elif message_parts[0] == '!part':
            if len(message_parts) == 1:
                self.bot.ircsend(f'PART {channel}')
                self.channel_manager.remove_channel(channel)
            else:
                self.bot.ircsend(f'PART {message_parts[1]}')
                self.channel_manager.remove_channel(message_parts[1])

        elif message_parts[0] == "!quit":
            if len(message_parts) == 0:
                quit_message = 'EliteBot!'
            else:
                quit_message = message[len(message_parts[0]) + 1:]
            self.bot.ircsend(f'QUIT :{quit_message}')
            self.bot.ircsock.close()
            self.bot.connected = False
            sys.exit()

        elif message_parts[0] == "!raw":
            if len(message_parts) > 1:
                if message_parts[1].upper() == "PRIVMSG" and len(message_parts) > 3:
                    raw_command = ' '.join(message_parts[1:3]) + " :" + ' '.join(message_parts[3:])
                else:
                    raw_command = ' '.join(message_parts[1:])
                self.bot.ircsend(raw_command)

        elif message_parts[0] == '!me':
            if len(message_parts) > 1:
                action_message = ' '.join(message_parts[1:])
                self.bot.ircsend(f'PRIVMSG {channel} :\x01ACTION {action_message}\x01')
            else:
                self.bot.ircsend(f'PRIVMSG {channel} :Please specify an action')

        elif message_parts[0] == '!ping':
            if len(message_parts) > 1:
                self.bot.ircsend(f'PRIVMSG {channel} :Pinging {message_parts[1]}')
            else:
                self.bot.ircsend(f'PRIVMSG {channel} :Please specify a nick to ping')