Support numbering defined by paragraph style

1 files changed, 24 insertions, 3 deletions

diff --git a/lib/docx/numbering-xml.js b/lib/docx/numbering-xml.js
index 90e479e..64c4210 100644
--- a/lib/docx/numbering-xml.js
+++ b/lib/docx/numbering-xml.js
@@ -1,8 +1,21 @@
+var _ = require("underscore");
+
 exports.readNumberingXml = readNumberingXml;
 exports.Numbering = Numbering;
-exports.defaultNumbering = new Numbering({});
+exports.defaultNumbering = new Numbering({}, {});
 
 function Numbering(nums, abstractNums, styles) {
+    var allLevels = _.flatten(_.values(abstractNums).map(function(abstractNum) {
+        return _.values(abstractNum.levels);
+    }));
+
+    var levelsByParagraphStyleId = _.indexBy(
+        allLevels.filter(function(level) {
+            return level.paragraphStyleId != null;
+        }),
+        "paragraphStyleId"
+    );
+
     function findLevel(numId, level) {
         var num = nums[numId];
         if (num) {
@@ -18,8 +31,13 @@ function Numbering(nums, abstractNums, styles) {
         }
     }
 
+    function findLevelByParagraphStyleId(styleId) {
+        return levelsByParagraphStyleId[styleId] || null;
+    }
+
     return {
-        findLevel: findLevel
+        findLevel: findLevel,
+        findLevelByParagraphStyleId: findLevelByParagraphStyleId
     };
 }
 
@@ -47,9 +65,12 @@ function readAbstractNum(element) {
     element.getElementsByTagName("w:lvl").forEach(function(levelElement) {
         var levelIndex = levelElement.attributes["w:ilvl"];
         var numFmt = levelElement.first("w:numFmt").attributes["w:val"];
+        var paragraphStyleId = levelElement.firstOrEmpty("w:pStyle").attributes["w:val"];
+
         levels[levelIndex] = {
             isOrdered: numFmt !== "bullet",
-            level: levelIndex
+            level: levelIndex,
+            paragraphStyleId: paragraphStyleId
         };
     });