1 files changed, 96 insertions, 0 deletions
diff --git a/src/main/java/de/thedevstack/conversationsplus/entities/Roster.java b/src/main/java/de/thedevstack/conversationsplus/entities/Roster.java
new file mode 100644
index 00000000..ec1a9426
--- /dev/null
+++ b/src/main/java/de/thedevstack/conversationsplus/entities/Roster.java
@@ -0,0 +1,96 @@
+package de.thedevstack.conversationsplus.entities;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+
+import de.thedevstack.conversationsplus.xmpp.jid.Jid;
+
+public class Roster {
+	final Account account;
+	final HashMap<Jid, Contact> contacts = new HashMap<>();
+	private String version = null;
+
+	public Roster(Account account) {
+		this.account = account;
+	}
+
+	public Contact getContactFromRoster(Jid jid) {
+		if (jid == null) {
+			return null;
+		}
+		synchronized (this.contacts) {
+			Contact contact = contacts.get(jid.toBareJid());
+			if (contact != null && contact.showInRoster()) {
+				return contact;
+			} else {
+				return null;
+			}
+		}
+	}
+
+	public Contact getContact(final Jid jid) {
+		synchronized (this.contacts) {
+			if (!contacts.containsKey(jid.toBareJid())) {
+				Contact contact = new Contact(jid.toBareJid());
+				contact.setAccount(account);
+				contacts.put(contact.getJid().toBareJid(), contact);
+				return contact;
+			}
+			return contacts.get(jid.toBareJid());
+		}
+	}
+
+	public void clearPresences() {
+		for (Contact contact : getContacts()) {
+			contact.clearPresences();
+		}
+	}
+
+	public void markAllAsNotInRoster() {
+		for (Contact contact : getContacts()) {
+			contact.resetOption(Contact.Options.IN_ROSTER);
+		}
+	}
+
+	public List<Contact> getWithSystemAccounts() {
+		List<Contact> with = getContacts();
+		for(Iterator<Contact> iterator = with.iterator(); iterator.hasNext();) {
+			Contact contact = iterator.next();
+			if (contact.getSystemAccount() == null) {
+				iterator.remove();
+			}
+		}
+		return with;
+	}
+
+	public List<Contact> getContacts() {
+		synchronized (this.contacts) {
+			return new ArrayList<>(this.contacts.values());
+		}
+	}
+
+	public void initContact(final Contact contact) {
+		if (contact == null) {
+			return;
+		}
+		contact.setAccount(account);
+		contact.setOption(Contact.Options.IN_ROSTER);
+		synchronized (this.contacts) {
+			contacts.put(contact.getJid().toBareJid(), contact);
+		}
+	}
+
+	public void setVersion(String version) {
+		this.version = version;
+	}
+
+	public String getVersion() {
+		return this.version;
+	}
+
+	public Account getAccount() {
+		return this.account;
+	}
+}