From 957a7469187090d1e12837ee0f8bb726beb92f27 Mon Sep 17 00:00:00 2001
From: agirbal <antoine@10gen.com>
Date: Wed, 2 Oct 2013 12:37:17 -0700
Subject: [PATCH] - issue168: allow to save / open binary from / to a file

---
 resource/xml/umongo.xml                       | 25 +++++++---
 .../edgytech/umongo/DocumentFieldMenu.java    | 50 ++++++++++++++++---
 src/com/edgytech/umongo/EditBinaryDialog.java | 30 +++++++++--
 3 files changed, 87 insertions(+), 18 deletions(-)

diff --git a/resource/xml/umongo.xml b/resource/xml/umongo.xml
index a00c9e6..c0b33a8 100644
--- a/resource/xml/umongo.xml
+++ b/resource/xml/umongo.xml
@@ -27,13 +27,24 @@
             <MenuItem id="set" call="set"/>
             <MenuItem id="unset" call="unset"/>
             <MenuSeparator/>
-            <MenuItem id="compressionStats" call="compressionStats" showDialog="false">
-                <InfoDialog id="compressionStatsDialog" text="/html/about.html"/>
+            <MenuItem id="compressionStats" call="compressionStats" showDialog="false" buttons="OK">
+                <FormDialog>
+                    <Text id="compressionStatsOriginal" label="Original" field="true"/>
+                    <Text id="compressionStatsDeflate" label="Deflate" field="true"/>
+                    <Text id="compressionStatsGZip" label="GZip" field="true"/>
+                </FormDialog>
             </MenuItem>
             <MenuItem id="saveBinaryToFile" call="saveBinaryToFile">
-                <FileSelectorField id="saveBinaryOutputFile">
-                    <FileSelectorDialog type="SAVE" mode="FILES_ONLY"/>
-                </FileSelectorField>
+                <FormDialog>
+                    <FileSelectorField id="saveBinaryOutputFile">
+                        <FileSelectorDialog type="SAVE" mode="FILES_ONLY"/>
+                    </FileSelectorField>
+                </FormDialog>
+            </MenuItem>
+            <MenuItem id="decodeBinary" call="decodeBinary">
+                <FormDialog buttons="OK" preferredSize="640;480">
+                    <TextArea id="expandTextArea" field="false" zone="CENTER" editable="false"/>
+                </FormDialog>
             </MenuItem>
             <MenuSeparator/>
             <MenuItem id="copyField" call="copyField"/>
@@ -827,7 +838,9 @@
             </Zone>
         </com.edgytech.umongo.RouterPanel>
         <com.edgytech.umongo.EditBinaryDialog>
-            <IntSpinner id="size"/>
+            <FileSelectorField id="inputFile">
+                <FileSelectorDialog type="OPEN" mode="FILES_ONLY"/>
+            </FileSelectorField>
         </com.edgytech.umongo.EditBinaryDialog>
         <com.edgytech.umongo.EditBooleanDialog>
             <CheckBox id="value" label="True"/>
diff --git a/src/com/edgytech/umongo/DocumentFieldMenu.java b/src/com/edgytech/umongo/DocumentFieldMenu.java
index 6ee18d2..9ceed93 100644
--- a/src/com/edgytech/umongo/DocumentFieldMenu.java
+++ b/src/com/edgytech/umongo/DocumentFieldMenu.java
@@ -24,6 +24,8 @@
 import java.awt.datatransfer.Clipboard;
 import java.awt.datatransfer.StringSelection;
 import java.io.ByteArrayOutputStream;
+import java.io.FileOutputStream;
+import java.io.FileWriter;
 import java.io.IOException;
 import java.nio.charset.Charset;
 import java.util.zip.DeflaterOutputStream;
@@ -40,9 +42,13 @@ enum Item {
         set,
         unset,
         compressionStats,
-        compressionStatsDialog,
+        compressionStatsOriginal,
+        compressionStatsDeflate,
+        compressionStatsGZip,
         saveBinaryToFile,
         saveBinaryOutputFile,
+        decodeBinary,
+        decodeBinaryText,
         copyField,
         copyValue
     }
@@ -186,7 +192,7 @@ public void compressionStats(ButtonBase button) throws IOException {
             bytes = MongoUtils.getJSON(value).toString().getBytes(Charset.forName("UTF-8"));
         }
         
-        String msg = "Initial Size: " + bytes.length + "\n";
+        setStringFieldValue(Item.compressionStatsOriginal, String.valueOf(bytes.length));
         ByteArrayOutputStream baos = new ByteArrayOutputStream(bytes.length);
 //        System.out.println("Bytes before: " + bytes.length);
         
@@ -195,7 +201,7 @@ public void compressionStats(ButtonBase button) throws IOException {
         dos.write(bytes);
         dos.close();
         byte[] deflate = baos.toByteArray();
-        msg += "Deflate: " + deflate.length + "\n";
+        setStringFieldValue(Item.compressionStatsDeflate, String.valueOf(deflate.length));
 //        System.out.println("Bytes deflate: " + deflate.length);
         
         baos.reset();
@@ -203,16 +209,46 @@ public void compressionStats(ButtonBase button) throws IOException {
         gos.write(bytes);
         gos.close();
         byte[] gzip = baos.toByteArray();
-        msg += "GZip: " + gzip.length + "\n";
+        setStringFieldValue(Item.compressionStatsGZip, String.valueOf(gzip.length));
 //        System.out.println("Bytes gzip: " + gzip.length);
         
-        InfoDialog dia = (InfoDialog)getBoundComponentUnit(Item.compressionStatsDialog);
-        dia.text = msg;
+        Showable dia = ((MenuItem)getBoundComponentUnit(Item.compressionStats)).getDialog();
         dia.show();
     }
 
     public void saveBinaryToFile(ButtonBase button) throws IOException {
+        String path = ((FileSelectorField)getBoundComponentUnit(Item.saveBinaryOutputFile)).getPath();
+
+        final DocView dv = (DocView) (UMongo.instance.getTabbedResult().getSelectedUnit());
+        TreeNodeDocumentField node = (TreeNodeDocumentField) dv.getSelectedNode().getUserObject();
+        Object value = node.getValue();
+        byte[] bytes = null;
+        if (value instanceof byte[]) {
+            bytes = (byte[]) value;
+        } else if (value instanceof Binary) {
+            bytes = ((Binary)value).getData();
+        } else {
+            bytes = MongoUtils.getJSON(value).toString().getBytes(Charset.forName("UTF-8"));
+        }
         
+        FileOutputStream fos = new FileOutputStream(path);
+        fos.write(bytes);
+        fos.close();
     }
-    
+
+//    public void decodeBinary(ButtonBase button) throws IOException {
+//        final DocView dv = (DocView) (UMongo.instance.getTabbedResult().getSelectedUnit());
+//        TreeNodeDocumentField node = (TreeNodeDocumentField) dv.getSelectedNode().getUserObject();
+//        Object value = node.getValue();
+//        byte[] bytes = null;
+//        if (value instanceof byte[]) {
+//            bytes = (byte[]) value;
+//        } else if (value instanceof Binary) {
+//            bytes = ((Binary)value).getData();
+//        } else {
+//            bytes = MongoUtils.getJSON(value).toString().getBytes(Charset.forName("UTF-8"));
+//        }
+//        
+//        BinaryDecoder
+//    }
 }
diff --git a/src/com/edgytech/umongo/EditBinaryDialog.java b/src/com/edgytech/umongo/EditBinaryDialog.java
index 2cc9e2a..829718c 100644
--- a/src/com/edgytech/umongo/EditBinaryDialog.java
+++ b/src/com/edgytech/umongo/EditBinaryDialog.java
@@ -16,7 +16,14 @@
 
 package com.edgytech.umongo;
 
+import com.edgytech.swingfast.FileSelectorField;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 import org.bson.types.Binary;
+import sun.misc.IOUtils;
 
 /**
  *
@@ -24,7 +31,7 @@
  */
 public class EditBinaryDialog extends EditFieldDialog {
     enum Item {
-        size
+        inputFile
     }
 
     public EditBinaryDialog() {
@@ -33,13 +40,26 @@ public EditBinaryDialog() {
 
     @Override
     public Object getValue() {
-        int size = getIntFieldValue(Item.size);
-        return new Binary((byte)0, new byte[size]);
+        FileInputStream fis = null;
+        try {
+            String path = ((FileSelectorField) getBoundComponentUnit(Item.inputFile)).getPath();
+            fis = new FileInputStream(path);
+            byte[] bytes = IOUtils.readFully(fis, -1, true);
+            return new Binary((byte)0, bytes);
+        } catch (Exception ex) {
+            getLogger().log(Level.WARNING, null, ex);
+        } finally {
+            try {
+                fis.close();
+            } catch (IOException ex) {
+                Logger.getLogger(EditBinaryDialog.class.getName()).log(Level.SEVERE, null, ex);
+            }
+        }
+        return null;
     }
 
     @Override
     public void setValue(Object value) {
-        Binary bin = (Binary) value;
-        setIntFieldValue(Item.size, bin.length());
+        // nothing to do here...
     }
 }