1 files changed, 38 insertions, 0 deletions
diff --git a/src/com/isode/stroke/serializer/MessageSerializer.java b/src/com/isode/stroke/serializer/MessageSerializer.java
new file mode 100644
index 0000000..27d5c73
--- /dev/null
+++ b/src/com/isode/stroke/serializer/MessageSerializer.java
@@ -0,0 +1,38 @@
+/*
+ * Copyright (c) 2010, Isode Limited, London, England.
+ * All rights reserved.
+ */
+/*
+ * Copyright (c) 2010, Remko Tron�on.
+ * All rights reserved.
+ */
+
+package com.isode.stroke.serializer;
+
+import com.isode.stroke.elements.Message;
+import com.isode.stroke.elements.Stanza;
+import com.isode.stroke.serializer.xml.XMLElement;
+
+public class MessageSerializer extends GenericStanzaSerializer<Message>{
+
+    public MessageSerializer(PayloadSerializerCollection payloadSerializers) {
+        super(Message.class, "message", payloadSerializers);
+    }
+
+    @Override
+    void setStanzaSpecificAttributesGeneric(Message message, XMLElement element) {
+        if (message.getType().equals(Message.Type.Chat)) {
+		element.setAttribute("type", "chat");
+	}
+	else if (message.getType().equals(Message.Type.Groupchat)) {
+		element.setAttribute("type", "groupchat");
+	}
+	else if (message.getType().equals(Message.Type.Headline)) {
+		element.setAttribute("type", "headline");
+	}
+	else if (message.getType().equals(Message.Type.Error)) {
+		element.setAttribute("type", "error");
+	}
+    }
+    
+}