local wait4Rez = require('wait4rez')

-- add following call into your bot loop
wait4Rez()

--- @type Mq
local mq = require('mq')

local Wait4RezStates = {
  Idle = "IDLE",
  Waiting4Rez = "WAITING4REZ",
  Looting = "LOOTING",
}

local state = Wait4RezStates.Idle

local function log (message, ...)
  local logMessage = string.format(message, ...)
  print(logMessage)
end

local function diedEvent()
        log("%s died, awaiting rezs.", mq.TLO.Me.Name())

  mq.cmd("/beep")
  state = Wait4RezStates.Waiting4Rez
end

mq.event("slain", "You have been slain by #*#", diedEvent)
mq.event("died", "You died.", diedEvent)

local function doEvents()
  mq.doevents("slain")
  mq.doevents("died")
end

local function ensureTarget(targetId)
  if not targetId then
    log("Invalid <targetId>")
    return false
  end

  if mq.TLO.Target.ID() ~= targetId then
    if mq.TLO.SpawnCount("id "..targetId)() > 0 then
      mq.cmdf("/mqtarget id %s", targetId)
      mq.delay("3s", function() return mq.TLO.Target.ID() == targetId end)
    else
      log("EnsureTarget has no spawncount for target id <%d>", targetId)
    end
  end

  return mq.TLO.Target.ID() == targetId
end

local function waitToZone()
        log("Waiting to zone.")

  local me = mq.TLO.Me.Name()
  repeat
    mq.delay(100)
  until mq.TLO.Spawn(me.."'s").ID()

  mq.delay(500)

  log("Completed zoneing to corpse.")
end

local function doLoot()
  local me = mq.TLO.Me.Name()
  if ensureTarget(mq.TLO.Spawn(me.."'s").ID()) then
    log("Corpse distance <%s>", mq.TLO.Target.Distance())
    if mq.TLO.Target.Distance() < 100 then
      while mq.TLO.Target.Distance() > 15 do
        mq.cmd("/corpse")
        mq.delay(20)
      end

      mq.cmd("/loot")
      mq.delay("5s", function() return mq.TLO.Window("LootWnd") and mq.TLO.Window("LootWnd").Open() end)
      mq.delay("5s", function() return mq.TLO.Corpse.Items() ~= nil end)
      mq.delay(500)
      if not mq.TLO.Window("LootWnd") or not mq.TLO.Corpse.Items then
        log("Could not open loot window.")
      else
        mq.cmd("/notify LootWnd LootAllButton leftmouseup")
        mq.delay("30s", function() return not mq.TLO.Window("LootWnd").Open() end)
      end
    else
      log("Corpse out of range. Could not loot.")
    end
  end
  state = Wait4RezStates.Idle
end

local function doWait4Rez()
  mq.cmd("/bc Ready for rezz.")
  mq.cmd("/consent guild")

  repeat
    mq.delay(10)
  until mq.TLO.Window("ConfirmationDialogBox").Open()

  mq.cmd("/nomodkey /notify ConfirmationDialogBox Yes_Button leftmouseup")
  waitToZone()
  doLoot()
  state = Wait4RezStates.Idle
  log("\ag[SUCCESS]\axRezzed, looted corpse and ready for action.")
end


local function createAliases()
  mq.unbind('/wait4rez')
  mq.unbind('/waitforrez')
  mq.unbind('/dead')
  mq.unbind('/lootCorpse')
  mq.bind("/wait4rez", function() state = Wait4RezStates.Waiting4Rez end)
  mq.bind("/waitforrez", function() state = Wait4RezStates.Waiting4Rez end)
  mq.bind("/dead", function() state = Wait4RezStates.Waiting4Rez end)
  mq.bind("/lootCorpse", function() state = Wait4RezStates.Looting end)
end

createAliases()

local function wait4RezStateMachine()
  doEvents()

  if state == Wait4RezStates.Idle then
    return
  end

  if state == Wait4RezStates.Waiting4Rez then
    doWait4Rez()
  end

  if state == Wait4RezStates.Looting then
    doLoot()
  end
end

return wait4RezStateMachine