FurCData.lua

local currentChar = FurC.CharacterName
local task = LibAsync:Create("FurnitureCatalogue_ScanDataFiles")

local lastLink = nil
local recipeArray = nil

local ver = FurC.Constants.Versioning
local src = FurC.Constants.ItemSources

local function getCurrentChar()
  currentChar = currentChar or zo_strformat(GetUnitName("player"))
  return currentChar
end

-- GetItemLinkItemId doesn't work the way I need it
-- ToDo: fix this, should only take one type of link (not nil, number, string, links)
local function getItemId(itemLink)
  if nil == itemLink or "" == itemLink then
    return
  end
  if type(itemLink) == "number" and itemLink > 9999 then
    return itemLink
  end
  local _, _, _, itemId = ZO_LinkHandler_ParseLink(itemLink)
  return tonumber(itemId)
end
FurC.GetItemId = getItemId

--- Get item link from id
--- @param itemId any
--- @return string link or empty string
-- ToDo: fix this, should only take one type of input
local function getItemLink(itemId)
  if nil == itemId or #tostring(itemId) < 4 then
    return ""
  end
  itemId = tostring(itemId)
  if #itemId > 55 then
    return itemId
  end
  return zo_strformat("|H1:item:<<1>>:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0|h|h", itemId)
end
FurC.GetItemLink = getItemLink

local function printItemLink(itemId)
  if nil == itemId then
    return
  end
  itemId = tostring(itemId)
  local itemLink = nil
  if #itemId > 55 then
    itemLink = itemId
  end
  itemLink = itemLink or zo_strformat("|H1:item:<<1>>:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0:0|h|h", itemId)
  FurC.Logger:Info("[%s] = '',\t\t-- %s", itemId, GetItemLinkName(itemLink))
end
FurC.PrintItemLink = printItemLink

local function addDatabaseEntry(recipeKey, recipeArray)
  if recipeKey and recipeArray and {} ~= recipeArray then
    if FurC.settings.data[recipeKey] ~= nil then
      for k, v in pairs(recipeArray) do
        FurC.settings.data[recipeKey][k] = v
      end
    else
      FurC.settings.data[recipeKey] = recipeArray
    end
  end
end

local function makeMaterial(recipeKey, recipeArray, tryPlaintext, forcePlaintext)
  if
    nil == recipeArray
    or (nil == recipeArray.blueprint and nil == recipeArray.recipeIndex and nil == recipeArray.recipeListIndex)
  then
    return "couldn't get material list, please re-scan character knowledge"
  end
  local ret = ""
  local ingredients = FurC.GetIngredients(recipeKey, recipeArray)
  forcePlaintext = forcePlaintext or tryPlaintext and NonContiguousCount(ingredients) > 4
  for ingredientLink, qty in pairs(ingredients) do
    -- auto-capitalize because for some reason the ZOS API doesn't
    local itemText = (
      forcePlaintext and string.gsub(" " .. GetItemLinkName(ingredientLink), "%W%l", string.upper):sub(2)
    ) or ingredientLink
    ret = zo_strformat("<<1>> <<2>>x <<3>>, ", ret, qty, itemText)
  end
  return ret:sub(0, -3)
end
FurC.GetMats = makeMaterial

function FurC.GetIngredients(itemLink, recipeArray)
  recipeArray = recipeArray or FurC.Find(itemLink)
  local ingredients = {}
  if {} ~= recipeArray and recipeArray.blueprint then
    local blueprintLink = FurC.GetItemLink(recipeArray.blueprint)
    numIngredients = GetItemLinkRecipeNumIngredients(blueprintLink)
    for ingredientIndex = 1, numIngredients do
      local name, _, qty = GetItemLinkRecipeIngredientInfo(blueprintLink, ingredientIndex)
      local ingredientLink = GetItemLinkRecipeIngredientItemLink(blueprintLink, ingredientIndex, LINK_STYLE_DEFAULT)
      ingredients[ingredientLink] = qty
    end
  else
    local _, name, numIngredients = GetRecipeInfo(recipeArray.recipeListIndex, recipeArray.recipeIndex)
    for ingredientIndex = 1, numIngredients do
      local name, _, qty =
        GetRecipeIngredientItemInfo(recipeArray.recipeListIndex, recipeArray.recipeIndex, ingredientIndex)
      local ingredientLink = GetRecipeIngredientItemLink(
        recipeArray.recipeListIndex,
        recipeArray.recipeIndex,
        ingredientIndex,
        LINK_STYLE_DEFAULT
      )
      ingredients[ingredientLink] = qty
    end
  end
  return ingredients
end

local function parseFurnitureItem(itemLink, override) -- saves to DB, returns recipeArray
  if
    not (override or IsItemLinkPlaceableFurniture(itemLink) or GetItemLinkItemType(itemLink) == ITEMTYPE_FURNISHING)
  then
    return
  end

  local recipeKey = GetItemLinkItemId(itemLink)
  local recipeArray = FurC.settings.data[recipeKey]
  if nil ~= recipeArray then
    return recipeArray
  end

  recipeArray = {}

  addDatabaseEntry(recipeKey, recipeArray)

  return recipeArray
end

local function parseBlueprint(blueprintLink) -- saves to DB, returns recipeArray
  local itemLink = GetItemLinkRecipeResultItemLink(blueprintLink, LINK_STYLE_BRACKETS)
  local blueprintId = getItemId(blueprintLink)
  local recipeKey = getItemId(itemLink)
  if
    nil == recipeKey -- we don't have a key to access the database
    or nil == itemLink -- we don't have an item link to parse
    or nil == GetItemLinkName(itemLink) -- we didn't find an item result for our recipe
  then
    return
  end

  local recipeArray = FurC.settings.data[recipeKey] or {}
  recipeArray.origin = recipeArray.origin or src.CRAFTING
  recipeArray.craftingSkill = recipeArray.craftingSkill or GetItemLinkCraftingSkillType(blueprintLink)
  recipeArray.blueprint = recipeArray.blueprint or getItemId(blueprintLink)

  if IsItemLinkRecipeKnown(blueprintLink) then
    local recipeChars = recipeArray.characters
    if nil ~= recipeChars then
      recipeChars[FurC.CharacterName] = true
    else
      recipeArray.characters = {}
      recipeArray.characters[FurC.CharacterName] = true
    end
  end
  addDatabaseEntry(recipeKey, recipeArray)
  return recipeArray
end

---sets recipeArray, returns it
---@param itemOrBlueprintLink any
---@return table
function FurC.Find(itemOrBlueprintLink)
  if tonumber(itemOrBlueprintLink) == itemOrBlueprintLink then
    itemOrBlueprintLink = FurC.GetItemLink(itemOrBlueprintLink)
  end
  -- do not return empty arrays. If this returns nil, abort!
  if nil == itemOrBlueprintLink or #itemOrBlueprintLink == 0 then
    return {}
  end

  if itemOrBlueprintLink == lastLink and nil ~= recipeArray then
    return recipeArray
  else
    recipeArray = nil
    lastLink = itemOrBlueprintLink
  end

  if IsItemLinkFurnitureRecipe(itemOrBlueprintLink) then
    recipeArray = parseBlueprint(itemOrBlueprintLink)
  elseif IsItemLinkPlaceableFurniture(itemOrBlueprintLink) then
    recipeArray = parseFurnitureItem(itemOrBlueprintLink)
  else
    itemId = getItemId(itemOrBlueprintLink)
    if itemId ~= nil and tonumber(itemId) > 0 and FurC.settings.data ~= nil then
      recipeArray = FurC.settings.data[itemId]
    end
  end

  return recipeArray or {}
end

function FurC.Delete(itemOrBlueprintLink) -- sets recipeArray, returns it - calls scanItemLink
  local recipeArray = FurC.GetItemId(itemOrBlueprintLink)
  if nil == recipeArray then
    return
  end
  if nil ~= recipeArray.itemId then
    FurC.settings.data[recipeArray.itemId] = nil
  end
end

function FurC.GetEntry(itemOrBlueprintLink)
  local itemLink = (IsItemLinkFurnitureRecipe(itemOrBlueprintLink) and GetRecipeResultItemLink(itemOrBlueprintLink))
    or itemOrBlueprintLink
  local recipeArray = FurC.Find(itemLink)
  FurC.Logger:Debug("Trying to get entry for %s: %s", itemLink, recipeArray)
  if not recipeArray then
    return
  end
  local itemId = getItemId(itemOrBlueprintLink)
  if recipeArray.blueprint then
    itemId = getItemId(GetItemLinkRecipeResultItemLink(blueprintLink))
  end
  return itemId, recipeArray
end

function FurC.IsFavorite(itemLink, recipeArray)
  recipeArray = recipeArray or FurC.Find(itemLink)
  return recipeArray.favorite
end

-- ToDo: move to fave reference table?
function FurC.Fave(itemLink, recipeArray)
  recipeArray = recipeArray or FurC.Find(itemLink)
  recipeArray.favorite = not recipeArray.favorite
  if not recipeArray.favorite then
    recipeArray.favorite = nil
  end

  FurC.UpdateGui()
end

local function scanRecipeIndices(recipeListIndex, recipeIndex) -- returns recipeArray or nil, initialises
  local itemLink = GetRecipeResultItemLink(recipeListIndex, recipeIndex, LINK_STYLE_BRACKETS)
  if nil == itemLink or #itemLink == 0 or not IsItemLinkPlaceableFurniture(itemLink) then
    return
  end

  local recipeKey = getItemId(itemLink)

  local recipeArray = FurC.settings.data[recipeKey] or {}
  recipeArray.origin = src.CRAFTING
  recipeArray.version = recipeArray.version or 2
  recipeArray.recipeListIndex = recipeArray.recipeListIndex or recipeListIndex
  recipeArray.recipeIndex = recipeArray.recipeIndex or recipeIndex

  recipeArray.characters = recipeArray.characters or {}

  if GetRecipeInfo(recipeListIndex, recipeIndex) then
    recipeArray.characters[getCurrentChar()] = true
    FurC.settings.accountCharacters = FurC.settings.accountCharacters or {}
    FurC.settings.accountCharacters[getCurrentChar()] = FurC.settings.accountCharacters[getCurrentChar()] or true
  end

  addDatabaseEntry(recipeKey, recipeArray)
  return recipeArray
end

function FurC.TryCreateRecipeEntry(recipeListIndex, recipeIndex) -- returns scanRecipeIndices, called from Events.lua
  return scanRecipeIndices(recipeListIndex, recipeIndex)
end

function FurC.IsAccountKnown(recipeKey, recipeArray)
  if recipeKey == nil and recipeArray == nil then
    return false
  end
  recipeArray = recipeArray or FurC.settings.data[recipeKey]
  return not (nil == recipeArray or nil == recipeArray.characters or NonContiguousCount(recipeArray.characters) == 0)
end

function FurC.CanCraft(recipeKey, recipeArray)
  if recipeKey == nil and recipeArray == nil then
    return false
  end
  recipeArray = recipeArray or FurC.settings.data[recipeKey]
  if FurC.IsAccountKnown(recipeKey, recipeArray) then
    return recipeArray.characters[getCurrentChar()]
  end
  return false
end

function FurC.GetCraftingSkillType(recipeKey, recipeArray)
  local itemLink = FurC.GetItemLink(recipeKey)
  local craftingSkillType = GetItemLinkCraftingSkillType(itemLink)

  if 0 == craftingSkillType and recipeArray.blueprint then
    craftingSkillType = GetItemLinkRecipeCraftingSkillType(FurC.GetItemLink(recipeArray.blueprint))
  elseif 0 == craftingSkillType and recipeArray.recipeListIndex and recipeArray.recipeIndex then
    _, _, _, _, _, _, craftingSkillType = GetRecipeInfo(recipeArray.recipeListIndex, recipeArray.recipeIndex)
  end

  return craftingSkillType
end

local function scanCharacter()
  local listName, numRecipes
  for recipeListIndex = 1, GetNumRecipeLists() do
    listName, numRecipes = GetRecipeListInfo(recipeListIndex)
    for recipeIndex = 1, numRecipes do
      scanRecipeIndices(recipeListIndex, recipeIndex) --  returns true on success
    end
  end
  FurC.Logger:Debug(GetString(SI_FURC_DEBUG_CHARSCANCOMPLETE))
end
FurC.ScanCharacter = scanCharacter

function FurC.RescanRumourRecipes()
  local function rescan()
    for itemId, recipeArray in pairs(FurC.settings.data) do
      if recipeArray.source == src.RUMOUR then
        local itemLink = recipeArray[itemLink]
        if not FurC.RumourRecipes[itemLink] then
          recipeArray.source = src.CRAFTING
          recipeArray.origin = nil
        end
      end
    end
  end

  if nil ~= task then
    task:Call(rescan):Then(FurC.UpdateGui)
  else
    rescan()
    FurC.UpdateGui()
  end
end

local recipeArray
local function scanFromFiles(shouldScanCharacter)
  local function parseZoneData(zoneName, zoneData, versionNumber, origin)
    for vendorName, vendorData in pairs(zoneData) do
      for itemId, itemData in pairs(vendorData) do
        recipeArray = parseFurnitureItem(FurC.GetItemLink(itemId), true)
        if not recipeArray then
          FurC.Logger:Debug("Error when scanning %s", itemId)
        else
          recipeArray.origin = origin
          recipeArray.version = versionNumber
          addDatabaseEntry(itemId, recipeArray)
        end
      end
    end
  end

  local function scanRecipeFile()
    local recipeKey, recipeArray

    local function makeKeySet(versionData)
      local keySet = {}
      for k, v in pairs(versionData) do
        table.insert(keySet, k)
      end
      return keySet
    end

    local function scanArray(ary, versionNumber, origin)
      if nil == ary then
        return
      end

      for _, recipeId in ipairs(ary) do
        local recipeLink = FurC.GetItemLink(recipeId)
        local itemLink = GetItemLinkRecipeResultItemLink(recipeLink) or FurC.GetItemLink(recipeId)
        recipeArray = FurC.Find(itemLink) or parseBlueprint(recipeLink) or parseFurnitureItem(itemLink)
        local recipeListIndex, recipeIndex = GetItemLinkGrantedRecipeIndices(recipeLink)
        if nil == recipeArray then
          FurC.Logger:Debug("scanRecipeFile: error for ID %s - %s", recipeId, itemLink)
        else
          recipeKey = getItemId(itemLink)
          recipeArray.version = versionNumber
          if not recipeArray.origin or origin ~= src.RUMOUR then
            recipeArray.origin = origin
          end
          recipeArray.blueprint = recipeId
          addDatabaseEntry(recipeKey, recipeArray)
        end
      end
    end

    for versionNumber, versionData in pairs(FurC.Recipes) do
      scanArray(versionData, versionNumber, src.CRAFTING)
    end

    for versionNumber, versionData in pairs(FurC.RolisRecipes) do
      scanArray(makeKeySet(versionData), versionNumber, src.CRAFTING)
    end

    for versionNumber, versionData in pairs(FurC.FaustinaRecipes) do
      scanArray(makeKeySet(versionData), versionNumber, src.CRAFTING)
    end
  end

  local function scanRolis()
    for versionNumber, versionData in pairs(FurC.Rolis) do
      for itemId, itemSource in pairs(versionData) do
        recipeArray = parseFurnitureItem(FurC.GetItemLink(itemId), true)
        if nil ~= recipeArray then
          recipeArray.version = versionNumber
          recipeArray.origin = src.ROLIS
          addDatabaseEntry(itemId, recipeArray)
        end
      end
    end
    for versionNumber, versionData in pairs(FurC.Faustina) do
      for itemId, itemSource in pairs(versionData) do
        recipeArray = parseFurnitureItem(FurC.GetItemLink(itemId), true)
        if nil ~= recipeArray then
          recipeArray.version = versionNumber
          recipeArray.origin = src.ROLIS
          addDatabaseEntry(itemId, recipeArray)
        end
      end
    end
  end

  local function scanFestivalFiles()
    for versionNumber, versionData in pairs(FurC.EventItems) do
      for eventName, eventData in pairs(versionData) do
        for eventItemSource, eventItemData in pairs(eventData) do
          for itemId, _ in pairs(eventItemData) do
            recipeArray = {}
            recipeArray.craftable = false
            recipeArray.version = versionNumber
            recipeArray.origin = src.FESTIVAL_DROP
            addDatabaseEntry(itemId, recipeArray)
          end
        end
      end
    end
  end

  local function scanMiscItemFile()
    for versionNumber, versionData in pairs(FurC.MiscItemSources) do
      for origin, originData in pairs(versionData) do
        for itemId, itemSource in pairs(originData) do
          local itemLink = FurC.GetItemLink(itemId)
          recipeArray = parseFurnitureItem(FurC.GetItemLink(itemId))
          if nil ~= recipeArray then
            recipeArray.version = versionNumber

            -- make sure that we don't set rumour source from data file.
            if not recipeArray.origin or origin ~= src.RUMOUR or recipeArray.origin == src.RUMOUR then
              recipeArray.origin = origin
            end -- 3.5: moved src.RUMOUR to beginning of table so it'll get overwritten

            addDatabaseEntry(itemId, recipeArray)
          else
            if origin == src.RUMOUR then
              FurC.Logger:Debug("invalid rumour item: %s (%s)", itemId, FurC.GetItemLink(itemId))
            else
              FurC.Logger:Debug("scanMiscItemFile: Error when scanning item ID %s (origin %s)", itemId, origin)
            end
          end
        end
      end
    end
  end

  local function scanVendorFiles()
    FurC.InitAchievementVendorList()
    local recipeKey, recipeArray, itemSource

    for versionNumber, versionData in pairs(FurC.AchievementVendors) do
      for zoneName, zoneData in pairs(versionData) do
        parseZoneData(zoneName, zoneData, versionNumber, src.VENDOR)
      end
    end

    for versionNumber, vendorData in pairs(FurC.LuxuryFurnisher) do
      for itemId, itemData in pairs(vendorData) do
        local recipeArray = {}

        recipeArray.origin = src.LUXURY
        recipeArray.version = versionNumber
        addDatabaseEntry(itemId, recipeArray)
      end
    end

    for versionNumber, versionData in pairs(FurC.PVP) do
      for zoneName, zoneData in pairs(versionData) do
        parseZoneData(zoneName, zoneData, versionNumber, src.PVP)
      end
    end
  end

  local function scanRumours()
    for versionNumber, items in pairs(FurC.Rumours) do
      for itemId, itemSource in pairs(items) do
        recipeArray = parseFurnitureItem(FurC.GetItemLink(itemId), true)
        if nil ~= recipeArray then
          recipeArray.version = versionNumber
          recipeArray.origin = src.RUMOUR
          addDatabaseEntry(itemId, recipeArray)
        end
      end
    end
    for index, blueprintId in pairs(FurC.RumourRecipes) do
      local blueprintLink = FurC.GetItemLink(blueprintId)
      local itemLink = GetItemLinkRecipeResultItemLink(blueprintLink, LINK_STYLE_BRACKETS)
      if #itemLink == 0 then
        itemLink = blueprintLink
      end
      local itemId = getItemId(itemLink)
      recipeArray = parseBlueprint(blueprintLink) or parseFurnitureItem(itemLink) or {}
      if blueprintId ~= itemId then
        recipeArray.blueprint = blueprintId
      end
      recipeArray.recipeListIndex, recipeArray.recipeIndex = GetItemLinkGrantedRecipeIndices(blueprintLink)
      recipeArray.origin = src.RUMOUR
      recipeArray.verion = recipeArray.version or ver.HOMESTEAD
      addDatabaseEntry(itemId, recipeArray)
    end
  end

  local function scanCharacterOrMaybeNot()
    if shouldScanCharacter then
      scanCharacter()
    end
  end

  FurC.IsLoading(true)

  if nil ~= task then
    task
      :Call(scanRecipeFile)
      :Then(scanMiscItemFile)
      :Then(scanVendorFiles)
      :Then(scanRolis)
      :Then(scanFestivalFiles)
      :Then(scanCharacterOrMaybeNot)
      :Then(scanRumours)
      :Then(FurC.UpdateGui)
  else
    scanRecipeFile()
    scanMiscItemFile()
    scanVendorFiles()
    scanRolis()
    scanFestivalFiles()
    scanCharacterOrMaybeNot()
    scanRumours()
    FurC.UpdateGui()
  end
end
FurC.ScanFromFiles = scanFromFiles

local function getScanFromFiles()
  if FurC.settings.version < FurC.version then
    FurC.settings.version = FurC.version
    return true
  end

  return FurC.settings.data == {}
end

local function getScanCharacter()
  if nil == FurC.settings.accountCharacters[FurC.CharacterName] then
    FurC.settings.accountCharacters[FurC.CharacterName] = false
    return true
  end
end

function FurC.ScanRecipes(shouldScanFiles, shouldScanCharacter) -- returns database
  shouldScanFiles = shouldScanFiles or getScanFromFiles()
  shouldScanCharacter = (shouldScanCharacter or getScanCharacter())
  if shouldScanFiles then
    FurC.Logger:Debug(GetString(SI_FURC_VERBOSE_SCANNING_DATA_FILE))
    scanFromFiles(shouldScanCharacter)
  elseif shouldScanCharacter then
    FurC.Logger:Debug(GetString(SI_FURC_VERBOSE_SCANNING_CHARS))
    scanCharacter()
  end
end

function FurC.GetItemDescription(recipeKey, recipeArray, stripColor, attachItemLink)
  recipeKey = getItemId(recipeKey)
  FurC.settings.emptyItemSources = FurC.settings.emptyItemSources or {}
  recipeArray = recipeArray or FurC.Find(recipeKey)
  if {} == recipeArray then
    return ""
  end

  local origin = recipeArray.origin
  if origin == src.CRAFTING or origin == src.WRIT_VENDOR then
    return FurC.GetMats(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.ROLIS then
    return FurC.getRolisSource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.LUXURY then
    return FurC.getLuxurySource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.GUILDSTORE then
    return GetString(SI_FURC_SEEN_IN_GUILDSTORE)
  end
  if origin == src.VENDOR then
    return FurC.getAchievementVendorSource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.FESTIVAL_DROP then
    return FurC.getEventDropSource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.PVP then
    return FurC.getPvpSource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  if origin == src.RUMOUR then
    return FurC.getRumourSource(recipeKey, recipeArray, stripColor, attachItemLink)
  end
  return FurC.getMiscItemSource(recipeKey, recipeArray, stripColor, attachItemLink)
end

function FurC.ShouldBeInFurC(link)
  link = FurC.GetItemLink(link)
  if not link then
    return false
  end

  if IsItemLinkPlaceableFurniture(link) then
    return nil == FurC.settings.data[getItemId(link)]
  end

  -- if not IsItemLinkFurnitureRecipe(link) then	return false end

  local resultLink = GetItemLinkRecipeResultItemLink(link, LINK_STYLE_BRACKETS)
  if not resultLink then
    return false
  end

  local resultId = getItemId(resultLink)
  local recipeId = getItemId(link)
  if not resultId or not recipeId or not IsItemLinkPlaceableFurniture(resultLink) then
    return false
  end

  for _, versionData in pairs(FurC.Recipes) do
    for _, id in ipairs(versionData) do
      if id == recipeId then
        return false
      end
    end
  end
  for _, versionData in pairs(FurC.FaustinaRecipes) do
    if versionData[recipeId] then
      return false
    end
  end
  for _, versionData in pairs(FurC.RolisRecipes) do
    if versionData[recipeId] then
      return false
    end
  end

  -- yeah okay, it should actually return false, but this is a util function for datamining
  return nil == FurC.settings.data[resultId]
end