-- All Data
local Menus = {}
local Components = {}
local Resources = {}

-- Current Data
local OpenedMenu = nil
local HoveredIndex = 1

function IsResourceStarted(resource)
  for _, v in pairs(Resources) do
    if v == resource then
      return true
    end
  end
  return false
end

function AddMenu(name, resource)
  local index = Utils.GenerateUUID()
  table.insert(Resources, resource)

  Menus[index] = {
    name = name,
    type = "base",
    components = {},
    resource = resource
  }

  return index
end

function AddSubMenu(name, menu, resource)
  local index = Utils.GenerateUUID()

  Menus[index] = {
    name = name,
    type = "submenu",
    components = {},
    parent = menu,
    resource = resource
  }

  table.insert(Menus[menu].components, {
    index = index,
    name = name,
    parent = menu,
    type = "submenu",
    resource = resource
  })

  return index
end

function AddButton(name, menu, callback, resource)
  local index = Utils.GenerateUUID()

  Components[index] = {
    name = name,
    type = "button",
    action = callback,
    resource = resource
  }

  table.insert(Menus[menu].components, {
    index = index,
    name = name,
    type = "button"
  })
end

function AddCheckbox(name, state, menu, callback, resource)
  local index = Utils.GenerateUUID()

  Components[index] = {
    name = name,
    type = "checkbox",
    action = callback,
    state = state,
    resource = resource
  }

  table.insert(Menus[menu].components, {
    index = index,
    name = name,
    type = "checkbox",
    state = state
  })
end

function AddList(name, menu, list, callback, resource)
  local index = Utils.GenerateUUID()

  Components[index] = {
    name = name,
    type = "list",
    action = callback,
    list = list,
    listIndex = 1,
    resource = resource
  }

  table.insert(Menus[menu].components, {
    index = index,
    name = name,
    type = "list",
    list = list,
    listIndex = 1
  })
end

function OpenMenu(menu, color)
  local menuToOpen = Menus[menu]
  if menuToOpen then
    SendNUIMessage({
      type = "open_menu",
      data = {
        name = menuToOpen.name,
        components = menuToOpen.components,
        option = 1,
        color = color
      }
    })
    OpenedMenu = menuToOpen
    HoveredIndex = 1
  end
end

function CloseMenu()
  if OpenedMenu then
    SendNUIMessage({ type = "close_menu" })
    OpenedMenu = nil
    HoveredIndex = 1
  end
end

function IsAnyMenuOpen()
  if OpenedMenu then
    return true
  end
  return false
end

function IsMenuOpened(menu)
  if OpenedMenu.index == menu then
    return true
  end
  return false
end

function GetOpenedMenu()
  if OpenedMenu then
    return OpenedMenu.index
  end
  return nil
end

function ClearMenu(menu)
  for k, v in pairs(Menus[menu].components) do
    local index = v.index
    for a = 1, #Components do
      table.remove(Components, index, 1)
    end
    table.remove(Menus[menu].components, k, 1)
  end
end

-- EXPORTS
exports("AddMenu", AddMenu)
exports("AddSubMenu", AddSubMenu)
exports("AddButton", AddButton)
exports("OpenMenu", OpenMenu)
exports("AddCheckbox", AddCheckbox)
exports("AddList", AddList)
exports("CloseMenu", CloseMenu)
exports("IsAnyMenuOpen", IsAnyMenuOpen)
exports("IsMenuOpen", IsMenuOpened)
exports("GetOpenedMenu", GetOpenedMenu)
exports("ClearMenu", ClearMenu)

-- EVENTS
AddEventHandler("onResourceStop", function(resource)
  if resource == GetCurrentResourceName() then return end
  if not IsResourceStarted(resource) then return end

  for k, v in pairs(Menus) do
    if v.resource == resource then
      Menus[k] = nil
    end
  end

  for k, v in pairs(Components) do
    if v.resource == resource then
      Components[k] = nil
    end
  end

  for k, v in pairs(Resources) do
    if v == resource then
      Resources[k] = nil
    end
  end

end)

-- CONTROLS
function GoUp()
  local prev = HoveredIndex - 1

  if prev < 1 then
    prev = #OpenedMenu.components
  end

  HoveredIndex = prev
  SendNUIMessage({
    type = "set_menu_option",
    data = {
      option = HoveredIndex
    }
  })
end

function GoDown()
  local next = HoveredIndex + 1

  if next > #OpenedMenu.components then
    next = 1
  end

  HoveredIndex = next
  SendNUIMessage({
    type = "set_menu_option",
    data = {
      option = HoveredIndex
    }
  })
end

function GoLeft()
  local selected = OpenedMenu.components[HoveredIndex]
  if selected then
    if selected.type == "list" then
      local comp = Components[selected.index]
      local next = selected.listIndex - 1

      if next < 1 then
        next = #selected.list
      end

      selected.listIndex = next
      comp.listIndex = next
      SendNUIMessage({
        type = "set_list_item",
        data = {
          index = selected.index,
          listIndex = next
        }
      })
    end
  end
end

function GoRight()
  local selected = OpenedMenu.components[HoveredIndex]
  if selected then
    if selected.type == "list" then
      local comp = Components[selected.index]
      local next = selected.listIndex + 1

      if next > #selected.list then
        next = 1
      end

      selected.listIndex = next
      comp.listIndex = next
      SendNUIMessage({
        type = "set_list_item",
        data = {
          index = selected.index,
          listIndex = next
        }
      })
    end
  end
end

function Enter()
  local selected = OpenedMenu.components[HoveredIndex]
  if selected then
    if selected.type == "submenu" then
      OpenMenu(selected.index)
    else
      local component = Components[selected.index]
      if selected.type == "checkbox" then
        local newState = not component.state
        component.state = newState
        selected.state = newState
        component.action(newState)
        SendNUIMessage({
          type = "set_checkbox_state",
          data = {
            id = selected.index,
            state = newState
          }
        })
      elseif selected.type == "list" then
        local comp = Components[selected.index]
        comp.action(comp.list[comp.listIndex])
      elseif selected.type == "button" then
        Components[selected.index].action()
      end
    end
  end
end

function Backspace()
  if OpenedMenu then
    if not OpenedMenu.parent then
      CloseMenu()
    else
      OpenMenu(OpenedMenu.parent)
    end
  end
end

Citizen.CreateThread(function()
  while true do
    Citizen.Wait(0)
    if OpenedMenu then

      if IsControlJustPressed(0, 172) then
        GoUp()
      elseif IsControlJustPressed(0, 173) then
        GoDown()
      elseif IsControlJustPressed(0, 174) then
        GoLeft()
      elseif IsControlJustPressed(0, 175) then
        GoRight()
      elseif IsControlJustPressed(0, 176) then
        Enter()
      elseif IsControlJustPressed(0, 177) then
        Backspace()
      end

    end
  end
end)