From 756d2036744da4c0867fcfd13419afb1c2505460 Mon Sep 17 00:00:00 2001
From: Mjonir <github@mjonir.com>
Date: Tue, 6 Jun 2023 15:56:20 +0200
Subject: [PATCH] Add files via upload

---
 cybersyn/scripts/gui/main.lua | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/cybersyn/scripts/gui/main.lua b/cybersyn/scripts/gui/main.lua
index 7628c42..f7de53c 100644
--- a/cybersyn/scripts/gui/main.lua
+++ b/cybersyn/scripts/gui/main.lua
@@ -60,9 +60,10 @@ end
 
 
 
-function manager_gui.on_player_created(e)
-	local player = game.get_player(e.player_index)
+local function create_player(player_index)
+	local player = game.get_player(player_index)
 	if not player then return end
+
 	local player_data = {
 		search_network_mask = -1,
 		trains_orderings = {},
@@ -71,12 +72,16 @@ function manager_gui.on_player_created(e)
 		refs = manager.create(player),
 		selected_tab = "stations_tab",
 	}
-	global.manager.players[e.player_index] = player_data
+	global.manager.players[player_index] = player_data
 
 	--manager.update(global, player, player_data)
 	--top_left_button_update(player, player_data)
 end
 
+function manager_gui.on_player_created(e)
+	create_player(player_index)
+end
+
 function manager_gui.on_player_removed(e)
 	global.manager.players[e.player_index] = nil
 end
@@ -147,9 +152,16 @@ end
 
 
 function manager_gui.on_migration()
+	for i, p in pairs(game.players) do
+		if global.manager.players[player_index] == nil then
+			create_player(i)
+		end
+	end
+	
 	for i, v in pairs(global.manager.players) do
 		manager_gui.reset_player(i, v)
 	end
+
 	init_items(global.manager)
 end