Add new hook that treat all messages, not only PRIVMSG ; can respond with a response.Hook to create a hook

channel.py

@@ -1,71 +1,102 @@
 # coding=utf-8
 
+# Nemubot is a modulable IRC bot, built around XML configuration files.
+# Copyright (C) 2012  Mercier Pierre-Olivier
+#
+# 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, node, srv):
-    self.node = node
-    self.name = node["name"]
-    self.password = node["password"]
-    self.people = dict()
-    self.srv = srv
-    self.topic = ""
+    def __init__(self, name, password=None):
+        self.name = name
+        self.password = password
+        self.people = dict()
+        self.topic = ""
 
-  def join(self, nick, level = 0):
-    #print ("%s arrive sur %s" % (nick, self.name))
-    self.people[nick] = level
+    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 chtopic(self, newtopic):
-    """Send command to change the topic"""
-    self.srv.send_msg(self.name, newtopic, "TOPIC")
-    self.topic = newtopic
+    def join(self, nick, level = 0):
+        """Someone join the channel"""
+        #print ("%s arrive sur %s" % (nick, self.name))
+        self.people[nick] = level
 
-  def nick(self, oldnick, newnick):
-    #print ("%s change de nom pour %s" % (oldnick, newnick))
-    if oldnick in self.people:
-      lvl = self.people[oldnick]
-      del self.people[oldnick]
-    else:
-      lvl = 0
-    self.people[newnick] = lvl
+    def chtopic(self, newtopic):
+        """Send command to change the topic"""
+        self.srv.send_msg(self.name, newtopic, "TOPIC")
+        self.topic = newtopic
 
-  def part(self, nick):
-    #print ("%s vient de quitter %s" % (self.sender, self.channel))
-    if nick in self.people:
-      del self.people[nick]
+    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 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 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 parse332(self, msg):
-    self.topic = msg.content
+    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 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.people[p] = 0
-        continue
-      self.people[p[1:]] = level
+    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)