sc2 0.1.6

organelle networks for StarCraft II Client API
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


use organelle;
use organelle::{ ResultExt, Cell, Protocol, Constraint };

use super::{ Result, Soma, Message, Role, Race, Difficulty, PlayerSetup };

/// cell that acts as the built-in SC2 AI
pub enum ComputerCell {
    /// initialize the soma
    Init(Init),
    /// respond to the setup queries
    Setup(Setup),
}

impl ComputerCell {
    /// create a new computer cell
    pub fn new(race: Race, difficulty: Difficulty) -> Result<Self> {
        Ok(
            ComputerCell::Init(
                Init {
                    soma: Soma::new(
                        vec![
                            Constraint::RequireOne(Role::Controller),
                            Constraint::RequireOne(Role::InstanceProvider),
                        ],
                        vec![ ],
                    )?,

                    setup: PlayerSetup::Computer {
                        race: race,
                        difficulty: difficulty,
                    },
                }
            )
        )
    }
}

impl Cell for ComputerCell {
    type Message = Message;
    type Role = Role;

    fn update(self, msg: Protocol<Message, Role>) -> organelle::Result<Self> {
        match self {
            ComputerCell::Init(state) => state.update(msg),
            ComputerCell::Setup(state) => state.update(msg),
        }.chain_err(
            || organelle::ErrorKind::CellError
        )
    }
}

pub struct Init {
    soma:           Soma,

    setup:          PlayerSetup,
}

impl Init {
    fn update(mut self, msg: Protocol<Message, Role>) -> Result<ComputerCell> {
        if let Some(msg) = self.soma.update(msg)? {
            match msg {
                Protocol::Start => Setup::setup(self.soma, self.setup),


                Protocol::Message(_, msg) => {
                    bail!("unexpected message {:#?}", msg)
                },
                _ => bail!("unexpected protocol message")
            }
        }
        else {
            Ok(ComputerCell::Init(self))
        }
    }
}

pub struct Setup {
    soma:           Soma,

    setup:          PlayerSetup,
}

impl Setup {
    fn setup(soma: Soma, setup: PlayerSetup) -> Result<ComputerCell> {
        Ok(ComputerCell::Setup(Setup { soma: soma, setup: setup }))
    }

    fn update(mut self, msg: Protocol<Message, Role>) -> Result<ComputerCell> {
        if let Some(msg) = self.soma.update(msg)? {
            match msg {
                Protocol::Message(_, Message::RequestPlayerSetup(_)) => {
                    self.soma.send_req_input(
                        Role::Controller, Message::PlayerSetup(self.setup)
                    )?;

                    Ok(ComputerCell::Setup(self))
                },

                Protocol::Message(_, msg) => {
                    bail!("unexpected message {:#?}", msg)
                },
                _ => bail!("unexpected protocol message")
            }
        }
        else {
            Ok(ComputerCell::Setup(self))
        }
    }
}