新增分组的hook模板配置

exts/yapi-plugin-export-markdown-template/client.js

@@ -31,5 +31,14 @@ module.exports = function() {
         name: 'Markdown',
         component: Services
       }
-})
+    })
+
+    this.bindHook('add_group_tab', function (panes) {
+      panes.push({
+        key: 'Markdown',
+        title: 'Markdown设置',
+        component: mdTemplate,
+        roles: ['admin', 'owner']
+      })
+    })
 };

exts/yapi-plugin-export-markdown-template/configModel.js

@@ -10,13 +10,14 @@ class configModel extends baseModel {
   getSchema() {
     return {
       uid: { type: Number},
-      project_id: { type: Number, required: true },
+      project_id: { type: Number, required: false },
+      group_id: { type: Number, required: false },
       //是否开启自动同步
       is_export_by_interface: { type: Boolean, default: false },
       // 模板
       template_data: String,
       add_time: Number,
-      up_time: Number,
+      up_time: Number
     };
   }
 
@@ -26,6 +27,11 @@ class configModel extends baseModel {
     })
   }
 
+  getByGroupId(id) {
+    return this.model.findOne({
+      group_id: id
+    })
+  }
   delByProjectId(project_id){
     return this.model.remove({
       project_id: project_id

exts/yapi-plugin-export-markdown-template/controller.js

@@ -3,6 +3,7 @@ const interfaceModel = require('models/interface.js');
 const projectModel = require('models/project.js');
 const interfaceCatModel = require('models/interfaceCat.js');
 const userModel = require("models/user.js");
+const groupModel = require("models/group.js");
 const yapi = require('yapi.js');
 const uuid = require('uuid');
 const configModel = require("./configModel");
@@ -17,6 +18,7 @@ class exportMarkdownController extends baseController {
     this.projectModel = yapi.getInst(projectModel);
     this.configModel = yapi.getInst(configModel);
     this.userModel = yapi.getInst(userModel);
+    this.groupModel = yapi.getInst(groupModel);
   }
 
   /*
@@ -82,16 +84,19 @@ class exportMarkdownController extends baseController {
     }
 
     let templateData = await this.configModel.getByProjectId(pid);
+    let curProject = await this.projectModel.get(pid);
+    if (!templateData){
+      templateData = await this.configModel.getByGroupId(curProject.group_id);
+    }
     // console.log(result);
-    if (!templateData.is_export_by_interface) {
+    if (!templateData || !templateData.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);
@@ -153,8 +158,9 @@ class exportMarkdownController extends baseController {
   async upConfig(ctx) {
     let requestBody = ctx.request.body;
     let projectId = requestBody.project_id;
-    if (!projectId) {
-      return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id'));
+    let groupId = requestBody.group_id;
+    if (!projectId && !groupId) {
+      return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id/组id'));
     }
 
     if ((await this.checkAuth(projectId, 'project', 'edit')) !== true) {
@@ -176,11 +182,17 @@ class exportMarkdownController extends baseController {
    * @param {*} ctx
    */
   async getConfig(ctx) {
-    let projectId = ctx.query.project_id;
-    if (!projectId) {
-      return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id'));
+    let projectId = ctx.query.project_id || ctx.query.projectId;
+    let groupId = ctx.query.group_id || ctx.query.groupId;
+    if (!projectId && !groupId) {
+      return (ctx.body = yapi.commons.resReturn(null, 408, '缺少项目Id/组id'));
+    }
+    let  result = null;
+    if (projectId){
+      result = await this.configModel.getByProjectId(projectId);
+    }else {
+      result= await this.configModel.getByGroupId(groupId);
     }
-    let result = await this.configModel.getByProjectId(projectId);
     return (ctx.body = yapi.commons.resReturn(result));
   }
 
@@ -196,12 +208,15 @@ class exportMarkdownController extends baseController {
     if (!interfaceData) {
       return (ctx.body = yapi.commons.resReturn(null, 200, ''));
     }
+    let project = await this.projectModel.get(interfaceData.project_id);
     let templateData = await this.configModel.getByProjectId(interfaceData.project_id);
+    if (!templateData || !templateData.template_data) {
+      templateData = await this.configModel.getByGroupId(project.group_id);
+    }
     if (!templateData || !templateData.template_data) {
       return (ctx.body = yapi.commons.resReturn(null, 200, ''));
     }
     let interfaceCat = await this.catModel.get(interfaceData.catid);
-    let project = await this.projectModel.get(interfaceData.project_id);
     let {result, errMsg} = await this.executeJsRender(templateData.template_data, project, interfaceData, interfaceCat, userData);
     if (errMsg){
       ctx.body = yapi.commons.resReturn(errMsg, 502, '渲染出错');

exts/yapi-plugin-export-markdown-template/mdTemplate/Services.js

@@ -69,10 +69,10 @@ class MDTemplateServices extends Component {
     const extraContent = (
       <Fragment>
         <Button size={"small"} type={"link"} icon="copy"
-                onClick={(e) => this.preCopy(this.state.render_data.data)}>拷贝原文</Button>
+                onClick={() => this.preCopy(this.state.render_data.data)}>拷贝原文</Button>
 
         <Button size={"small"} type={"link"} icon="table"
-                onClick={(e) => {
+                onClick={() => {
                   this.setState({
                     isHtml: !this.state.isHtml
                   })

exts/yapi-plugin-export-markdown-template/mdTemplate/mdTemplate.js

@@ -44,6 +44,7 @@ export default class ProjectInterfaceSync extends Component {
   static propTypes = {
     form: PropTypes.object,
     match: PropTypes.object,
+    groupId: PropTypes.number,
     projectId: PropTypes.number,
     projectMsg: PropTypes.object,
     handleSwaggerUrlData: PropTypes.func
@@ -55,11 +56,14 @@ export default class ProjectInterfaceSync extends Component {
       config_data: { is_export_by_interface: false, template_data:"",_id:null }
     };
   }
-
+  componentDidMount() {
+    this.getSyncData();
+  }
   handleSubmit = async () => {
-    const { form, projectId } = this.props;
+    const { form, projectId, groupId } = this.props;
     let params = {
       project_id: projectId,
+      group_id: groupId,
       is_export_by_interface: this.state.config_data.is_export_by_interface,
       template_data: this.state.config_data.template_data,
       uid: this.props.projectMsg.uid
@@ -97,12 +101,10 @@ export default class ProjectInterfaceSync extends Component {
     this.setState({
       config_data: {}
     });
-    this.getSyncData();
   }
 
   async getSyncData() {
-    let projectId = this.props.projectMsg._id;
-    let result = await axios.get('/api/plugin/mdConfig/get?project_id=' + projectId);
+    let result = await axios.get('/api/plugin/mdConfig/get' + (this.props.projectId ? `?project_id=${this.props.projectId}` : `?groupId=${this.props.groupId}`));
     if (result.data.errcode === 0) {
       if (result.data.data) {
         this.setState({