Skip to main content

manabrew_engine/trigger/
trigger_rolled_die.rs

1use serde::{Deserialize, Serialize};
2
3use crate::event::RunParams;
4use crate::game::GameState;
5use crate::parsing::compare::compare_expr;
6use crate::parsing::{keys, Params};
7use crate::spellability::SpellAbility;
8use crate::trigger::TriggerType;
9
10use super::trigger::TriggerBehavior;
11
12#[derive(Debug, Clone, Serialize, Deserialize)]
13pub struct TriggerRolledDie {
14    pub valid_player: Option<crate::parsing::CompiledSelector>,
15    pub valid_result: Option<String>,
16    pub valid_sides: Option<String>,
17    pub number: Option<i32>,
18    pub natural: bool,
19    pub rolled_to_visit_attractions: bool,
20}
21
22impl TriggerRolledDie {
23    pub fn parse(params: &Params) -> Box<dyn TriggerBehavior> {
24        Box::new(Self {
25            valid_player: params.selector_cloned(keys::VALID_PLAYER),
26            valid_result: params.get_cloned(keys::VALID_RESULT),
27            valid_sides: params.get_cloned(keys::VALID_SIDES),
28            number: params.get("Number").and_then(|n| n.parse::<i32>().ok()),
29            natural: params.is_true("Natural"),
30            rolled_to_visit_attractions: params.has("RolledToVisitAttractions"),
31        })
32    }
33}
34
35#[typetag::serde]
36impl TriggerBehavior for TriggerRolledDie {
37    fn trigger_type(&self) -> TriggerType {
38        TriggerType::RolledDie
39    }
40
41    fn perform_test(
42        &self,
43        trigger: &super::trigger::Trigger,
44        params: &RunParams,
45        game: &GameState,
46    ) -> bool {
47        let _host_controller = trigger.base.card_trait_base.host_controller(game);
48        if !trigger.matches_optional_valid_player_filter(&self.valid_player, params.player, game) {
49            return false;
50        }
51        if self.rolled_to_visit_attractions && params.rolled_to_visit_attractions != Some(true) {
52            return false;
53        }
54        if let Some(filter) = self.valid_result.as_ref() {
55            let result = if self.natural {
56                params.natural_result
57            } else {
58                params.die_result
59            };
60            let Some(result) = result else {
61                return false;
62            };
63            if !matches_die_filter(filter, result, params.die_sides) {
64                return false;
65            }
66        }
67        if let Some(filter) = self.valid_sides.as_ref() {
68            let Some(sides) = params.die_sides else {
69                return false;
70            };
71            if !compare_expr(sides, filter) {
72                return false;
73            }
74        }
75        if let Some(expected_number) = self.number {
76            if params.number != Some(expected_number) {
77                return false;
78            }
79        }
80        true
81    }
82
83    fn set_triggering_objects(
84        &self,
85        _trigger: &super::trigger::Trigger,
86        sa: &mut SpellAbility,
87        params: &RunParams,
88        _game: &GameState,
89    ) {
90        if let Some(result) = params.die_result {
91            sa.set_triggering_object(crate::ability::AbilityKey::Result, result.to_string());
92        }
93        if let Some(p) = params.player {
94            sa.set_triggering_object(crate::ability::AbilityKey::Player, p.0.to_string());
95        }
96    }
97
98    fn get_important_stack_objects(
99        &self,
100        _trigger: &super::trigger::Trigger,
101        sa: &SpellAbility,
102    ) -> String {
103        format!(
104            "Player: {}, Result: {}",
105            sa.trigger_objects
106                .get(&crate::ability::AbilityKey::Player)
107                .map(|s| s.as_str())
108                .unwrap_or(""),
109            sa.trigger_objects
110                .get(&crate::ability::AbilityKey::Result)
111                .map(|s| s.as_str())
112                .unwrap_or("")
113        )
114    }
115}
116
117fn matches_die_filter(filter: &str, result: i32, sides: Option<i32>) -> bool {
118    for entry in filter
119        .split(',')
120        .map(str::trim)
121        .filter(|entry| !entry.is_empty())
122    {
123        if entry.eq_ignore_ascii_case("Highest") {
124            if sides == Some(result) {
125                return true;
126            }
127            continue;
128        }
129        if let Ok(value) = entry.parse::<i32>() {
130            if value == result {
131                return true;
132            }
133            continue;
134        }
135        if entry.len() >= 3 && compare_expr(result.max(0), entry) {
136            return true;
137        }
138    }
139    false
140}