Added option to export database

Author: Vinayaga Sundar

debug-db/src/main/assets/app.js

@@ -7,6 +7,8 @@ $( document ).ready(function() {
     });
 });
 
+var isDatabaseSelected = true;
+
 function getData(tableName) {
 
    $.ajax({url: "getAllDataFromTheTable?tableName="+tableName, success: function(result){
@@ -31,6 +33,13 @@ function queryFunction() {
 
 }
 
+function downloadDb() {
+    if (isDatabaseSelected) {
+        var url = window.location.href;
+        window.open(url+"downloadDb", "_blank")
+    }
+}
+
 
 function getDBList() {
 
@@ -59,9 +68,15 @@ function openDatabaseAndGetTableList(db) {
     if("APP_SHARED_PREFERENCES" == db) {
         $('#run-query').removeClass('active');
         $('#run-query').addClass('disabled');
+        $('#export-db').removeClass('active');
+        $('#export-db').addClass('disabled');
+        isDatabaseSelected = false;
     } else {
         $('#run-query').removeClass('disabled');
         $('#run-query').addClass('active');
+        $('#export-db').removeClass('disabled');
+        $('#export-db').addClass('active');
+        isDatabaseSelected = true;
     }
 
    $("#selected-db-info").text("Selected Database : "+db);

debug-db/src/main/assets/index.html

@@ -109,6 +109,9 @@
             <button id="run-query" type="submit" onclick="queryFunction()" class="btn btn-primary pull-right disabled">Run
                 Query
             </button>
+            <button id="export-db" type="button" onclick="downloadDb()" class="btn btn-primary">
+                Export DB
+            </button>
         </div>
     </div>
 

debug-db/src/main/java/com/amitshekhar/server/ClientServer.java

@@ -40,10 +40,12 @@
 import java.io.BufferedReader;
 import java.io.ByteArrayOutputStream;
 import java.io.File;
+import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
+import java.io.OutputStream;
 import java.io.PrintStream;
 import java.net.ServerSocket;
 import java.net.Socket;
@@ -84,6 +86,11 @@ public class ClientServer implements Runnable {
     private Gson mGson;
     private boolean isDbOpenned;
 
+    /**
+     * Hold the selected database name
+     */
+    private String mSelectedDatabase = null;
+
     /**
      * WebServer constructor.
      */
@@ -226,13 +233,17 @@ private void handle(Socket socket) throws IOException {
                 if (Constants.APP_SHARED_PREFERENCES.equals(database)) {
                     response = getAllPrefTableName();
                     closeDatabase();
+                    mSelectedDatabase = null;
                 } else {
                     openDatabase(database);
                     response = getAllTableName();
+                    mSelectedDatabase = database;
                 }
 
                 String data = mGson.toJson(response);
                 bytes = data.getBytes();
+            } else if (route.startsWith("downloadDb")) {
+                bytes = getDatabaseFile();
             } else {
                 bytes = loadContent(route);
             }
@@ -246,7 +257,12 @@ private void handle(Socket socket) throws IOException {
             // Send out the content.
             output.println("HTTP/1.0 200 OK");
             output.println("Content-Type: " + detectMimeType(route));
-            output.println("Content-Length: " + bytes.length);
+
+            if (route.startsWith("downloadDb")) {
+                output.println("Content-Disposition: attachment; filename=" + mSelectedDatabase);
+            } else {
+                output.println("Content-Length: " + bytes.length);
+            }
             output.println();
             output.write(bytes);
             output.flush();
@@ -318,6 +334,32 @@ private String detectMimeType(String fileName) {
         }
     }
 
+    private byte[] getDatabaseFile() {
+        if (TextUtils.isEmpty(mSelectedDatabase)) {
+            return null;
+        }
+
+        File file = new File(mDatabaseDir, mSelectedDatabase);
+
+        byte[] byteArray = null;
+        try {
+            InputStream inputStream = new FileInputStream(file);
+            ByteArrayOutputStream bos = new ByteArrayOutputStream();
+            byte[] b = new byte[(int) file.length()];
+            int bytesRead = 0;
+
+            while ((bytesRead = inputStream.read(b)) != -1) {
+                bos.write(b, 0, bytesRead);
+            }
+
+            byteArray = bos.toByteArray();
+        } catch (IOException e) {
+            Log.e(TAG, "getDatabaseFile: ", e);
+        }
+
+        return byteArray;
+    }
+
     private void getDatabaseDir() {
         File root = mContext.getFilesDir().getParentFile();
         File dbRoot = new File(root, "/databases");