from paircd.client import Client
from paircd.command_handler import CommandHandler
from paircd.message import Message
from paircd.server import Server


class QuitHandler(CommandHandler):
    def __init__(self) -> None:
        super().__init__("QUIT")

    async def handle(self, server: Server, client: Client, msg: Message) -> None:
        quit_msg = ""
        if len(msg.args) == 1:
            quit_msg = msg.args[0]
        await client.quit(server, f"Quit: {quit_msg}")