From ed7924efa8a6ee07dbd0c8270b49839230ce10b4 Mon Sep 17 00:00:00 2001
From: suxiaoxin <hellosean1025@163.com>
Date: Fri, 29 Sep 2017 14:58:07 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E5=90=8E=E7=AB=AF?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client/components/Postman/Postman.js |  2 +-
 client/plugin-module.js              |  2 +-
 server/controllers/interface.js      |  8 ++++
 server/controllers/interfaceCol.js   | 60 ++++++++++++++++++++++++++++
 server/router.js                     | 10 +++++
 5 files changed, 80 insertions(+), 2 deletions(-)

diff --git a/client/components/Postman/Postman.js b/client/components/Postman/Postman.js
index 20af53a1..d7046ec7 100755
--- a/client/components/Postman/Postman.js
+++ b/client/components/Postman/Postman.js
@@ -552,7 +552,7 @@ export default class Run extends Component {
                 <div>
                   <a
                     target="blank"
-                    href="/attachment/cross-request-v2.0.1.zip"
+                    href="/api/interface/download_crx"
                   > [手动下载] </a>
                   <span> zip 文件解压后将 crx 文件拖入到 chrome://extensions/ </span>
                   <a
diff --git a/client/plugin-module.js b/client/plugin-module.js
index 7c6d6c73..c3cbdea7 100644
--- a/client/plugin-module.js
+++ b/client/plugin-module.js
@@ -1 +1 @@
-module.exports = {}
\ No newline at end of file
+module.exports = {qsso : require('plugins/yapi-plugin-qsso/client.js')}
\ No newline at end of file
diff --git a/server/controllers/interface.js b/server/controllers/interface.js
index b8221dd0..69090f18 100755
--- a/server/controllers/interface.js
+++ b/server/controllers/interface.js
@@ -255,6 +255,14 @@ class interfaceController extends baseController {
         }
     }
 
+    async downloadCrx(ctx){
+        let filename = 'crossRequest.zip';
+        let dataBuffer = yapi.fs.readFileSync(yapi.path.join(yapi.WEBROOT, 'static/attachment/cross-request-v2.0.1.zip'));
+        ctx.set('Content-disposition', 'attachment; filename=' + filename);
+        ctx.set('Content-Type', 'application/zip');
+        ctx.body = dataBuffer;
+    }
+
     async listByCat(ctx) {
         let catid = ctx.request.query.catid;
         if (!catid) {
diff --git a/server/controllers/interfaceCol.js b/server/controllers/interfaceCol.js
index bc461ba8..5eeb11d6 100755
--- a/server/controllers/interfaceCol.js
+++ b/server/controllers/interfaceCol.js
@@ -236,6 +236,66 @@ class interfaceColController extends baseController{
         }
     }
 
+    async addCaseList(ctx){
+        try{
+            let params = ctx.request.body;
+            params = yapi.commons.handleParams(params, {
+                project_id: 'number',
+                col_id: 'number'
+            });
+            if(!params.interface_list || !Array.isArray(params.interface_list)){
+                return ctx.body = yapi.commons.resReturn(null, 400, 'interface_list 参数有误');
+            }
+
+            if (!params.project_id) {
+                return ctx.body = yapi.commons.resReturn(null, 400, '项目id不能为空');
+            }
+
+            let auth = await this.checkAuth(params.project_id, 'project', 'edit');
+            if (!auth) {
+                return ctx.body = yapi.commons.resReturn(null, 400, '没有权限');
+            }
+
+
+            if (!params.col_id) {
+                return ctx.body = yapi.commons.resReturn(null, 400, '接口集id不能为空');
+            }
+
+            let data = {
+                uid: this.getUid(),
+                index: 0,
+                add_time: yapi.commons.time(),
+                up_time: yapi.commons.time(),
+                project_id: params.project_id,
+                col_id: params.col_id
+            }
+
+            for(let i=0; i<params.interface_list.length; i++){
+                let interfaceData = await this.interfaceModel.getBaseinfo(params.interface_list[i]);
+                data.interface_id = params.interface_list[i];
+                data.casename = interfaceData.title;
+                await this.caseModel.save(data);
+                let username = this.getUsername();
+                this.colModel.get(params.col_id).then((col)=>{
+                    yapi.commons.saveLog({
+                        content: `用户 "${username}" 在接口集 "${col.name}" 下导入了接口 ${data.casename}`,
+                        type: 'project',
+                        uid: this.getUid(),
+                        username: username,
+                        typeid: params.project_id
+                    });
+                });
+            }
+
+            this.projectModel.up(params.project_id,{up_time: new Date().getTime()}).then();
+
+            ctx.body = yapi.commons.resReturn('ok');
+
+        }catch(e){
+            ctx.body = yapi.commons.resReturn(null, 402, e.message);
+        }
+    }
+
     /**
      * 更新一个接口用例
      * @interface /col/up_case
diff --git a/server/router.js b/server/router.js
index 2a39ff8b..212cae5a 100755
--- a/server/router.js
+++ b/server/router.js
@@ -244,6 +244,11 @@ let routerConfig = {
 			"path": "add",
 			"method": "post"
 		},
+		{
+			"action": "downloadCrx",
+			"path"  : "download_crx",
+			"method": "get"
+		},
 		{
 			"action": "getCatMenu",
 			"path": "getCatMenu",
@@ -320,6 +325,11 @@ let routerConfig = {
 		action: "addCol",
 		path: "add_col",
 		method: "post"
+	},{
+		action: 'addCaseList',
+		path: 'add_case_list',
+		method: 'post'	
+		
 	}, {
 		action: "list",
 		path: "list",