From acee7556b0c90d96f9fea9b7370f1bd707c4468f Mon Sep 17 00:00:00 2001
From: Daniel Schadt <kingdread@gmx.de>
Date: Thu, 25 Nov 2021 13:20:22 +0100
Subject: more categories: WvW and Twisted Castle

---
 src/categories.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'src')

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
-- 
cgit v1.2.3