more categories: WvW and Twisted Castle

1 files changed, 8 insertions, 2 deletions

diff --git a/src/categories.rs b/src/categories.rs
index 20d540d..ef8ab00 100644
--- a/src/categories.rs
+++ b/src/categories.rs
@@ -1,4 +1,4 @@
-use evtclib::{Encounter, Log};
+use evtclib::{Encounter, GameMode, Log};
 
 pub trait Categorizable {
     fn category(&self) -> &'static str;
@@ -6,6 +6,10 @@ pub trait Categorizable {
 
 impl Categorizable for Log {
     fn category(&self) -> &'static str {
+        if self.game_mode() == Some(GameMode::WvW) {
+            return "World versus World";
+        }
+
         if let Some(encounter) = self.encounter() {
             match encounter {
                 Encounter::ValeGuardian | Encounter::Gorseval | Encounter::Sabetha => {
@@ -14,7 +18,9 @@ impl Categorizable for Log {
                 Encounter::Slothasor | Encounter::BanditTrio | Encounter::Matthias => {
                     "Wing 2 (Salvation Pass)"
                 }
-                Encounter::KeepConstruct | Encounter::Xera => "Wing 3 (Stronghold of the Faithful)",
+                Encounter::KeepConstruct | Encounter::TwistedCastle | Encounter::Xera => {
+                    "Wing 3 (Stronghold of the Faithful)"
+                }
                 Encounter::Cairn
                 | Encounter::MursaatOverseer
                 | Encounter::Samarog