local MER, F, E, I, V, P, G, L = unpack(ElvUI_MerathilisUI)
F.String = {}

local error = error
local type, unpack = type, unpack
local char = string.char
local strbyte, strfind, strlen, strsub = strbyte, strfind, strlen, strsub
local utf8len, utf8lower, utf8sub, utf8upper = string.utf8len, string.utf8lower, string.utf8sub, string.utf8upper
local tinsert = tinsert

function F.String.Color(msg, color)
	if type(color) == "string" then
		if #color == 8 then
			return "|c" .. color .. msg .. "|r"
		else
			return "|cff" .. color .. msg .. "|r"
		end
	else
		return "|cff" .. I.Strings.Colors[color] .. msg .. "|r"
	end
end

function F.String.HexToRGB(hex)
	local r, g, b, a = strmatch(hex, "^#?(%x%x)(%x%x)(%x%x)(%x?%x?)$")
	if not r then
		return 0, 0, 0, nil
	end
	return tonumber(r, 16) / 255,
		tonumber(g, 16) / 255,
		tonumber(b, 16) / 255,
		(a ~= "") and (tonumber(a, 16) / 255) or nil
end

function F.String.CharBytes(s, i)
	-- argument defaults
	i = i or 1

	-- argument checking
	if type(s) ~= "string" then
		error("bad argument #1 to 'F.CharBytes' (string expected, got " .. type(s) .. ")")
	end
	if type(i) ~= "number" then
		error("bad argument #2 to 'F.CharBytes' (number expected, got " .. type(i) .. ")")
	end

	local c = strbyte(s, i)

	-- determine bytes needed for character, based on RFC 3629
	-- validate byte 1
	if c > 0 and c <= 127 then
		-- UTF8-1
		return 1
	elseif c >= 194 and c <= 223 then
		-- UTF8-2
		local c2 = strbyte(s, i + 1)

		if not c2 then
			error("UTF-8 string terminated early")
		end

		-- validate byte 2
		if c2 < 128 or c2 > 191 then
			error("Invalid UTF-8 character")
		end

		return 2
	elseif c >= 224 and c <= 239 then
		-- UTF8-3
		local c2 = strbyte(s, i + 1)
		local c3 = strbyte(s, i + 2)

		if not c2 or not c3 then
			error("UTF-8 string terminated early")
		end

		-- validate byte 2
		if c == 224 and (c2 < 160 or c2 > 191) then
			error("Invalid UTF-8 character")
		elseif c == 237 and (c2 < 128 or c2 > 159) then
			error("Invalid UTF-8 character")
		elseif c2 < 128 or c2 > 191 then
			error("Invalid UTF-8 character")
		end

		-- validate byte 3
		if c3 < 128 or c3 > 191 then
			error("Invalid UTF-8 character")
		end

		return 3
	elseif c >= 240 and c <= 244 then
		-- UTF8-4
		local c2 = strbyte(s, i + 1)
		local c3 = strbyte(s, i + 2)
		local c4 = strbyte(s, i + 3)

		if not c2 or not c3 or not c4 then
			error("UTF-8 string terminated early")
		end

		-- validate byte 2
		if c == 240 and (c2 < 144 or c2 > 191) then
			error("Invalid UTF-8 character")
		elseif c == 244 and (c2 < 128 or c2 > 143) then
			error("Invalid UTF-8 character")
		elseif c2 < 128 or c2 > 191 then
			error("Invalid UTF-8 character")
		end

		-- validate byte 3
		if c3 < 128 or c3 > 191 then
			error("Invalid UTF-8 character")
		end

		-- validate byte 4
		if c4 < 128 or c4 > 191 then
			error("Invalid UTF-8 character")
		end

		return 4
	else
		error("Invalid UTF-8 character")
	end
end

function F.String.Replace(s, mapping)
	-- argument checking
	if type(s) ~= "string" then
		error("bad argument #1 to 'F.Replace' (string expected, got " .. type(s) .. ")")
	end
	if type(mapping) ~= "table" then
		error("bad argument #2 to 'F.Replace' (table expected, got " .. type(mapping) .. ")")
	end

	local pos = 1
	local bytes = strlen(s)
	local charbytes
	local newstr = ""

	while pos <= bytes do
		charbytes = F.String.CharBytes(s, pos)
		local c = strsub(s, pos, pos + charbytes - 1)

		newstr = newstr .. (mapping[c] or c)

		pos = pos + charbytes
	end

	return newstr
end

function F.String.Split(subject, delimiter)
	if not subject or subject == "" then
		return {}
	end

	local length = strlen(delimiter)
	local results = {}

	local i = 0
	local j = 0

	while true do
		j = strfind(subject, delimiter, i + length)
		if strlen(subject) == i then
			break
		end

		if j == nil then
			tinsert(results, strsub(subject, i))
			break
		end

		tinsert(results, strsub(subject, i, j - 1))
		i = j + length
	end

	return unpack(results)
end

function F.String.FastRGB(r, g, b)
	return format("%02x%02x%02x", r * 255, g * 255, b * 255)
end

function F.String.FastRGBA(r, g, b, a)
	return format("%02x%02x%02x%02x", (a or 1) * 255, r * 255, g * 255, b * 255)
end

function F.String.RGB(msg, colors)
	if colors.r then
		return F.String.Color(msg, F.String.FastRGB(colors.r, colors.g, colors.b))
	else
		return F.String.Color(msg, F.String.FastRGB(colors[1], colors[2], colors[3]))
	end
end

function F.String.Uppercase(text)
	if type(text) ~= "string" then
		return text
	end
	return utf8upper(text)
end

function F.String.Lowercase(text)
	if type(text) ~= "string" then
		return text
	end
	return utf8lower(text)
end

function F.String.UppercaseFirstLetter(text)
	if type(text) ~= "string" then
		return text
	end
	return utf8upper(utf8sub(text, 1, 1)) .. utf8sub(text, 2)
end

function F.String.UppercaseFirstLetterOnly(text)
	if type(text) ~= "string" then
		return text
	end
	return utf8upper(utf8sub(text, 1, 1)) .. utf8lower(utf8sub(text, 2))
end

function F.String.LowercaseEnum(text)
	if type(text) ~= "string" then
		return text
	end
	return strtrim(text):gsub("_", " "):gsub("(%a)([%w_']*)", function(a, b)
		return F.String.Uppercase(a) .. F.String.Lowercase(b)
	end)
end

function F.String.ColorFirstLetter(text)
	if type(text) ~= "string" then
		return text
	end

	return F.String.MERATHILISUI(utf8upper(utf8sub(text, 1, 1))) .. "|cfff5feff" .. utf8sub(text, 2) .. "|r"
end

function F.String.StripTexture(text)
	if type(text) ~= "string" then
		return text
	end
	return gsub(text, "(%s?)(|?)|[TA].-|[ta](%s?)", function(w, x, y)
		if x == "" then
			return (w ~= "" and w) or (y ~= "" and y) or ""
		end
	end)
end

function F.String.StripColor(text)
	if type(text) ~= "string" then
		return text
	end

	-- Remove |c...|r format
	text = text:gsub("|c%x%x%x%x%x%x%x%x(.-)|r", "%1")
	-- Remove |cn...: format
	text = text:gsub("|cn.-:(.-)|r", "%1")
	-- Remove any remaining |r
	text = text:gsub("|r", "")
	-- Remove any remaining |
	text = text:gsub("|", "")

	return text
end

function F.String.Strip(text)
	if type(text) ~= "string" then
		return text
	end
	return F.String.StripColor(F.String.StripTexture(text))
end

function F.String.Trim(text)
	return strmatch(text, "^%s*(.*%S)") or ""
end

function F.String.RemoveRuneOfThePrefix(text)
	return text:gsub(".* the ", ""):gsub(".* of ", "")
end

function F.String.RemoveEveryOfTheAndEverythingAfter(text)
	return text:gsub(" of (.*)", ""):gsub(" the (.*)", "")
end

-- Capture the following strings
-- .+%s(.+)$
-- . represents all characters
-- + makes us capture 1 or more repetitions of the previous character/symbol will always match the longest possible part
-- %s represents all space characters
-- $ at the end makes the pattern match to the end of the string
-- Example: "Lightweave Embroidery" captures "Embroidery"
function F.String.GetTheLastWordOfAString(text)
	return strmatch(text, ".+%s(.+)$")
end

-- Capture the following string
-- ^[%s%p]*
-- ^ forces us to start capturing at the start of the string
-- %s represents all space characters
-- %p represents all punctuation characters
-- * matches 0 or more repetitions of the previous character/symbol/pattern
-- [] is a capture group
-- This would capture the start of the string and replace all spaces with nothing (but most likely isn't working)
function F.String.RemoveAllWhitespaceCharacters(text)
	return text:gsub("^[%s%p]*", "")
end

-- Capture the following string
-- %d+
-- %d represents all digits
-- this would capture the longest number chain in a string
function F.String.ContainsNumericalCharacters(text)
	return strmatch(text, "%d+")
end

-- Capture the following string
-- %d+
-- %d represents all digits
-- this would capture the longest number chain in a string
function F.String.RemoveTheLongestNumericalChain(text)
	return text:gsub("%D+", "")
end

function F.String.Abbreviate(text)
	if type(text) ~= "string" or text == "" then
		return text
	end

	-- if string has Rune at the start it is almost 100% a DK Rune and needs some different initial logic.
	if strmatch(text, "^Rune") then
		text = F.String.RemoveRuneOfThePrefix(text)
	else
		text = F.String.RemoveEveryOfTheAndEverythingAfter(text)
	end

	local letters = ""
	local lastWord = F.String.GetTheLastWordOfAString(text)
	if not lastWord then
		return text
	end

	-- split the string on each space and loop through them
	-- If we have a string that contains numbers we will add them differently to the stringbuilder
	-- If we have an alphabetical word we check if the first letter is Uppercase, if this is the case add it to the resulting string with a . after it
	-- Else we ignore the word
	for word in gmatch(text, ".-%s") do
		local firstLetter = F.String.RemoveAllWhitespaceCharacters(word)

		if not F.String.ContainsNumericalCharacters(firstLetter) then
			firstLetter = utf8sub(firstLetter, 1, 1)
			if firstLetter ~= utf8lower(firstLetter) then
				-- combine letters value with firstletter value, and then add a . and space
				letters = format("%s%s. ", letters, firstLetter)
			end
		else
			firstLetter = F.String.RemoveTheLongestNumericalChain(firstLetter)
			-- combine letters value with firstLetter value and then add a space
			letters = format("%s%s ", letters, firstLetter)
		end
	end

	-- Combine the build string in the loop and the complete last word
	return format("%s%s", letters, lastWord)
end

E.TagFunctions.Abbrev = F.String.Abbreviate

function F.String.FastGradient(text, r1, g1, b1, r2, g2, b2)
	local msg, len, idx = "", utf8len(text), 0

	for i = 1, len do
		local x = utf8sub(text, i, i)
		if strmatch(x, "%s") then
			msg = msg .. x
			idx = idx + 1
		else
			local relperc = (idx / len)

			if not r2 then
				msg = msg .. "|cff" .. F.String.FastRGB(r1, g1, b1) .. x .. "|r"
			else
				local r, g, b = F.FastColorGradient(relperc, r1, g1, b1, r2, g2, b2)
				msg = msg .. "|cff" .. F.String.FastRGB(r, g, b) .. x .. "|r"
				idx = idx + 1
			end
		end
	end

	return msg
end

function F.String.FastGradientHex(text, h1, h2)
	local r2, g2, b2
	local r1, g1, b1 = F.String.HexToRGB(h1)

	if h2 then
		r2, g2, b2 = F.String.HexToRGB(h2)
	else
		local h, s, l = F.ConvertToHSL(r1, g1, b1)
		r1, g1, b1 = F.ConvertToRGB(F.ClampToHSL(h, s * 0.95, l * 1.2))
		r2, g2, b2 = F.ConvertToRGB(F.ClampToHSL(h, s * 1.35, l * 0.85))
	end

	return F.String.FastGradient(text, r1, g1, b1, r2, g2, b2)
end

function F.String.FastColorGradientHex(percentage, h1, h2)
	local r1, g1, b1 = F.String.HexToRGB(h1)
	local r2, g2, b2 = F.String.HexToRGB(h2)

	return F.FastColorGradient(percentage, r1, g1, b1, r2, g2, b2)
end

function F.String.GradientClass(text, class, reverse)
	if not text or text == "" then
		text = E.myLocalizedClass
	end

	local unitClass = class or E.myclass

	local db = E.db and E.db.mui and E.db.mui.themes
	if not db then
		db = {}
	end
	if db.classColorMap then
		local colorMap = E.db.mui.themes.classColorMap

		-- check if class is an actual valid class that we have gradients for, if not, fallback to player's class
		if not colorMap[1][unitClass] then
			unitClass = E.myclass
		end

		local left = colorMap[1][unitClass] -- Left (player UF)
		local right = colorMap[2][unitClass] -- Right (player UF)

		if left and left.r and right and right.r then
			if not reverse then
				return F.String.FastGradient(text, left.r, left.g, left.b, right.r, right.g, right.b)
			else
				return F.String.FastGradient(text, right.r, right.g, right.b, left.r, left.g, left.b)
			end
		else
			return text
		end
	else
		return text
	end
end

function F.String.Class(msg, class)
	local finalClass = class or E.myclass

	local color = E:ClassColor(finalClass, true)
	return F.String.Color(msg, F.String.FastRGB(color.r, color.g, color.b))
end

function F.String.MERATHILISUI(msg)
	return F.String.Color(msg, I.Enum.Colors.MER)
end

function F.String.Details(msg)
	if not msg or msg == "" then
		return F.String.Color(L["Details"], I.Enum.Colors.DETAILS)
	end

	return F.String.Color(msg, I.Enum.Colors.DETAILS)
end

function F.String.BigWigs(msg)
	if not msg or msg == "" then
		return F.String.Color(L["BigWigs"], I.Enum.Colors.BIGWIGS)
	end

	return F.String.Color(msg, I.Enum.Colors.BIGWIGS)
end

function F.String.OmniCD(msg)
	if not msg or msg == "" then
		return F.String.Color(L["OmniCD"], I.Enum.Colors.OMNICD)
	end

	return F.String.Color(msg, I.Enum.Colors.OMNICD)
end

function F.String.WindTools(msg)
	if not msg or msg == "" then
		return F.String.Color(L["WindTools"], I.Enum.Colors.WT)
	end

	return F.String.Color(msg, I.Enum.Colors.WT)
end

function F.String.FCT(msg)
	if not msg or msg == "" then
		return F.String.Color(L["FCT"], I.Enum.Colors.FCT)
	end

	return F.String.Color(msg, I.Enum.Colors.FCT)
end

function F.String.AS(msg)
	if not msg or msg == "" then
		return F.String.Color(L["AddOnSkins"], I.Enum.Colors.AS)
	end

	return F.String.Color(msg, I.Enum.Colors.AS)
end

function F.String.ElvUI(msg)
	if not msg or msg == "" then
		return F.String.Color(L["ElvUI"], I.Enum.Colors.ELVUI)
	end

	return F.String.Color(msg, I.Enum.Colors.ELVUI)
end

function F.String.ElvUIValue(msg)
	return F.String.RGB(msg, E.media.rgbvaluecolor)
end

function F.String.Error(msg)
	return F.String.Color(msg, I.Enum.Colors.ERROR)
end

function F.String.Good(msg)
	return F.String.Color(msg, I.Enum.Colors.GOOD)
end

function F.String.Warning(msg)
	return F.String.Color(msg, I.Enum.Colors.WARNING)
end

function F.String.GradientString()
	return F.String.FastGradient("Gradient", 0, 0.6, 1, 0, 0.9, 1)
end

function F.String.Legendary(msg)
	return F.String.Color(msg, I.Enum.Colors.LEGENDARY)
end

function F.String.Epic(msg)
	return F.String.Color(msg, I.Enum.Colors.EPIC)
end

function F.String.Rare(msg)
	return F.String.Color(msg, I.Enum.Colors.RARE)
end

function F.String.Beta(msg)
	return F.String.Color(msg, I.Enum.Colors.BETA)
end

function F.String.Grey(msg)
	return F.String.Color(msg, I.Enum.Colors.GREY)
end

-- Credits to WunderUI
function F.String.ConvertGlyph(unicode)
	if unicode <= 0x7F then
		return char(unicode)
	end
	if unicode <= 0x7FF then
		return char(0xC0 + floor(unicode / 0x40), 0x80 + (unicode % 0x40))
	end
	if unicode <= 0xFFFF then
		return char(0xE0 + floor(unicode / 0x1000), 0x80 + (floor(unicode / 0x40) % 0x40), 0x80 + (unicode % 0x40))
	end
	error(L["Could not convert unicode "] .. tostring(unicode))
end
