LALALALALALALALALA

2 files changed, 74 insertions, 0 deletions

diff --git a/src/contact.py b/src/contact.py
new file mode 100644
index 00000000..c3b07fc2
--- /dev/null
+++ b/src/contact.py
@@ -0,0 +1,29 @@
+# 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 sleekxmpp.xmlstream.jid import JID
+
+class Contact(object):
+    """
+    Defines a roster item
+    """
+    def __init__(self, jid):
+        self._jid = JID(jid)         # a SleekXMPP jid object
+        self._display_name = None
+        self.groups = []        # a list of groups the contact is in
+
+    def getJid(self):
+        return self._jid
diff --git a/src/roster.py b/src/roster.py
new file mode 100644
index 00000000..421e05e4
--- /dev/null
+++ b/src/roster.py
@@ -0,0 +1,45 @@
+# 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 contact import Contact
+
+class Roster(object):
+    """
+    Defines the roster
+    """
+    def __init__(self):
+        self._contacts = {}
+
+    def addContactToList(self, contact):
+        assert isinstance(contact, Contact)
+        assert contact not in self._contacts
+        self._contacts[contact.getJid().bare] = contact
+
+    def getContacts(self):
+        """
+        returns all the contacts in a list
+        TODO: sorted
+        TODO: only some contacts (online only for example)
+        """
+        return [contact for contact in self._contacts.keys()]
+
+    def __len__(self):
+        return len(self._contacts)
+
+    def getContact(self, bare_jid):
+        if bare_jid not in self._contacts:
+            return None
+        return self._contacts[bare_jid]