Privileged jokers

Author: V-rtualized

api/connection.lua

@@ -20,6 +20,7 @@ MPAPI.connection_state = {
 	is_temp = false,
 	use_discord_name = false,
 	preferred_joker = 'j_joker',
+	privileges = {},
 }
 
 local _mqtt_instance = nil
@@ -137,6 +138,10 @@ MPAPI.is_ready = function()
 	return _ready
 end
 
+MPAPI.get_privileges = function()
+	return MPAPI.connection_state.privileges
+end
+
 -----------------------------
 -- INTERNAL FUNCTIONS
 -----------------------------
@@ -274,6 +279,7 @@ connection_on_state_change = function(new_state, context)
 		MPAPI.connection_state.is_temp = _connection.is_temp or false
 		MPAPI.connection_state.use_discord_name = _connection.use_discord_name or false
 		MPAPI.connection_state.preferred_joker = _connection.preferred_joker or 'j_joker'
+		MPAPI.connection_state.privileges = _connection.privileges
 	end
 	set_connection_state_status_text()
 
@@ -306,6 +312,7 @@ reset_connection_state_variables = function()
 	MPAPI.connection_state.is_temp = false
 	MPAPI.connection_state.use_discord_name = false
 	MPAPI.connection_state.preferred_joker = 'j_joker'
+	MPAPI.connection_state.privileges = nil
 end
 
 set_connection_state_status_text = function()

lib/util.lua

@@ -6,3 +6,22 @@ MPAPI.shallow_copy = function(t)
 	end
 	return out
 end
+
+-- Merges two tables with unique values, preserves order
+MPAPI.merge_unique = function(a, b)
+	local seen = {}
+	local out = {}
+	for _, v in ipairs(a) do
+		if not seen[v] then
+			seen[v] = true
+			out[#out + 1] = v
+		end
+	end
+	for _, v in ipairs(b) do
+		if not seen[v] then
+			seen[v] = true
+			out[#out + 1] = v
+		end
+	end
+	return out
+end

networking/connection.lua

@@ -22,6 +22,7 @@ function connection.new(opts)
 		display_name = nil,
 		use_discord_name = false,
 		preferred_joker = 'j_joker',
+		privileges = {},
 		steam_id = nil,
 		discord_name = nil,
 		is_temp = false,
@@ -69,6 +70,9 @@ function connection:_handle_auth_success(data)
 		self.display_name = data.player.displayName or self.steam_name
 		self.use_discord_name = data.player.useDiscordName or false
 		self.preferred_joker = data.player.preferredJoker or 'j_joker'
+		if data.player.privileges then
+			self.privileges = data.player.privileges
+		end
 	end
 
 	self.lobby_data = data.lobby or nil

ui/avatar_selection.lua

@@ -46,36 +46,15 @@ local AVATAR_JOKERS = {
 	'j_idol',
 	'j_matador',
 	'j_stuntman',
-	'j_hack',
-	'j_hologram',
-	'j_vagabond',
-	'j_photograph',
-	'j_hallucination',
-	'j_baseball',
-	'j_sock_and_buskin',
-	'j_blueprint',
-	'j_invisible',
-	'j_brainstorm',
-	'j_caino',
-	'j_triboulet',
-	'j_yorick',
-	'j_chicot',
-	'j_perkeo',
 }
 
-local avatar_centers = {}
-for _, id in ipairs(AVATAR_JOKERS) do
-	if G.P_CENTERS[id] then
-		avatar_centers[#avatar_centers + 1] = G.P_CENTERS[id]
-	end
-end
-
 -----------------------------
 -- STATE VARIABLES
 -----------------------------
 
 local _card_rows = {}
 local _joker_tables = {}
+local _avatar_centers = {}
 
 -----------------------------
 -- UI FUNCTIONS
@@ -84,10 +63,11 @@ local _joker_tables = {}
 local create_UIBox_avatar_selection = function()
 	_joker_tables = {}
 
+	calculate_avatar_centers(avatar_jokers_to_use)
 	create_card_rows()
 	populate_page(1)
 
-	local total_pages = math.ceil(#avatar_centers / CARDS_PER_PAGE)
+	local total_pages = math.ceil(#_avatar_centers / CARDS_PER_PAGE)
 	local page_options = {}
 	for i = 1, total_pages do
 		page_options[#page_options + 1] = localize('k_page') .. ' ' .. tostring(i) .. '/' .. tostring(total_pages)
@@ -122,6 +102,18 @@ local create_UIBox_avatar_selection = function()
 	})
 end
 
+calculate_avatar_centers = function()
+	local privilege_jokers = MPAPI.get_privileges()
+	local avatar_jokers_to_use = privilege_jokers.jokers and MPAPI.merge_unique(AVATAR_JOKERS, privilege_jokers.jokers) or AVATAR_JOKERS
+
+	_avatar_centers = {}
+	for _, id in ipairs(avatar_jokers_to_use) do
+		if G.P_CENTERS[id] then
+			_avatar_centers[#_avatar_centers + 1] = G.P_CENTERS[id]
+		end
+	end
+end
+
 create_card_rows = function()
 	_card_rows = {}
 	for j = 1, ROWS do
@@ -150,7 +142,7 @@ populate_page = function(page)
 	local page_offset = CARDS_PER_PAGE * (page - 1)
 	for i = 1, COLS do
 		for j = 1, ROWS do
-			local center = avatar_centers[i + (j - 1) * COLS + page_offset]
+			local center = _avatar_centers[i + (j - 1) * COLS + page_offset]
 			if not center then
 				break
 			end