Add content type for embedded style map

1 files changed, 34 insertions, 5 deletions

diff --git a/lib/docx/style-map.js b/lib/docx/style-map.js
index 04597a2..ea88cc7 100644
--- a/lib/docx/style-map.js
+++ b/lib/docx/style-map.js
@@ -1,7 +1,6 @@
 var _ = require("underscore");
 
 var promises = require("../promises");
-var officeXml = require("./office-xml-reader");
 var xml = require("../xml");
 
 exports.writeStyleMap = writeStyleMap;
@@ -10,16 +9,20 @@ exports.readStyleMap = readStyleMap;
 
 var schema = "http://schemas.zwobble.org/mammoth/style-map";
 var styleMapPath = "mammoth/style-map";
+var styleMapAbsolutePath = "/" + styleMapPath;
 
 function writeStyleMap(docxFile, styleMap) {
     docxFile.write(styleMapPath, styleMap);
-    return updateRelationships(docxFile);
+    return updateRelationships(docxFile).then(function() {
+        return updateContentTypes(docxFile);
+    });
 }
 
 function updateRelationships(docxFile) {
     var path = "word/_rels/document.xml.rels";
     var relationshipsUri = "http://schemas.openxmlformats.org/package/2006/relationships";
-    return officeXml.readXmlFromZipFile(docxFile, path)
+    return docxFile.read(path, "utf8")
+        .then(xml.readString)
         .then(function(relationshipsDocument) {
             var relationshipsContainer = relationshipsDocument.root;
             var relationships = relationshipsContainer.children;
@@ -27,7 +30,7 @@ function updateRelationships(docxFile) {
             var attributes = {
                 "Id": relationshipId,
                 "Type": schema,
-                "Target": "/" + styleMapPath
+                "Target": styleMapAbsolutePath
             };
             
             var existingRelationship = _.find(relationships, function(relationship) {
@@ -41,7 +44,33 @@ function updateRelationships(docxFile) {
             var namespaces = {"": relationshipsUri};
             return docxFile.write(path, xml.writeString(relationshipsContainer, namespaces));
         });
-    
+}
+
+function updateContentTypes(docxFile) {
+    var path = "[Content_Types].xml";
+    var contentTypesUri = "http://schemas.openxmlformats.org/package/2006/content-types";
+    var overrideName = "{" + contentTypesUri + "}Override";
+    return docxFile.read(path, "utf8")
+        .then(xml.readString)
+        .then(function(typesDocument) {
+            var typesElement = typesDocument.root;
+            var children = typesElement.children;
+            var attributes = {
+                "PartName": styleMapAbsolutePath,
+                "ContentType": "text/prs.mammoth.style-map"
+            };
+            
+            var existingOverride = _.find(children, function(child) {
+                return child.name === overrideName && child.attributes.PartName === styleMapAbsolutePath;
+            });
+            if (existingOverride) {
+                existingOverride.attributes = attributes;
+            } else {
+                children.push(xml.element(overrideName, attributes));
+            }
+            var namespaces = {"": contentTypesUri};
+            return docxFile.write(path, xml.writeString(typesElement, namespaces));
+        });
 }
 
 function readStyleMap(docxFile) {