sc2 0.3.3

Rust implementation of the StarCraft II Client API
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

use ctrlc;
use futures::prelude::*;
use futures::sync;
use organelle::{Axon, Impulse, Soma};

use super::{Error, Result};
use synapses::Synapse;

/// soma that stops the organelle upon Ctrl-C
pub struct CtrlcBreakerSoma;

impl CtrlcBreakerSoma {
    /// create a new Ctrl-C breaker soma
    pub fn axon() -> Axon<Self> {
        Axon::new(Self {}, vec![], vec![])
    }
}

impl Soma for CtrlcBreakerSoma {
    type Synapse = Synapse;
    type Error = Error;

    #[async(boxed)]
    fn update(self, imp: Impulse<Self::Synapse>) -> Result<Self> {
        match imp {
            Impulse::Start(_, tx, handle) => {
                let (sync_tx, sync_rx) = sync::mpsc::channel(1);

                ctrlc::set_handler(move || {
                    if let Err(e) = sync_tx.clone().send(()).wait() {
                        eprintln!("unable to send Ctrl-C signal {:?}", e);
                    }
                })?;

                handle.spawn(
                    sync_rx
                        .and_then(move |_| {
                            tx.clone().send(Impulse::Stop).map_err(|_| ())
                        })
                        .into_future()
                        .map(|_| ())
                        .map_err(|_| ()),
                );

                Ok(self)
            },

            _ => Ok(self),
        }
    }
}