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
|
extern crate serde_json;
use std::collections::HashMap;
use std::io::Write;
use crate::constants;
use crate::item::ItemType;
use crate::mass::{Mass, MassType};
use crate::modules::construction::Construction;
use crate::modules::construction::ConstructionStatus;
use crate::modules::types::ModuleType;
use crate::server::connection::{receive, ServerConnection};
use crate::storage::Storage;
#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct ConstructionData {
pub status: ConstructionStatus,
pub has_enough: bool,
}
impl ServerConnection {
pub fn server_construction(&mut self, masses: &mut HashMap<String, Mass>) {
let mut ship = masses.remove(&self.name).unwrap();
if let MassType::Ship {
ref mut construction,
ref mut storage,
..
} = ship.mass_type
{
let construction = construction.as_mut().unwrap();
let construction_data = get_construction_data(storage, construction);
let send = serde_json::to_string(&construction_data).unwrap() + "\n";
self.open = self.stream.write(send.as_bytes()).is_ok();
match receive(&mut self.buff_r) {
Some(recv) => {
if let "c" = recv.as_str() {
if construction_data.has_enough {
construction.toggle();
}
}
}
None => self.open = false,
}
if construction_data.status == ConstructionStatus::Constructed {
storage.take_items(ItemType::Iron, constants::SHIP_CONSTRUCTION_IRON_COST);
masses.insert(
"Station".to_string(),
Mass::new_station(
ModuleType::Refinery,
ship.position.clone(),
ship.velocity.clone(),
),
);
construction.taken();
}
}
masses.insert(self.name.clone(), ship);
}
}
fn get_construction_data(storage: &Storage, construction: &Construction) -> ConstructionData {
ConstructionData {
status: construction.status.clone(),
has_enough: storage
.items
.iter()
.filter(|item| item.itemtype == ItemType::Iron)
.count()
>= constants::SHIP_CONSTRUCTION_IRON_COST,
}
}
|
