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

//! Interface for Screeps [`RawMemory`] global object.
//!
//! [`RawMemory`]: https://docs.screeps.com/api/#RawMemory

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});
    }
}