From 0e039515f6ab2c91f2ec4d6451d283dbc0cdaf1a Mon Sep 17 00:00:00 2001
From: Evil Factory <36804725+evilfactory@users.noreply.github.com>
Date: Mon, 28 Jun 2021 17:34:22 -0300
Subject: [PATCH] hotfix

---
 Example/Lua/traitormod/autorun/traitormod.lua | 22 ++++++++++++++++---
 Example/Lua/traitormod/traitorconfig.lua      |  2 ++
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/Example/Lua/traitormod/autorun/traitormod.lua b/Example/Lua/traitormod/autorun/traitormod.lua
index b389b46c3..7688a0a07 100644
--- a/Example/Lua/traitormod/autorun/traitormod.lua
+++ b/Example/Lua/traitormod/autorun/traitormod.lua
@@ -19,9 +19,17 @@ local warningClients = {}
 local config = dofile("lua/traitormod/traitorconfig.lua")
 local util = dofile("lua/traitormod/util.lua")
 
+local assassinationChooseFunc 
+
 Game.OverrideTraitors(config.enableTraitors) -- shutup old traitors
 Game.AllowWifiChat(config.enableWifiChat) -- deprecated
 
+if config.chooseBotsAsTraitorTargets then
+    assassinationChooseFunc = util.GetValidPlayersNoTraitors
+else
+    assassinationChooseFunc = util.GetValidPlayersNoBotsAndNoTraitors
+end
+
 traitormod.config = config
 
 traitormod.setPercentage = function(client, amount)
@@ -181,7 +189,15 @@ traitormod.assignNormalTraitors = function(amount)
 
     for key, value in pairs(traitors) do traitormod.roundtraitors[value] = {} end
 
-    local targets = util.GetValidPlayersNoTraitors(traitormod.roundtraitors)
+    local targets = assassinationChooseFunc(traitormod.roundtraitors)
+
+    if #targets == 0 then
+        for key, value in pairs(traitors) do
+            traitormod.sendTraitorMessage("Looks like you are a traitor without targets.", util.clientChar(value))
+            traitormod.roundtraitors[value].objectiveType = "nothing"
+            return
+        end
+    end
 
     for key, value in pairs(traitors) do
         traitormod.roundtraitors[value].objectiveType = "kill"
@@ -222,7 +238,7 @@ traitormod.assignNormalTraitors = function(amount)
 end
 
 traitormod.chooseNextObjective = function(key, value)
-    local players = util.GetValidPlayersNoTraitors(traitormod.roundtraitors)
+    local players = assassinationChooseFunc(traitormod.roundtraitors)
 
     if #players == 0 then
         traitormod.sendTraitorMessage("Good job agent, You did it.",
@@ -251,7 +267,7 @@ end
 Hook.Add("roundStart", "traitor_start", function()
 
     Game.SendMessage(
-        "We are using Custom Traitors Plugin by EvilFactory (https://steamcommunity.com/id/evilfactory/)\n Join discord.gg/f9zvNNuxu9",
+        "We are using TraitorMod Plugin by EvilFactory (https://steamcommunity.com/id/evilfactory/)\n Join discord.gg/f9zvNNuxu9",
         3)
 
     local players = util.GetValidPlayersNoBots()
diff --git a/Example/Lua/traitormod/traitorconfig.lua b/Example/Lua/traitormod/traitorconfig.lua
index ac2d56f39..df7b1b4de 100644
--- a/Example/Lua/traitormod/traitorconfig.lua
+++ b/Example/Lua/traitormod/traitorconfig.lua
@@ -14,6 +14,7 @@ config.codewords = codewords
 config.amountCodewords = 2
 config.traitorSpawnDelay = 60
 config.nextMissionDelay = 60
+config.chooseBotsAsTraitorTargets = false
 
 config.endRoundWhenAllTraitorsDie = false
 config.endRoundDelayInSeconds = 60
@@ -49,4 +50,5 @@ end
 -- shipTraitors and normal traitors will be selected equally
 config.getAmountShipTraitors = config.getAmountTraitors
 
+
 return config;
\ No newline at end of file