100 lines
2.5 KiB
JavaScript
100 lines
2.5 KiB
JavaScript
const Koa = require('koa');
|
|
const path = require('path');
|
|
const url = require('url');
|
|
const swaggerGenerator = require('@mtfos/swagger-generator');
|
|
const { copyObject } = require('src/utils/index.js');
|
|
|
|
const app = new Koa();
|
|
const config = require('src/config/index.js');
|
|
|
|
// @ts-ignore
|
|
app.proxy = true;
|
|
|
|
// const server = app.listen(config.server.port, () => {
|
|
// console.log(`server start on port ${server.address().port}`); // eslint-disable-line
|
|
// });
|
|
|
|
// load middleware module
|
|
const koaLogger = require('koa-logger');
|
|
const koaCors = require('@koa/cors');
|
|
const koaBody = require('koa-body');
|
|
const koaMount = require('koa-mount');
|
|
const koaStatic = require('koa-static');
|
|
const rootRouter = require('src/routes/index.js');
|
|
|
|
const packageJSON = require('./package.json');
|
|
|
|
let swaggerDoc = null;
|
|
// generate swagger document
|
|
(async () => {
|
|
const swaggerSpec = {
|
|
swaggerDefinition: {
|
|
info: {
|
|
description: 'KeyCloak OAuth Demo Server',
|
|
title: 'KeyCloak',
|
|
version: packageJSON.version,
|
|
},
|
|
host: new url.URL(config.server.url).host,
|
|
basePath: '',
|
|
produces: ['application/json'],
|
|
schemes: ['http', 'https'],
|
|
securityDefinitions: {
|
|
JWT: {
|
|
type: 'apiKey',
|
|
in: 'header',
|
|
name: 'Authorization',
|
|
description: 'Bearer token',
|
|
},
|
|
},
|
|
},
|
|
basedir: __dirname,
|
|
files: ['./routes/**/*.js'],
|
|
};
|
|
swaggerDoc = await swaggerGenerator.generateSpec(swaggerSpec);
|
|
})().catch(err => {
|
|
console.error('[Error] Generate swagger doc failed, ', err);
|
|
process.exit(1);
|
|
});
|
|
|
|
// production 不掛上swagger ui
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
app.use(koaMount('/api-docs/', koaStatic(path.resolve(__dirname, 'public', 'swagger-ui'), { index: 'index.html' })));
|
|
rootRouter.get('/api-docs', async c => {
|
|
if (!/\/$/.test(c.url)) c.redirect(`${c.url}/`);
|
|
});
|
|
}
|
|
|
|
// set swagger doc route
|
|
rootRouter.get('/api-docs.json', async c => {
|
|
c.type = 'application/json';
|
|
const spec = copyObject(swaggerDoc || {});
|
|
|
|
if (c.protocol === 'https') {
|
|
spec.schemes = ['https'];
|
|
}
|
|
|
|
c.body = spec;
|
|
});
|
|
|
|
app.use(koaLogger());
|
|
app.use(
|
|
koaCors({
|
|
credentials: true,
|
|
// allow all origin
|
|
origin: ctx => ctx.get('origin'),
|
|
})
|
|
);
|
|
app.use(
|
|
koaBody({
|
|
multipart: true,
|
|
formidable: {
|
|
maxFileSize: 100 * 1024 * 1024, // 100 mb
|
|
},
|
|
})
|
|
);
|
|
app.use(koaMount('/', koaStatic(path.resolve(__dirname, 'public', 'html'))));
|
|
app.use(rootRouter.allowedMethods());
|
|
app.use(rootRouter.routes());
|
|
|
|
module.exports = app;
|