paircd/paircd/handler/privmsg.py

import logging

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


class PrivmsgHandler(CommandHandler):
    def __init__(self) -> None:
        super().__init__("PRIVMSG", 2)

    async def handle(self, server: Server, client: Client, msg: Message) -> None:
        recipient = msg.args[0]
        raw_msg = msg.args[1]
        out = Message("PRIVMSG", [recipient, f":{raw_msg}"], prefix=client.id())

        for name, other_client in server.clients_by_nick.items():
            if name == recipient:
                other_client.write_message(out)
                return

        for name, channel in server.channels_by_name.items():
            if name == recipient:
                channel.write_message(out)
                return

        client.log("unknown recipient", level=logging.WARN)