chore: add sort by is_melee, is_ranged, is_spellcaster

RakuJa · Dec 9, 2024 · ab95c49 · ab95c49
1 parent 2fdafc1
commit ab95c49
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/db/bestiary_proxy.rs b/src/db/bestiary_proxy.rs
@@ -97,6 +97,26 @@ pub async fn get_paginated_creatures(
                 .essential
                 .family
                 .cmp(&b.core_data.essential.family),
+            CreatureSortEnum::Alignment => a
+                .core_data
+                .essential
+                .alignment
+                .cmp(&b.core_data.essential.alignment),
+            CreatureSortEnum::Melee => a
+                .core_data
+                .derived
+                .is_melee
+                .cmp(&b.core_data.derived.is_melee),
+            CreatureSortEnum::Ranged => a
+                .core_data
+                .derived
+                .is_ranged
+                .cmp(&b.core_data.derived.is_ranged),
+            CreatureSortEnum::SpellCaster => a
+                .core_data
+                .derived
+                .is_spell_caster
+                .cmp(&b.core_data.derived.is_spell_caster),
         };
         match pagination
             .bestiary_sort_data

diff --git a/src/models/bestiary_structs.rs b/src/models/bestiary_structs.rs
@@ -29,6 +29,14 @@ pub enum CreatureSortEnum {
     Rarity,
     #[serde(alias = "family", alias = "FAMILY")]
     Family,
+    #[serde(alias = "alignment", alias = "ALIGNMENT")]
+    Alignment,
+    #[serde(alias = "melee", alias = "MELEE")]
+    Melee,
+    #[serde(alias = "ranged", alias = "RANGED")]
+    Ranged,
+    #[serde(alias = "SPELLCASTER", alias = "SPELLCASTER")]
+    SpellCaster,
 }
 
 #[derive(Serialize, Deserialize, IntoParams, ToSchema, Eq, PartialEq, Hash, Default)]