Exit nonzero from readme cmd if section is absent

Author: Anand Thakker

lib/commands/readme.js

@@ -51,7 +51,10 @@ function readme(documentation, parsedArgs) {
       throw err;
     }
     var readmeContent = fs.readFileSync(readmeFile, 'utf8');
-    mdast.use(plugin, { toInject: docsAst }).process(readmeContent, onInjected.bind(null, readmeContent));
+    mdast.use(plugin, {
+      section: readmeOptions.section,
+      toInject: docsAst
+    }).process(readmeContent, onInjected.bind(null, readmeContent));
   }
 
   function onInjected(readmeContent, err, file, content) {
@@ -81,7 +84,9 @@ function readme(documentation, parsedArgs) {
 // wrap the inject utility as an mdast plugin
 function plugin(mdast, options) {
   return function transform(targetAst, file, next) {
-    inject('API', targetAst, options.toInject);
+    if (!inject(options.section, targetAst, options.toInject)) {
+      return next(new Error('Heading ' + options.section + ' not found.'));
+    }
     next();
   };
 }

package.json

@@ -28,7 +28,7 @@
     "jsdoc-inline-lex": "^1.0.1",
     "mdast": "^2.0.0",
     "mdast-toc": "^1.1.0",
-    "mdast-util-inject": "^1.0.0",
+    "mdast-util-inject": "^1.1.0",
     "micromatch": "^2.1.6",
     "mime": "^1.3.4",
     "module-deps": "^4.0.2",

test/bin-readme.js

@@ -87,6 +87,14 @@ test('readme command', function (group) {
       });
     });
 
+    group.test('errors if specified readme section is missing', function (t) {
+      documentation(['readme index.js -s DUMMY'], {cwd: d}, function (err, stdout, stderr) {
+        t.ok(err);
+        t.ok(err.code !== 0, 'exit nonzero');
+        t.end();
+      });
+    });
+
     group.end();
   });
 });