1 files changed, 110 insertions, 0 deletions
diff --git a/sourcemod/scripting/gokz-core/demofix.sp b/sourcemod/scripting/gokz-core/demofix.sp
new file mode 100644
index 0000000..84a9307
--- /dev/null
+++ b/sourcemod/scripting/gokz-core/demofix.sp
@@ -0,0 +1,110 @@
+static ConVar CV_EnableDemofix;
+static Handle H_DemofixTimer;
+static bool mapRunning;
+
+void OnPluginStart_Demofix()
+{
+	AddCommandListener(Command_Demorestart, "demorestart");
+	CV_EnableDemofix = AutoExecConfig_CreateConVar("gokz_demofix", "1", "Whether GOKZ applies demo record fix to server. (0 = Disabled, 1 = Update warmup period once, 2 = Regularly reset warmup period)", _, true, 0.0, true, 2.0);
+	CV_EnableDemofix.AddChangeHook(OnDemofixConVarChanged);
+	// If the map is tweaking the warmup value, we need to rerun the fix again.
+	FindConVar("mp_warmuptime").AddChangeHook(OnDemofixConVarChanged);
+	// We assume that the map is already loaded on late load.
+	if (gB_LateLoad)
+	{
+		mapRunning = true;
+	}
+}
+
+void OnMapStart_Demofix()
+{
+	mapRunning = true;
+}
+
+void OnMapEnd_Demofix()
+{
+	mapRunning = false;
+}
+
+void OnRoundStart_Demofix()
+{
+	DoDemoFix();
+}
+
+public Action Command_Demorestart(int client, const char[] command, int argc)
+{
+	FixRecord(client);
+	return Plugin_Continue;
+}
+
+static void FixRecord(int client)
+{
+	// For some reasons, demo playback speed is absolute trash without a round_start event.
+	// So whenever the client starts recording a demo, we create the event and send it to them.
+	Event e = CreateEvent("round_start", true);
+	int timelimit = FindConVar("mp_timelimit").IntValue;
+	e.SetInt("timelimit", timelimit);
+	e.SetInt("fraglimit", 0);
+	e.SetString("objective", "demofix");
+
+	e.FireToClient(client);
+	delete e;
+}
+
+public void OnDemofixConVarChanged(ConVar convar, const char[] oldValue, const char[] newValue)
+{
+	DoDemoFix();
+}
+
+public Action Timer_EnableDemoRecord(Handle timer)
+{
+	EnableDemoRecord();
+	return Plugin_Continue;
+}
+
+static void DoDemoFix()
+{
+	if (H_DemofixTimer != null)
+	{
+		delete H_DemofixTimer;
+	}
+	// Setting the cvar value to 1 can avoid clogging the demo file and slightly increase performance.
+	switch (CV_EnableDemofix.IntValue)
+	{
+		case 0:
+		{
+			if (!mapRunning)
+			{
+				return;
+			}
+
+			GameRules_SetProp("m_bWarmupPeriod", 0);
+		}
+		case 1:
+		{
+			// Set warmup time to 2^31-1, effectively forever
+			if (FindConVar("mp_warmuptime").IntValue != 2147483647)
+			{
+				FindConVar("mp_warmuptime").SetInt(2147483647);
+			}
+			EnableDemoRecord();
+		}
+		case 2:
+		{
+			H_DemofixTimer = CreateTimer(1.0, Timer_EnableDemoRecord, _, TIMER_REPEAT);
+		}
+	}
+}
+
+static void EnableDemoRecord()
+{
+	// Enable warmup to allow demo recording
+	// m_fWarmupPeriodEnd is set in the past to hide the timer UI
+	if (!mapRunning)
+	{
+		return;
+	}
+	GameRules_SetProp("m_bWarmupPeriod", 1);
+	GameRules_SetPropFloat("m_fWarmupPeriodStart", GetGameTime() - 1.0);
+	GameRules_SetPropFloat("m_fWarmupPeriodEnd", GetGameTime() - 1.0);
+}
+\ No newline at end of file