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

//! Get global Screeps resources.
//!
//! This contains all functionality from the [`Game`] object in Screeps. That
//! generally means all state which is true this tick throughout the world.
//!
//! [`Game`]: http://docs.screeps.com/api/#Game
use crate::{
    macros::*,
    objects::{HasId, RoomObject, SizedRoomObject},
    traits::TryInto,
    ConversionError,
};

pub mod cpu;
pub mod gcl;
pub mod map;
pub mod market;
pub mod shards;

/// See [http://docs.screeps.com/api/#Game.constructionSites]
///
/// [http://docs.screeps.com/api/#Game.constructionSites]: http://docs.screeps.com/api/#Game.constructionSites
pub mod construction_sites {
    game_map_access!(objects::ConstructionSite, Game.constructionSites);
}

/// See [http://docs.screeps.com/api/#Game.creeps]
///
/// [http://docs.screeps.com/api/#Game.creeps]: http://docs.screeps.com/api/#Game.creeps
pub mod creeps {
    game_map_access!(objects::Creep, Game.creeps);
}

/// See [http://docs.screeps.com/api/#Game.flags]
///
/// [http://docs.screeps.com/api/#Game.flags]: http://docs.screeps.com/api/#Game.flags
pub mod flags {
    game_map_access!(objects::Flag, Game.flags);
}

// TODO: See [http://docs.screeps.com/api/#Game.resources]
///
/// [http://docs.screeps.com/api/#Game.resources]: http://docs.screeps.com/api/#Game.resources
/// See [http://docs.screeps.com/api/#Game.rooms]
///
/// [http://docs.screeps.com/api/#Game.rooms]: http://docs.screeps.com/api/#Game.rooms
pub mod rooms {
    use std::collections::HashMap;

    use crate::{local::RoomName, macros::*, objects::Room};

    /// Retrieve the full `HashMap<RoomName, Room>`.
    pub fn hashmap() -> HashMap<RoomName, Room> {
        // `TryFrom<Value>` is only implemented for `HashMap<String, V>`.
        //
        // See https://github.com/koute/stdweb/issues/359.
        let map: HashMap<String, Room> = js_unwrap!(Game.rooms);
        map.into_iter()
            .map(|(key, val)| {
                (
                    key.parse()
                        .expect("expected room name in Game.rooms to be valid"),
                    val,
                )
            })
            .collect()
    }

    /// Retrieve the string keys of this object.
    pub fn keys() -> Vec<String> {
        js_unwrap!(Object.keys(Game.rooms))
    }

    /// Retrieve all values in this object.
    pub fn values() -> Vec<Room> {
        js_unwrap_ref!(Object.values(Game.rooms))
    }

    /// Retrieve a specific value by key.
    pub fn get(name: RoomName) -> Option<Room> {
        js_unwrap_ref!(Game.rooms[@{name}])
    }
}

/// See [http://docs.screeps.com/api/#Game.spawns]
///
/// [http://docs.screeps.com/api/#Game.spawns]: http://docs.screeps.com/api/#Game.spawns
pub mod spawns {
    game_map_access!(objects::StructureSpawn, Game.spawns);
}

/// See [http://docs.screeps.com/api/#Game.structures]
///
/// [http://docs.screeps.com/api/#Game.structures]: http://docs.screeps.com/api/#Game.structures
pub mod structures {
    game_map_access!(objects::Structure, Game.structures);
}

/// See [http://docs.screeps.com/api/#Game.time]
///
/// [http://docs.screeps.com/api/#Game.time]: http://docs.screeps.com/api/#Game.time
pub fn time() -> u32 {
    js_unwrap!(Game.time)
}

/// See [http://docs.screeps.com/api/#Game.getObjectById]
///
/// This gets an object expecting a specific type and will return a
/// `ConversionError` if the type does not match.
///
/// If all you want to assume is that something has an ID, use
/// [`get_object_erased`].
///
/// [http://docs.screeps.com/api/#Game.getObjectById]: http://docs.screeps.com/api/#Game.getObjectById
pub fn get_object_typed<T>(id: &str) -> Result<Option<T>, ConversionError>
where
    T: HasId + SizedRoomObject,
{
    js!(return Game.getObjectById(@{id});).try_into()
}

/// See [http://docs.screeps.com/api/#Game.getObjectById]
///
/// This gets the object in 'erased' form - all that is known about it is that
/// it's a RoomObject.
///
/// If a more specific type is expected, [`get_object_typed`] can be used.
///
/// [http://docs.screeps.com/api/#Game.getObjectById]: http://docs.screeps.com/api/#Game.getObjectById
pub fn get_object_erased(id: &str) -> Option<RoomObject> {
    js_unwrap_ref!(Game.getObjectById(@{id}))
}

pub fn notify(message: &str, group_interval: Option<u32>) {
    js! { @(no_return)
        Game.notify(@{message}, @{group_interval.unwrap_or(0)});
    }
}