feat: export source in csv files (#2) from StephaneBour/export_csv

StephaneBour · Nov 8, 2017 · 6b4b7c1 · 6b4b7c1
2 parents 368e412 + ad6b812
commit 6b4b7c1
Show file tree

Hide file tree

Showing 4 changed files with 87 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -20,19 +20,20 @@ Have fun !
 - Autocompletion on fields
 - Add "filter" in autocomplete
 - Add "Saved query"
+- Export source data in CSV file 
 
 ### Screenshots
 
-Main screen
+#### Main screen
 
 ![Main screen](https://github.com/StephaneBour/sense-chrome/raw/master/screenshots/main.jpg)
 
-Autocomplete
+#### Autocomplete
 
 ![Main screen](https://github.com/StephaneBour/sense-chrome/raw/master/screenshots/autocomplete.jpg)
 
 
-Save your query
+#### Save your query
 
 ![Save 1](https://github.com/StephaneBour/sense-chrome/raw/master/screenshots/saved-1.jpg)
 

diff --git a/index.html b/index.html
@@ -22,6 +22,8 @@
         </button>
         <button id="saved_btn" class="btn pull-right-btn" data-toggle="modal" data-target="#saved_popup">Saved query
         </button>
+        <button id="export_csv" class="btn btn-info hide pull-right-btn">Export source in CSV
+        </button>
 
     </div>
 </div>

diff --git a/lib/src-noconflict/ace.js b/lib/src-noconflict/ace.js
@@ -4386,6 +4386,19 @@ var EditSession = function(text, mode) {
         this.$deltasFold = [];
         this.getUndoManager().reset();
     };
+    this.setCsv = function(text) {
+        if(text != '' && text != null) {
+            this.$csv = text;
+            $('#export_csv').removeClass('hide');
+        } else {
+            $('#export_csv').addClass('hide');
+        }
+    };
+
+    this.getCsv = function () {
+        return this.$csv;
+    }
+
     this.getValue =
     this.toString = function() {
         return this.doc.getValue();
@@ -7310,6 +7323,7 @@ var Anchor = require("./anchor").Anchor;
 
 var Document = function(text) {
     this.$lines = [];
+    this.$csv = '';
     if (text.length == 0) {
         this.$lines = [""];
     } else if (Array.isArray(text)) {
@@ -10500,6 +10514,7 @@ var VirtualRenderer = function(container, theme) {
 
     this.content = dom.createElement("div");
     this.content.className = "ace_content";
+
     this.scroller.appendChild(this.content);
 
     this.setHighlightGutterLine(true);

diff --git a/src/base.js b/src/base.js
@@ -1,5 +1,5 @@
 if (!sense)
-    sense = { };
+    sense = {};
 
 sense.VERSION = "0.9.0";
 
@@ -106,19 +106,22 @@ function submitCurrentRequestToES() {
             if (typeof xhr.status == "number" &&
                 ((xhr.status >= 400 && xhr.status < 600) ||
                     (xhr.status >= 200 && xhr.status < 300)
-                    )) {
+                )) {
                 // we have someone on the other side. Add to history
                 sense.history.addToHistory(es_server, es_url, es_method, es_data);
 
 
                 var value = xhr.responseText;
                 try {
+                    var inJson = JSON.parse(value);
                     value = JSON.stringify(JSON.parse(value), null, 3);
+
                 }
                 catch (e) {
 
                 }
                 sense.output.getSession().setValue(value);
+                sense.output.getSession().setCsv(ConvertSourceToCSV(inJson.hits.hits));
             }
             else {
                 sense.output.getSession().setValue("Request failed to get to the server (status code: " + xhr.status + "):" + xhr.responseText);
@@ -148,7 +151,7 @@ function reformatData(data, indent) {
         }
     }
 
-    return { changed: changed, data: formatted_data}
+    return {changed: changed, data: formatted_data}
 }
 
 
@@ -237,6 +240,40 @@ function handleCURLPaste(text) {
 var CURRENT_REQ_RANGE = null;
 
 
+function ConvertSourceToCSV(objArray) {
+    var array = typeof objArray != 'object' ? JSON.parse(objArray) : objArray;
+    var str = '';
+    var headers = [];
+
+    for (var i = 0; i < array.length; i++) {
+        var line = '';
+        var source = array[i]._source;
+
+        if (str == '') {
+            for (var index in source) {
+                str += index + ',';
+                headers.push(index);
+            }
+
+            str = str.slice(0, -1);
+            str += "\r\n";
+        }
+
+        for (var h = 0; h < headers.length; h++) {
+            if (typeof source[headers[h]] == 'undefined') {
+                source[headers[h]] = '';
+            }
+            line += source[headers[h]] + ',';
+        }
+
+        line = line.slice(0, -1);
+
+        str += line + '\r\n';
+    }
+
+    return str;
+}
+
 function saveEditorState() {
     try {
         var content = sense.editor.getValue();
@@ -248,6 +285,25 @@ function saveEditorState() {
     }
 }
 
+function exportCsv() {
+    var csv = sense.output.getSession().getCsv();
+    saveAs(csv, "export-sense.csv");
+}
+
+var saveAs = (function () {
+    var a = document.createElement("a");
+    document.body.appendChild(a);
+    a.style = "display: none";
+    return function (data, fileName) {
+        var blob = new Blob([data], {type: "text/plain;charset=utf-8"}),
+            url = window.URL.createObjectURL(blob);
+        a.href = url;
+        a.download = fileName;
+        a.click();
+        window.URL.revokeObjectURL(url);
+    };
+}());
+
 function updateEditorActionsBar() {
     var editor_actions = $("#editor_actions");
 
@@ -290,7 +346,7 @@ function highlighCurrentRequestAndUpdateActionBar() {
     if (new_current_req_range != null && CURRENT_REQ_RANGE != null &&
         new_current_req_range.start.row == CURRENT_REQ_RANGE.start.row &&
         new_current_req_range.end.row == CURRENT_REQ_RANGE.end.row
-        ) {
+    ) {
         // same request, now see if we are on the first line and update the action bar
         var cursorRow = sense.editor.getCursorPosition().row;
         if (cursorRow == CURRENT_REQ_RANGE.start.row) {
@@ -467,6 +523,12 @@ function init() {
         e.preventDefault();
     });
 
+
+    $("#export_csv").click(function (e) {
+        exportCsv();
+        e.preventDefault();
+    });
+
     var help_popup = $("#help_popup");
 
     help_popup.on('shown', function () {