battler 0.9.1

Pokémon battle engine for Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

use alloc::{
    borrow::ToOwned,
    vec::Vec,
};

use anyhow::Result;
use battler_data::Id;
use hashbrown::HashMap;

use crate::{
    battle::{
        Context,
        EffectContext,
    },
    effect::{
        AppliedEffectHandle,
        AppliedEffectLocation,
        EffectHandle,
    },
    error::WrapOptionError,
};

/// Object for managing applied effects in battle and how they link to one another.
///
/// A linked effect ends when another effect ends. Linking always goes both ways.
pub struct LinkedEffectsManager {
    effects: HashMap<u32, AppliedEffectHandle>,
}

impl LinkedEffectsManager {
    /// Creates a new linked effects manager.
    pub fn new() -> Self {
        Self {
            effects: HashMap::default(),
        }
    }

    fn get_linked_id(context: &mut Context, effect: &AppliedEffectHandle) -> Result<Option<u32>> {
        let connector = match effect.effect_state_connector() {
            Some(connector) => connector,
            None => return Ok(None),
        };
        if !connector.exists(context)? {
            return Ok(None);
        }
        if let Some(id) = connector.get_mut(context)?.linked_id() {
            return Ok(Some(id));
        }
        let id = context.battle_mut().next_effect_linked_id();
        connector.get_mut(context)?.set_linked_id(id);
        context
            .battle_mut()
            .linked_effects_manager
            .effects
            .insert(id, effect.to_owned());
        Ok(Some(id))
    }

    /// Links one effect to another.
    pub fn link(
        context: &mut Context,
        from: &AppliedEffectHandle,
        to: &AppliedEffectHandle,
    ) -> Result<bool> {
        let from_uuid = match Self::get_linked_id(context, from)? {
            Some(uuid) => uuid,
            None => return Ok(false),
        };
        let to_uuid = match Self::get_linked_id(context, to)? {
            Some(uuid) => uuid,
            None => return Ok(false),
        };
        from.effect_state_connector()
            .wrap_expectation("expected applied effect to have effect state after getting link id")?
            .get_mut(context)?
            .add_link(to_uuid);
        to.effect_state_connector()
            .wrap_expectation("expected applied effect to have effect state after getting link id")?
            .get_mut(context)?
            .add_link(from_uuid);
        Ok(true)
    }

    /// Removes an effect by ID, effectively ending all linked effects.
    pub fn remove_by_id(
        context: &mut EffectContext,
        id: &Id,
        location: AppliedEffectLocation,
    ) -> Result<()> {
        let effect_handle = context.battle_mut().get_effect_handle_by_id(id)?.clone();
        Self::remove(context, effect_handle, location)
    }

    /// Removes an effect, effectively ending all linked effects.
    pub fn remove(
        context: &mut EffectContext,
        effect: EffectHandle,
        location: AppliedEffectLocation,
    ) -> Result<()> {
        let effect = AppliedEffectHandle::new(effect, location);
        let connector = match effect.effect_state_connector() {
            Some(connector) => connector,
            None => return Ok(()),
        };
        if !connector.exists(context.as_battle_context_mut())? {
            return Ok(());
        }

        let linked_id = match connector
            .get_mut(context.as_battle_context_mut())?
            .linked_id()
        {
            Some(linked_id) => linked_id,
            None => return Ok(()),
        };

        context
            .battle_mut()
            .linked_effects_manager
            .effects
            .remove(&linked_id);

        let mut context = context.forward_effect_context(effect.effect_handle.clone())?;
        for link in connector
            .get_mut(context.as_battle_context_mut())?
            .linked_to()
            .cloned()
            .collect::<Vec<_>>()
        {
            Self::remove_link(&mut context, link, linked_id)?;
        }

        Ok(())
    }

    fn remove_link(context: &mut EffectContext, link: u32, remove: u32) -> Result<()> {
        let linked_effect = match context.battle().linked_effects_manager.effects.get(&link) {
            Some(effect) => effect.clone(),
            None => return Ok(()),
        };
        let connector = match linked_effect.effect_state_connector() {
            Some(connector) => connector,
            None => return Ok(()),
        };

        if !connector.exists(context.as_battle_context_mut())? {
            return Ok(());
        }

        let effect_state = connector.get_mut(context.as_battle_context_mut())?;
        effect_state.remove_link(remove);

        linked_effect.end(context)?;

        Ok(())
    }
}