Fixed getting more then one kit / Cooldown

RocketModPlugins · Apr 6, 2015 · dc94253 · dc94253
1 parent d99cd75
commit dc94253
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 11 deletions.
diff --git a/CommandKit.cs b/CommandKit.cs
@@ -59,10 +59,10 @@ public void Execute(RocketPlayer caller, string command)
                 }
             }
 
-            KeyValuePair<string, IndividualKitCooldown> individualCooldown = Kits.InvididualCooldown.Where(k => k.Key == caller.ToString() && k.Value.Kit == kit.Name).FirstOrDefault();
-            if (!individualCooldown.Equals(default(KeyValuePair<string, IndividualKitCooldown>)))
+            KeyValuePair<string, DateTime> individualCooldown = Kits.InvididualCooldown.Where(k => k.Key == (caller.ToString() + kit.Name)).FirstOrDefault();
+            if (!individualCooldown.Equals(default(KeyValuePair<string, DateTime>)))
             {
-                double individualCooldownSeconds = (DateTime.Now - individualCooldown.Value.Cooldown).TotalSeconds;
+                double individualCooldownSeconds = (DateTime.Now - individualCooldown.Value).TotalSeconds;
                 if (individualCooldownSeconds < kit.Cooldown)
                 {
                     RocketChatManager.Say(caller, Kits.Instance.Translate("command_kit_cooldown_kit", (int)(kit.Cooldown - individualCooldownSeconds)));
@@ -79,8 +79,23 @@ public void Execute(RocketPlayer caller, string command)
             }
             RocketChatManager.Say(caller, Kits.Instance.Translate("command_kit_success", kit.Name));
 
-            Kits.GlobalCooldown.Add(caller.ToString(), DateTime.Now);
-            Kits.InvididualCooldown.Add(caller.ToString(), new IndividualKitCooldown() { Cooldown = DateTime.Now, Kit = kit.Name });
+            if (Kits.GlobalCooldown.ContainsKey(caller.ToString()))
+            {
+                Kits.GlobalCooldown[caller.ToString()] = DateTime.Now;
+            }
+            else
+            {
+                Kits.GlobalCooldown.Add(caller.ToString(), DateTime.Now);
+            }
+
+            if (Kits.GlobalCooldown.ContainsKey(caller.ToString()))
+            {
+                Kits.InvididualCooldown[caller.ToString() + kit.Name] = DateTime.Now;
+            }
+            else
+            {
+                Kits.InvididualCooldown.Add(caller.ToString() + kit.Name, DateTime.Now);
+            }
         }
     }
 }
diff --git a/Kits.cs b/Kits.cs
@@ -10,17 +10,12 @@
 
 namespace unturned.ROCKS.Kits
 {
-    public class IndividualKitCooldown {
-        public DateTime Cooldown;
-        public string Kit;
-    }
-
     public class Kits : RocketPlugin<KitsConfiguration>
     {
         public static Kits Instance = null;
 
         public static Dictionary<string, DateTime> GlobalCooldown = new Dictionary<string,DateTime>();
-        public static Dictionary<string, IndividualKitCooldown> InvididualCooldown = new Dictionary<string,IndividualKitCooldown>();
+        public static Dictionary<string, DateTime> InvididualCooldown = new Dictionary<string, DateTime>();
 
         protected override void Load()
         {