From 1e533e6420f9e763a9254d76a786d0c9058f4aeb Mon Sep 17 00:00:00 2001
From: Tretiner <alexandirina@inbox.ru>
Date: Sat, 27 Jul 2024 22:13:58 +0500
Subject: [PATCH] Added bug types

---
 .../Domain/BugCheckers/BugCheckerBase.cs                   | 3 +++
 .../Domain/BugCheckers/FastT2Checker.cs                    | 3 +++
 .../Domain/BugCheckers/GenBugChecker.cs                    | 3 +++
 SoulstormReplayReader.Core/Enums/BugType.cs                | 7 +++++++
 4 files changed, 16 insertions(+)
 create mode 100644 SoulstormReplayReader.Core/Enums/BugType.cs

diff --git a/SoulstormReplayReader.Core/Domain/BugCheckers/BugCheckerBase.cs b/SoulstormReplayReader.Core/Domain/BugCheckers/BugCheckerBase.cs
index 8ba4bae..fde9bfa 100644
--- a/SoulstormReplayReader.Core/Domain/BugCheckers/BugCheckerBase.cs
+++ b/SoulstormReplayReader.Core/Domain/BugCheckers/BugCheckerBase.cs
@@ -1,9 +1,12 @@
 using SoulstormReplayReader.Core.Domain.Action;
+using SoulstormReplayReader.Core.Enums;
 
 namespace SoulstormReplayReader.Core.Domain.BugCheckers;
 
 public abstract class BugCheckerBase(int playerId, RaceEnum playerRace)
 {
+    public abstract BugType BugType { get; }
+
     public readonly int PlayerId = playerId;
     public readonly RaceEnum PlayerRace = playerRace;
 
diff --git a/SoulstormReplayReader.Core/Domain/BugCheckers/FastT2Checker.cs b/SoulstormReplayReader.Core/Domain/BugCheckers/FastT2Checker.cs
index 7731969..9e72990 100644
--- a/SoulstormReplayReader.Core/Domain/BugCheckers/FastT2Checker.cs
+++ b/SoulstormReplayReader.Core/Domain/BugCheckers/FastT2Checker.cs
@@ -1,10 +1,13 @@
 using SoulstormReplayReader.Core.Domain.Action;
+using SoulstormReplayReader.Core.Enums;
 using SoulstormReplayReader.Core.Utils;
 
 namespace SoulstormReplayReader.Core.Domain.BugCheckers;
 
 public sealed class FastT2Checker(int playerId, RaceEnum playerRace) : BugCheckerBase(playerId, playerRace)
 {
+    public override BugType BugType => BugType.FastT2;
+
     private byte _gensBuilt = 0;
     private bool _isAspectPortalBuilt = false;
     private bool _isSoulShrineBuilt = false;
diff --git a/SoulstormReplayReader.Core/Domain/BugCheckers/GenBugChecker.cs b/SoulstormReplayReader.Core/Domain/BugCheckers/GenBugChecker.cs
index 782c6ad..b7fcb34 100644
--- a/SoulstormReplayReader.Core/Domain/BugCheckers/GenBugChecker.cs
+++ b/SoulstormReplayReader.Core/Domain/BugCheckers/GenBugChecker.cs
@@ -1,9 +1,12 @@
 using SoulstormReplayReader.Core.Domain.Action;
+using SoulstormReplayReader.Core.Enums;
 
 namespace SoulstormReplayReader.Core.Domain.BugCheckers;
 
 public sealed class GenBugChecker(int playerId, RaceEnum playerRace) : BugCheckerBase(playerId, playerRace)
 {
+    public override BugType BugType => BugType.GenBug;
+
     private const int TicksRange = 4;
     private const int GensToTrigger = 3;
 
diff --git a/SoulstormReplayReader.Core/Enums/BugType.cs b/SoulstormReplayReader.Core/Enums/BugType.cs
new file mode 100644
index 0000000..824a797
--- /dev/null
+++ b/SoulstormReplayReader.Core/Enums/BugType.cs
@@ -0,0 +1,7 @@
+namespace SoulstormReplayReader.Core.Enums;
+
+public enum BugType : byte
+{
+    GenBug,
+    FastT2
+}
\ No newline at end of file