markdown 模板导出
This commit is contained in:
200
exts/yapi-plugin-export-markdown-template/controller.js
Normal file
200
exts/yapi-plugin-export-markdown-template/controller.js
Normal file
@@ -0,0 +1,200 @@
|
||||
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 uuid = require('uuid');
|
||||
const configModel = require("./configModel");
|
||||
const Safeify = require('safeify').default;
|
||||
const JSZip = require("jszip");
|
||||
|
||||
class exportMarkdownController extends baseController {
|
||||
constructor(ctx) {
|
||||
super(ctx);
|
||||
this.catModel = yapi.getInst(interfaceCatModel);
|
||||
this.interModel = yapi.getInst(interfaceModel);
|
||||
this.projectModel = yapi.getInst(projectModel);
|
||||
this.configModel = yapi.getInst(configModel);
|
||||
}
|
||||
|
||||
/*
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
|
||||
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 result = await this.configModel.getByProjectId(pid);
|
||||
// console.log(result);
|
||||
if (!result.is_export_by_interface){
|
||||
console.log("重定向")
|
||||
ctx.status = 302;
|
||||
ctx.redirect(`/api/plugin/export?type=markdown&pid=${pid}`);
|
||||
return ;
|
||||
}
|
||||
let curProject;
|
||||
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)
|
||||
// console.log("list", list)
|
||||
|
||||
|
||||
let model;
|
||||
|
||||
// const tmpPath = fs.mkdtempSync(path.join(os.tmpdir(),uuid.v4()));
|
||||
// console.log("tmpPath=", tmpPath)
|
||||
const zip = JSZip();
|
||||
|
||||
for (let item of data) {
|
||||
for(let interfaceItem of item.list){
|
||||
const safeVm = new Safeify({
|
||||
timeout: 3000, //超时时间
|
||||
asyncTimeout: 10000, //包含异步操作的超时时间
|
||||
unrestricted: true,
|
||||
quantity: 4, //沙箱进程数量，默认同 CPU 核数
|
||||
memoryQuota: 500, //沙箱最大能使用的内存（单位 m），默认 500m
|
||||
cpuQuota: 0.5, //沙箱的 cpu 资源配额（百分比），默认 50%
|
||||
});
|
||||
// console.log("curProject", curProject)
|
||||
const vmContext = {
|
||||
projectTmp: JSON.stringify(curProject),
|
||||
interfaceTmp: JSON.stringify(interfaceItem),
|
||||
categoryTmp: JSON.stringify(item)
|
||||
};
|
||||
try {
|
||||
safeVm.preset("const project = JSON.parse(projectTmp);const interface = JSON.parse(interfaceTmp);const category=JSON.parse(categoryTmp)")
|
||||
model = await safeVm.run(result.template_data, vmContext);
|
||||
// fs.writeFileSync(`${tmpPath}/${interfaceItem.title}.md`, model.toString())
|
||||
zip.file(`${interfaceItem.title}.md`, model.toString())
|
||||
safeVm.destroy();
|
||||
} catch (e) {
|
||||
console.log(e)
|
||||
return ;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ctx.set('Content-Disposition', `attachment; filename=${encodeURIComponent(curProject.name)}.zip`);
|
||||
let dataaa = await zip.generateAsync({
|
||||
type: "nodebuffer",
|
||||
// 压缩算法
|
||||
compression: "DEFLATE",
|
||||
streamFiles: true,
|
||||
compressionOptions: {
|
||||
level: 9
|
||||
}
|
||||
});
|
||||
ctx.body = dataaa
|
||||
// ctx.set('Content-Length', dataaa.length);
|
||||
|
||||
// dataaa.copy(ctx.body)
|
||||
|
||||
// fs.unlinkSync(tmpPath)
|
||||
} catch (error) {
|
||||
yapi.commons.log(error, 'error');
|
||||
ctx.body = yapi.commons.resReturn(null, 502, '下载出错');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存配置信息
|
||||
* @param {*} ctx
|
||||
*/
|
||||
async upConfig(ctx) {
|
||||
let requestBody = ctx.request.body;
|
||||
let projectId = requestBody.project_id;
|
||||
if (!projectId) {
|
||||
return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id'));
|
||||
}
|
||||
|
||||
if ((await this.checkAuth(projectId, 'project', 'edit')) !== true) {
|
||||
return (ctx.body = yapi.commons.resReturn(null, 405, '没有权限'));
|
||||
}
|
||||
|
||||
let result;
|
||||
if (requestBody.id) {
|
||||
result = await this.configModel.up(requestBody);
|
||||
} else {
|
||||
result = await this.configModel.save(requestBody);
|
||||
}
|
||||
|
||||
return (ctx.body = yapi.commons.resReturn(result));
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询配置信息
|
||||
* @param {*} ctx
|
||||
*/
|
||||
async getConfig(ctx) {
|
||||
let projectId = ctx.query.project_id;
|
||||
if (!projectId) {
|
||||
return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id'));
|
||||
}
|
||||
let result = await this.configModel.getByProjectId(projectId);
|
||||
return (ctx.body = yapi.commons.resReturn(result));
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = exportMarkdownController;
|
||||
Reference in New Issue
Block a user