Lobby api

Author: V-rtualized

api/connection.lua

@@ -122,9 +122,9 @@ end
 
 MPAPI.is_connected = function()
 	if _connection then
-		return _connection:get_state()
+		return _connection:get_state() == 'connected'
 	end
-	return 'disconnected'
+	return false
 end
 
 MPAPI.get_mqtt = function()
@@ -297,6 +297,11 @@ connection_on_state_change = function(new_state, context)
 		log_state_update(new_state, context)
 	end
 
+	-- Auto-create lobby object on reconnection
+	if new_state == 'connected' and context.reconnected_lobby and MPAPI._internal.create_reconnected_lobby then
+		MPAPI._internal.create_reconnected_lobby(context.reconnected_lobby)
+	end
+
 	run_new_state_user_callbacks(new_state, context)
 
 	for _, fn in ipairs(_state_change_callbacks) do

api/lobby.lua

@@ -0,0 +1,364 @@
+-- Forward declarations for helper functions
+local create_lobby_callback
+local create_lobby_object
+local subscribe_all
+local handle_event
+local handle_metadata
+local handle_own_state
+local cleanup_lobby
+
+-----------------------------
+-- STATE VARIABLES
+-----------------------------
+
+local _current_lobby = nil
+
+-----------------------------
+-- API FUNCTIONS
+-----------------------------
+
+MPAPI.create_lobby = function(mod_id, opts)
+	opts = opts or {}
+	local conn = MPAPI.get_connection()
+	local mqtt = MPAPI.get_mqtt()
+
+	if not conn or conn:get_state() ~= 'connected' then
+		MPAPI.sendWarnMessage('create_lobby: not connected')
+		return nil
+	end
+
+	if not mqtt then
+		MPAPI.sendWarnMessage('create_lobby: MQTT not available')
+		return nil
+	end
+
+	local lobby = create_lobby_object({
+		mod_id = mod_id,
+		player_id = conn.player_id,
+		mqtt = mqtt,
+		api = conn.api,
+		connection = conn,
+	})
+
+	_current_lobby = lobby
+
+	conn.api:create_lobby(conn.jwt_token, mod_id, opts.max_players, create_lobby_callback)
+
+	return lobby
+end
+
+MPAPI.join_lobby = function(mod_id, code, opts)
+	opts = opts or {}
+	local conn = MPAPI.get_connection()
+	local mqtt = MPAPI.get_mqtt()
+
+	if not conn or conn:get_state() ~= 'connected' then
+		MPAPI.sendWarnMessage('join_lobby: not connected')
+		return nil
+	end
+
+	if not mqtt then
+		MPAPI.sendWarnMessage('join_lobby: MQTT not available')
+		return nil
+	end
+
+	local lobby = create_lobby_object({
+		mod_id = mod_id,
+		player_id = conn.player_id,
+		mqtt = mqtt,
+		api = conn.api,
+		connection = conn,
+	})
+
+	_current_lobby = lobby
+	
+	conn.api:join_lobby(conn.jwt_token, code, )
+
+	return lobby
+end
+
+MPAPI.get_current_lobby = function()
+	return _current_lobby
+end
+
+-----------------------------
+-- INTERNAL FUNCTIONS
+-----------------------------
+
+MPAPI._internal.create_reconnected_lobby = function(lobby_data)
+	local conn = MPAPI.get_connection()
+	local mqtt = MPAPI.get_mqtt()
+
+	if not conn or not mqtt then
+		return
+	end
+
+	local lobby = create_lobby_object({
+		code = lobby_data.code,
+		mod_id = lobby_data.modId,
+		is_host = lobby_data.isHost or false,
+		player_id = conn.player_id,
+		mqtt = mqtt,
+		api = conn.api,
+		connection = conn,
+		metadata = lobby_data.metadata or {},
+	})
+
+	subscribe_all(lobby)
+	_current_lobby = lobby
+	lobby:_fire('connected')
+end
+
+-----------------------------
+-- HELPER FUNCTIONS
+-----------------------------
+
+join_lobby_callback = function(err, data)
+	if err then
+		_current_lobby:_fire('error', err)
+		return
+	end
+
+	conn.jwt_token = data.token
+
+	_current_lobby.code = data.lobby.code
+	_current_lobby.is_host = data.lobby.isHost or false
+	lo_current_lobbybby._metadata = data.lobby.metadata or {}
+
+	subscribe_all(_current_lobby)
+	_current_lobby:_fire('connected')
+end
+
+create_lobby_callback = function(err, data)
+	if err then
+		_current_lobby:_fire('error', err)
+		return
+	end
+
+	conn.jwt_token = data.token
+
+	_current_lobby.code = data.lobby.code
+	_current_lobby.is_host = true
+	_current_lobby._metadata = data.lobby.metadata or {}
+
+	subscribe_all(_current_lobby)
+	_current_lobby:_fire('connected')
+end
+
+create_lobby_object = function(opts)
+	local lobby = {
+		code = opts.code,
+		mod_id = opts.mod_id,
+		is_host = opts.is_host or false,
+		player_id = opts.player_id,
+		_mqtt = opts.mqtt,
+		_api = opts.api,
+		_connection = opts.connection,
+		_metadata = opts.metadata or {},
+		_player_state = nil,
+		_players = {},
+		_event_handlers = {},
+		_destroyed = false,
+	}
+
+	function lobby:on(event_name, handler)
+		if not self._event_handlers[event_name] then
+			self._event_handlers[event_name] = {}
+		end
+		local handlers = self._event_handlers[event_name]
+		handlers[#handlers + 1] = handler
+	end
+
+	function lobby:_fire(event_name, ...)
+		local handlers = self._event_handlers[event_name]
+		if not handlers then
+			return
+		end
+		for _, handler in ipairs(handlers) do
+			local ok, err = pcall(handler, ...)
+			if not ok then
+				MPAPI.sendWarnMessage('Lobby event "' .. event_name .. '" handler error: ' .. tostring(err))
+			end
+		end
+	end
+
+	function lobby:set_metadata(tbl)
+		if self._destroyed then
+			return
+		end
+		if not self.is_host then
+			self:_fire('error', 'Only the host can set metadata')
+			return
+		end
+		self._api:set_lobby_metadata(self._connection.jwt_token, self.code, tbl, function(err, data)
+			if err then
+				self:_fire('error', err)
+				return
+			end
+			if data and data.metadata then
+				self._metadata = data.metadata
+			end
+		end)
+	end
+
+	function lobby:get_metadata()
+		return self._metadata
+	end
+
+	function lobby:set_player_state(tbl)
+		if self._destroyed then
+			return
+		end
+		local topic = self._mqtt:lobby_topic(self.code, 'players/' .. self.player_id .. '/state')
+		self._mqtt:publish(topic, MPAPI.json_encode(tbl), 1, true)
+		self._player_state = tbl
+	end
+
+	function lobby:get_player_state()
+		return self._player_state
+	end
+
+	function lobby:get_players()
+		local result = {}
+		for _, player in pairs(self._players) do
+			result[#result + 1] = player
+		end
+		return result
+	end
+
+	function lobby:leave()
+		if self._destroyed then
+			return
+		end
+		self._api:leave_lobby(self._connection.jwt_token, self.code, function(err, data)
+			if err then
+				self:_fire('error', err)
+				return
+			end
+			if data and data.token then
+				self._connection.jwt_token = data.token
+			end
+			cleanup_lobby(self)
+			self:_fire('disconnected')
+		end)
+	end
+
+	return lobby
+end
+
+subscribe_all = function(lobby)
+	if not lobby._mqtt or not lobby.code then
+		return
+	end
+
+	local events_topic = lobby._mqtt:lobby_topic(lobby.code, 'events')
+	lobby._mqtt:subscribe(events_topic, 1, function(topic, payload)
+		handle_event(lobby, payload)
+	end)
+
+	local metadata_topic = lobby._mqtt:lobby_topic(lobby.code, 'metadata')
+	lobby._mqtt:subscribe(metadata_topic, 1, function(topic, payload)
+		handle_metadata(lobby, payload)
+	end)
+
+	local state_topic = lobby._mqtt:lobby_topic(lobby.code, 'players/' .. lobby.player_id .. '/state')
+	lobby._mqtt:subscribe(state_topic, 1, function(topic, payload)
+		handle_own_state(lobby, payload)
+	end)
+end
+
+handle_event = function(lobby, payload)
+	if lobby._destroyed then
+		return
+	end
+
+	local ok, data = pcall(MPAPI.json_decode, payload)
+	if not ok or not data or not data.type then
+		return
+	end
+
+	local event_type = data.type
+
+	if event_type == 'player_joined' then
+		if data.playerId then
+			lobby._players[data.playerId] = {
+				id = data.playerId,
+				displayName = data.displayName,
+				is_away = false,
+			}
+		end
+		lobby:_fire('player_joined', data.playerId)
+	elseif event_type == 'player_left' then
+		if data.playerId then
+			lobby._players[data.playerId] = nil
+		end
+		lobby:_fire('player_left', data.playerId)
+	elseif event_type == 'player_disconnected' then
+		if data.playerId and lobby._players[data.playerId] then
+			lobby._players[data.playerId].is_away = true
+		end
+		lobby:_fire('player_disconnected', data.playerId)
+	elseif event_type == 'player_reconnected' then
+		if data.playerId and lobby._players[data.playerId] then
+			lobby._players[data.playerId].is_away = false
+		end
+		lobby:_fire('player_reconnected', data.playerId)
+	elseif event_type == 'metadata_changed' then
+		if data.data then
+			lobby._metadata = data.data
+		end
+		lobby:_fire('metadata_changed', lobby._metadata)
+	elseif event_type == 'host_changed' then
+		if data.playerId then
+			lobby.is_host = (data.playerId == lobby.player_id)
+		end
+		lobby:_fire('host_changed', data.playerId)
+	elseif event_type == 'lobby_closed' then
+		cleanup_lobby(lobby)
+		lobby:_fire('disconnected')
+	end
+end
+
+handle_metadata = function(lobby, payload)
+	if lobby._destroyed then
+		return
+	end
+
+	local ok, data = pcall(MPAPI.json_decode, payload)
+	if not ok or not data then
+		return
+	end
+
+	lobby._metadata = data
+	lobby:_fire('metadata_changed', lobby._metadata)
+end
+
+handle_own_state = function(lobby, payload)
+	if lobby._destroyed then
+		return
+	end
+
+	local ok, data = pcall(MPAPI.json_decode, payload)
+	if not ok or not data then
+		return
+	end
+
+	lobby._player_state = data
+end
+
+cleanup_lobby = function(lobby)
+	if lobby._destroyed then
+		return
+	end
+	lobby._destroyed = true
+
+	if lobby._mqtt and lobby.code then
+		lobby._mqtt:unsubscribe(lobby._mqtt:lobby_topic(lobby.code, 'events'))
+		lobby._mqtt:unsubscribe(lobby._mqtt:lobby_topic(lobby.code, 'metadata'))
+		lobby._mqtt:unsubscribe(lobby._mqtt:lobby_topic(lobby.code, 'players/' .. lobby.player_id .. '/state'))
+	end
+
+	if _current_lobby == lobby then
+		_current_lobby = nil
+	end
+end