preprocessor config, fix for #212

lib/yuidoc.js

@@ -43,7 +43,8 @@ YUI.add('yuidoc', function (Y) {
         version: '0.1.0',
         paths: [],
         themedir: path.join(__dirname, 'themes', 'default'),
-        syntaxtype: 'js'
+        syntaxtype: 'js',
+        preprocessor: undefined
     };
 
     /**
@@ -245,8 +246,40 @@ YUI.add('yuidoc', function (Y) {
             }
         },
 
+        /**
+         * Applies preprocessors to the data tree.
+         * @method applyPreprocessors
+         * @private
+         */
+        applyPreprocessors: function (data) {
+            var preprocessors, preprocessor, p,
+                preprocessorAbsFile, preprocessorModule;
+
+            if (this.options.preprocessor) {
+                preprocessors = [].concat(this.options.preprocessor);
+
+                for (p=0; p<preprocessors.length; p++) {
+                    preprocessor=preprocessors[p];
+
+                    // Check if path is absolute, from:
+                    // http://stackoverflow.com/questions/21698906/how-to-check-if-a-path-is-absolute-or-relative
+                    if (path.resolve(preprocessor)===preprocessor) {
+                        preprocessorAbsFile=preprocessor;
+                    }
+
+                    else {
+                        preprocessorAbsFile=path.join(process.cwd(),preprocessor);
+                    }
+
+                    preprocessorModule=require(preprocessorAbsFile);
+                    preprocessorModule(data);
+                }
+            }
+        },
+
         /**
          * Writes the parser JSON data to disk.
+         * Applies preprocessors, if any.
          * @method writeJSON
          * @param {Object} parser The DocParser instance to use
          * @private
@@ -260,6 +293,8 @@ YUI.add('yuidoc', function (Y) {
 
             data = parser.data;
 
+            this.applyPreprocessors(data);
+
             data.warnings = parser.warnings;
 
             if (self.selleck && self.options.selleck && data.files && data.modules) {

package.json

@@ -49,7 +49,7 @@
     },
     "scripts": {
         "pretest": "jshint ./lib/*.js ./tests/*.js",
-        "test": "istanbul cover --print=both --yui ytestrunner -- --include ./tests/options.js --include ./tests/builder.js --include ./tests/parser.js --include ./tests/parser_coffee.js --include ./tests/files.js --include ./tests/utils.js"
+        "test": "istanbul cover --print=both --yui ytestrunner -- --include ./tests/options.js --include ./tests/builder.js --include ./tests/parser.js --include ./tests/parser_coffee.js --include ./tests/files.js --include ./tests/utils.js --include ./tests/preprocessor.js"
     },
     "preferGlobal": "true",
     "licenses":[

scripts/test.sh

@@ -19,7 +19,8 @@ wait
     ./builder.js \
     ./options.js \
     ./utils.js \
-    ./files.js
+    ./files.js \
+    ./preprocessor.js
 
 exit $?
 

tests/input/preprocessor/preprocessortest.js

@@ -0,0 +1,5 @@
+/**
+ * This class is for testing the preprocessor option.
+ * @class TestPreprocessor
+ * @customtag hello
+ */

tests/lib/testpreprocessor.js

@@ -0,0 +1,9 @@
+module.exports=function(data) {
+	global.testPreprocessorCallCount++;
+
+	for (className in data.classes) {
+		classData=data.classes[className];
+
+		classData.customtagPlusStar=classData.customtag+"*";
+	}
+}

tests/preprocessor.js

@@ -0,0 +1,71 @@
+/*global Y:true */
+var YUITest = require('yuitest'),
+    Assert = YUITest.Assert,
+    path = require('path'),
+    Y = require(path.join(__dirname, '../', 'lib', 'index'));
+
+//Move to the test dir before running the tests.
+process.chdir(__dirname);
+
+var suite = new YUITest.TestSuite({
+    name: 'Preprocessor Test Suite',
+    setUp: function () {
+        process.chdir(__dirname);
+    }
+});
+
+suite.add(new YUITest.TestCase({
+    name: "Preprocessor",
+    'test: single preprocessor': function () {
+        global.testPreprocessorCallCount=0;
+
+        var json = (new Y.YUIDoc({
+            quiet: true,
+            paths: ['input/preprocessor'],
+            outdir: './out',
+            preprocessor: 'lib/testpreprocessor.js'
+        })).run();
+
+        Assert.isObject(json);
+        Assert.areSame(global.testPreprocessorCallCount,1,"the preprocessor was not called");
+    },
+    'test: single preprocessor with absolute path': function () {
+        global.testPreprocessorCallCount=0;
+
+        var json = (new Y.YUIDoc({
+            quiet: true,
+            paths: ['input/preprocessor'],
+            outdir: './out',
+            preprocessor: path.join(process.cwd(),'lib/testpreprocessor.js')
+        })).run();
+
+        Assert.isObject(json);
+        Assert.areSame(global.testPreprocessorCallCount,1,"the preprocessor was not called when an absolute path was used");
+    },
+    'test: several preprocessors': function () {
+        global.testPreprocessorCallCount=0;
+
+        var json = (new Y.YUIDoc({
+            quiet: true,
+            paths: ['input/preprocessor'],
+            outdir: './out',
+            preprocessor: ['lib/testpreprocessor.js','./lib/testpreprocessor']
+        })).run();
+
+       Assert.isObject(json);
+       Assert.areSame(global.testPreprocessorCallCount,2,"the preprocessor was not called twice");
+    },
+    'test: the test preprocessor does its job': function () {
+        var json = (new Y.YUIDoc({
+            quiet: true,
+            paths: ['input/preprocessor'],
+            outdir: './out',
+            preprocessor: 'lib/testpreprocessor.js'
+        })).run();
+
+        Assert.isObject(json);
+        Assert.areSame(json.classes.TestPreprocessor.customtagPlusStar,"hello*","the preprocessor did not modify the data");
+    }
+}));
+
+YUITest.TestRunner.add(suite);