summaryrefslogtreecommitdiff
path: root/src/server/refinery.rs
blob: bb0627f49cccc654b2d7f3f31269b48d318ca47e (plain)
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
extern crate serde_json;

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

use item::Item;
use mass::{Mass, MassType};
use server::connection::ServerConnection;

#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct RefineryData {
    pub has_minerals    : bool,
    pub status          : bool,
}

impl ServerConnection {
    pub fn server_refinery(&mut self, masses : &mut HashMap<String, Mass>) -> bool {
        let mut ship = masses.remove(&self.name).unwrap();
        let ship_clone = ship.clone();
        let mut refine = false;
        let mut connection_good = true;

        if let MassType::Ship{ref mut refinery, ..} = ship.mass_type {
            let mut refinery = refinery.as_mut().unwrap();

            let refinery_data = RefineryData {
                has_minerals    : ship_clone.has_minerals(),
                status          : refinery.status,
            };

            let send = serde_json::to_string(&refinery_data).unwrap() + "\n";
            match self.stream.write(send.as_bytes()) {
                Ok(_result) => (),
                Err(_error) => connection_good = false,
            }

            let mut recv = String::new();
            match self.buff_r.read_line(&mut recv) {
                Ok(result) => match recv.as_bytes() {
                    b"R\n" => {
                        if refinery_data.has_minerals {
                            refinery.toggle();
                        }
                    },
                    _ => {
                        if result == 0 {
                            connection_good = false;
                        }
                    },
                }
                Err(_error) => (),
            }

            if !refinery_data.has_minerals {
                refinery.off();
            }

            if refinery.status && refinery.ready {
                refinery.take();
                refine = true;
            }
        }
    
        if refine {
            ship.take("Iron");
            ship.give(Item::new("Refined Iron", 1));
        }

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