From a6ddbae4983c2051ad2a036adfb6bcd2afb9dee5 Mon Sep 17 00:00:00 2001
From: Drew Regitsky <drew@nylas.com>
Date: Tue, 1 Dec 2015 15:35:14 -0800
Subject: [PATCH] fix(examples): Small fixes to QuickSchedule plugin

---
 examples/N1-Quick-Schedule/calendar.html | 10 ++++----
 examples/N1-Quick-Schedule/lib/main.cjsx | 29 +++++++++++++++---------
 2 files changed, 23 insertions(+), 16 deletions(-)

diff --git a/examples/N1-Quick-Schedule/calendar.html b/examples/N1-Quick-Schedule/calendar.html
index 9f88079bc..f65a45716 100644
--- a/examples/N1-Quick-Schedule/calendar.html
+++ b/examples/N1-Quick-Schedule/calendar.html
@@ -5,11 +5,11 @@
     <title></title>
     <link rel='stylesheet' href='bootstrap.min.css'/>
     <link rel='stylesheet' href='fullcalendar/fullcalendar.css'/>
-    <script src='nylas://N1-Quick-Schedule/fullcalendar/lib/jquery.min.js'></script>
-    <script src='nylas://N1-Quick-Schedule/fullcalendar/lib/moment.min.js'></script>
-    <script src='nylas://N1-Quick-Schedule/fullcalendar/lib/moment-timezone-with-data.min.js'></script>
-    <script src='nylas://N1-Quick-Schedule/fullcalendar/fullcalendar.js'></script>
-    <script src='nylas://N1-Quick-Schedule/bootstrap.min.js'></script>
+    <script src='fullcalendar/lib/jquery.min.js'></script>
+    <script src='fullcalendar/lib/moment.min.js'></script>
+    <script src='fullcalendar/lib/moment-timezone-with-data.min.js'></script>
+    <script src='fullcalendar/fullcalendar.js'></script>
+    <script src='bootstrap.min.js'></script>
 
     <script>
 
diff --git a/examples/N1-Quick-Schedule/lib/main.cjsx b/examples/N1-Quick-Schedule/lib/main.cjsx
index df17e93fa..78fb7792f 100644
--- a/examples/N1-Quick-Schedule/lib/main.cjsx
+++ b/examples/N1-Quick-Schedule/lib/main.cjsx
@@ -77,15 +77,21 @@ module.exports =
     # `DraftStoreExtension`.
     DraftStore.registerExtension AvailabilityDraftExtension
 
-    # Subscribe to the ipc event `fromRenderer`, which will be published
-    # elsewhere in the package.
-    protocol.registerStringProtocol 'quick-schedule', (request, callback) =>
-      {host:event,query:rawQuery} = url.parse(request.url)
-      stringArgs = qs.parse(rawQuery)
-      data = {}
-      for own k,v of stringArgs
-        data[k] = JSON.parse(v)
-      response = @_onCalendarEvent(event,data,callback)
+    # Register a protocol that allows the calendar window to pass data back to the plugin
+    # with web requests
+    if NylasEnv.isMainWindow()
+      # First unregister the protocol, in case it has already been registered with a callback that's no
+      # longer valid (e.g. if the main window has reloaded). If the protocol is not registered, this line
+      # does nothing.
+      protocol.unregisterProtocol('quick-schedule')
+      # Now register the new protocol
+      protocol.registerStringProtocol 'quick-schedule', (request, callback) =>
+        {host:event,query:rawQuery} = url.parse(request.url)
+        stringArgs = qs.parse(rawQuery)
+        data = {}
+        for own k,v of stringArgs
+          data[k] = JSON.parse(v)
+        response = @_onCalendarEvent(event,data,callback)
 
   # Serialize is called when your package is about to be unmounted.
   # You can return a state object that will be passed back to your package
@@ -100,8 +106,9 @@ module.exports =
   #
   deactivate: ->
     ComponentRegistry.unregister CalendarButton
-    DraftStore.unregister AvailabilityDraftExtension
-    protocol.unregisterProtocol('quick-schedule')
+    DraftStore.unregisterExtension AvailabilityDraftExtension
+    if NylasEnv.isMainWindow()
+      protocol.unregisterProtocol('quick-schedule')
 
   ### Internal Methods ###