From 59b5664439269fc3ad4a0d84d9761837dd4af0a0 Mon Sep 17 00:00:00 2001
From: shavit <the@shav.it>
Date: Sat, 13 Jan 2018 09:48:23 +0200
Subject: [PATCH] Added Shavit_GetReplayBotType and ReplayStatus enum. (#574)

---
 addons/sourcemod/scripting/include/shavit.inc | 20 ++++++++++++++++---
 addons/sourcemod/scripting/shavit-replay.sp   |  6 ++++++
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/addons/sourcemod/scripting/include/shavit.inc b/addons/sourcemod/scripting/include/shavit.inc
index 0948258b..c21d22b2 100644
--- a/addons/sourcemod/scripting/include/shavit.inc
+++ b/addons/sourcemod/scripting/include/shavit.inc
@@ -44,7 +44,7 @@
 
 // for reference, not used anymore
 // game types
-enum ServerGame(+=1)
+enum ServerGame
 {
 	Game_CSS = 0,
 	Game_CSGO,
@@ -52,14 +52,14 @@ enum ServerGame(+=1)
 };
 
 // status
-enum TimerStatus(+=1)
+enum TimerStatus
 {
 	Timer_Stopped = 0,
 	Timer_Running,
 	Timer_Paused
 };
 
-enum ReplayStatus(+=1)
+enum ReplayStatus
 {
 	Replay_Start = 0,
 	Replay_Running,
@@ -67,6 +67,12 @@ enum ReplayStatus(+=1)
 	Replay_Idle
 };
 
+enum ReplayBotType
+{
+	Replay_Central = 0,
+	Replay_Legacy
+};
+
 enum
 {
 	sStyleName,
@@ -786,6 +792,13 @@ native int Shavit_GetReplayBotStyle(int client);
  */
 native int Shavit_GetReplayBotTrack(int client);
 
+/**
+ * Gets the replay bot type setting of the server.
+ *
+ * @return							See ReplayBotType enum.
+ */
+native ReplayBotType Shavit_GetReplayBotType();
+
 /**
  * Retrieve the replay bot's current played frame.
  *
@@ -1125,6 +1138,7 @@ public void __pl_shavit_SetNTVOptional()
 	MarkNativeAsOptional("Shavit_GetReplayBotIndex");
 	MarkNativeAsOptional("Shavit_GetReplayBotStyle");
 	MarkNativeAsOptional("Shavit_GetReplayBotTrack");
+	MarkNativeAsOptional("Shavit_GetReplayBotType");
 	MarkNativeAsOptional("Shavit_GetReplayData");
 	MarkNativeAsOptional("Shavit_GetReplayFrameCount");
 	MarkNativeAsOptional("Shavit_GetReplayLength");
diff --git a/addons/sourcemod/scripting/shavit-replay.sp b/addons/sourcemod/scripting/shavit-replay.sp
index 08efffbe..fcb65c30 100644
--- a/addons/sourcemod/scripting/shavit-replay.sp
+++ b/addons/sourcemod/scripting/shavit-replay.sp
@@ -140,6 +140,7 @@ public APLRes AskPluginLoad2(Handle myself, bool late, char[] error, int err_max
 	CreateNative("Shavit_GetReplayBotIndex", Native_GetReplayBotIndex);
 	CreateNative("Shavit_GetReplayBotStyle", Native_GetReplayBotStyle);
 	CreateNative("Shavit_GetReplayBotTrack", Native_GetReplayBotTrack);
+	CreateNative("Shavit_GetReplayBotType", Native_GetReplayBotType);
 	CreateNative("Shavit_GetReplayData", Native_GetReplayData);
 	CreateNative("Shavit_GetReplayFrameCount", Native_GetReplayFrameCount);
 	CreateNative("Shavit_GetReplayLength", Native_GetReplayLength);
@@ -421,6 +422,11 @@ public int Native_GetReplayBotTrack(Handle handler, int numParams)
 	return GetReplayTrack(GetNativeCell(1));
 }
 
+public int Native_GetReplayBotType(Handle handler, int numParams)
+{
+	return view_as<int>((gB_CentralBot)? Replay_Central:Replay_Legacy);
+}
+
 public void Shavit_OnDatabaseLoaded()
 {
 	gH_SQL = Shavit_GetDatabase();