Mode mode (and mode arg) parsing to IRCChannel.py, add IRCChannel.mode_str

1 files changed, 40 insertions, 1 deletions

diff --git a/src/IRCChannel.py b/src/IRCChannel.py
index 82f8468d..e195248b 100644
--- a/src/IRCChannel.py
+++ b/src/IRCChannel.py
@@ -1,7 +1,9 @@
-import typing, uuid
+import re, typing, uuid
 from src import EventManager, IRCBot, IRCBuffer, IRCObject, IRCServer, IRCUser
 from src import utils
 
+RE_MODES = re.compile(r"[-+]\w+")
+
 class Channel(IRCObject.Object):
     name = ""
     def __init__(self, name: str, id, server: "IRCServer.Server",
@@ -50,6 +52,26 @@ class Channel(IRCObject.Object):
     def has_user(self, user: IRCUser.User) -> bool:
         return user in self.users
 
+    def mode_str(self) -> str:
+        modes = [] # type: typing.List[typing.Tuple[str, typing.List[str]]]
+        # sorta alphanumerically by mode char
+        modes_iter = sorted(self.modes.items(), key=lambda mode: mode[0])
+
+        for mode, args in modes_iter:
+            # not list mode (e.g. +b) and not prefix mode (e.g. +o)
+            if (not mode in self.server.channel_list_modes and
+                    not mode in self.server.prefix_modes):
+                args_list = typing.cast(typing.List[str], list(args))
+                modes.append((mode, args_list))
+
+        # move modes with args to the front
+        modes.sort(key=lambda mode: not bool(mode[1]))
+
+        out_modes = "".join(mode for mode, args in modes)
+        out_args = " ".join(args[0] for mode, args in modes if args)
+
+        return "+%s%s" % (out_modes, " %s" % out_args if out_args else "")
+
     def add_mode(self, mode: str, arg: str=None):
         if not mode in self.modes:
             self.modes[mode] = set([])
@@ -87,6 +109,23 @@ class Channel(IRCObject.Object):
         else:
             self.add_mode(mode, arg)
 
+    def parse_modes(self, modes: str, args: typing.List[str]):
+        for chunk in RE_MODES.findall(modes):
+            remove = chunk[0] == "-"
+            for mode in chunk[1:]:
+                if mode in self.server.channel_list_modes:
+                    args.pop(0)
+                elif (mode in self.server.channel_paramatered_modes or
+                        mode in self.server.prefix_modes):
+                    self.change_mode(remove, mode, args.pop(0))
+                elif mode in self.server.channel_setting_modes:
+                    if remove:
+                        self.change_mode(remove, mode)
+                    else:
+                        self.change_mode(remove, mode, args.pop(0))
+                elif mode in self.server.channel_modes:
+                    self.change_mode(remove, mode)
+
     def set_setting(self, setting: str, value: typing.Any):
         self.bot.database.channel_settings.set(self.id, setting, value)
     def get_setting(self, setting: str, default: typing.Any=None