git-svn-id: https://joinup.ec.europa.eu/svn/mocca/trunk@546 8a26b1a7-26f0-462f-b9ef-d0e30c41f5a4

2 files changed, 235 insertions, 0 deletions

diff --git a/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/MimeFilter.java b/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/MimeFilter.java
new file mode 100644
index 00000000..c0385dce
--- /dev/null
+++ b/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/MimeFilter.java
@@ -0,0 +1,115 @@
+/*
+* Copyright 2008 Federal Chancellery Austria and
+* Graz University of Technology
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+package at.gv.egiz.bku.gui.viewer;
+
+import java.io.File;
+import java.util.ResourceBundle;
+import javax.swing.filechooser.FileFilter;
+
+/**
+ *
+ * @author clemens
+ */
+class MimeFilter extends FileFilter {
+
+  private static final String MIMETYPE_DESC_XML = "mimetype.desc.xml";
+  private static final String MIMETYPE_DESC_HTML = "mimetype.desc.html";
+  private static final String MIMETYPE_DESC_XHTML = "mimetype.desc.xhtml";
+  private static final String MIMETYPE_DESC_TXT = "mimetype.desc.txt";
+  private static final String MIMETYPE_DESC_PDF = "mimetype.desc.pdf";
+  private static final String MIMETYPE_DESC_BIN = "mimetype.desc.bin";
+  private static final String MIMETYPE_DESC_UNKNOWN = "mimetype.desc.unknown";
+
+  protected String mimeType;
+  protected ResourceBundle messages;
+
+  public MimeFilter(String mimeType, ResourceBundle messages) {
+    this.mimeType = mimeType;
+    this.messages = messages;
+  }
+
+  @Override
+  public boolean accept(File f) {
+
+    if (f.isDirectory()) {
+      return true;
+    }
+
+    String ext = getExtension(f);
+    if ("text/xml".equals(mimeType)) {
+      return "xml".equalsIgnoreCase(ext);
+    } else if ("text/html".equals(mimeType)) {
+      return "html".equalsIgnoreCase(ext) || "htm".equalsIgnoreCase(ext);
+    } else if ("application/xhtml+xml".equals(mimeType)) {
+      return "xhtml".equalsIgnoreCase(ext);
+    } else if ("text/plain".equals(mimeType)) {
+      return "txt".equalsIgnoreCase(ext);
+    } else if ("application/pdf".equals(mimeType)) {
+      return "pdf".equalsIgnoreCase(ext);
+    } else {
+      return true;
+    }
+  }
+
+  private String getExtension(File f) {
+    String ext = null;
+    String s = f.getName();
+    int i = s.lastIndexOf('.');
+
+    if (i > 0 && i < s.length() - 1) {
+      ext = s.substring(i + 1).toLowerCase();
+    }
+    return ext;
+  }
+
+  @Override
+  public String getDescription() {
+    if ("text/xml".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_XML);
+    } else if ("text/html".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_HTML);
+    } else if ("application/xhtml+xml".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_XHTML);
+    } else if ("text/plain".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_TXT);
+    } else if ("application/pdf".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_PDF);
+    } else if ("application/octet-stream".equals(mimeType)) {
+      return messages.getString(MIMETYPE_DESC_BIN);
+    } else {
+      return messages.getString(MIMETYPE_DESC_UNKNOWN);
+    }
+  }
+
+  public static String getExtension(String mimeType) {
+    if ("text/xml".equals(mimeType)) {
+      return ".xml";
+    } else if ("text/html".equals(mimeType)) {
+      return ".html";
+    } else if ("application/xhtml+xml".equals(mimeType)) {
+      return ".xhtml";
+    } else if ("text/plain".equals(mimeType)) {
+      return ".txt";
+    } else if ("application/pdf".equals(mimeType)) {
+      return ".pdf";
+    } else if ("application/octet-stream".equals(mimeType)) {
+      return ".bin";
+    } else {
+      return "";
+    }
+  }
+}
\ No newline at end of file
diff --git a/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/SecureViewerSaveDialog.java b/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/SecureViewerSaveDialog.java
new file mode 100644
index 00000000..40133f95
--- /dev/null
+++ b/BKUCommonGUI/src/main/java/at/gv/egiz/bku/gui/viewer/SecureViewerSaveDialog.java
@@ -0,0 +1,120 @@
+package at.gv.egiz.bku.gui.viewer;
+
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+import java.io.BufferedOutputStream;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.text.MessageFormat;
+import java.util.Locale;
+import java.util.ResourceBundle;
+
+import javax.swing.JFileChooser;
+import javax.swing.JOptionPane;
+import javax.swing.SwingUtilities;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+
+import at.gv.egiz.bku.gui.BKUGUIFacade;
+import at.gv.egiz.stal.HashDataInput;
+
+public class SecureViewerSaveDialog {
+
+  protected static final Log log = LogFactory.getLog(SecureViewerSaveDialog.class);
+    
+  public static void showSaveDialog(final HashDataInput hashDataInput, final ResourceBundle messages,
+      final ActionListener okListener, final String okCommand) {
+
+    log.debug("[" + Thread.currentThread().getName()
+        + "] scheduling save dialog");
+
+    SwingUtilities.invokeLater(new Runnable() {
+
+      @Override
+      public void run() {
+
+        log
+            .debug("[" + Thread.currentThread().getName()
+                + "] show save dialog");
+
+        String userHome = System.getProperty("user.home");
+
+        JFileChooser fileDialog = new JFileChooser(userHome);
+        fileDialog.setMultiSelectionEnabled(false);
+        fileDialog.setDialogType(JFileChooser.SAVE_DIALOG);
+        fileDialog.setFileHidingEnabled(true);
+        fileDialog.setDialogTitle(messages
+            .getString(BKUGUIFacade.WINDOWTITLE_SAVE));
+        fileDialog.setFileSelectionMode(JFileChooser.FILES_ONLY);
+        String mimeType = hashDataInput.getMimeType();
+        MimeFilter mimeFilter = new MimeFilter(mimeType, messages);
+        fileDialog.setFileFilter(mimeFilter);
+        String filename = messages
+            .getString(BKUGUIFacade.SAVE_HASHDATAINPUT_PREFIX)
+            + MimeFilter.getExtension(mimeType);
+        fileDialog.setSelectedFile(new File(userHome, filename));
+
+        // parent contentPane -> placed over applet
+        switch (fileDialog.showSaveDialog(fileDialog)) {
+        case JFileChooser.APPROVE_OPTION:
+          File file = fileDialog.getSelectedFile();
+          String id = hashDataInput.getReferenceId();
+          if (file.exists()) {
+            String msgPattern = messages
+                .getString(BKUGUIFacade.MESSAGE_OVERWRITE);
+            int overwrite = JOptionPane.showConfirmDialog(fileDialog,
+                MessageFormat.format(msgPattern, file), messages
+                    .getString(BKUGUIFacade.WINDOWTITLE_OVERWRITE),
+                JOptionPane.OK_CANCEL_OPTION);
+            if (overwrite != JOptionPane.OK_OPTION) {
+              break;
+            }
+          }
+          if (log.isDebugEnabled()) {
+            log.debug("writing hashdata input " + id + " (" + mimeType
+                + ") to file " + file);
+          }
+          FileOutputStream fos = null;
+          try {
+            fos = new FileOutputStream(file);
+            BufferedOutputStream bos = new BufferedOutputStream(fos);
+            InputStream hdi = hashDataInput.getHashDataInput();
+            int b;
+            while ((b = hdi.read()) != -1) {
+              bos.write(b);
+            }
+            bos.flush();
+            bos.close();
+          } catch (IOException ex) {
+            log.error("Failed to write " + file + ": " + ex.getMessage());
+            log.debug(ex);
+            String errPattern = messages
+                .getString(BKUGUIFacade.ERR_WRITE_HASHDATA);
+            JOptionPane.showMessageDialog(fileDialog, MessageFormat.format(
+                errPattern, ex.getMessage()), messages
+                .getString(BKUGUIFacade.WINDOWTITLE_ERROR),
+                JOptionPane.ERROR_MESSAGE);
+          } finally {
+            try {
+              if (fos != null) {
+                fos.close();
+              }
+            } catch (IOException ex) {
+            }
+          }
+          break;
+        case JFileChooser.CANCEL_OPTION:
+          log.debug("cancelled save dialog");
+          break;
+        }
+        if (okListener != null) {
+          okListener.actionPerformed(new ActionEvent(fileDialog,
+              ActionEvent.ACTION_PERFORMED, okCommand));
+        }
+      }
+    });
+  }
+}