From a16a5fc01529365c4c792e7f5701be2d00809d5c Mon Sep 17 00:00:00 2001
From: Will Berry <73126355+wdberry@users.noreply.github.com>
Date: Fri, 10 Mar 2023 12:45:40 -0500
Subject: [PATCH] Add logic from upstream to only refresh train tab if selected
 and allow for query limt

---
 cybersyn/scripts/gui/trains.lua | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/cybersyn/scripts/gui/trains.lua b/cybersyn/scripts/gui/trains.lua
index f9c9487..82309c3 100644
--- a/cybersyn/scripts/gui/trains.lua
+++ b/cybersyn/scripts/gui/trains.lua
@@ -16,7 +16,7 @@ function trains_tab.create(widths)
 			--caption = #trains_sorted == 0 and { "cybersyn-gui.trains" } or { "cybersyn-gui.trains", #train_list },
 			caption = { "cybersyn-gui.trains" },
 			--badge_text = format.number(#ltn_data.sorted_trains.composition),
-			--handler = trains_tab.handle.change_tab, --on_click
+			handler = trains_tab.handle.on_trains_tab_selected, --on_click
 			tags = { tab = "trains_tab" },
 		},
 		content = {
@@ -281,6 +281,12 @@ function trains_tab.handle.open_train_gui(player, player_data, refs, e)
 	train_util.open_gui(player.index, train_entity)
 end
 
+---@param player LuaPlayer
+---@param player_data PlayerData
+function trains_tab.handle.on_trains_tab_selected(player, player_data)
+    player_data.selected_tab = "trains_tab"
+end
+
 gui.add_handlers(trains_tab.handle, trains_tab.wrapper)
 
 return trains_tab