preprocessor config, fix for #212

lib/yuidoc.js

@@ -245,8 +245,33 @@ YUI.add('yuidoc', function (Y) {
             }
         },
 
+        /**
+         * Applies preprocessors to the data tree.
+         * @method applyPreprocessors
+         * @private
+         */
+        applyPreprocessors: function (data) {
+            var preprocessors,
+                preprocessorsRelativeTo;
+
+            // If the preprocessor paths are relative they will be searched for relative
+            // to the process working directory. This is consistent with how other paths
+            // are treated by yuidoc, such as the config options 'paths' and 'outdir'.
+            preprocessorsRelativeTo = process.cwd();
+
+            if (this.options.preprocessor) {
+                preprocessors = [].concat(this.options.preprocessor);
+
+                preprocessors.forEach(function (preprocessor) {
+                    var preprocessorModule = require(path.resolve(preprocessorsRelativeTo, preprocessor));
+                    preprocessorModule(data);
+                });
+            }
+        },
+
         /**
          * Writes the parser JSON data to disk.
+         * Applies preprocessors, if any.
          * @method writeJSON
          * @param {Object} parser The DocParser instance to use
          * @private
@@ -262,6 +287,8 @@ YUI.add('yuidoc', function (Y) {
 
             data.warnings = parser.warnings;
 
+            this.applyPreprocessors(data);
+
             if (self.selleck && self.options.selleck && data.files && data.modules) {
                 Object.keys(self.selleck).forEach(function (file) {
                     Object.keys(data.files).forEach(function (f) {

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);