Added functionality for adding custom tags

Sethen Maleno · Sethen Maleno · commit 152257ed4e8e · 2015-05-24T15:41:57.000-04:00
diff --git a/bin/cli.js b/bin/cli.js
@@ -1,5 +1,8 @@
 #!/usr/bin/env node
 
+/* eslint-env node */
+'use strict';
+
 var markdownJson = process.argv[2];
 var markdownInclude = require('../markdown-include');
 
diff --git a/markdown-include.js b/markdown-include.js
@@ -134,9 +134,13 @@ exports.compileFiles = function (path) {
 					self.build[file].parsedData = self.options.tableOfContents.heading + '\n\n' + self.tableOfContents + '\n\n' + self.build[file].parsedData;
 				}
 			}
+		}
 
-			deferred.resolve(self.writeFile(self.build[file].parsedData));
+		if (self.customTags && self.customTags.length) {
+			self.build[file].parsedData = self.resolveCustomTags(self.build[file].parsedData);
 		}
+
+		deferred.resolve(self.writeFile(self.build[file].parsedData));
 	});
 
 	return deferred.promise;
@@ -336,6 +340,28 @@ exports.replaceWith = function (obj) {
 	return replaced;
 };
 
+exports.resolveCustomTags = function (data) {
+	if (data) {
+		var k;
+
+		for (k = 0; k < this.customTags.length; k += 1) {
+			var customTagObj = this.customTags[k];
+			var replacedData;
+
+			if (replacedData) {
+				customTagObj.data = replacedData;
+			}
+			else {
+				customTagObj.data = data;
+			}
+
+			replacedData = this.stripTagsInFile(customTagObj);
+		}
+
+		return replacedData;
+	}
+};
+
 /**
  * For stripping a pattern out of a tag
  * @param  {Object} obj Key value pair include tag and pattern to strip
