From b1c5d88edc57b99975365b6c6f75f536cd4345ec Mon Sep 17 00:00:00 2001
From: Andrew Kwon <andrewshky@gmail.com>
Date: Sun, 15 Sep 2019 14:01:35 -0700
Subject: [PATCH 1/3] Allow transitions to URL mode from entity mode

---
 src/init.lua | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/init.lua b/src/init.lua
index c02f5f0..596e688 100644
--- a/src/init.lua
+++ b/src/init.lua
@@ -309,6 +309,11 @@ Ki.defaultTransitionEvents = {
             function() Ki.state:exitMode() end,
             { "Entity Mode", "Exit to Normal Mode" },
         },
+        {
+            {"cmd"}, "u",
+            function() Ki.state:enterUrlMode() end,
+            { "Entity Mode", "Transition to URL Mode" },
+        },
         {
             {"cmd"}, "f",
             function() Ki.state:enterFileMode() end,
@@ -406,6 +411,7 @@ Ki._defaultStateEvents = {
     { name = "enterBrightnessControlMode", from = "normal", to = "brightness" },
     { name = "enterUrlMode", from = "normal", to = "url" },
     { name = "enterUrlMode", from = "select", to = "url" },
+    { name = "enterUrlMode", from = "entity", to = "url" },
     { name = "exitMode", from = "normal", to = "desktop" },
     { name = "exitMode", from = "entity", to = "desktop" },
     { name = "exitMode", from = "file", to = "desktop" },

From fbfcb92371dea142fc57ac903893173b026ecddd Mon Sep 17 00:00:00 2001
From: Andrew Kwon <andrewshky@gmail.com>
Date: Sun, 15 Sep 2019 14:42:30 -0700
Subject: [PATCH 2/3] Implement menu item selection event helper method

---
 docs/html/Application.html                  | 33 ++++++++++++
 docs/markdown/Application.md                |  9 ++++
 spec/application.spec.lua                   |  3 ++
 src/application.lua                         | 52 +++++++++++++++++-
 src/defaults.lua                            |  1 +
 src/entities/preview.lua                    | 27 ++--------
 src/entities/quicktime-player.lua           | 58 +++++++++++++++------
 src/entities/text-edit.lua                  | 27 ++--------
 src/osascripts/quicktime-player.applescript | 42 +++++++++++++++
 9 files changed, 186 insertions(+), 66 deletions(-)
 create mode 100644 src/osascripts/quicktime-player.applescript

diff --git a/docs/html/Application.html b/docs/html/Application.html
index 3ae62f1..5192d41 100644
--- a/docs/html/Application.html
+++ b/docs/html/Application.html
@@ -27,6 +27,7 @@ <h3>API Overview</h3>
         <li>Methods - API calls which can only be made on an object returned by a constructor</li>
           <ul>
             <li><a href="#createMenuItemEvent">createMenuItemEvent</a></li>
+            <li><a href="#createMenuItemSelectionEvent">createMenuItemSelectionEvent</a></li>
             <li><a href="#focus">focus</a></li>
             <li><a href="#getApplication">getApplication</a></li>
             <li><a href="#getMenuItemList">getMenuItemList</a></li>
@@ -87,6 +88,38 @@ <h5><a href="#createMenuItemEvent">createMenuItemEvent</a></h5>
 <ul>
 <li>None</li>
 </ul>
+</td>
+              </tr>
+            </table>
+          </section>
+          <section id="createMenuItemSelectionEvent">
+            <a name="//apple_ref/cpp/Method/createMenuItemSelectionEvent" class="dashAnchor"></a>
+            <h5><a href="#createMenuItemSelectionEvent">createMenuItemSelectionEvent</a></h5>
+            <table>
+              <tr>
+                <th>Signature</th>
+                <td><code>Application.createMenuItemSelectionEvent(menuItem[, shouldFocusAfter, shouldFocusBefore])</code></td>
+              </tr>
+              <tr>
+                <th>Type</th>
+                <td>Method</td>
+              </tr>
+              <tr>
+                <th>Description</th>
+                <td><p>Convenience method to create an event handler that presents a selection modal containing menu items that are nested/expandable underneath at the provided <code>menuItem</code> path, with optionally specified behavior on how the menu item selection occurs</p>
+<p>Parameters:</p>
+<ul>
+<li><code>menuItem</code> - a table list of strings that represent a path to a menu item that expands to menu item list, i.e., <code>{ "File", "Open Recent" }</code></li>
+<li><code>options</code> - a optional table containing some or all of the following fields to define the behavior for the menu item selection event:<ul>
+<li><code>focusBefore</code> - an optional boolean denoting whether to focus the application before the menu item is selected</li>
+<li><code>focusAfter</code> - an optional boolean denoting whether to focus the application after the menu item is selected</li>
+</ul>
+</li>
+</ul>
+<p>Returns:</p>
+<ul>
+<li>None</li>
+</ul>
 </td>
               </tr>
             </table>
diff --git a/docs/markdown/Application.md b/docs/markdown/Application.md
index aea9662..63acb4e 100644
--- a/docs/markdown/Application.md
+++ b/docs/markdown/Application.md
@@ -9,6 +9,7 @@ Application class that subclasses [Entity](Entity.html) to represent some automa
  * [behaviors](#behaviors)
 * Methods - API calls which can only be made on an object returned by a constructor
  * [createMenuItemEvent](#createMenuItemEvent)
+ * [createMenuItemSelectionEvent](#createMenuItemSelectionEvent)
  * [focus](#focus)
  * [getApplication](#getApplication)
  * [getMenuItemList](#getMenuItemList)
@@ -36,6 +37,14 @@ Application class that subclasses [Entity](Entity.html) to represent some automa
 | **Parameters**                              | <ul><li>`menuItem` - the menu item to select, specified as either a string or a table</li><li>`options` - a optional table containing some or all of the following fields to define the behavior for the menu item selection event:</li><li>   `isRegex` - a boolean denoting whether there is a regular expression within the menu item name(s)</li><li>   `isToggleable` - a boolean denoting whether the menu item parameter is passed in as a list of two items to toggle between, i.e., `{ "Play", "Pause" }`</li><li>   `focusBefore` - an optional boolean denoting whether to focus the application before the menu item is selected</li><li>   `focusAfter` - an optional boolean denoting whether to focus the application after the menu item is selected</li></ul> |
 | **Returns**                                 | <ul><li>None</li></ul>          |
 
+| [createMenuItemSelectionEvent](#createMenuItemSelectionEvent)         |                                                                                     |
+| --------------------------------------------|-------------------------------------------------------------------------------------|
+| **Signature**                               | `Application.createMenuItemSelectionEvent(menuItem[, shouldFocusAfter, shouldFocusBefore])`                                                                    |
+| **Type**                                    | Method                                                                     |
+| **Description**                             | Convenience method to create an event handler that presents a selection modal containing menu items that are nested/expandable underneath at the provided `menuItem` path, with optionally specified behavior on how the menu item selection occurs                                                                     |
+| **Parameters**                              | <ul><li>`menuItem` - a table list of strings that represent a path to a menu item that expands to menu item list, i.e., `{ "File", "Open Recent" }`</li><li>`options` - a optional table containing some or all of the following fields to define the behavior for the menu item selection event:</li><li>   `focusBefore` - an optional boolean denoting whether to focus the application before the menu item is selected</li><li>   `focusAfter` - an optional boolean denoting whether to focus the application after the menu item is selected</li></ul> |
+| **Returns**                                 | <ul><li>None</li></ul>          |
+
 | [focus](#focus)         |                                                                                     |
 | --------------------------------------------|-------------------------------------------------------------------------------------|
 | **Signature**                               | `Application:focus(app[, choice])`                                                                    |
diff --git a/spec/application.spec.lua b/spec/application.spec.lua
index 8a2c0e1..32862ce 100644
--- a/spec/application.spec.lua
+++ b/spec/application.spec.lua
@@ -51,6 +51,7 @@ local function getSelectionItemsTests()
                     text = "Test Window",
                     subText = "Window 1 ",
                     windowId = 1,
+                    windowIndex = 1,
                 },
             },
         },
@@ -69,6 +70,7 @@ local function getSelectionItemsTests()
                     text = "Test Window",
                     subText = "Window 1 (full screen)",
                     windowId = 1,
+                    windowIndex = 1,
                 },
             },
         },
@@ -87,6 +89,7 @@ local function getSelectionItemsTests()
                     text = "Test Window",
                     subText = "Window 1 (minimized)",
                     windowId = 1,
+                    windowIndex = 1,
                 },
             },
         },
diff --git a/src/application.lua b/src/application.lua
index a799a4e..3ddfece 100644
--- a/src/application.lua
+++ b/src/application.lua
@@ -21,7 +21,8 @@ if not _G.requirePackage then
 end
 -- luacov: enable
 
-local Entity = _G.requirePackage("Entity", true)
+local Util = _G.requirePackage("util", true)
+local Entity = _G.requirePackage("entity", true)
 local Application = Entity:subclass("Application")
 local ApplicationBehaviors = {}
 
@@ -118,6 +119,7 @@ function Application:getSelectionItems()
             text = window:title(),
             subText = "Window "..index.." "..state,
             windowId = window:id(),
+            windowIndex = index,
         })
     end
 
@@ -159,6 +161,53 @@ function Application.createMenuItemEvent(menuItem, options)
     end
 end
 
+--- Application.createMenuItemSelectionEvent(menuItem[, shouldFocusAfter, shouldFocusBefore])
+--- Method
+--- Convenience method to create an event handler that presents a selection modal containing menu items that are nested/expandable underneath at the provided `menuItem` path, with optionally specified behavior on how the menu item selection occurs
+---
+--- Parameters:
+---  * `menuItem` - a table list of strings that represent a path to a menu item that expands to menu item list, i.e., `{ "File", "Open Recent" }`
+---  * `options` - a optional table containing some or all of the following fields to define the behavior for the menu item selection event:
+---     * `focusBefore` - an optional boolean denoting whether to focus the application before the menu item is selected
+---     * `focusAfter` - an optional boolean denoting whether to focus the application after the menu item is selected
+---
+--- Returns:
+---  * None
+function Application.createMenuItemSelectionEvent(menuItem, options)
+    options = options or {}
+
+    return function(app, windowChoice)
+        local menuItemList = Application.getMenuItemList(app, menuItem)
+        local choices = {}
+
+        for _, item in pairs(menuItemList) do
+            if item.AXTitle and #item.AXTitle > 0 then
+                table.insert(choices, {
+                    text = item.AXTitle,
+                })
+            end
+        end
+
+        Application.showSelectionModal(choices, function(menuItemChoice)
+            if menuItemChoice then
+                if (options.focusBefore) then
+                    Application.focus(app, windowChoice)
+                end
+
+                local targetMenuItem = Util:clone(menuItem)
+
+                table.insert(targetMenuItem, menuItemChoice.text)
+
+                app:selectMenuItem(targetMenuItem)
+
+                if (options.focusAfter) then
+                    Application.focus(app, windowChoice)
+                end
+            end
+        end)
+    end
+end
+
 --- Application.getMenuItemList(app, menuItemPath) -> table or nil
 --- Method
 --- Gets a list of menu items from a hierarchical menu item path
@@ -177,7 +226,6 @@ function Application.getMenuItemList(app, menuItemPath)
         for _, menuItem in pairs(menuItems) do
             if menuItem.AXTitle == menuItemName then
                 menuItems = menuItem.AXChildren[1]
-
                 if menuItemName ~= menuItemPath[#menuItemPath] then
                     isFound = true
                 end
diff --git a/src/defaults.lua b/src/defaults.lua
index eaaff5c..5812d46 100644
--- a/src/defaults.lua
+++ b/src/defaults.lua
@@ -154,6 +154,7 @@ function Defaults.createEntityEvents()
         { nil, "m", entities.Messages, { "Select Events", "Select a Messages conversation" } },
         { nil, "n", entities.Notes, { "Select Events", "Select a Note" } },
         { nil, "p", entities.Preview, { "Select Events", "Select a Preview window" } },
+        { nil, "q", entities.QuickTimePlayer, { "Select Events", "QuickTime Player" } },
         { nil, "s", entities.Safari, { "Select Events", "Select a Safari tab or window" } },
         { nil, "t", entities.Terminal, { "Select Events", "Select a Terminal window" } },
         { nil, ",", entities.SystemPreferences, { "Entities", "Select a System Preferences pane" } },
diff --git a/src/entities/preview.lua b/src/entities/preview.lua
index 5632ae7..ab42c54 100644
--- a/src/entities/preview.lua
+++ b/src/entities/preview.lua
@@ -1,32 +1,13 @@
 local Application = dofile(_G.spoonPath.."/application.lua")
 local actions = {
+    close = Application.createMenuItemEvent("Close Window", { focusBefore = true }),
     find = Application.createMenuItemEvent({ "Find", "Find..." }, { focusAfter = true }),
     open = Application.createMenuItemEvent("Open...", { focusAfter = true }),
-    close = Application.createMenuItemEvent("Close Window", { focusBefore = true }),
+    openRecent = Application.createMenuItemSelectionEvent({ "File", "Open Recent" }, {
+        focusAfter = true,
+    }),
 }
 
-function actions.openRecent(app)
-    local menuItem = { "File", "Open Recent" }
-    local menuItemList = Application.getMenuItemList(app, menuItem)
-    local recentFileChoices = {}
-
-    for _, item in pairs(menuItemList) do
-        if item.AXTitle and #item.AXTitle > 0 then
-            table.insert(recentFileChoices, {
-                text = item.AXTitle,
-            })
-        end
-    end
-
-    local function selectMenuItemAndFocus(choice)
-        table.insert(menuItem, choice.text)
-        app:selectMenuItem(menuItem)
-        Application.focus(app)
-    end
-
-    Application.showSelectionModal(recentFileChoices, selectMenuItemAndFocus)
-end
-
 local shortcuts = {
     { nil, "o", actions.open, { "File", "Open..." } },
     { nil, "w", actions.close, { "File", "Close Window" } },
diff --git a/src/entities/quicktime-player.lua b/src/entities/quicktime-player.lua
index c800637..b6da40a 100644
--- a/src/entities/quicktime-player.lua
+++ b/src/entities/quicktime-player.lua
@@ -3,35 +3,59 @@ local actions = {
     newAudioRecording = Application.createMenuItemEvent("New Audio Recording"),
     newMovieRecording = Application.createMenuItemEvent("New Movie Recording"),
     openFile = Application.createMenuItemEvent("Open File...", { focusAfter = true }),
+    openRecent = Application.createMenuItemSelectionEvent({ "File", "Open Recent" }, {
+        focusAfter = true,
+    }),
     newScreenRecording = Application.createMenuItemEvent("New Screen Recording"),
+    trim = Application.createMenuItemEvent("Trim...", { focusBefore = true }),
+    exportAs = Application.createMenuItemSelectionEvent({ "File", "Export As" }, {
+        focusBefore = true,
+    }),
 }
 
-function actions.openRecent(app)
-    local menuItem = { "File", "Open Recent" }
-    local menuItemList = Application.getMenuItemList(app, menuItem)
-    local recentFileChoices = {}
-
-    for _, item in pairs(menuItemList) do
-        if item.AXTitle and #item.AXTitle > 0 then
-            table.insert(recentFileChoices, {
-                text = item.AXTitle,
-            })
-        end
+function actions.togglePlay(_, choice)
+    local target = choice
+        and "document "..choice.windowIndex
+        or "the first document"
+    local script = Application.renderScriptTemplate("quicktime-player", {
+        option = "toggle-play",
+        target = target,
+    })
+    local isOk, _, rawTable = hs.osascript.applescript(script)
+
+    if not isOk then
+        Application.notifyError("Error toggling play", rawTable.NSLocalizedFailureReason)
+    end
+end
+
+function actions.toggleFileLoop(_, choice)
+    local target = choice
+        and "document "..choice.windowIndex
+        or "the first document"
+    local script = Application.renderScriptTemplate("quicktime-player", {
+        option = "toggle-looping",
+        target = target,
+    })
+    local isOk, isLooping, rawTable = hs.osascript.applescript(script)
+
+    if not isOk then
+        Application.notifyError("Error toggling file loop option", rawTable.NSLocalizedFailureReason)
     end
 
-    Application.showSelectionModal(recentFileChoices, function(choice)
-        if choice then
-            table.insert(menuItem, choice.text)
-            app:selectMenuItem(menuItem)
-        end
-    end)
+    local toggleText = isLooping and "ON" or "OFF"
+
+    hs.notify.show("Ki", "QuickTime Player", "Turned File Loop option"..toggleText)
 end
 
 local shortcuts = {
     { nil, "a", actions.newAudioRecording, { "File", "New Audio Recording" } },
+    { nil, "e", actions.exportAs, { "File", "Export As" } },
+    { nil, "l", actions.toggleFileLoop, { "File", "Toggle File Loop Option" } },
     { nil, "m", actions.newMovieRecording, { "File", "New Movie Recording" } },
     { nil, "o", actions.openFile, { "File", "Open File..." } },
     { nil, "s", actions.newScreenRecording, { "File", "New Screen Recording" } },
+    { nil, "t", actions.trim, { "File", "Trim..." } },
+    { nil, "space", actions.togglePlay, { "Playback", "Toggle Play" } },
     { { "shift" }, "o", actions.openRecent, { "File", "Open Recent" } },
 }
 
diff --git a/src/entities/text-edit.lua b/src/entities/text-edit.lua
index 6eeb01e..3a21953 100644
--- a/src/entities/text-edit.lua
+++ b/src/entities/text-edit.lua
@@ -3,35 +3,14 @@ local actions = {
     newDocument = Application.createMenuItemEvent("New", { focusAfter = true }),
     saveDocument = Application.createMenuItemEvent("Save", { focusAfter = true }),
     openDocument = Application.createMenuItemEvent("Open...", { focusAfter = true }),
+    openRecent = Application.createMenuItemSelectionEvent({ "File", "Open Recent" }, {
+        focusAfter = true,
+    }),
     printDocument = Application.createMenuItemEvent("Print...", { focusBefore = true }),
     closeDocument = Application.createMenuItemEvent("Close", { focusBefore = true }),
     duplicateDocument = Application.createMenuItemEvent("Duplicate", { focusAfter = true }),
 }
 
-function actions.openRecent(app)
-    local menuItem = { "File", "Open Recent" }
-    local menuItemList = Application.getMenuItemList(app, menuItem)
-    local recentFileChoices = {}
-
-    for _, item in pairs(menuItemList) do
-        if item.AXTitle and #item.AXTitle > 0 then
-            table.insert(recentFileChoices, {
-                text = item.AXTitle,
-            })
-        end
-    end
-
-    local function selectMenuItemAndFocus(choice)
-        if choice then
-            table.insert(menuItem, choice.text)
-            app:selectMenuItem(menuItem)
-            Application.focus(app)
-        end
-    end
-
-    Application.showSelectionModal(recentFileChoices, selectMenuItemAndFocus)
-end
-
 local shortcuts = {
     { nil, "n", actions.newDocument, { "File", "New Document" } },
     { nil, "o", actions.openDocument, { "File", "Open Document" } },
diff --git a/src/osascripts/quicktime-player.applescript b/src/osascripts/quicktime-player.applescript
new file mode 100644
index 0000000..da0c94b
--- /dev/null
+++ b/src/osascripts/quicktime-player.applescript
@@ -0,0 +1,42 @@
+-- AppleScript template for various QuickTime Player options
+-- `option` - the name of the option
+-- `targetDocument` - the target QuickTime Player document
+set option to "{{option}}"
+
+tell application "QuickTime Player"
+
+    set target to {{target}}
+
+	if option is "toggle-play" then
+
+        if target is playing then
+
+            pause target
+
+        else
+
+            play target
+
+        end if
+
+    else if option is "stop" then
+
+        stop target
+
+    else if option is "toggle-looping" then
+
+        if the first document is looping then
+
+            set looping of target to false
+
+        else
+
+            set looping of target to true
+
+        end if
+
+        return looping of target
+
+	end if
+
+end tell

From e659d327a506bc9f8ee8a10ff7fdf8b92da40b9a Mon Sep 17 00:00:00 2001
From: Andrew Kwon <andrewshky@gmail.com>
Date: Sun, 15 Sep 2019 15:18:13 -0700
Subject: [PATCH 3/3] Fix dialog block alerts to focus correctly

---
 spec/entity.spec.lua |  4 ++++
 src/defaults.lua     | 36 ++++++++++++++++++++++++------------
 src/entity.lua       |  8 +++++---
 3 files changed, 33 insertions(+), 15 deletions(-)

diff --git a/spec/entity.spec.lua b/spec/entity.spec.lua
index df9bc91..2ad6d1a 100644
--- a/spec/entity.spec.lua
+++ b/spec/entity.spec.lua
@@ -179,10 +179,14 @@ local function triggerAfterConfirmationTests()
             local mockDialog = {
                 blockAlert = function() return test.answer end,
             }
+            local mockTimer = {
+                doAfter = function(_, callback) callback() end
+            }
 
             _G.hs = mock(mockHs({
                 dialog = mockDialog,
                 focus = focusSpy,
+                timer = mockTimer,
             }))
 
             entity.triggerAfterConfirmation(table.unpack(test.args))
diff --git a/src/defaults.lua b/src/defaults.lua
index 5812d46..e54eca3 100644
--- a/src/defaults.lua
+++ b/src/defaults.lua
@@ -366,11 +366,15 @@ function Defaults.createNormalEvents(Ki)
     function actions.logout()
         Ki.state:exitMode()
 
-        local answer = hs.dialog.blockAlert("Log out from your computer?", "", "Log out", "Cancel")
+        hs.timer.doAfter(0, function()
+            hs.focus()
 
-        if answer == "Log out" then
-            hs.osascript.applescript([[ tell application "System Events" to log out ]])
-        end
+            local answer = hs.dialog.blockAlert("Log out from your computer?", "", "Log out", "Cancel")
+
+            if answer == "Log out" then
+                hs.osascript.applescript([[ tell application "System Events" to log out ]])
+            end
+        end)
     end
 
     function actions.startScreenSaver()
@@ -389,21 +393,29 @@ function Defaults.createNormalEvents(Ki)
     function actions.restart()
         Ki.state:exitMode()
 
-        local answer = hs.dialog.blockAlert("Restart your computer?", "", "Restart", "Cancel")
+        hs.timer.doAfter(0, function()
+            hs.focus()
 
-        if answer == "Restart" then
-            hs.osascript.applescript([[ tell application "Finder" to restart ]])
-        end
+            local answer = hs.dialog.blockAlert("Restart your computer?", "", "Restart", "Cancel")
+
+            if answer == "Restart" then
+                hs.osascript.applescript([[ tell application "Finder" to restart ]])
+            end
+        end)
     end
 
     function actions.shutdown()
         Ki.state:exitMode()
 
-        local answer = hs.dialog.blockAlert("Shut down your computer?", "", "Shut Down", "Cancel")
+        hs.timer.doAfter(0, function()
+            hs.focus()
 
-        if answer == "Shut Down" then
-            hs.osascript.applescript([[ tell application "System Events" to shut down ]])
-        end
+            local answer = hs.dialog.blockAlert("Shut down your computer?", "", "Shut Down", "Cancel")
+
+            if answer == "Shut Down" then
+                hs.osascript.applescript([[ tell application "System Events" to shut down ]])
+            end
+        end)
     end
 
     local normalEvents = {
diff --git a/src/entity.lua b/src/entity.lua
index ee5d976..9f9690e 100644
--- a/src/entity.lua
+++ b/src/entity.lua
@@ -200,11 +200,13 @@ end
 --- Returns:
 ---   * None
 function Entity.triggerAfterConfirmation(question, action)
-    hs.focus()
+    hs.timer.doAfter(0, function()
+        hs.focus()
 
-    local answer = hs.dialog.blockAlert(question, "", "Confirm", "Cancel")
+        local answer = hs.dialog.blockAlert(question, "", "Confirm", "Cancel")
 
-    if answer == "Confirm" then action() end
+        if answer == "Confirm" then action() end
+    end)
 end
 
 --- Entity.selectionModalShortcuts