wf=hs.window.filter
ffBrowser = "Firefox"
chromiumBrowser = "Brave Browser"
-- browser = "Chromium"

-- Open new / focus existing Finder window in current desktop space

hs.hotkey.bind({"cmd", "ctrl"}, "X", function()
	local app = "Finder"

	wf_app = wf.new(false):setAppFilter(app, {currentSpace=true})

	local wins = wf_app:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end
	
	if (count > 0)
	then
		wf_app:getWindows()[1]:focus()
	else
		hs.osascript.applescript(string.format([[
			tell application "System Events" to tell process "%s"
				click menu item "New Finder Window" of menu "File" of menu bar 1
				set frontmost to true
			end tell
		]], app))
	end

end)

-- Open new / focus existing terminal window in current desktop space

-- hs.hotkey.bind({"cmd", "ctrl"}, "T", function()
--     -- Adjust the app filter to Warp
--     wf_warp = wf.new(false):setAppFilter("Warp", {currentSpace=true, visible=true})
-- 
--     local wins = wf_warp:getWindows()
--     local count = 0
--     for _ in pairs(wins) do count = count + 1 end
-- 
--     if (count > 0) then
--         wf_warp:getWindows()[1]:focus()
--     else
--         hs.osascript.applescript([[
--    			tell application "System Events" to tell process "Warp"
--    				click menu item "New Window" of menu "File" of menu bar 1
--    			end tell
--         ]])
--     end
-- end)

-- hs.hotkey.bind({"cmd", "ctrl"}, "T", function()
--  	wf_kitty = wf.new(false):setAppFilter("kitty", {currentSpace=true, visible=true})
--  	local wins = wf_kitty:getWindows()
--  	local count = 0
--  	for _ in pairs(wins) do count = count + 1 end
-- 
--  	if (count > 0)
--  	then
--  		wf_kitty:getWindows()[1]:focus()
--  	else
--  		hs.osascript.applescript([[
-- 			tell application "System Events" to tell process "kitty"
-- 				click menu item "New OS Window" of menu "Shell" of menu bar 1
-- 			end tell
--  		]])
--  	end
-- end)

hs.hotkey.bind({"cmd", "ctrl"}, "T", function()
	wf_iterm2 = wf.new(false):setAppFilter("iTerm2", {currentSpace=true, visible=true})

	local wins = wf_iterm2:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end

	if (count > 0)
	then
		wf_iterm2:getWindows()[1]:focus()
	else
		hs.osascript.applescript([[
			tell application "iTerm"
				create window with default profile
				activate
			end tell
		]])
	end

end)

-- Open new / focus existing browser window in current desktop space

function focusOrOpenFirefox()

	-- assumes wf_browser is the name of a firefox-based browser
	wf_browser = wf.new(false):setAppFilter(ffBrowser, {currentSpace=true, visible=true})

	local wins = wf_browser:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end
	
	if (count > 0)
	then
		wf_browser:getWindows()[1]:focus()
		-- print(count)
	else
		-- for firefox:
		hs.osascript.applescript(string.format([[
			tell application "System Events" to tell process "%s"
				click menu item "New Window" of menu "File" of menu bar 1
				set frontmost to true
			end tell
		]], ffBrowser))
	end

end

function focusOrOpenChromium()

	-- assumes wf_browser is the name of a chromium-based browser
	wf_browser = wf.new(false):setAppFilter(chromiumBrowser, {currentSpace=true, visible=true})

	local wins = wf_browser:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end
	
	if (count > 0)
	then
		wf_browser:getWindows()[1]:focus()
		-- print(count)
	else
		hs.osascript.applescript(string.format([[
			tell application "%s"
				make new window
				activate
			end tell
		]], chromiumBrowser))
	end

end

hs.hotkey.bind({"cmd", "ctrl"}, "W", focusOrOpenChromium)


-- CHROMIUM-BASED BROWSERS ONLY: Open new tab to right of current browser
-- NOTE: currently replaced with binding in chromiumsystem settings
-- hs.hotkey.bind({"cmd", "option"}, "T", function()
-- 
-- 	local focusedAppName = hs.window.focusedWindow():application():title()
-- 
-- 	if focusedAppName == chromiumBrowser 
-- 	then
-- 		hs.osascript.applescript(string.format([[
-- 			tell application "System Events" to tell process "%s"
-- 				click menu item "New Tab to the Right" of menu "Tab" of menu bar 1
-- 			end tell
-- 		]], focusedAppName))
-- 	else
-- 		hs.notify.new({title=string.format("%s not focused.", browser)}):send()
-- 	end
-- 
-- end)


-- Open new / focus existing vscode window in current desktop space
-- app = VSCodium, Code, or Cursor
function openVsCode(app)
	wf_app = wf.new(false):setAppFilter(app, {currentSpace=true, visible=true})

	local wins = wf_app:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end
	
	if (count > 0) then
		wf_app:getWindows()[1]:focus()
	else
        hs.osascript.applescript(string.format([[
            tell application "System Events" to tell process "%s"
                click menu item "New Window" of menu "File" of menu bar 1
                set frontmost to true
            end tell
        ]], app))
	end

end


-- superbinding to use both cursor and vscode
-- try to focus cursor (primarily) or vscode (secondarily), then open cursor if neither is open
function tryFocusCursorOrVSCodeThenOpenCursor()
	wf_cursor= wf.new(false):setAppFilter("Cursor", {currentSpace=true, visible=true})
	wf_vscode= wf.new(false):setAppFilter("Code", {currentSpace=true, visible=true})

	local cursor_wins = wf_cursor:getWindows()
	local cursor_count = 0
	for _ in pairs(cursor_wins) do cursor_count = cursor_count + 1 end

	local vscode_wins = wf_vscode:getWindows()
	local vscode_count = 0
	for _ in pairs(vscode_wins) do vscode_count = vscode_count + 1 end

	if (cursor_count > 0) then
		wf_cursor:getWindows()[1]:focus()
	elseif (vscode_count > 0) then
		wf_vscode:getWindows()[1]:focus()
	else
        hs.osascript.applescript(string.format([[
            tell application "System Events" to tell process "%s"
                click menu item "New Window" of menu "File" of menu bar 1
                set frontmost to true
            end tell
        ]], "Cursor"))
	end

end

hs.hotkey.bind({"cmd", "ctrl"}, "E", function()
	-- openVsCode("Cursor")
	-- openVsCode("Code")
	tryFocusCursorOrVSCodeThenOpenCursor()
end)

hs.hotkey.bind({"cmd", "ctrl"}, "V", function()
	openVsCode("Code")
end)


-- hs.hotkey.bind({"cmd", "ctrl"}, "V", function()
	-- openVsCode("Code")
	-- openVsCode("VSCodium")
	
	-- No neovide function because it doesn't support system events (yet)
	-- hs.application.open("Neovide")
-- end)



-- Open new / focus existing g/n/mac/vim(r) window in current desktop space

-- hs.hotkey.bind({"cmd", "ctrl"}, "E", function()
-- 	local app = "VimR"
-- 
-- 	wf_app = wf.new(false):setAppFilter(app, {currentSpace=true, visible=true})
-- 
-- 	local wins = wf_app:getWindows()
-- 	local count = 0
-- 	for _ in pairs(wins) do count = count + 1 end
-- 	
-- 	if (count > 0)
-- 	then
-- 		wf_app:getWindows()[1]:focus()
-- 	else
-- 		hs.osascript.applescript(string.format([[
-- 			tell application "System Events" to tell process "%s"
-- 			    click menu item "New Window" of menu "File" of menu bar 1
-- 				set frontmost to true
-- 			end tell
-- 		]], app))
-- 
-- 	end
-- 
-- end)

-- APP-AGNOSTIC GLOBAL OPEN/FOCUS BINDINGS

-- common apps
-- hs.hotkey.bind({"cmd", "ctrl"}, "R", function() hs.application.open("Obsidian") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "M", function() hs.application.open("Spotify") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "G", function() hs.application.open("Google Calendar") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "Z", function() hs.application.open("Preview") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "C", function() hs.application.open("Open WebUI") end) 

-- annoying apps (must manually open)
hs.hotkey.bind({"cmd", "ctrl"}, "A", function() switchToIfApplicationOpen("Telegram") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "S", function() switchToIfApplicationOpen("Signal") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "N", function() switchToIfApplicationOpen("Numi") end) 
-- hs.hotkey.bind({"cmd", "ctrl"}, "S", function() switchToIfApplicationOpen("Texts") end) 
-- hs.hotkey.bind({"cmd", "ctrl", "shift"}, "S", function() hs.application.open("Signal") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "K", function() switchToIfApplicationOpen("KiCad") end) 
-- hs.hotkey.bind({"cmd", "ctrl"}, "C", function() switchToIfApplicationOpen("CLion") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "J", function() switchToIfApplicationOpen("IntelliJ IDEA") end) 
-- hs.hotkey.bind({"cmd", "ctrl"}, "L", function() switchToIfApplicationOpen("Linear") end) 
hs.hotkey.bind({"cmd", "ctrl"}, "L", function() switchToIfApplicationOpen("LTspice") end) 


-- Clear clipboard

hs.hotkey.bind({"cmd", "shift", "ctrl"}, "C", function()
	hs.pasteboard.setContents("")
	hs.notify.new({title="Cleared clipboard."}):send()
end)


--------------------------------------
-- DESKTOP SWITCHING (axed)
-- Retroactive note: it should be harder, not easier, to context-switch :P
--------------------------------------

-- function gotoDesktopNumber(desktop)
--     local deskstr = string.format("Desktop %d", desktop)
-- 	local index = 0
-- 	for item in hs.spaces.spacesForScreen("Main") do
-- 		if item == deskstr then
-- 			break
-- 		else
-- 			index = index + 1
-- 		end
-- 	end
-- 	print("INDEX ", index)
-- 	hs.spaces.gotoSpace(index)
-- end
-- 
-- hs.hotkey.bind({"cmd", "ctrl"}, "1", function() gotoDesktopNumber(1) end)
-- hs.hotkey.bind({"cmd", "ctrl"}, "2", function() gotoDesktopNumber(2) end)
-- hs.hotkey.bind({"cmd", "ctrl"}, "3", function() gotoDesktopNumber(3) end)
-- hs.hotkey.bind({"cmd", "ctrl"}, "4", function() gotoDesktopNumber(4) end)

--------------------------------------
-- SWITCH TO IF OPEN
-- Switches to application if there's an instance of it open
--------------------------------------

function switchToIfWindowOpen(app)
	wf_app = wf.new(false):setAppFilter(app, {currentSpace=true, visible=true})

	local wins = wf_app:getWindows()
	local count = 0
	for _ in pairs(wins) do count = count + 1 end
	
	if (count > 0) then
		wf_app:getWindows()[1]:focus()
    end
end


--------------------------------------
-- SWITCH TO IF GLOBALLY OPEN
-- Switches to application if there's an instance of it open
--------------------------------------

function switchToIfApplicationOpen(appName)
    local runningApps = hs.application.runningApplications()
    local isOpen = false
    for _, app in pairs(runningApps) do
        if app:name() == appName then
            isOpen = true
            break
        end
    end

	if (isOpen) then
        hs.application.open(appName)
    end
end


--------------------------------------
-- KEY COMBO TO APPLICATION
-- Sends keystrokes but only if the specified application is focused
--------------------------------------

function sendKeyComboToApplication(appComboTable)

	for appComboPair in ipairs(appComboTable) do

		app = appComboPair[1]
		mods = appComboPair[2]
		key = appComboPair[3]
		
		wf_app = wf.new(false):setAppFilter(app, {currentSpace=true, visible=true})

		local wins = wf_app:getWindows()
		local count = 0
		for _ in pairs(wins) do count = count + 1 end

		if (count > 0)
		then
			hs.eventtap.keyStroke(mods, key)
		end

	end

end

-------------------
-- A R C H I V E --
-------------------

-- Desktop Action Hotkeys (MOVED TO SYSTEM SETTINGS)

-- -- Show Desktop (like windows or KDE)
-- 
-- hs.hotkey.bind({"alt"}, "d", function()
-- 	hs.eventtap.keyStroke({"fn"}, "f11")
-- end)
-- 
-- -- Switch Desktops (like windows or KDE)
-- 
-- hs.hotkey.bind({"ctrl", "alt"}, "left", function()
-- 	hs.eventtap.keyStroke({"ctrl"}, "left")
-- end)
-- 
-- hs.hotkey.bind({"ctrl", "alt"}, "right", function()
-- 	hs.eventtap.keyStroke({"ctrl"}, "right")
-- end)
-- 
-- 
-- -- Expose Windows (Mission Control)
-- 
-- hs.hotkey.bind({"alt"}, "tab", function() 
-- 	hs.application.open("Mission Control"):activate()
-- end)


-- Move windows between displays

-- hs.hotkey.bind({"ctrl", "shift"}, "right", function()
-- 	hs.window.focusedWindow():moveToScreen(hs.screen:next())
-- end)
-- 
-- hs.hotkey.bind({"ctrl", "shift"}, "left", function()
-- 	hs.window.focusedWindow():moveToScreen(hs.screen:previous())
-- end)


-- The ShiftIt Alternative (MOVED TO RECTANGLE)

-- units = {
--   right50       = { x = 0.50, y = 0.00, w = 0.50, h = 1.00 },
--   right50       = { x = 0.50, y = 0.00, w = 0.50, h = 1.00 },
--   left50        = { x = 0.00, y = 0.00, w = 0.50, h = 1.00 },
--   left50        = { x = 0.00, y = 0.00, w = 0.50, h = 1.00 },
--   top50         = { x = 0.00, y = 0.00, w = 1.00, h = 0.50 },
--   bot50         = { x = 0.00, y = 0.50, w = 1.00, h = 0.50 },
--   upright50     = { x = 0.50, y = 0.00, w = 0.50, h = 0.50 },
--   botright50    = { x = 0.50, y = 0.50, w = 0.50, h = 0.50 },
--   upleft50      = { x = 0.00, y = 0.00, w = 0.50, h = 0.50 },
--   botleft50     = { x = 0.00, y = 0.50, w = 0.50, h = 0.50 },
--   maximum       = { x = 0.00, y = 0.00, w = 1.00, h = 1.00 }
-- }
-- 
-- mash = { 'shift', 'ctrl', 'cmd' }
-- hs.hotkey.bind(mash, 'l', function() hs.window.focusedWindow():move(units.right50,    nil, true) end)
-- hs.hotkey.bind(mash, 'h', function() hs.window.focusedWindow():move(units.left50,     nil, true) end)
-- hs.hotkey.bind(mash, 'k', function() hs.window.focusedWindow():move(units.top50,      nil, true) end)
-- hs.hotkey.bind(mash, 'j', function() hs.window.focusedWindow():move(units.bot50,      nil, true) end)
-- hs.hotkey.bind(mash, ']', function() hs.window.focusedWindow():move(units.upright50,  nil, true) end)
-- hs.hotkey.bind(mash, '[', function() hs.window.focusedWindow():move(units.upleft50,   nil, true) end)
-- hs.hotkey.bind(mash, ';', function() hs.window.focusedWindow():move(units.botleft50,  nil, true) end)
-- hs.hotkey.bind(mash, "'", function() hs.window.focusedWindow():move(units.botright50, nil, true) end)