From 8cbd1cc4aab38dfdc35887a664d9c4c8e726aa94 Mon Sep 17 00:00:00 2001
From: Alistair McIntyre <ajm412@uclive.ac.nz>
Date: Tue, 25 Apr 2017 17:38:26 +1200
Subject: [PATCH] Added support to import XML packet to XML Document object.
 #story[820]

---
 .../seng302/models/parsers/StreamParser.java  | 47 ++++++++++++++++---
 1 file changed, 40 insertions(+), 7 deletions(-)

diff --git a/src/main/java/seng302/models/parsers/StreamParser.java b/src/main/java/seng302/models/parsers/StreamParser.java
index 4fb6d4e3..2a6037cf 100644
--- a/src/main/java/seng302/models/parsers/StreamParser.java
+++ b/src/main/java/seng302/models/parsers/StreamParser.java
@@ -1,25 +1,24 @@
 package seng302.models.parsers;
 
 
-import org.w3c.dom.Element;
+import org.w3c.dom.Document;
+import org.xml.sax.InputSource;
 import org.xml.sax.SAXException;
 
+import javax.xml.parsers.DocumentBuilder;
 import javax.xml.parsers.DocumentBuilderFactory;
 import javax.xml.parsers.ParserConfigurationException;
 import java.io.ByteArrayInputStream;
 import java.io.IOException;
-import java.nio.ByteBuffer;
-import java.nio.ByteOrder;
-import java.util.ArrayList;
+import java.io.StringReader;
 import java.util.Arrays;
-import java.util.Map;
 
 /**
  * Created by kre39 on 23/04/17.
  */
 public class StreamParser {
 
-    static void parsePacket(StreamPacket packet) {
+     static void parsePacket(StreamPacket packet) {
         switch (packet.getType()){
             case HEARTBEAT:
                 extractHeartBeat(packet);
@@ -74,7 +73,41 @@ public class StreamParser {
 
     }
 
-    private static void extractXmlMessage(StreamPacket packet){
+    static void extractXmlMessage(StreamPacket packet){
+
+        byte[] payload = packet.getPayload();
+        String xmlMessage = "";
+
+        ByteArrayInputStream payloadStream = new ByteArrayInputStream(payload);
+
+        //Bunch of data we don't want (Message Version Number, AckNumber, Timestamp)
+        payloadStream.skip(9);
+        int xmlMessageSubType = payloadStream.read();
+        payloadStream.skip(2);
+
+        //checks the length of the xml message itself
+        int xmlMessageLength = payloadStream.read() | payloadStream.read() << 8;
+
+        //Converts XML message to string to be parsed
+        int currentChar;
+        while (payloadStream.available() > 0 && (currentChar = payloadStream.read()) != 0) {
+        xmlMessage += (char)currentChar;
+        }
+
+        //Create XML document Object
+        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
+        DocumentBuilder db = null;
+        try {
+            db = dbf.newDocumentBuilder();
+            Document doc = db.parse(new InputSource(new StringReader(xmlMessage)));
+            // TODO: 25/04/17 ajm412: Check that the object matches expected structure and return Document object.
+        } catch (ParserConfigurationException e) {
+            e.printStackTrace();
+        } catch (SAXException e) {
+            e.printStackTrace();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
 
     }