新增Postman导出

This commit is contained in:
2024-03-01 21:53:37 +08:00
parent 5cc1c349cd
commit 2aca2c3dbe
6 changed files with 354 additions and 2 deletions

View File

@@ -1 +1 @@
module.exports = {"import-postman" : {module: require('exts/yapi-plugin-import-postman/client.js'),options: null},"import-har" : {module: require('exts/yapi-plugin-import-har/client.js'),options: null},"advanced-mock" : {module: require('exts/yapi-plugin-advanced-mock/client.js'),options: null},"import-swagger" : {module: require('exts/yapi-plugin-import-swagger/client.js'),options: null},"statistics" : {module: require('exts/yapi-plugin-statistics/client.js'),options: null},"export-data" : {module: require('exts/yapi-plugin-export-data/client.js'),options: null},"gen-services" : {module: require('exts/yapi-plugin-gen-services/client.js'),options: null},"export-swagger2-data" : {module: require('exts/yapi-plugin-export-swagger2-data/client.js'),options: null},"import-yapi-json" : {module: require('exts/yapi-plugin-import-yapi-json/client.js'),options: null},"wiki" : {module: require('exts/yapi-plugin-wiki/client.js'),options: null},"swagger-auto-sync" : {module: require('exts/yapi-plugin-swagger-auto-sync/client.js'),options: null}}
module.exports = {"import-postman" : {module: require('exts/yapi-plugin-import-postman/client.js'),options: null},"import-har" : {module: require('exts/yapi-plugin-import-har/client.js'),options: null},"advanced-mock" : {module: require('exts/yapi-plugin-advanced-mock/client.js'),options: null},"import-swagger" : {module: require('exts/yapi-plugin-import-swagger/client.js'),options: null},"statistics" : {module: require('exts/yapi-plugin-statistics/client.js'),options: null},"export-data" : {module: require('exts/yapi-plugin-export-data/client.js'),options: null},"gen-services" : {module: require('exts/yapi-plugin-gen-services/client.js'),options: null},"export-swagger2-data" : {module: require('exts/yapi-plugin-export-swagger2-data/client.js'),options: null},"import-yapi-json" : {module: require('exts/yapi-plugin-import-yapi-json/client.js'),options: null},"wiki" : {module: require('exts/yapi-plugin-wiki/client.js'),options: null},"swagger-auto-sync" : {module: require('exts/yapi-plugin-swagger-auto-sync/client.js'),options: null},"export-postman" : {module: require('exts/yapi-plugin-export-postman/client.js'),options: null}}

View File

@@ -21,9 +21,11 @@ module.exports = {
name: 'wiki'
}, {
name: 'swagger-auto-sync'
}, {
name: 'export-postman'
}
// {
// name: 'test'
// }
]
}
}

View File

@@ -0,0 +1,12 @@
function exportData(exportDataModule, pid) {
console.log("inject ")
exportDataModule.postman = {
name: 'postman collection',
route: `/api/plugin/exportPostman?pid=${pid}`,
desc: '导出项目接口文档为Postman Json文件'
};
}
module.exports = function() {
this.bindHook('export_data', exportData);
};

View File

@@ -0,0 +1,322 @@
const baseController = require('controllers/base.js');
const interfaceModel = require('models/interface.js');
const projectModel = require('models/project.js');
const interfaceCatModel = require('models/interfaceCat.js');
const yapi = require('yapi.js');
const uid = require("uuid")
class exportPostmanController extends baseController {
constructor(ctx) {
super(ctx);
this.catModel = yapi.getInst(interfaceCatModel);
this.interModel = yapi.getInst(interfaceModel);
this.projectModel = yapi.getInst(projectModel);
}
/*
handleListClass,handleExistId is same as the exportController(yapi-plugin-export-data).
No DRY,but i have no idea to optimize it.
*/
async handleListClass(pid, status) {
let result = await this.catModel.list(pid),
newResult = [];
for (let i = 0, item, list; i < result.length; i++) {
item = result[i].toObject();
list = await this.interModel.listByInterStatus(item._id, status);
list = list.sort((a, b) => {
return a.index - b.index;
});
if (list.length > 0) {
item.list = list;
newResult.push(item);
}
}
return newResult;
}
handleExistId(data) {
function delArrId(arr, fn) {
if (!Array.isArray(arr)) return;
arr.forEach(item => {
delete item._id;
delete item.__v;
delete item.uid;
delete item.edit_uid;
delete item.catid;
delete item.project_id;
if (typeof fn === 'function') fn(item);
});
}
delArrId(data, function (item) {
delArrId(item.list, function (api) {
delArrId(api.req_body_form);
delArrId(api.req_params);
delArrId(api.req_query);
delArrId(api.req_headers);
if (api.query_path && typeof api.query_path === 'object') {
delArrId(api.query_path.params);
}
});
});
return data;
}
dealMenuOrInterface(item) {
// console.log(item)
return {
name: item.name,
item: item.list.map(it => {
return {
name: it.title,
request: {
method: it.method,
header: it.req_headers.map(hit => {
return {
key: hit.name,
value: hit.value,
type: "text"
}
}),
url:{
raw: `{{BaseHost}}/${it.path}`.replace(/\/+/g,"/"),
host: ["{{BaseHost}}"],
// port:"",
path: it.path.split("/").filter(it => it.length > 0)
},
response:[]
}
}
}),
event: [
{
listen: "prerequest",
script: {
type: "text/javascript",
exec: [
""
]
}
},
{
listen: "test",
script: {
type: "text/javascript",
exec: [
""
]
}
}
],
variable: [
{
key: "BaseHost",
value: ""
}
]
};
}
async exportData(ctx) {
let pid = ctx.request.query.pid;
let status = ctx.request.query.status;
if (!pid) {
ctx.body = yapi.commons.resReturn(null, 200, 'pid 不为空');
}
let curProject;
let tp = '';
const that = this
try {
curProject = await this.projectModel.get(pid);
// ctx.set('Content-Type', 'application/json');
ctx.set('Content-Type', 'application/octet-stream');
const list = await this.handleListClass(pid, status);
let data = this.handleExistId(list);
// console.log("curProject", curProject)
let model = await convertToPostmanModel(data);
tp = JSON.stringify(model, null, 2);
ctx.set('Content-Disposition', `attachment; filename=${encodeURIComponent(curProject.name)}.postman_collection.json`);
return (ctx.body = tp);
} catch (error) {
yapi.commons.log(error, 'error');
ctx.body = yapi.commons.resReturn(null, 502, '下载出错');
}
//Convert to SwaggerV2.0 (OpenAPI 2.0)
async function convertToPostmanModel(list) {
const postmanObj = {
info:{
_postman_id: uid.v4(),
name:curProject.name,
schema:"https://schema.getpostman.com/json/collection/v2.1.0/collection.json"
},
item:(() => {
return list.map(it => that.dealMenuOrInterface(it))
})(),
paths: (() => {
let apisObj = {};
for (let aptTag of list) { //list of category
for (let api of aptTag.list) //list of api
{
if (apisObj[api.path] == null) {
apisObj[api.path] = {};
}
apisObj[api.path][api.method.toLowerCase()] = (() => {
let apiItem = {};
apiItem['tags'] = [aptTag.name];
apiItem['summary'] = api.title;
apiItem['description'] = api.markdown;
switch (api.req_body_type) {
case 'form':
case 'file':
apiItem['consumes'] = ['multipart/form-data']; //form data required
break;
case 'json':
apiItem['consumes'] = ['application/json'];
break;
case 'raw':
apiItem['consumes'] = ['text/plain'];
break;
default:
break;
}
apiItem['parameters'] = (() => {
let paramArray = [];
for (let p of api.req_headers) //Headers parameters
{
//swagger has consumes proprety, so skip proprety "Content-Type"
if (p.name === 'Content-Type') {
continue;
}
paramArray.push({
name: p.name,
in: 'header',
description: `${p.name} (Only:${p.value})`,
required: Number(p.required) === 1,
type: 'string', //always be type string
default: p.value
});
}
for (let p of api.req_params) //Path parameters
{
paramArray.push({
name: p.name,
in: 'path',
description: p.desc,
required: true, //swagger path parameters required proprety must be always true,
type: 'string' //always be type string
});
}
for (let p of api.req_query) //Query parameters
{
paramArray.push({
name: p.name,
in: 'query',
required: Number(p.required) === 1,
description: p.desc,
type: 'string' //always be type string
});
}
switch (api.req_body_type) //Body parameters
{
case 'form':
{
for (let p of api.req_body_form) {
paramArray.push({
name: p.name,
in: 'formData',
required: Number(p.required) === 1,
description: p.desc,
type: p.type === 'text' ? 'string' : 'file' //in this time .formData type have only text or file
});
}
break;
}
case 'json':
{
if (api.req_body_other) {
let jsonParam = JSON.parse(api.req_body_other);
if (jsonParam) {
paramArray.push({
name: 'root',
in: 'body',
description: jsonParam.description,
schema: jsonParam //as same as swagger's format
});
}
}
break;
}
case 'file':
{
paramArray.push({
name: 'upfile',
in: 'formData', //use formData
description: api.req_body_other,
type: 'file'
});
break;
}
case 'raw':
{
paramArray.push({
name: 'raw',
in: 'body',
description: 'raw paramter',
schema: {
type: 'string',
format: 'binary',
default: api.req_body_other
}
});
break;
}
default:
break;
}
return paramArray;
})();
apiItem['responses'] = {
'200': {
description: 'successful operation',
schema: (() => {
let schemaObj = {};
if (api.res_body_type === 'raw') {
schemaObj['type'] = 'string';
schemaObj['format'] = 'binary';
schemaObj['default'] = api.res_body;
} else if (api.res_body_type === 'json') {
if (api.res_body) {
let resBody = JSON.parse(api.res_body);
if (resBody !== null) {
//schemaObj['type']=resBody.type;
schemaObj = resBody; //as the parameters,
}
}
}
return schemaObj;
})()
}
};
return apiItem;
})();
}
}
return apisObj;
})()
};
return postmanObj;
}
}
}
module.exports = exportPostmanController;

View File

@@ -0,0 +1,4 @@
module.exports = {
server: true,
client: true
}

View File

@@ -0,0 +1,12 @@
const exportSwaggerController = require('./controller');
module.exports = function(){
this.bindHook('add_router', function(addRouter){
addRouter({
controller: exportSwaggerController,
method: 'get',
path: 'exportPostman',
action: 'exportData'
})
})
}