Main menu stuff

Author: V-rtualized

api/connection.lua

@@ -28,6 +28,7 @@ local _connection = nil
 local _ready = false
 local _ready_callbacks = {}
 local _last_opts = nil
+local _state_change_callbacks = {}
 
 local SERVER_DEFAULTS = {
 	api_url = 'http://localhost:8788',
@@ -55,6 +56,10 @@ MPAPI.on_loaded = function(fn)
 	_ready_callbacks[#_ready_callbacks + 1] = fn
 end
 
+MPAPI.on_connection_state_change = function(fn)
+	_state_change_callbacks[#_state_change_callbacks + 1] = fn
+end
+
 MPAPI.connect = function(opts)
 	opts = opts or {}
 	_last_opts = opts
@@ -293,6 +298,13 @@ connection_on_state_change = function(new_state, context)
 	end
 
 	run_new_state_user_callbacks(new_state, context)
+
+	for _, fn in ipairs(_state_change_callbacks) do
+		local ok, err = pcall(fn, new_state, context)
+		if not ok then
+			MPAPI.sendWarnMessage('on_connection_state_change callback error: ' .. tostring(err))
+		end
+	end
 end
 
 log_state_update = function(new_state, context)

api/mod_registry.lua

@@ -47,10 +47,13 @@ MPAPI.register_mod = function(opts)
 		return
 	end
 
+	local hide_logo = opts.hide_logo == true
+
 	local existing = _registered_mods[opts.id]
 	if existing and existing.is_official then
 		existing.main_menu_ui = opts.main_menu_ui
 		existing.server_config = opts.server_config
+		existing.hide_logo = hide_logo
 		if opts.name then
 			existing.name = opts.name
 		end
@@ -65,6 +68,7 @@ MPAPI.register_mod = function(opts)
 			server_config = opts.server_config,
 			main_menu_ui = opts.main_menu_ui,
 			download_url = opts.download_url,
+			hide_logo = hide_logo,
 			is_official = false,
 		}
 		_mod_order[#_mod_order + 1] = opts.id
@@ -119,7 +123,7 @@ MPAPI._internal.activate_mod = function(id)
 
 	_active_mod = id
 
-	replace_main_menu(mod.main_menu_ui)
+	replace_main_menu(mod.main_menu_ui, mod.hide_logo)
 	update_account_button()
 end
 
@@ -160,10 +164,25 @@ update_account_button = function()
 	end
 end
 
-replace_main_menu = function(build_fn)
+replace_main_menu = function(build_fn, hide_logo)
 	if G.MAIN_MENU_UI then
 		G.MAIN_MENU_UI:remove()
 	end
+	if G.PROFILE_BUTTON then
+		G.PROFILE_BUTTON:remove()
+		G.PROFILE_BUTTON = nil
+	end
+	if hide_logo then
+		if G.SPLASH_LOGO then
+			G.SPLASH_LOGO.states.visible = false
+		end
+		if G.title_top then
+			G.title_top.states.visible = false
+			for _, card in ipairs(G.title_top.cards or {}) do
+				card.states.visible = false
+			end
+		end
+	end
 	G.MAIN_MENU_UI = UIBox({
 		definition = build_fn(),
 		config = { align = 'bmi', offset = { x = 0, y = 10 }, major = G.ROOM_ATTACH, bond = 'Weak' },
@@ -172,10 +191,23 @@ replace_main_menu = function(build_fn)
 	G.MAIN_MENU_UI:align_to_major()
 end
 
+MPAPI._internal.replace_main_menu = function(build_fn, hide_logo)
+	replace_main_menu(build_fn, hide_logo)
+end
+
 restore_main_menu = function()
 	if G.MAIN_MENU_UI then
 		G.MAIN_MENU_UI:remove()
 	end
+	if G.SPLASH_LOGO then
+		G.SPLASH_LOGO.states.visible = true
+	end
+	if G.title_top then
+		G.title_top.states.visible = true
+		for _, card in ipairs(G.title_top.cards or {}) do
+			card.states.visible = true
+		end
+	end
 	if _original_set_main_menu_UI then
 		_original_set_main_menu_UI()
 	end

localization/en-us.lua

@@ -22,6 +22,11 @@ return {
 			k_unlink_discord = 'Unlink Discord',
 			b_open_download_page = 'Open Download Page',
 			b_retry_connection = 'Retry Connection',
+			b_create_lobby_cap = { 'CREATE', 'LOBBY' },
+			b_join_lobby_cap = 'JOIN LOBBY',
+			b_by_code_cap = 'BY CODE',
+			b_from_clipboard_cap = 'FROM CLIPBOARD',
+			b_find_game_cap = 'FIND GAME',
 		},
 	},
 }

ui/main_menu.lua

@@ -1,7 +1,6 @@
 -- Forward declarations for helper functions
 local build_mod_buttons
 local build_mod_button
-local set_modded_main_menu_ui
 local attach_account_button
 
 -----------------------------
@@ -140,18 +139,6 @@ build_mod_button = function(mod, inner_width)
 	}
 end
 
-set_modded_main_menu_ui = function(mod)
-	if G.MAIN_MENU_UI then
-		G.MAIN_MENU_UI:remove()
-	end
-	G.MAIN_MENU_UI = UIBox({
-		definition = mod.main_menu_ui(),
-		config = { align = 'bmi', offset = { x = 0, y = 10 }, major = G.ROOM_ATTACH, bond = 'Weak' },
-	})
-	G.MAIN_MENU_UI.alignment.offset.y = 0
-	G.MAIN_MENU_UI:align_to_major()
-end
-
 attach_account_button = function()
 	G.E_MANAGER:add_event(Event({
 		blockable = false,
@@ -207,7 +194,7 @@ set_main_menu_UI = function()
 	local active = MPAPI.get_active_mod_data()
 
 	if active and active.main_menu_ui then
-		set_modded_main_menu_ui(active)
+		MPAPI._internal.replace_main_menu(active.main_menu_ui, active.hide_logo)
 	else
 		_set_main_menu_UI_ref()
 	end