Introducing nemubot v4 directories architecture

lib/channel.py

@@ -0,0 +1,102 @@
+# coding=utf-8
+
+# Nemubot is a smart and modulable IM bot.
+# Copyright (C) 2012-2014  nemunaire
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+class Channel:
+    def __init__(self, name, password=None):
+        self.name = name
+        self.password = password
+        self.people = dict()
+        self.topic = ""
+
+    def treat(self, cmd, msg):
+        if cmd == "353":
+            self.parse353(msg)
+        elif cmd == "332":
+            self.parse332(msg)
+        elif cmd == "MODE":
+            self.mode(msg)
+        elif cmd == "JOIN":
+            self.join(msg.nick)
+        elif cmd == "NICK":
+            self.nick(msg.nick, msg.content)
+        elif cmd == "PART" or cmd == "QUIT":
+            self.part(msg.nick)
+        elif cmd == "TOPIC":
+            self.topic = self.content
+
+    def join(self, nick, level = 0):
+        """Someone join the channel"""
+        #print ("%s arrive sur %s" % (nick, self.name))
+        self.people[nick] = level
+
+    def chtopic(self, newtopic):
+        """Send command to change the topic"""
+        self.srv.send_msg(self.name, newtopic, "TOPIC")
+        self.topic = newtopic
+
+    def nick(self, oldnick, newnick):
+        """Someone change his nick"""
+        if oldnick in self.people:
+            #print ("%s change de nom pour %s sur %s" % (oldnick, newnick, self.name))
+            lvl = self.people[oldnick]
+            del self.people[oldnick]
+            self.people[newnick] = lvl
+
+    def part(self, nick):
+        """Someone leave the channel"""
+        if nick in self.people:
+            #print ("%s vient de quitter %s" % (nick, self.name))
+            del self.people[nick]
+
+    def mode(self, msg):
+        if msg.content[0] == "-k":
+            self.password = ""
+        elif msg.content[0] == "+k":
+            if len(msg.content) > 1:
+                self.password = ' '.join(msg.content[1:])[1:]
+            else:
+                self.password = msg.content[1]
+        elif msg.content[0] == "+o":
+            self.people[msg.nick] |= 4
+        elif msg.content[0] == "-o":
+            self.people[msg.nick] &= ~4
+        elif msg.content[0] == "+h":
+            self.people[msg.nick] |= 2
+        elif msg.content[0] == "-h":
+            self.people[msg.nick] &= ~2
+        elif msg.content[0] == "+v":
+            self.people[msg.nick] |= 1
+        elif msg.content[0] == "-v":
+            self.people[msg.nick] &= ~1
+
+    def parse332(self, msg):
+        self.topic = msg.content
+
+    def parse353(self, msg):
+        for p in msg.content:
+            p = p.decode()
+            if p[0] == "@":
+                level = 4
+            elif p[0] == "%":
+                level = 2
+            elif p[0] == "+":
+                level = 1
+            else:
+                self.join(p, 0)
+                continue
+            self.join(p[1:], level)