src/server/engines.rs


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

extern crate serde_json;

use std::collections::HashMap;
use std::io::BufRead;
use std::io::Write;

use crate::mass::{Mass, MassType};
use crate::modules::navigation::NavigationStatus;
use crate::server::connection::ServerConnection;

impl ServerConnection {
    pub fn server_engines(&mut self, masses: &mut HashMap<String, Mass>) {
        if self.open {
            let mut ship = masses.remove(&self.name).unwrap();
            let ship_clone = ship.clone();

            if let MassType::Ship {
                ref mut engines,
                ref navigation,
                ..
            } = ship.mass_type
            {
                let navigation = navigation.clone().unwrap();
                let engines = engines.as_mut().unwrap();
                let targeted = navigation.status == NavigationStatus::Targeted;

                let send = serde_json::to_string(&targeted).unwrap() + "\n";
                if let Err(_err) = self.stream.write(send.as_bytes()) {
                    self.open = false;
                }

                let target = match navigation.target_name {
                    Some(name) => masses.get(&name),
                    None => None,
                };

                let mut recv = String::new();
                if let Ok(result) = self.buff_r.read_line(&mut recv) {
                    engines.give_client_data(&ship_clone, target, recv);
                    if result == 0 {
                        self.open = false;
                    }
                }
            }

            masses.insert(self.name.clone(), ship);
        }
    }
}