From b3a5579b6eda3710151f2c8bf73e5e845d1ea4cb Mon Sep 17 00:00:00 2001
From: teknomunk <teknomunk@protonmail.com>
Date: Tue, 9 Apr 2024 00:40:04 +0000
Subject: [PATCH] Fix crashes in train logic, allow breaking apart trains

---
 mods/ENTITIES/mcl_minecarts/movement.lua |  2 +-
 mods/ENTITIES/mcl_minecarts/train.lua    | 32 ++++++++++++++++++++++--
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/mods/ENTITIES/mcl_minecarts/movement.lua b/mods/ENTITIES/mcl_minecarts/movement.lua
index 1b84504f1..7210e22ce 100644
--- a/mods/ENTITIES/mcl_minecarts/movement.lua
+++ b/mods/ENTITIES/mcl_minecarts/movement.lua
@@ -121,7 +121,7 @@ local function handle_cart_collision(cart1, prev_pos, next_dir)
 	for uuid,v in pairs(carts) do
 		-- Clean up dead carts
 		local data = get_cart_data(uuid)
-		if not data then
+		if not data or not data.connected_at then
 			carts[uuid] = nil
 			dirty = true
 			uuid = nil
diff --git a/mods/ENTITIES/mcl_minecarts/train.lua b/mods/ENTITIES/mcl_minecarts/train.lua
index cac57fa7d..b3648f026 100644
--- a/mods/ENTITIES/mcl_minecarts/train.lua
+++ b/mods/ENTITIES/mcl_minecarts/train.lua
@@ -4,6 +4,7 @@ local mod = mcl_minecarts
 
 -- Imports
 local get_cart_data = mod.get_cart_data
+local save_cart_data = mod.save_cart_data
 local MAX_TRAIN_LENGTH = mod.MAX_TRAIN_LENGTH
 
 -- Follow .behind to the back end of a train
@@ -62,6 +63,27 @@ function mod.distance_between_cars(car1, car2)
 end
 local distance_between_cars = mod.distance_between_cars
 
+local function break_train_at(cart)
+	if cart.ahead then
+		local ahead = get_cart_data(cart.ahead)
+		if ahead then
+			ahead.behind = nil
+			cart.ahead = nil
+			save_cart_data(ahead.uuid)
+		end
+	end
+	if cart.behind then
+		local behind = get_cart_data(cart.behind)
+		if behind then
+			behind.ahead = nil
+			cart.behind = nil
+			save_cart_data(behind.uuid)
+		end
+	end
+	save_cart_data(cart.uuid)
+end
+mod.break_train_at = break_train_at
+
 function mod.update_train(cart)
 	local staticdata = cart._staticdata
 
@@ -88,11 +110,17 @@ function mod.update_train(cart)
 		local e = 0
 		local separation
 		local cart_velocity = velocity
-		if behind then
+		if not c.connected_at then
+			break_train_at(c)
+		elseif behind then
 			separation = distance_between_cars(behind, c)
 			local e = 0
-			if separation > 1.6 then
+			if not separation then
+				break_train_at(c)
+			elseif separation > 1.6 then
 				cart_velocity = velocity * 0.9
+			elseif separation > 2.5 then
+				break_train_at(c)
 			elseif separation < 1.15 then
 				cart_velocity = velocity * 1.1
 			end