build: update auto-doc to support generate command

Author: clydin

tools/publish/src/generate-docs.ts

@@ -1,6 +1,9 @@
 import * as fs from 'fs';
 import * as path from 'path';
-import { logging, tags } from '@angular-devkit/core';
+import { logging, strings, tags } from '@angular-devkit/core';
+import {
+  getEngineHost,
+} from '../../../packages/@angular/cli/utilities/schematics';
 
 export default async function () {
   const commandsPath = __dirname + '/../../../packages/@angular/cli/commands';
@@ -17,70 +20,117 @@ export default async function () {
       continue;
     }
 
-    try {
-      await command.initialize({});
-    } catch (e) {
-      console.log(`initialize failed [${commandFile}]: ` + e.toString());
-    }
+    generateDoc(command, commandFile);
+
+    if (command.name === 'generate') {
+      const host = getEngineHost();
+      const collection = host.createCollectionDescription('@schematics/angular');
 
-    let optionText;
-    if (!command.options) {
-      optionText = '';
-    } else {
-      optionText = (command.options as any[])
-        .filter(option => !option.hidden)
-        .map(option => {
-          let defaultText = '';
-          if (option.default) {
-            defaultText = `<em>default value: ${option.default}</em>`;
-          }
-          let aliasText = '';
-          if (option.aliases && option.aliases.length > 0) {
-            aliasText = (option.aliases as string[])
-              .map(alias => '<code>' + (alias.length === 1 ? '-' : '--') + alias + '</code>')
-              .join(',');
-            aliasText = ` (alias: ${aliasText})`;
-          }
-
-          return tags.stripIndent`
-            <details>
-              <summary>${option.name}</summary>
-              <p>
-                <code>--${option.name}</code>${aliasText} ${defaultText}
-              </p>
-              <p>
-                ${option.description}
-              </p>
-            </details>
-          `;
-        }).join('\n');
+      for (const schematicName in collection.schematics) {
+        const schematic = collection.schematics[schematicName];
+        if (schematic.hidden || schematic.private) {
+          continue;
+        }
+        const generateCommand = new commandConstructor(
+          { project: { root: path.join(__dirname, '../fake_root/') } },
+          new logging.NullLogger(),
+        );
+        generateDoc(
+          generateCommand,
+          commandFile,
+          { _: [`${collection.name}:${schematicName}`] },
+          {
+            name: strings.dasherize(schematicName),
+            description: schematic.description,
+            postfix: '-' + schematicName,
+          },
+        );
+      }
     }
+  }
+}
+
+interface DocInfo {
+  name: string;
+  description: string;
+  postfix: string;
+}
+async function generateDoc(
+  command: any,
+  commandFile: string,
+  options: any = {},
+  info?: Partial<DocInfo>,
+) {
+  const docInfo = {
+    name: command.name,
+    description: command.description,
+    postfix: '',
+    ...info,
+  };
 
-    const docFile = path.join(
-      __dirname,
-      '../../../docs/documentation/',
-      path.basename(commandFile, '.ts') + '.md');
+  try {
+    await command.initialize(options);
+  } catch (e) {
+    console.log(`initialize failed [${commandFile}]: ` + e);
+  }
 
-    let docText;
-    if (fs.existsSync(docFile)) {
-      docText = fs.readFileSync(docFile, 'utf8');
-      docText = docText.slice(0, docText.indexOf('## Options') + 10);
-    } else {
-      // tslint:disable:max-line-length
-      docText = tags.stripIndent`
-        <!-- Links in /docs/documentation should NOT have \`.md\` at the end, because they end up in our wiki at release. -->
+  let optionText;
+  if (!command.options) {
+    optionText = '';
+  } else {
+    optionText = (command.options as any[])
+      .filter(option => !option.hidden)
+      .map(option => {
+        let defaultText = '';
+        if (option.default) {
+          defaultText = ` <em>default value: ${option.default}</em>`;
+        }
+        let aliasText = '';
+        if (option.aliases && option.aliases.length > 0) {
+          aliasText = (option.aliases as string[])
+            .map(alias => '<code>' + (alias.length === 1 ? '-' : '--') + alias + '</code>')
+            .join(',');
+          aliasText = ` (alias: ${aliasText})`;
+        }
 
-        # ng ${command.name}
+        return tags.stripIndent`
+          <details>
+            <summary>${option.name}</summary>
+            <p>
+              <code>--${option.name}</code>${aliasText}${defaultText}
+            </p>
+            <p>
+              ${option.description}
+            </p>
+          </details>
+        `;
+      }).join('\n');
+  }
 
-        ## Overview
-        ${command.description}
+  const docFile = path.join(
+    __dirname,
+    '../../../docs/documentation/',
+    path.basename(commandFile, '.ts') + docInfo.postfix + '.md');
 
-        ## Options
-      `;
-      // tslint:enable:max-line-length
-    }
+  let docText;
+  if (fs.existsSync(docFile)) {
+    docText = fs.readFileSync(docFile, 'utf8');
+    docText = docText.slice(0, docText.indexOf('## Options') + 10);
+  } else {
+    // tslint:disable:max-line-length
+    docText = tags.stripIndent`
+      <!-- Links in /docs/documentation should NOT have \`.md\` at the end, because they end up in our wiki at release. -->
 
-    const finalText = docText + '\n' + (optionText ? optionText : 'None.') + '\n';
-    fs.writeFileSync(docFile, finalText);
+      # ng ${docInfo.name}
+
+      ## Overview
+      ${docInfo.description}
+
+      ## Options
+    `;
+    // tslint:enable:max-line-length
   }
+
+  const finalText = docText + '\n' + (optionText ? optionText : 'None.') + '\n';
+  fs.writeFileSync(docFile, finalText);
 }