Provide a method to specify bundle documents when build app bundle (Haavar Valeur)

See:
arduino/Arduino#1665
https://java.net/jira/browse/APPBUNDLER-1

Author: cmaglie

appbundler/src/com/oracle/appbundler/AbstractKeyValue.java

@@ -0,0 +1,21 @@
+package com.oracle.appbundler;
+
+/**
+ * Created by haavar on 1/8/14.
+ */
+public class AbstractKeyValue {
+    private String value = null;
+
+    public String getValue() {
+        return value;
+    }
+
+    public void addText(String value) {
+        this.value = value;
+    }
+
+    @Override
+    public String toString() {
+        return value;
+    }
+}

appbundler/src/com/oracle/appbundler/AppBundlerTask.java

@@ -41,6 +41,7 @@
 import java.nio.file.Path;
 import java.nio.file.StandardCopyOption;
 import java.util.ArrayList;
+import java.util.List;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
 
@@ -79,6 +80,7 @@ public class AppBundlerTask extends Task {
     private ArrayList<FileSet> libraryPath = new ArrayList<>();
     private ArrayList<String> options = new ArrayList<>();
     private ArrayList<String> arguments = new ArrayList<>();
+    private List<DocumentType> documentTypes = new ArrayList<>();
 
     private static final String EXECUTABLE_NAME = "JavaAppLauncher";
     private static final String DEFAULT_ICON_NAME = "GenericApp.icns";
@@ -186,6 +188,10 @@ public void addConfiguredArgument(Argument argument) throws BuildException {
         arguments.add(value);
     }
 
+    public void addDocumentType(DocumentType documentType) {
+         documentTypes.add(documentType);
+    }
+
     @Override
     public void execute() throws BuildException {
         // Validate required properties
@@ -243,6 +249,15 @@ public void execute() throws BuildException {
             throw new IllegalStateException("Main class name is required.");
         }
 
+        for (DocumentType documentType : documentTypes) {
+            if(documentType.getName() == null) {
+                throw new IllegalStateException("Name is required for document type.");
+            }
+            if (!documentType.getIcon().exists()) {
+                throw new IllegalStateException("Document icon does not exist.");
+            }
+        }
+
         // Create the app bundle
         try {
             System.out.println("Creating app bundle: " + name);
@@ -297,6 +312,12 @@ public void execute() throws BuildException {
 
             // Copy icon to Resources folder
             copyIcon(resourcesDirectory);
+
+            for(DocumentType documentType : documentTypes) {
+                if (documentType.getIcon() != null) {
+                    copy(documentType.getIcon(), new File(resourcesDirectory, documentType.getIcon().getName()));
+                }
+            }
         } catch (IOException exception) {
             throw new BuildException(exception);
         }
@@ -451,6 +472,25 @@ private void writeInfoPlist(File file) throws IOException {
                 writeProperty(xout, "LSApplicationCategoryType", applicationCategory);
             }
 
+            if ( ! documentTypes.isEmpty()) {
+                writeKey(xout, "CFBundleDocumentTypes");
+                xout.writeStartElement(ARRAY_TAG);
+                xout.writeCharacters("\n");
+                for (DocumentType documentType : documentTypes) {
+                    writeProperty(xout, "CFBundleTypeName", documentType.getName());
+                    writeProperty(xout, "CFBundleTypeRole", documentType.getRole().name());
+                    if (documentType.getIcon() != null) {
+                        writeProperty(xout, "CFBundleTypeIconFile", documentType.getIcon().getName());
+                    }
+                    writeStringList(xout, "CFBundleTypeExtensions", documentType.getExtensions());
+                    writeStringList(xout, "CFBundleTypeMIMETypes", documentType.getMimeTypes());
+                    writeStringList(xout, "CFBundleTypeOSTypes", documentType.getOsTypes());
+
+                }
+                xout.writeEndElement();
+                xout.writeCharacters("\n");
+            }
+
             // Write runtime
             if (runtime != null) {
                 writeProperty(xout, "JVMRuntime", runtime.getDir().getParentFile().getParentFile().getName());
@@ -524,6 +564,19 @@ private void writeProperty(XMLStreamWriter xout, String key, String value) throw
         writeString(xout, value);
     }
 
+    private void writeStringList(XMLStreamWriter xout, String key, List<? extends AbstractKeyValue> list) throws XMLStreamException {
+        if ( ! list.isEmpty()) {
+            writeKey(xout, key);
+            xout.writeStartElement(ARRAY_TAG);
+            xout.writeCharacters("\n");
+            for (AbstractKeyValue extension : list) {
+                writeString(xout, extension.getValue());
+            }
+            xout.writeEndElement();
+            xout.writeCharacters("\n");
+        }
+    }
+
     private void writePkgInfo(File file) throws IOException {
         Writer out = new BufferedWriter(new FileWriter(file));
 

appbundler/src/com/oracle/appbundler/DocumentType.java

@@ -0,0 +1,77 @@
+package com.oracle.appbundler;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.List;
+
+public class DocumentType {
+    public enum Role{Editor, Viewer, Shell, None}
+
+    private String name;
+    private File icon;
+    private Role role = Role.None;
+    private List<DocumentExtension> extensions = new ArrayList<>();
+    private List<MimeType> mimeTypes = new ArrayList<>();
+    private List<OsType> osTypes = new ArrayList<>();
+
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public void setIcon(File icon) {
+          this.icon = icon;
+    }
+
+    public void setRole(Role role) {
+        this.role = role;
+    }
+
+    public void addExtension(DocumentExtension extension) {
+        extensions.add(extension);
+    }
+
+    public void addMimeType(MimeType mimeType) {
+        mimeTypes.add(mimeType);
+    }
+
+    public void addOsType(OsType osType) {
+        osTypes.add(osType);
+    }
+
+
+    public String getName() {
+        return name;
+    }
+
+    public File getIcon() {
+        return icon;
+    }
+
+    public Role getRole() {
+        return role;
+    }
+
+    public List<? extends AbstractKeyValue> getExtensions() {
+        return extensions;
+    }
+
+    public List<? extends AbstractKeyValue> getMimeTypes() {
+        return mimeTypes;
+    }
+
+    public List<? extends AbstractKeyValue> getOsTypes() {
+        return osTypes;
+    }
+
+
+    public static class DocumentExtension extends AbstractKeyValue  {
+    }
+
+    public static class MimeType extends AbstractKeyValue {
+    }
+
+    public static class OsType extends AbstractKeyValue {
+    }
+
+}