swagger-egg-plus/app.js

swagger document generator for egg.

'use strict';

const pathToSwaggerUi = require('swagger-ui-dist').absolutePath();
const Swagger = require('./lib/swagger');

class AppBootHook {
  constructor(app) {
    this.app = app;
  }

  configWillLoad() {
    let staticDirs = this.app.config.static.dir;
    // CAUTION: index.html under 'app/public' directory will be overwrite by 'swagger-ui-dist'
    if (Array.isArray(staticDirs)) staticDirs.unshift(pathToSwaggerUi);
    else staticDirs = [ pathToSwaggerUi, staticDirs ];
    this.app.config.static.dir = staticDirs;
  }

  didLoad() {
    try {
      new Swagger(this.app, pathToSwaggerUi);
    } catch (error) {
      error.message = `[swagger-egg]: ${error.message}`;
      throw error;
    }
  }
}

module.exports = AppBootHook;