Add support for multiple yamls, uses openapi to bundle yamls (#148)

rohit-gohri · Mar 13, 2022 · ad66f92 · ad66f92
1 parent 937e73a
commit ad66f92
Show file tree

Hide file tree

Showing 8 changed files with 1,325 additions and 29 deletions.
diff --git a/.changeset/new-poems-return.md b/.changeset/new-poems-return.md
@@ -0,0 +1,5 @@
+---
+'docusaurus-plugin-redoc': minor
+---
+
+Add support for bundling multiple yamls that reference each other
diff --git a/packages/docusaurus-plugin-redoc/package.json b/packages/docusaurus-plugin-redoc/package.json
@@ -28,9 +28,9 @@
   "dependencies": {
     "@docusaurus/types": "^2.0.0-beta.17",
     "@docusaurus/utils": "^2.0.0-beta.17",
+    "@redocly/openapi-core": "^1.0.0-beta.87",
     "joi": "^17.5.0",
-    "redoc": "^2.0.0-rc.64",
-    "yaml": "^1.10.2"
+    "redoc": "^2.0.0-rc.64"
   },
   "engines": {
     "node": ">=14"

diff --git a/packages/docusaurus-plugin-redoc/src/index.ts b/packages/docusaurus-plugin-redoc/src/index.ts
@@ -6,16 +6,25 @@ import type {
   OptionValidationContext,
 } from '@docusaurus/types';
 import { normalizeUrl } from '@docusaurus/utils';
-import YAML from 'yaml';
 import { loadAndBundleSpec } from 'redoc';
+import {
+  formatProblems,
+  getTotals,
+  Config,
+  bundle,
+  loadConfig,
+  stringifyYaml,
+} from '@redocly/openapi-core';
 
 import {
   PluginOptionSchema,
   PluginOptions,
   PluginOptionsWithDefault,
   DEFAULT_OPTIONS,
 } from './options';
-import { ParsedSpec, SpecProps, ApiDocProps } from './types/common';
+import { SpecProps, ApiDocProps } from './types/common';
+// eslint-disable-next-line @typescript-eslint/no-var-requires
+const version = require('../package.json').version;
 
 export { PluginOptions };
 
@@ -25,50 +34,69 @@ export default function redocPlugin(
 ): Plugin<Record<string, unknown>> {
   const { baseUrl } = context.siteConfig;
   const options: PluginOptionsWithDefault = { ...DEFAULT_OPTIONS, ...opts };
-  const { debug, spec, url: downloadUrl } = options;
+  const { debug, spec, url: downloadUrl, config } = options;
+
   let url = downloadUrl;
+  const isSpecFile = fs.existsSync(spec);
+  const fileName = path.join(
+    'redocusaurus',
+    `${options.id || 'api-spec'}.yaml`,
+  );
+
   if (debug) {
     console.error('[REDOCUSAURUS_PLUGIN] Opts Input:', opts);
     console.error('[REDOCUSAURUS_PLUGIN] Options:', options);
   }
   return {
     name: 'docusaurus-plugin-redoc',
     async loadContent() {
-      let parsedSpec: ParsedSpec | null = null;
-      // If local file
-      if (fs.existsSync(spec)) {
+      if (!isSpecFile) {
+        // If spec is a remote url then add it as download url also as a default
+        url = url || spec;
         if (debug) {
-          console.log('[REDOCUSAURUS_PLUGIN] reading file: ', spec);
+          console.log('[REDOCUSAURUS_PLUGIN] bundling spec from url', spec);
         }
+        return loadAndBundleSpec(spec!);
+      }
+
+      // If local file
+      if (debug) {
+        console.log('[REDOCUSAURUS_PLUGIN] reading file: ', spec);
+      }
 
-        const file = fs.readFileSync(spec).toString();
+      let redoclyConfig: Config;
 
-        if (spec.endsWith('.yaml') || spec.endsWith('.yml')) {
-          parsedSpec = YAML.parse(file);
-        } else parsedSpec = JSON.parse(file);
+      if (config) {
+        if (typeof config === 'string') {
+          redoclyConfig = await loadConfig(config);
+        } else {
+          redoclyConfig = new Config(config);
+        }
       } else {
-        // If spec is a remote url then add it as download url
-        url = spec;
+        redoclyConfig = await loadConfig();
       }
 
-      if (debug) {
-        console.log('[REDOCUSAURUS_PLUGIN] bundling spec');
+      const { bundle: bundledSpec, problems } = await bundle({
+        ref: spec,
+        config: redoclyConfig,
+      });
+
+      if (problems?.length) {
+        console.error('[REDOCUSAURUS_PLUGIN] errors while bundling spec', spec);
+
+        formatProblems(problems, {
+          totals: getTotals(problems),
+          version,
+        });
       }
-      const content = await loadAndBundleSpec(parsedSpec || spec!);
 
       if (debug) {
-        console.log('[REDOCUSAURUS_PLUGIN] Content loaded');
+        console.log('[REDOCUSAURUS_PLUGIN] File Bundled');
       }
 
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      return content as any;
-    },
-    getPathsToWatch() {
-      if (!spec) {
-        return [];
-      }
-      const contentPath = path.resolve(context.siteDir, spec);
-      return [contentPath];
+      // If download url is not provided then use bundled yaml as a static file (see `postBuild`)
+      url = url || fileName;
+      return bundledSpec.parsed;
     },
     async contentLoaded({ content, actions }) {
       const { createData, addRoute, setGlobalData } = actions;
@@ -114,6 +142,27 @@ export default function redocPlugin(
         addRoute(routeOptions);
       }
     },
+    async postBuild(props) {
+      if (!isSpecFile || downloadUrl) {
+        return;
+      }
+      // Create a static file from bundled spec
+      const staticFile = path.join(context.outDir, fileName);
+      fs.mkdirSync(path.dirname(staticFile));
+      console.error(
+        '[REDOCUSAURUS_PLUGIN] creating static bundle copy for download',
+        staticFile,
+      );
+      // create bundled url
+      const bundledYaml = stringifyYaml(props.content);
+      fs.writeFileSync(staticFile, bundledYaml);
+    },
+    getPathsToWatch() {
+      if (!isSpecFile) {
+        return [];
+      }
+      return [path.resolve(spec)];
+    },
   };
 }
 

diff --git a/packages/docusaurus-plugin-redoc/src/options.ts b/packages/docusaurus-plugin-redoc/src/options.ts
@@ -1,4 +1,5 @@
 import * as Joi from 'joi';
+import { RawConfig } from '@redocly/openapi-core';
 
 type LayoutProps = {
   title?: string;
@@ -21,6 +22,11 @@ export interface PluginOptions {
   route?: string;
   layout?: LayoutProps;
   debug?: boolean;
+  /**
+   * Redocly config to bundle file
+   * @see https://redocly.com/docs/cli/configuration/configuration-file/
+   */
+  config?: string | Partial<RawConfig>;
 }
 
 export interface PluginOptionsWithDefault extends PluginOptions {
@@ -39,4 +45,5 @@ export const PluginOptionSchema = Joi.object<PluginOptions>({
   layout: Joi.any().default(DEFAULT_OPTIONS.layout),
   debug: Joi.boolean().default(DEFAULT_OPTIONS.debug),
   route: Joi.string().uri({ relativeOnly: true }).optional(),
+  config: Joi.any().optional(),
 });
diff --git a/website/docusaurus.config.js b/website/docusaurus.config.js
@@ -16,6 +16,12 @@ const redocusaurus = [
         url: `/openapi-page.yaml`,
         route: '/examples/using-spec-yaml/',
       },
+      {
+        id: 'using-multi-file-yaml',
+        // Local File
+        spec: 'examples/multi-yaml/index.yaml',
+        route: '/examples/using-multi-file-yaml/',
+      },
       {
         id: 'using-spec-url',
         // Remote File
@@ -119,6 +125,10 @@ const config = {
               label: 'Custom Layout',
               to: '/examples/custom-layout/',
             },
+            {
+              label: 'Using Multiple YAMLs',
+              to: '/examples/using-multi-file-yaml/',
+            },
             {
               label: 'Using Spec URL',
               to: '/examples/using-spec-url/',