local fontColor
local BACKGROUNDS = {
  {
    title     = "Ancestral Knowledge",
    url       = "http://cloud-3.steamusercontent.com/ugc/1915746489207287888/2F9F6F211ED0F98E66C9D35D93221E4C7FB6DD3C/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Astronomical Atlas",
    url       = "http://cloud-3.steamusercontent.com/ugc/1754695853007989004/9153BC204FC707AE564ECFAC063A11CB8C2B5D1E/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Backpack",
    url       = "http://cloud-3.steamusercontent.com/ugc/2018212896278691928/F55BEFFC2540109C6333179532F583B367FF2EBC/",
    fontcolor = { 0, 0, 0 }
  },
  {
    title     = "Crystallizer of Dreams",
    url       = "http://cloud-3.steamusercontent.com/ugc/1915746489207280958/100F16441939E5E23818651D1EB5C209BF3125B9/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Diana Stanley",
    url       = "http://cloud-3.steamusercontent.com/ugc/1754695635919071208/1AB7222850201630826BFFBA8F2BD0065E2D572F/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Gloria Goldberg",
    url       = "http://cloud-3.steamusercontent.com/ugc/1754695635919102502/453D4426118C8A6DE2EA281184716E26CA924C84/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Sefina Rousseau",
    url       = "http://cloud-3.steamusercontent.com/ugc/1754695635919099826/3C3CBFFAADB2ACA9957C736491F470AE906CC953/",
    fontcolor = { 0, 0, 0 }
  },
  {
    title     = "Stick to the Plan",
    url       = "http://cloud-3.steamusercontent.com/ugc/2018214163838897493/8E38B96C5A8D703A59009A932432CBE21ABE63A2/",
    fontcolor = { 1, 1, 1 }
  },
  {
    title     = "Wooden Sledge",
    url       = "http://cloud-3.steamusercontent.com/ugc/1750192233783143973/D526236AAE16BDBB98D3F30E27BAFC1D3E21F4AC/",
    fontcolor = { 0, 0, 0 }
  }
}

-- save state and options to restore onLoad
function onSave() return JSON.encode({ cardsInBag, showCost, showIcons }) end

-- load variables and create context menu
function onLoad(savedData)
  local loadedData = JSON.decode(savedData)
  cardsInBag       = loadedData[1] or {}
  showCost         = loadedData[2] or true
  showIcons        = loadedData[3] or true
  fontColor        = getFontColor()
  recreateButtons()

  self.addContextMenuItem("Select image", selectImage)
  self.addContextMenuItem("Toggle cost", function(color)
    showCost = not showCost
    printToColor("Show cost of cards: " .. tostring(showCost), color, "White")
    refresh()
  end)

  self.addContextMenuItem("Toggle skill icons", function(color)
    showIcons = not showIcons
    printToColor("Show skill icons of cards: " .. tostring(showIcons), color, "White")
    refresh()
  end)
end

-- gets the font color based on background url
function getFontColor()
  local customInfo = self.getCustomObject()
  for i = 1, #BACKGROUNDS do
    if BACKGROUNDS[i].url == customInfo.diffuse then
      return BACKGROUNDS[i].fontcolor
    end
  end
  return { 1, 1, 1 }
end

-- called by context menu to change background image
function selectImage(color)
  -- generate list of options
  local options = {}
  for i = 1, #BACKGROUNDS do
    options[i] = BACKGROUNDS[i].title
  end

  -- prompt user to select option
  Player[color].showOptionsDialog("Select image:", options, 1, function(_, optionIndex)
    local customInfo = self.getCustomObject()
    customInfo.diffuse = BACKGROUNDS[optionIndex].url
    self.setCustomObject(customInfo)
    self.reload()
  end)
end

-- only allow cards to enter, split decks and reject other objects
function onObjectEnterContainer(container, object)
  if container ~= self then return end
  if object.tag == "Deck" then
    takeDeckOut(object.getGUID(), self.getPosition() + Vector(0, 0.1, 0))
  elseif object.tag ~= "Card" then
    broadcastToAll("The 'Attachment Helper' is meant to be used for cards.", "White")
  else
    findCard(object.getGUID(), object.getName(), object.getGMNotes())
    recreateButtons()
  end
end

-- takes the deck out and splits in into single cards
function takeDeckOut(guid, pos)
  local deck = self.takeObject({ guid = guid, position = pos, smooth = false })
  for i = 1, #deck.getObjects() do
    self.putObject(deck.takeObject({ position = pos + Vector(0, 0.1 * i, 0), smooth = false }))
  end
end

-- removes leaving cards from the "cardInBag" table
function onObjectLeaveContainer(container, object)
  if container == self then
    local guid = object.getGUID()
    local found = false
    for i, card in ipairs(cardsInBag) do
      if card.id == guid then
        table.remove(cardsInBag, i)
        found = true
        break
      end
    end

    if found ~= true then
      local name = object.getName()
      for i, card in ipairs(cardsInBag) do
        if card.name == name then
          table.remove(cardsInBag, i)
          break
        end
      end
    end
    recreateButtons()
  end
end

-- refreshes displayed buttons based on contained cards
function refresh()
  cardsInBag = {}
  for _, object in ipairs(self.getObjects()) do
    findCard(object.guid, object.name, object.gm_notes)
  end
  recreateButtons()
end

-- gets cost and icons for a card
function findCard(guid, name, GMNotes)
  local cost = ""
  local icons = {}
  local metadata = {}
  local displayName = name

  if displayName == nil or displayName == "" then displayName = "unnamed" end
  if showCost or showIcons then metadata = JSON.decode(GMNotes) end

  if showCost then
    if GMNotes ~= "" then cost = metadata.cost end
    if cost == nil or cost == "" then cost = "–" end
    displayName = "[" .. cost .. "] " .. displayName
  end

  if showIcons then
    if GMNotes ~= "" then
      icons[1] = metadata.wildIcons
      icons[2] = metadata.willpowerIcons
      icons[3] = metadata.intellectIcons
      icons[4] = metadata.combatIcons
      icons[5] = metadata.agilityIcons
    end

    local IconTypes = { "Wild", "Willpower", "Intellect", "Combat", "Agility" }
    local found = false
    for i = 1, 5 do
      if icons[i] ~= nil and icons[i] ~= "" then
        if found == false then
          displayName = displayName .. "\n" .. IconTypes[i] .. ": " .. icons[i]
          found = true
        else
          displayName = displayName .. " " .. IconTypes[i] .. ": " .. icons[i]
        end
      end
    end
  end
  table.insert(cardsInBag, { name = name, displayName = displayName, id = guid })
end

-- recreates buttons with up-to-date labels
function recreateButtons()
  self.clearButtons()
  local verticalPosition = 1.65

  for _, card in ipairs(cardsInBag) do
    local id = card.id
    local funcName = "removeCard" .. id
    self.setVar(funcName, function() removeCard(id) end)
    self.createButton({
      label          = card.displayName,
      click_function = funcName,
      function_owner = self,
      position       = { 0, -0.1, verticalPosition },
      height         = 200,
      width          = 1200,
      font_size      = string.len(card.displayName) > 20 and 75 or 100
    })
    verticalPosition = verticalPosition - 0.5
  end

  local countLabel = #cardsInBag
  local fontSize = 250
  if #cardsInBag == 0 then
    countLabel = "Attachment Helper"
    fontSize   = 150
  end

  self.createButton({
    label          = countLabel,
    click_function = "none",
    function_owner = self,
    position       = { 0, -0.1, -1.7 },
    height         = 0,
    width          = 0,
    font_size      = fontSize,
    font_color     = fontColor
  })
end

-- click-function for buttons to take a card out of the bag
function removeCard(cardGUID)
  self.takeObject({
    guid = cardGUID,
    rotation = self.getRotation(),
    position = self.getPosition() + Vector(0, 0.25, 0),
    callback_function = function(obj) obj.resting = true end
  })
end