bot6/util.py

import irctokens


class Util:
    def __init__(self, config, sock):
        self.sock = sock
        self.config = config
        self.target = ""

    def send(self, raw: str):
        print(f"> {raw}")
        self.sock.sendall(f"{raw}\r\n".encode("utf8"))

    def quit(self, msg=None):
        if msg != None:
            self.send("QUIT :" + msg)
        else:
            self.send("QUIT")

    def nick(self, nick=None):
        if nick == None:
            self.send("NICK " + self.config.self.nick)
        else:
            self.send("NICK " + nick)

    def mesg(self, msg: str, t=None):
        if t == None:
            t = self.target
        msg = str(msg).partition("\n")[0]
        if len(msg) >= 900:
            msg = msg[:900]
            self.mesg("message too long!")
        self.send(irctokens.build("PRIVMSG", [t, str(msg)]).format())