2 files changed, 124 insertions, 0 deletions
diff --git a/src/room.py b/src/room.py
new file mode 100644
index 00000000..87a78107
--- /dev/null
+++ b/src/room.py
@@ -0,0 +1,85 @@
+#!/usr/bin/python
+# -*- coding:utf-8 -*-
+#
+# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
+#
+# This file is part of Poezio.
+#
+# Poezio is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3 of the License.
+#
+# Poezio 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Poezio.  If not, see <http://www.gnu.org/licenses/>.
+
+from datetime import datetime
+from random import randrange
+from config import config
+from logging import logger
+
+class Room(object):
+    """
+    """
+    def __init__(self, name, nick, number):
+        self.name = name
+        self.own_nick = nick
+        self.color_state = 11   # color used in RoomInfo
+        self.nb = number       # number used in RoomInfo
+        self.joined = False     # false until self presence is received
+        self.users = []
+        self.lines = []         # (time, nick, msg) or (time, info)
+        self.topic = ''
+
+    def disconnect(self):
+        self.joined = False
+        self.users = []
+
+    def add_message(self, nick, msg, date=None):
+        if not date:
+            date = datetime.now()
+        color = None
+        self.set_color_state(12)
+        if nick != self.own_nick and self.joined: # do the highlight thing
+            if self.own_nick in msg:
+                self.set_color_state(13)
+                color = 3
+            else:
+                highlight_words = config.get('highlight_on', '').split(':')
+                for word in highlight_words:
+                    if word.lower() in msg.lower() and word != '':
+                        self.set_color_state(13)
+                        color = 3
+        if not msg:
+            logger.info('msg is None..., %s' % (nick))
+            return
+        self.lines.append((date, nick.encode('utf-8'),
+                          msg.encode('utf-8'), color))
+        if self.joined:         # log only NEW messages, not the history received on join
+            logger.message(self.name, nick.encode('utf-8'), msg.encode('utf-8'))
+        return color
+
+    def add_info(self, info, date=None):
+        """ info, like join/quit/status messages"""
+        if not date:
+            date = datetime.now()
+        try:
+            self.lines.append((date, info.encode('utf-8')))
+            return info.encode('utf-8')
+        except:
+            self.lines.append((date, info))
+            return info
+
+    def get_user_by_name(self, nick):
+        for user in self.users:
+            if user.nick == nick.encode('utf-8'):
+                return user
+        return None
+
+    def set_color_state(self, color):
+        if self.color_state < color or color == 11:
+            self.color_state = color
diff --git a/src/user.py b/src/user.py
new file mode 100644
index 00000000..0dee9e42
--- /dev/null
+++ b/src/user.py
@@ -0,0 +1,39 @@
+#!/usr/bin/python
+# -*- coding:utf-8 -*-
+#
+# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
+#
+# This file is part of Poezio.
+#
+# Poezio is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3 of the License.
+#
+# Poezio 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Poezio.  If not, see <http://www.gnu.org/licenses/>.
+
+from random import randrange
+from config import config
+
+class User(object):
+    """
+    keep trace of an user in a Room
+    """
+    def __init__(self, nick, affiliation, show, status, role):
+        self.update(affiliation, show, status, role)
+        self.change_nick(nick)
+        self.color = randrange(2, 10)
+
+    def update(self, affiliation, show, status, role):
+        self.affiliation = affiliation
+        self.show = show
+        self.status = status
+        self.role = role
+
+    def change_nick(self, nick):
+        self.nick = nick.encode('utf-8')