pr.lua

SMODS.CSS = {}
SMODS.CSS_Buffer = {}
SMODS.CSS_Selectors = {}
SMODS.CSS_Selector = SMODS.GameObject:extend({
	obj_table = SMODS.CSS_Selectors,
	obj_buffer = {},
	set = "CSS_Selector",
	enabled = true,

	required_params = {
		"key",
		"selector",
		"get_config",
	},

	create_parsed_selector = function(self, selector)
		local result = {
			n = nil,
			id = nil,
			classes = {},

			check_n = false,
			check_id = false,
			check_classes = false,
			allow_all = false,

			priority = { 0, 0, 0 },
		}

		if selector == "*" then
			result.allow_all = true
			result.priority[3] = 1
			return result
		end

		local i = 1
		local n = #selector

		while i <= n do
			local ch = selector:sub(i, i)

			if ch == "#" then
				local j = i + 1
				while j <= n and not selector:sub(j, j):match("[#.]") do
					j = j + 1
				end
				result.id = selector:sub(i + 1, j - 1)
				result.check_id = true
				result.priority[1] = 1
				i = j
			elseif ch == "." then
				local j = i + 1
				while j <= n and not selector:sub(j, j):match("[#.]") do
					j = j + 1
				end
				result.classes[selector:sub(i + 1, j - 1)] = true
				result.check_classes = true
				result.priority[2] = result.priority[2] + 1
				i = j
			else
				local j = i
				while j <= n and not selector:sub(j, j):match("[#.]") do
					j = j + 1
				end
				result.n = selector:sub(i, j - 1)
				result.check_n = true
				result.priority[3] = 1
				i = j
			end
		end

		return result
	end,
	create_element_selector = function(self, parsed_selector)
		return function(self, node)
			if parsed_selector.allow_all then
				return true
			end
			local any_match = false
			if parsed_selector.check_n then
				if node.n ~= G.UIT[parsed_selector.n] then
					return false
				end
				any_match = true
			end
			if parsed_selector.check_id then
				if (node.config and node.config.id) ~= parsed_selector.id then
					return false
				end
				any_match = true
			end
			if parsed_selector.check_classes then
				if not (node.config and node.config.class) then
					return false
				end

				local node_classes = {}
				for cls in node.config.class:gmatch("%S+") do
					node_classes[cls] = true
				end
				for class in pairs(parsed_selector.classes) do
					if not node_classes[class] then
						return false
					end
				end
				any_match = true
			end
			return any_match
		end
	end,
	get_element_hash_selector = function(self, node)
		local r = tostring(node.n)
		if node.config then
			if node.config.id then
				r = r .. "#" .. tostring(node.config.id)
			end
			if node.config.class then
				r = r .. "." .. tostring(node.config.class)
			end
		end
		return r
	end,

	pre_inject_class = function(self)
		SMODS.CSS = {}
		SMODS.CSS_Buffer = {}
	end,
	inject = function(self)
		self.parsed_selector = SMODS.CSS_Selector:create_parsed_selector(self.selector)
		self.element_selector = SMODS.CSS_Selector:create_element_selector(self.parsed_selector)
		self.selector_priority = self.parsed_selector.priority or { 0, 0, 0 }
		self.order = #self.obj_buffer
		table.insert(SMODS.CSS, self)
	end,
	post_inject_class = function(self)
		table.sort(SMODS.CSS, function(a, b)
			local pa = a.selector_priority
			local pb = b.selector_priority

			if pa[1] ~= pb[1] then
				return pa[1] < pb[1]
			end
			if pa[2] ~= pb[2] then
				return pa[2] < pb[2]
			end
			if pa[3] ~= pb[3] then
				return pa[3] < pb[3]
			end
			return a.order > b.order
		end)
	end,

	apply = function(self, node)
		if type(node) ~= "table" then
			return node
		end
		local hash_selector = SMODS.CSS_Selector:get_element_hash_selector(node)
		local css_config = {}
		if SMODS.CSS_Buffer[hash_selector] then
			for _, selector in ipairs(SMODS.CSS_Buffer[hash_selector]) do
				if selector.enabled then
					css_config = SMODS.merge_defaults(selector:get_config(node) or {}, css_config)
				end
			end
		else
			local selectors = {}
			for _, selector in ipairs(SMODS.CSS) do
				if selector:element_selector(node) then
					table.insert(selectors, selector)
					if selector.enabled then
						css_config = SMODS.merge_defaults(selector:get_config(node) or {}, css_config)
					end
				end
			end
			SMODS.CSS_Buffer[hash_selector] = selectors
		end
		node.config = node.config or {}
		SMODS.merge_defaults(node.config, css_config)
		return node
	end,
})

local old_set_parent_child = UIBox.set_parent_child
function UIBox:set_parent_child(node, ...)
	SMODS.CSS_Selector:apply(node)
	return old_set_parent_child(self, node, ...)
end

-- SMODS.CSS_Selector({
-- 	key = "test1",
-- 	selector = "C",
-- 	get_config = function(self)
-- 		return { colour = G.C.ORANGE }
-- 	end,
-- })
-- SMODS.CSS_Selector({
-- 	key = "test2",
-- 	selector = "R",
-- 	get_config = function(self)
-- 		return { colour = G.C.GREEN }
-- 	end,
-- })
-- SMODS.CSS_Selector({
-- 	key = "test3",
-- 	selector = "C#test_1",
-- 	get_config = function(self)
-- 		return { colour = G.C.MULT }
-- 	end,
-- })
-- SMODS.CSS_Selector({
-- 	key = "test4",
-- 	selector = "C#test_2",
-- 	get_config = function(self)
-- 		return { colour = G.C.CHIPS }
-- 	end,
-- })
-- SMODS.CSS_Selector({
-- 	key = "test5",
-- 	selector = "T#my_title.large_text.special_text",
-- 	get_config = function(self)
-- 		return { scale = 0.75, colour = G.C.ORANGE, shadow = true }
-- 	end,
-- })

-- SMODS.CSS_Selector({
-- 	key = "test6",
-- 	selector = "*",
-- 	get_config = function(self)
-- 		return { colour = G.C.GREEN }
-- 	end,
-- })