From ff39470d3765b54d2487c7943dcfd0fcd84ff229 Mon Sep 17 00:00:00 2001
From: Jeffrey Morgan <jmorganca@gmail.com>
Date: Fri, 19 Jun 2015 22:24:16 -0700
Subject: [PATCH] Fix bug where plist would not be modifyed for try on desktop

---
 Gruntfile.js | 42 +++++++++++++++++++++++++++++++++++++++---
 package.json |  1 +
 2 files changed, 40 insertions(+), 3 deletions(-)

diff --git a/Gruntfile.js b/Gruntfile.js
index 15bb64d1d5..7fcf577d2b 100644
--- a/Gruntfile.js
+++ b/Gruntfile.js
@@ -97,7 +97,8 @@ module.exports = function (grunt) {
           platform: 'darwin',
           arch: 'x64',
           asar: true,
-          'app-bundle-id': 'com.kitematic.kitematic'
+          'app-bundle-id': 'com.kitematic.kitematic',
+          'app-version': packagejson.version
         }
       }
     },
@@ -269,6 +270,41 @@ module.exports = function (grunt) {
       }
     },
 
+    plistbuddy: {
+      addBundleURLTypes: {
+        method: 'Add',
+        entry: 'CFBundleURLTypes',
+        type: 'array',
+        src: '<%= OSX_FILENAME %>/Contents/Info.plist'
+      },
+      addBundleURLTypesDict: {
+        method: 'Add',
+        entry: 'CFBundleURLTypes:0',
+        type: 'dict',
+        src: '<%= OSX_FILENAME %>/Contents/Info.plist'
+      },
+      addBundleURLTypesDictName: {
+        method: 'Add',
+        entry: 'CFBundleURLTypes:0:CFBundleURLName',
+        type: 'string',
+        value: 'Docker App Protocol',
+        src: '<%= OSX_FILENAME %>/Contents/Info.plist'
+      },
+      addBundleURLTypesDictSchemes: {
+        method: 'Add',
+        entry: 'CFBundleURLTypes:0:CFBundleURLSchemes',
+        type: 'array',
+        src: '<%= OSX_FILENAME %>/Contents/Info.plist'
+      },
+      addBundleURLTypesDictSchemesDocker: {
+        method: 'Add',
+        entry: 'CFBundleURLTypes:0:CFBundleURLSchemes:0',
+        type: 'string',
+        value: 'docker',
+        src: '<%= OSX_FILENAME %>/Contents/Info.plist'
+      }
+    },
+
     shell: {
       electron: {
         command: electron + ' ' + 'build',
@@ -287,7 +323,7 @@ module.exports = function (grunt) {
           'codesign --deep -v -f -s "<%= IDENTITY %>" <%= OSX_FILENAME_ESCAPED %>/Contents/Frameworks/*',
           'codesign -v -f -s "<%= IDENTITY %>" <%= OSX_FILENAME_ESCAPED %>',
           'codesign -vvv --display <%= OSX_FILENAME_ESCAPED %>',
-          'codesign -v --verify <%= OSX_FILENAME_ESCAPED %>',
+          'codesign -v --verify <%= OSX_FILENAME_ESCAPED %>'
         ].join(' && '),
       },
       zip: {
@@ -328,7 +364,7 @@ module.exports = function (grunt) {
   if (process.platform === 'win32') {
     grunt.registerTask('release', ['clean:release', 'download-binary', 'download-boot2docker-iso', 'babel', 'less', 'copy:dev', 'electron:windows', 'copy:windows', 'rcedit:exes', 'create-windows-installer', 'rename:installer']);
   } else {
-    grunt.registerTask('release', ['clean:release', 'download-binary', 'download-boot2docker-iso', 'babel', 'less', 'copy:dev', 'electron:osx', 'copy:osx', 'shell:sign', 'shell:zip']);
+    grunt.registerTask('release', ['clean:release', 'download-binary', 'download-boot2docker-iso', 'babel', 'less', 'copy:dev', 'electron:osx', 'copy:osx', 'plistbuddy', 'shell:sign', 'shell:zip']);
   }
 
   process.on('SIGINT', function () {
diff --git a/package.json b/package.json
index 7f4a2eca18..bcc69e9ad8 100644
--- a/package.json
+++ b/package.json
@@ -80,6 +80,7 @@
     "grunt-electron-installer": "^0.33.0",
     "grunt-if-missing": "^1.0.0",
     "grunt-newer": "^1.1.1",
+    "grunt-plistbuddy": "^0.1.1",
     "grunt-rcedit": "^0.3.1",
     "grunt-rename": "^0.1.4",
     "grunt-shell": "^1.1.2",