-- ------------------------------------------------------------------------------ --
--                                TradeSkillMaster                                --
--                          https://tradeskillmaster.com                          --
--    All Rights Reserved - Detailed license information included with addon.     --
-- ------------------------------------------------------------------------------ --

local _, TSM = ...
local SavedSearches = TSM.Shopping:NewPackage("SavedSearches")
local Log = TSM.Include("Util.Log")
local Database = TSM.Include("Util.Database")
local TempTable = TSM.Include("Util.TempTable")
local Settings = TSM.Include("Service.Settings")
local private = {
	settings = nil,
	db = nil,
}
local MAX_RECENT_SEARCHES = 2000



-- ============================================================================
-- Module Functions
-- ============================================================================

function SavedSearches.OnInitialize()
	private.settings = Settings.NewView()
		:AddKey("global", "userData", "savedShoppingSearches")

	-- remove duplicates
	local seen = TempTable.Acquire()
	for i = #private.settings.savedShoppingSearches.filters, 1, -1 do
		local filter = private.settings.savedShoppingSearches.filters[i]
		local filterLower = strlower(private.settings.savedShoppingSearches.filters[i])
		if seen[filterLower] then
			tremove(private.settings.savedShoppingSearches.filters, i)
			private.settings.savedShoppingSearches.name[filter] = nil
			private.settings.savedShoppingSearches.isFavorite[filter] = nil
		else
			seen[filterLower] = true
		end
	end
	TempTable.Release(seen)

	-- remove old recent searches
	local remainingRecentSearches = MAX_RECENT_SEARCHES
	local numRemoved = 0
	for i = #private.settings.savedShoppingSearches.filters, 1, -1 do
		local filter = private.settings.savedShoppingSearches.filters
		if not private.settings.savedShoppingSearches.isFavorite[filter] then
			if remainingRecentSearches > 0 then
				remainingRecentSearches = remainingRecentSearches - 1
			else
				tremove(private.settings.savedShoppingSearches.filters, i)
				private.settings.savedShoppingSearches.name[filter] = nil
				numRemoved = numRemoved + 1
			end
		end
	end
	if numRemoved > 0 then
		Log.Info("Removed %d old recent searches", numRemoved)
	end

	private.db = Database.NewSchema("SHOPPING_SAVED_SEARCHES")
		:AddUniqueNumberField("index")
		:AddStringField("name")
		:AddBooleanField("isFavorite")
		:AddStringField("filter")
		:AddIndex("index")
		:AddIndex("name")
		:Commit()
	private.RebuildDB()
end

function SavedSearches.CreateRecentSearchesQuery()
	return private.db:NewQuery()
		:OrderBy("index", false)
end

function SavedSearches.CreateFavoriteSearchesQuery()
	return private.db:NewQuery()
		:Equal("isFavorite", true)
		:OrderBy("name", true)
end

function SavedSearches.SetSearchIsFavorite(dbRow, isFavorite)
	local filter = dbRow:GetField("filter")
	private.settings.savedShoppingSearches.isFavorite[filter] = isFavorite or nil
	dbRow:SetField("isFavorite", isFavorite)
		:Update()
end

function SavedSearches.RenameSearch(dbRow, newName)
	local filter = dbRow:GetField("filter")
	private.settings.savedShoppingSearches.name[filter] = newName ~= filter and newName or nil
	dbRow:SetField("name", newName)
		:Update()
end

function SavedSearches.DeleteSearch(dbRow)
	local index, filter = dbRow:GetFields("index", "filter")
	tremove(private.settings.savedShoppingSearches.filters, index)
	private.settings.savedShoppingSearches.name[filter] = nil
	private.settings.savedShoppingSearches.isFavorite[filter] = nil
	private.RebuildDB()
end

function SavedSearches.RecordFilterSearch(filter)
	for i, existingFilter in ipairs(private.settings.savedShoppingSearches.filters) do
		if strlower(existingFilter) == strlower(filter) then
			-- move this to the end of the list and rebuild the DB
			-- insert the existing filter so we don't need to update the isFavorite and name tables
			tremove(private.settings.savedShoppingSearches.filters, i)
			tinsert(private.settings.savedShoppingSearches.filters, existingFilter)
			private.RebuildDB()
			return
		end
	end

	-- didn't find an existing entry, so add a new one
	tinsert(private.settings.savedShoppingSearches.filters, filter)
	private.db:NewRow()
		:SetField("index", #private.settings.savedShoppingSearches.filters)
		:SetField("name", filter)
		:SetField("isFavorite", false)
		:SetField("filter", filter)
		:Create()
end



-- ============================================================================
-- Private Helper Functions
-- ============================================================================

function private.RebuildDB()
	private.db:TruncateAndBulkInsertStart()
	for index, filter in ipairs(private.settings.savedShoppingSearches.filters) do
		local name = private.settings.savedShoppingSearches.name[filter] or filter
		local isFavorite = private.settings.savedShoppingSearches.isFavorite[filter] and true or false
		private.db:BulkInsertNewRow(index, name, isFavorite, filter)
	end
	private.db:BulkInsertEnd()
end