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
|
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>) {
let mut ship = masses.remove(&self.name).unwrap();
let ship_clone = ship.clone();
let mut refine = false;
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,
};
if self.open {
if self.txrx_refinery(&refinery_data) {
refinery.toggle();
}
}
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);
}
fn txrx_refinery(&mut self, refinery_data : &RefineryData) -> bool {
let send = serde_json::to_string(refinery_data).unwrap() + "\n";
match self.stream.write(send.as_bytes()) {
Err(_error) => self.open = 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 {
return true
}
},
_ => {
if result == 0 {
self.open = false;
}
},
}
_ => (),
}
false
}
}
|