1 files changed, 38 insertions, 0 deletions
diff --git a/Swiften/Roster/SetPresence.h b/Swiften/Roster/SetPresence.h
new file mode 100644
index 0000000..a18ae6d
--- /dev/null
+++ b/Swiften/Roster/SetPresence.h
@@ -0,0 +1,38 @@
+#ifndef SWIFTEN_SetPresence_H
+#define SWIFTEN_SetPresence_H
+
+#include "Swiften/Elements/Presence.h"
+#include "Swiften/JID/JID.h"
+#include "Swiften/Roster/RosterItemOperation.h"
+#include "Swiften/Roster/ContactRosterItem.h"
+
+namespace Swift {
+
+class RosterItem;
+
+class SetPresence : public RosterItemOperation {
+	public:
+		SetPresence(boost::shared_ptr<Presence> presence, JID::CompareType compareType = JID::WithoutResource) : presence_(presence), compareType_(compareType) {
+		}
+
+		virtual void operator() (RosterItem* item) const {
+			ContactRosterItem* contact = dynamic_cast<ContactRosterItem*>(item);
+			if (contact && contact->getJID().equals(presence_->getFrom(), compareType_)) {
+				if (presence_->getType() != Presence::Available) {
+					contact->setStatusShow(StatusShow::None);
+					contact->setStatusText(presence_->getStatus());
+				} else {
+					contact->setStatusShow(presence_->getShow());
+					contact->setStatusText(presence_->getStatus());
+				}
+			}
+		}
+	
+	private:
+		boost::shared_ptr<Presence> presence_;
+		JID::CompareType compareType_;
+};
+
+}
+#endif
+