db: 导出查询数据

source/language/en.js

@@ -398,6 +398,10 @@ module.exports = {
         query: (err) => antSword.noxss(`Failure to execute SQL!\n${err}`),
         parse: 'Return data format is incorrect!',
         noresult: 'No query results!'
+      },
+      dump: {
+        title: "Export Data",
+        success: "Export success",
       }
     },
     notsupport: 'Not support the current database type',

source/language/zh.js

@@ -399,6 +399,10 @@ module.exports = {
         query: (err) => antSword.noxss(`执行SQL失败！\n${err}`),
         parse: '返回数据格式不正确！',
         noresult: '没有查询结果！'
+      },
+      dump: {
+        title: "导出查询结果",
+        success: "导出成功",
       }
     },
     notsupport: '该功能暂不支持当前类型数据库',

source/modules/database/asp/index.js

@@ -543,13 +543,32 @@ class ASP {
       'rows': grid_data
     }, 'json');
     // 启用导出按钮
-    // this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
+    this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
+  }
+
+  // 导出查询数据
+  dumpResult() {
+    const grid = this.manager.result.layout.getAttachedObject();
+    let filename = `${this.core.__opts__.ip}_${new Date().format("yyyyMMddhhmmss")}.csv`;
+    antSword['test'] = this;
+    dialog.showSaveDialog({
+      title: LANG['result']['dump']['title'],
+      defaultPath: filename
+    },(filePath) => {
+      if (!filePath) { return; };
+      let headerStr = grid.hdrLabels.join(',');
+      let dataStr = grid.serializeToCSV();
+      let tempDataBuffer = new Buffer(headerStr+'\n'+dataStr);
+      fs.writeFileSync(filePath, tempDataBuffer);
+      toastr.success(LANG['result']['dump']['success'], LANG_T['success']);
+    });
   }
 
   // 禁用toolbar按钮
   disableToolbar() {
     this.manager.list.toolbar.disableItem('del');
     this.manager.list.toolbar.disableItem('edit');
+    this.manager.result.toolbar.disableItem('dump');
   }
 
   // 启用toolbar按钮

source/modules/database/custom/index.js

@@ -543,13 +543,32 @@ class CUSTOM {
       'rows': grid_data
     }, 'json');
     // 启用导出按钮
-    // this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
+    this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
+  }
+
+  // 导出查询数据
+  dumpResult() {
+    const grid = this.manager.result.layout.getAttachedObject();
+    let filename = `${this.core.__opts__.ip}_${new Date().format("yyyyMMddhhmmss")}.csv`;
+    antSword['test'] = this;
+    dialog.showSaveDialog({
+      title: LANG['result']['dump']['title'],
+      defaultPath: filename
+    },(filePath) => {
+      if (!filePath) { return; };
+      let headerStr = grid.hdrLabels.join(',');
+      let dataStr = grid.serializeToCSV();
+      let tempDataBuffer = new Buffer(headerStr+'\n'+dataStr);
+      fs.writeFileSync(filePath, tempDataBuffer);
+      toastr.success(LANG['result']['dump']['success'], LANG_T['success']);
+    });
   }
 
   // 禁用toolbar按钮
   disableToolbar() {
     this.manager.list.toolbar.disableItem('del');
     this.manager.list.toolbar.disableItem('edit');
+    this.manager.result.toolbar.disableItem('dump');
   }
 
   // 启用toolbar按钮

source/modules/database/index.js

@@ -158,14 +158,21 @@ class Database {
     layout.setText(`<i class="fa fa-inbox"></i> ${LANG['result']['title']}`);
     // layout.hideHeader();
 
-    // const toolbar = layout.attachToolbar();
-    // toolbar.loadStruct([
-    //   { id: 'dump', text: '导出', icon: 'upload', type: 'button', disabled: true },
-    //   { type: 'separator' }
-    // ]);
+    const toolbar = layout.attachToolbar();
+    toolbar.loadStruct([
+      { id: 'dump', text: '导出', icon: 'upload', type: 'button', disabled: true },
+      { type: 'separator' }
+    ]);
+    toolbar.attachEvent('onClick', (id) => {
+      switch(id) {
+        case 'dump':
+          this.drive.dumpResult();
+          break;
+      }
+    });
     return {
       layout: layout,
-      // toolbar: toolbar
+      toolbar: toolbar
     };
   }
 

source/modules/database/php/index.js

@@ -5,6 +5,8 @@
 
 const LANG = antSword['language']['database'];
 const LANG_T = antSword['language']['toastr'];
+const dialog = antSword.remote.dialog;
+const fs = require('fs');
 
 class PHP {
 
@@ -1485,13 +1487,31 @@ class PHP {
       'rows': grid_data
     }, 'json');
     // 启用导出按钮
-    // this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
+    this.manager.result.toolbar[grid_data.length > 0 ? 'enableItem' : 'disableItem']('dump');
   }
   
+  // 导出查询数据
+  dumpResult() {
+    const grid = this.manager.result.layout.getAttachedObject();
+    let filename = `${this.core.__opts__.ip}_${new Date().format("yyyyMMddhhmmss")}.csv`;
+    antSword['test'] = this;
+    dialog.showSaveDialog({
+      title: LANG['result']['dump']['title'],
+      defaultPath: filename
+    },(filePath) => {
+      if (!filePath) { return; };
+      let headerStr = grid.hdrLabels.join(',');
+      let dataStr = grid.serializeToCSV();
+      let tempDataBuffer = new Buffer(headerStr+'\n'+dataStr);
+      fs.writeFileSync(filePath, tempDataBuffer);
+      toastr.success(LANG['result']['dump']['success'], LANG_T['success']);
+    });
+  }
   // 禁用toolbar按钮
   disableToolbar() {
     this.manager.list.toolbar.disableItem('del');
     this.manager.list.toolbar.disableItem('edit');
+    this.manager.result.toolbar.disableItem('dump');
   }
 
   // 启用toolbar按钮