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

use js_sys::Array;
use wasm_bindgen::{prelude::*, JsCast};

use crate::{
    constants::EffectType,
    local::Position,
    objects::{Room, RoomPosition},
    prelude::*,
};

#[wasm_bindgen]
extern "C" {
    /// Parent class for all objects in rooms in the game world.
    ///
    /// [Screeps documentation](https://docs.screeps.com/api/#Room)
    #[derive(Clone, Debug)]
    pub type RoomObject;

    /// Effects applied to the object.
    ///
    /// [Screeps documentation](https://docs.screeps.com/api/#RoomObject.effects)
    #[wasm_bindgen(method, getter = effects)]
    fn effects_internal(this: &RoomObject) -> Option<Array>;

    /// Gets the [`RoomPosition`] of an object, which is a reference to an
    /// object in the javascript heap. In most cases, you'll likely want a
    /// native [`Position`] instead of using this function (see
    /// [`HasPosition::pos`]), there may be cases where this can provide
    /// some slight performance benefits due to reducing object churn in the js
    /// heap, so this is kept public.
    ///
    /// [Screeps documentation](https://docs.screeps.com/api/#RoomObject.pos)
    #[wasm_bindgen(method, getter = pos)]
    pub fn js_pos(this: &RoomObject) -> RoomPosition;

    /// A link to the room that the object is currently in, or `None` if the
    /// object is a power creep not spawned on the current shard, or a flag or
    /// construction site not in a visible room.
    ///
    /// [Screeps documentation](https://docs.screeps.com/api/#RoomObject.room)
    #[wasm_bindgen(method, getter)]
    pub fn room(this: &RoomObject) -> Option<Room>;
}

#[wasm_bindgen]
extern "C" {
    #[wasm_bindgen]
    #[derive(Debug)]
    pub type Effect;

    #[wasm_bindgen(method, getter)]
    pub fn effect(this: &Effect) -> EffectType;

    #[wasm_bindgen(method, getter)]
    pub fn level(this: &Effect) -> Option<u8>;

    #[wasm_bindgen(method, getter = ticksRemaining)]
    pub fn ticks_remaining(this: &Effect) -> EffectType;
}

impl<T> HasPosition for T
where
    T: AsRef<RoomObject>,
{
    fn pos(&self) -> Position {
        self.as_ref().js_pos().into()
    }
}

impl<T> RoomObjectProperties for T
where
    T: AsRef<RoomObject>,
{
    fn effects(&self) -> Vec<Effect> {
        RoomObject::effects_internal(self.as_ref())
            .map(|arr| arr.iter().map(JsCast::unchecked_into).collect())
            .unwrap_or_default()
    }

    fn effects_raw(&self) -> Option<Array> {
        RoomObject::effects_internal(self.as_ref())
    }

    fn room(&self) -> Option<Room> {
        RoomObject::room(self.as_ref())
    }
}