bitbot-3.11-fork/modules/commands/outs.py

75 lines
2.1 KiB
Python
Raw Normal View History

import re
from src import utils
STR_MORE = " (more...)"
STR_MORE_LEN = len(STR_MORE.encode("utf8"))
STR_CONTINUED = "(...continued) "
class Out(object):
def __init__(self, server, module_name, target, target_str, tags):
self.server = server
self.module_name = module_name
self._hide_prefix = False
self.target = target
self._target_str = target_str
self._text = ""
self.written = False
2019-05-21 11:56:05 +00:00
self._tags = tags
self._assured = False
def assure(self):
self._assured = True
def write(self, text):
self._text += text
self.written = True
return self
def send(self, method):
if self.has_text():
prefix = ""
if not self._hide_prefix:
prefix = utils.consts.RESET + "[%s] " % self.prefix()
full_text = "%s%s" % (prefix, self._text)
line_factory = None
if method == "PRIVMSG":
line_factory = utils.irc.protocol.privmsg
elif method == "NOTICE":
line_factory = utils.irc.protocol.notice
else:
raise ValueError("Unknown command method '%s'" % method)
line = line_factory(self._target_str, full_text, tags=self._tags)
if self._assured:
line.assure()
valid, truncated = line.truncate(self.server.hostmask())
if truncated:
truncated = valid[-STR_MORE_LEN:]+truncated
new_line = valid[:-STR_MORE_LEN]+STR_MORE
line = utils.irc.parse_line(new_line)
self._text = "%s%s" % (STR_CONTINUED, truncated)
else:
self._text = ""
sent_line = self.server.send(line)
def set_prefix(self, prefix):
self.module_name = prefix
def hide_prefix(self):
self._hide_prefix = True
def has_text(self):
return bool(self._text)
class StdOut(Out):
def prefix(self):
return utils.irc.color(self.module_name, utils.consts.GREEN)
class StdErr(Out):
def prefix(self):
return utils.irc.color(self.module_name, utils.consts.RED)