337 lines
14 KiB
JavaScript
337 lines
14 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
var fs = require("fs");
|
|
var path = require("path");
|
|
var Config_1 = require("./Config");
|
|
var OpenAPI_1 = require("./OpenAPI");
|
|
var CodePrinter_1 = require("./CodePrinter");
|
|
function codegenIHandler(funcs, config, cp) {
|
|
var schemasName = config.schemasName, utilsTSPath = config.utilsTSPath, clientOnly = config.clientOnly;
|
|
// import
|
|
cp.writeln("import * as Schemas from './" + schemasName + "'");
|
|
cp.writeln('import {FullDate, StrictTypeParser as STP, APIPromise} ' +
|
|
("from '" + utilsTSPath + "'"));
|
|
if (!clientOnly) {
|
|
cp.writeln('import {RouterContext as CTX} from \'@koa/router\'');
|
|
}
|
|
cp.writeln('import {AxiosResponse} from \'axios\'');
|
|
// api req, res types
|
|
cp.writeln("export type TAPI = {", 1);
|
|
for (var _i = 0, _a = Object.entries(funcs); _i < _a.length; _i++) {
|
|
var _b = _a[_i], funcName = _b[0], func = _b[1];
|
|
var reqTypes = func.reqTypes, resTypes = func.resTypes, method = func.method;
|
|
cp.writeln(funcName + ": {", 1);
|
|
// req
|
|
// req.path, ...
|
|
cp.writeln("req: {", 1);
|
|
for (var _c = 0, ELParameterIn_1 = OpenAPI_1.ELParameterIn; _c < ELParameterIn_1.length; _c++) {
|
|
var _in = ELParameterIn_1[_c];
|
|
var paras = reqTypes[_in];
|
|
if (paras == null)
|
|
continue;
|
|
cp.writeln(_in + ": {", 1);
|
|
for (var _d = 0, _e = Object.entries(paras); _d < _e.length; _d++) {
|
|
var _f = _e[_d], propName = _f[0], schemaType = _f[1];
|
|
cp.writeln(schemaType.forProp(propName) + ';');
|
|
}
|
|
cp.writeln('};', -1);
|
|
}
|
|
// body
|
|
var body = reqTypes.body;
|
|
if (body != null) {
|
|
// PATCH's req body: Partial
|
|
var typeName = body.typeName;
|
|
if (method == 'patch')
|
|
typeName = "Partial<" + typeName + ">";
|
|
cp.writeln("body" + (body.required ? '' : '?') + ": " + typeName + ";");
|
|
}
|
|
cp.writeln('}', -1); // req END
|
|
// res
|
|
cp.writeln("res: {", 1);
|
|
for (var _g = 0, _h = Object.entries(resTypes); _g < _h.length; _g++) {
|
|
var _j = _h[_g], status_1 = _j[0], schema = _j[1];
|
|
cp.writeln(schema.required ?
|
|
schema.forProp(status_1) + ";" : status_1 + ": void;");
|
|
}
|
|
cp.writeln('}', -1); // res END
|
|
// operation END
|
|
cp.writeln('}', -1);
|
|
}
|
|
// TAPI END
|
|
cp.writeln('}', -1);
|
|
// export IServerAPI
|
|
if (!clientOnly) {
|
|
cp.writeln('');
|
|
cp.writeln('type ValueOf<T> = T[keyof T];');
|
|
cp.writeln('type RServerAPI<T> = ValueOf<', 1);
|
|
cp.writeln('{[K in keyof T]: T[K] extends void ? [K, any?] : [K, T[K]]}>;', -1, false);
|
|
cp.writeln('export type IServerAPI<IState=any> = {[K in keyof TAPI]:', 1);
|
|
cp.writeln("(req: TAPI[K]['req'], state: IState, ctx: CTX) =>", 1);
|
|
cp.writeln("Promise<RServerAPI<TAPI[K]['res']>>}", -2, false);
|
|
}
|
|
// return
|
|
return cp.end();
|
|
}
|
|
function codegenRouter(funcs, config, cp) {
|
|
var schemasName = config.schemasName, IHandlerName = config.IHandlerName, ServerAPITSPath = config.ServerAPITSPath, utilsTSPath = config.utilsTSPath;
|
|
// import
|
|
cp.writeln("import * as Schemas from './" + schemasName + "'");
|
|
cp.writeln("import {IServerAPI} from './" + IHandlerName + "'");
|
|
cp.writeln("import * as Router from '@koa/router'");
|
|
cp.writeln("import {FullDate, StrictTypeParser as STP} from '" + utilsTSPath + "'");
|
|
cp.writeln("import * as bodyParser from 'koa-body'");
|
|
// api
|
|
cp.writeln("\nimport api from '" + ServerAPITSPath + "'");
|
|
cp.writeln("type IState = typeof api extends IServerAPI<infer T> ? T : any;");
|
|
// router
|
|
cp.writeln("type CTX = Router.RouterContext<IState>;");
|
|
cp.writeln("\nconst router = new Router<IState>();");
|
|
// function
|
|
var gcGetParams = {
|
|
path: function (attr) { return "ctx.params['" + attr + "']"; },
|
|
query: function (attr) { return "ctx.query['" + attr + "']"; },
|
|
header: function (attr) { return "ctx.headers['" + attr + "']"; },
|
|
cookie: function (attr) { return "ctx.cookies.get('" + attr + "')"; },
|
|
};
|
|
// route
|
|
for (var _i = 0, _a = Object.entries(funcs); _i < _a.length; _i++) {
|
|
var _b = _a[_i], funcName = _b[0], func = _b[1];
|
|
var method = func.method, url = func.url, reqTypes = func.reqTypes;
|
|
var isPartial = method === 'patch';
|
|
// TODO escape
|
|
var sURL = url.replace(/{(.*?)}/g, ':$1'); // {a} -> :a
|
|
var mid = '';
|
|
if (reqTypes.body) {
|
|
var maxSize = reqTypes.body.maxSize; // TODO doc
|
|
var config_1 = maxSize == null ? '' : "{jsonLimit: '" + maxSize + "'}";
|
|
mid = "bodyParser(" + config_1 + "), ";
|
|
}
|
|
cp.writeln("router." + method + "('" + sURL + "', " + mid + "async ctx => {", 1);
|
|
// req
|
|
if (Object.keys(reqTypes).length === 0) {
|
|
cp.writeln('const req = {};');
|
|
}
|
|
else {
|
|
cp.writeln('let req;');
|
|
cp.writeln('try {', 1);
|
|
cp.writeln('req = {', 1);
|
|
// paras
|
|
for (var _c = 0, ELParameterIn_2 = OpenAPI_1.ELParameterIn; _c < ELParameterIn_2.length; _c++) {
|
|
var _in = ELParameterIn_2[_c];
|
|
var paras = reqTypes[_in];
|
|
if (paras == null)
|
|
continue;
|
|
cp.writeln(_in + ": {", 1);
|
|
for (var _d = 0, _e = Object.entries(paras); _d < _e.length; _d++) {
|
|
var _f = _e[_d], name_1 = _f[0], schema = _f[1];
|
|
var pn = gcGetParams[_in](name_1);
|
|
var label = "req." + _in;
|
|
cp.writeln(name_1 + ": " + schema.stp(pn, label) + ",");
|
|
}
|
|
cp.writeln('},', -1);
|
|
}
|
|
// body
|
|
var body = reqTypes.body;
|
|
if (body != null) {
|
|
cp.writeln("body: " + body.stp('ctx.request.body', 'req.body', isPartial));
|
|
}
|
|
cp.writeln('}', -1);
|
|
cp.writeln('} catch(err) {', -1);
|
|
cp.tab(1);
|
|
cp.writeln('if (err instanceof STP.BadValueError)', 1);
|
|
cp.writeln('return ctx.throw(400, err.toString());');
|
|
cp.tab(-1);
|
|
cp.writeln('throw err;');
|
|
cp.writeln('}', -1);
|
|
}
|
|
// call
|
|
cp.writeln("const r = await api." + funcName + "(req, ctx.state, ctx);");
|
|
cp.writeln("ctx.status = r[0];");
|
|
cp.writeln("ctx.body = r[1] ?? '';");
|
|
// ctx END
|
|
cp.writeln('});', -1);
|
|
}
|
|
cp.writeln('\nexport default router;');
|
|
return cp.end();
|
|
}
|
|
function codegenClientAPI(funcs, config, cp) {
|
|
var IHandlerName = config.IHandlerName, schemasName = config.schemasName, utilsTSPath = config.utilsTSPath, validateStatus = config.validateStatus;
|
|
// import
|
|
cp.writeln("import {TAPI} from './" + IHandlerName + "'");
|
|
cp.writeln("import * as Schemas from './" + schemasName + "'");
|
|
cp.writeln("import {APIPromise, StrictTypeParser as STP, " +
|
|
("qStringify} from '" + utilsTSPath + "'"));
|
|
cp.writeln("import axios from 'axios'");
|
|
cp.writeln('');
|
|
// type
|
|
cp.writeln("type TSTP<T> = {[K in keyof T]: (data: any) =>" +
|
|
"T[K] extends void ? any : T[K]};");
|
|
cp.writeln("export type ExID<T> = {[K in keyof T]: " +
|
|
"'id' extends keyof T[K] ? {id: T[K]['id']} : T[K]};");
|
|
cp.writeln();
|
|
// axios
|
|
cp.writeln('const $http = axios.create({', 1);
|
|
cp.writeln('validateStatus: () => true,');
|
|
cp.writeln('paramsSerializer: params => qStringify(params),');
|
|
cp.writeln('});', -1);
|
|
// function
|
|
cp.writeln('\nfunction urlReplacer(url: string, ' +
|
|
'rules: {[_: string]: any}): string {', 1);
|
|
cp.writeln('for(const [attr, value] of Object.entries(rules))', 1);
|
|
cp.writeln("url = url.replace('{'+attr+'}', value)");
|
|
cp.writeln('return url;', -1);
|
|
cp.writeln('};', -1);
|
|
// implementation
|
|
// export default
|
|
cp.writeln('\nexport default {', 1);
|
|
// set $baseURL
|
|
cp.writeln('set $baseURL(url: string) {', 1);
|
|
cp.writeln('$http.interceptors.request.use(async config => {', 1);
|
|
cp.writeln('config.baseURL = url;');
|
|
cp.writeln('return config;');
|
|
cp.writeln('}, err => Promise.reject(err));', -1);
|
|
cp.writeln('},', -1);
|
|
var _loop_1 = function (funcName, func) {
|
|
var gcReq = function (_in) { return "TAPI['" + funcName + "']['req']['" + _in + "']"; };
|
|
var method = func.method, url = func.url, reqTypes = func.reqTypes, resTypes = func.resTypes;
|
|
var query = reqTypes.query, header = reqTypes.header, path_1 = reqTypes.path, body = reqTypes.body; // TODO cookie
|
|
// name
|
|
cp.writeln(funcName + ": (", 1);
|
|
// paras
|
|
for (var _i = 0, ELParameterIn_3 = OpenAPI_1.ELParameterIn; _i < ELParameterIn_3.length; _i++) {
|
|
var _in = ELParameterIn_3[_i];
|
|
var paras = reqTypes[_in];
|
|
if (paras == null)
|
|
continue;
|
|
var _required = false;
|
|
for (var _a = 0, _b = Object.values(paras); _a < _b.length; _a++) {
|
|
var required = _b[_a].required;
|
|
if (required) {
|
|
_required = true;
|
|
break;
|
|
}
|
|
}
|
|
cp.writeln(_in + ": " + gcReq(_in) + (_required ? '' : '={}') + ",");
|
|
}
|
|
// body
|
|
if (body != null) {
|
|
cp.writeln("body" + (body.required ? '' : '?') + ": ExID<" + gcReq('body') + ">,");
|
|
}
|
|
// return value
|
|
cp.tab(-1);
|
|
cp.writeln(") => APIPromise.init($http({", 1);
|
|
// req
|
|
cp.writeln("method: '" + method + "',");
|
|
var sURL = "'" + url + "'";
|
|
cp.writeln("url: " + (path_1 ? "urlReplacer(" + sURL + ", path)" : sURL) + ",");
|
|
if (query)
|
|
cp.writeln('params: query,');
|
|
if (header)
|
|
cp.writeln('header: header,');
|
|
if (body != null)
|
|
cp.writeln('data: body,');
|
|
cp.writeln('}), {', -1);
|
|
cp.tab(1);
|
|
// stp
|
|
for (var _c = 0, _d = Object.entries(resTypes); _c < _d.length; _c++) {
|
|
var _e = _d[_c], status_2 = _e[0], schema = _e[1];
|
|
var label = "ClientAPI[" + funcName + "][" + status_2 + "]";
|
|
cp.writeln(status_2 + ": x => " + schema.stp('x', label) + ",");
|
|
}
|
|
cp.writeln("} as TSTP<TAPI['" + funcName + "']['res']>,", -1);
|
|
cp.tab(1);
|
|
// kRsv
|
|
cp.writeln("[" + Object.keys(resTypes).filter(validateStatus).join(', ') + "]),", -1);
|
|
};
|
|
// functions
|
|
for (var _i = 0, _a = Object.entries(funcs); _i < _a.length; _i++) {
|
|
var _b = _a[_i], funcName = _b[0], func = _b[1];
|
|
_loop_1(funcName, func);
|
|
}
|
|
cp.writeln('}', -1);
|
|
return cp.end();
|
|
}
|
|
function codegenSchemas(schemas, config, cp) {
|
|
var _a;
|
|
var utilsTSPath = config.utilsTSPath;
|
|
// import
|
|
cp.writeln("import {FullDate, StrictTypeParser as STP} from '" + utilsTSPath + "'");
|
|
// schema
|
|
for (var _i = 0, _b = Object.entries(schemas); _i < _b.length; _i++) {
|
|
var _c = _b[_i], typeName = _c[0], rSchema = _c[1];
|
|
var schema = OpenAPI_1.resolveRef(rSchema, schemas, '#/components/schemas');
|
|
if (schema == null)
|
|
continue;
|
|
cp.writeln();
|
|
if (OpenAPI_1.isObjectSchema(schema)) {
|
|
// interface
|
|
cp.writeln("export interface " + typeName + " {", 1);
|
|
var propTypes = [];
|
|
var requireds = new Set((_a = schema.required) !== null && _a !== void 0 ? _a : []);
|
|
for (var _d = 0, _e = Object.entries(schema.properties); _d < _e.length; _d++) {
|
|
var _f = _e[_d], propName = _f[0], prop = _f[1];
|
|
var propType = new OpenAPI_1.SchemaType(prop, requireds.has(propName), true);
|
|
propTypes.push([propName, propType]);
|
|
cp.writeln(propType.forProp(propName) + ';');
|
|
}
|
|
cp.writeln('}', -1); // interface END
|
|
// const
|
|
cp.writeln("export const " + typeName + " = {", 1);
|
|
// .from
|
|
cp.writeln("from: (o: {[_: string]: any}): " + typeName + " => ({", 1);
|
|
for (var _g = 0, propTypes_1 = propTypes; _g < propTypes_1.length; _g++) {
|
|
var _h = propTypes_1[_g], n = _h[0], t = _h[1];
|
|
cp.writeln(n + ": " + t.stp("o." + n, typeName + '.' + n, false, true) + ",");
|
|
}
|
|
cp.writeln('}),', -1);
|
|
// Partial
|
|
cp.writeln("Partial: (o: {[_: string]: any}): Partial<" + typeName + "> => {", 1);
|
|
cp.writeln("const r: Partial<" + typeName + "> = {};");
|
|
var locPartial = "Partial<" + typeName + ">";
|
|
for (var _j = 0, propTypes_2 = propTypes; _j < propTypes_2.length; _j++) {
|
|
var _k = propTypes_2[_j], n = _k[0], t = _k[1];
|
|
cp.writeln("if (o." + n + " !== void 0) r." + n + " = " + t.stp("o." + n, locPartial + '.' + n, false, true) + ";");
|
|
}
|
|
cp.writeln('return r;');
|
|
cp.writeln('},', -1);
|
|
// fields
|
|
cp.writeln("fields: [", 1);
|
|
cp.writeln(propTypes.map(function (e) { return "'" + e[0] + "',"; }).join(' '));
|
|
cp.writeln("] as Array<keyof " + typeName + ">", -1);
|
|
// end of const
|
|
cp.writeln('}', -1);
|
|
}
|
|
else {
|
|
cp.writeln("export type " + typeName + " = " +
|
|
OpenAPI_1.SchemaType.typeNameOf(schema, true));
|
|
}
|
|
}
|
|
// return
|
|
return cp.end();
|
|
}
|
|
function codegen(openAPI, configUser) {
|
|
var _a;
|
|
var config = Object.assign({}, Config_1.configDefault, configUser);
|
|
// prepare
|
|
fs.mkdirSync(config.outputDir, { recursive: true });
|
|
var apiFuncs = OpenAPI_1.apiFunctionsOf(openAPI);
|
|
var gCP = function (fn) { return new CodePrinter_1.CodePrinter(fs.createWriteStream(path.join(config.outputDir, fn + '.ts')), config.indentString); };
|
|
var ps = [];
|
|
// write files
|
|
// handler
|
|
ps.push(codegenIHandler(apiFuncs, config, gCP(config.IHandlerName)));
|
|
// server
|
|
if (!config.clientOnly) {
|
|
ps.push(codegenRouter(apiFuncs, config, gCP(config.routerName)));
|
|
}
|
|
// client
|
|
ps.push(codegenClientAPI(apiFuncs, config, gCP(config.ClientAPIName)));
|
|
// schema
|
|
var schemas = (_a = openAPI.components) === null || _a === void 0 ? void 0 : _a.schemas;
|
|
if (schemas != null) {
|
|
ps.push(codegenSchemas(schemas, config, gCP(config.schemasName)));
|
|
}
|
|
// return
|
|
return Promise.all(ps);
|
|
}
|
|
exports.default = codegen;
|