1 files changed, 43 insertions, 0 deletions
diff --git a/Swiften/Elements/Message.h b/Swiften/Elements/Message.h
new file mode 100644
index 0000000..6d9171f
--- /dev/null
+++ b/Swiften/Elements/Message.h
@@ -0,0 +1,43 @@
+#pragma once
+
+#include <boost/optional.hpp>
+
+#include "Swiften/Base/String.h"
+#include "Swiften/Elements/Body.h"
+#include "Swiften/Elements/ErrorPayload.h"
+#include "Swiften/Elements/Stanza.h"
+
+namespace Swift
+{
+	class Message : public Stanza
+	{
+	  public:
+			enum Type { Normal, Chat, Error, Groupchat, Headline };
+
+			Message() : type_(Chat) { }
+
+			String getBody() const { 
+				boost::shared_ptr<Body> body(getPayload<Body>());
+				if (body) {
+					return body->getText();
+				}
+				return "";
+			}
+
+			void setBody(const String& body) { 
+				updatePayload(boost::shared_ptr<Body>(new Body(body)));
+			}
+
+			bool isError() {
+				boost::shared_ptr<Swift::ErrorPayload> error(getPayload<Swift::ErrorPayload>());
+				return getType() == Message::Error || error;
+			}
+
+			Type getType() const { return type_; }
+			void setType(Type type) { type_ = type; }
+
+		private:
+			String body_;
+			Type type_;
+	};
+}