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

//! Interface for Screeps `RawMemory` global object.

use serde::Deserialize;

use crate::macros::*;

#[derive(Deserialize, Debug)]
pub struct ForeignSegment {
    username: String,
    id: String,
    data: String,
}

js_deserializable!(ForeignSegment);

get_from_js!(get_active_segments -> {
    Object.keys(RawMemory.segments).map(Number)
} -> Vec<u32>);

/// Sets active segments (max 10 ids).
pub fn set_active_segments(ids: &[u32]) {
    assert!(
        ids.len() <= 10,
        "can't set more than 10 active segments at a time"
    );
    js! { @(no_return)
        RawMemory.setActiveSegments(@{ids});
    }
}

get_from_js!(get_segment(id: u32) -> {
    RawMemory.segments[@{id}]
} -> Option<String>);

pub fn set_segment(id: u32, data: &str) {
    js! { @(no_return)
        RawMemory.segments[@{id}] = @{data};
    }
}

/// This drops the reference to a segment; it doesn't affect the content of the
/// segment.
///
/// This is the equivalent of doing `delete RawMemory.segments[id]`. Again, this
/// only deletes the local view of the segment, not the serialized one. It may
/// be used to `set_segment` a new segment that wasn't part of the original 10
/// active segments.
pub fn drop_segment(id: u32) {
    js! { @(no_return)
        delete RawMemory.segments[@{id}];
    }
}

get_from_js!(get_foreign_segment -> {
    RawMemory.foreignSegment
} -> ForeignSegment);

/// Implements `RawMemory.setActiveForeignSegment`
///
/// To use the default public segment of `username` (as set with
/// [`set_default_public_segment`]), Use `None` instead of `Some(id)`.
///
/// To clear the foreign segment, pass the empty string `""` as a username.
pub fn set_active_foreign_segment(username: &str, id: Option<u32>) {
    if username == "" {
        js! { @(no_return)
            RawMemory.setActiveForeignSegment(null);
        }
    } else {
        match id {
            Some(id) => js! { @(no_return)
                RawMemory.setActiveForeignSegment(@{username}, @{id});
            },
            None => js! { @(no_return)
                RawMemory.setActiveForeignSegment(@{username});
            },
        };
    };
}

pub fn set_default_public_segment(id: u32) {
    js! { @(no_return)
        RawMemory.setDefaultPublicSegment(@{id});
    }
}

pub fn set_public_segments(ids: &[u32]) {
    js! { @(no_return)
        RawMemory.setPublicSegments(@{ids});
    }
}

get_from_js!(get -> {RawMemory.get()} -> String);

pub fn set(value: &str) {
    js! { @(no_return)
        RawMemory.set(@{value});
    }
}