Split command StdOut/StdErr in to their own file (commands)

1 files changed, 61 insertions, 0 deletions

diff --git a/modules/commands/outs.py b/modules/commands/outs.py
new file mode 100644
index 00000000..6d9efe38
--- /dev/null
+++ b/modules/commands/outs.py
@@ -0,0 +1,61 @@
+from src import utils
+
+class Out(object):
+    def __init__(self, server, module_name, target, msgid, statusmsg):
+        self.server = server
+        self.module_name = module_name
+        self._hide_prefix = False
+        self.target = target
+        self._text = ""
+        self.written = False
+        self._msgid = msgid
+        self._statusmsg = statusmsg
+
+    def write(self, text):
+        self._text += text
+        self.written = True
+        return self
+
+    def send(self, method):
+        if self.has_text():
+            text = self._text
+            text_encoded = text.encode("utf8")
+            if len(text_encoded) > OUT_CUTOFF:
+                text = "%s%s" % (text_encoded[:OUT_CUTOFF].decode("utf8"
+                    ).rstrip(), STR_MORE)
+                self._text = "%s%s" % (STR_CONTINUED, text_encoded[OUT_CUTOFF:
+                    ].decode("utf8").lstrip())
+            else:
+                self._text = ""
+
+
+            tags = {}
+            if self._msgid:
+                tags["+draft/reply"] = self._msgid
+
+            prefix = ""
+            if not self._hide_prefix:
+                prefix = utils.consts.RESET + "[%s] " % self.prefix()
+
+            target_str = "%s%s" % (self._statusmsg, self.target.name)
+            full_text = "%s%s" % (prefix, text)
+            if method == "PRIVMSG":
+                self.server.send_message(target_str, full_text, tags=tags)
+            elif method == "NOTICE":
+                self.server.send_notice(target_str, full_text, tags=tags)
+
+    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)
+