--
-- Author  : Dennis Eriksen <d@ennis.no>
-- File  : caff.lua
-- Created : 2024-12-03
--

if (config.sleepkill == nil and config.sleepstart == nil) then
  print("no caff")
  return
end

require("string")

local bt_state = hs.execute('/opt/homebrew/bin/blueutil -p')
bt_state = string.gsub(bt_state, '\n$', '')

local bluetooth = function(power)
  power = string.gsub(power, '\n$', '')

  if (power == "1") then
    power = "on"
  elseif (power == "0") then
    power = "off"
  end

  print("Setting bluetooth to " .. power)

  hs.task.new("/opt/homebrew/bin/blueutil",
    require("utils").taskCB,
    {'--power', "off"}
  ):start()
end



local watcherCB = function(event)

  --
  -- systemWillSleep -- SLEEP
  --
  if event == hs.caffeinate.watcher.systemWillSleep then

    -- Bluetooth
    if (config.getval('sleepkill.bluetooth')) then
      bt_state = hs.execute('/opt/homebrew/bin/blueutil -p')
      bt_state = string.gsub(bt_state, '\n$', '')
      bluetooth("off")
    end

    -- sleepkill apps
    if (#config.sleepkill.apps > 0) then
      for _, app in pairs(config.sleepkill.apps) do
        print("Trying to sleepkill " .. app)
        hs.execute('/usr/bin/killall ' .. app)
      end
    end

    -- WiFi
    if (config.sleepkill.wifi) then
      hs.wifi.setPower(false)
    end

  --
  -- screensDidWake -- WAKEUP
  --
  elseif event == hs.caffeinate.watcher.screensDidWake then

    -- Bluetooth
    if (config.sleepstart.bluetooth) then
      bluetooth("on")
    end

    -- WiFi
    if (config.sleepstart.wifi) then
      hs.wifi.setPower(true)
    end
  end
end

local caffWatcher = hs.caffeinate.watcher.new(watcherCB)
caffWatcher:start()