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

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

use crate::item::{Item, ItemType};
use crate::mass::{Mass, MassType};
use crate::modules::refinery::RefineryStatus;
use crate::server::connection::ServerConnection;

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

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

        if let MassType::Ship {
            ref mut refinery,
            ref mut storage,
            ..
        } = ship.mass_type
        {
            let refinery_data = RefineryData {
                has_crude_minerals: storage
                    .items
                    .iter()
                    .any(|item| item.itemtype == ItemType::CrudeMinerals),
                status: refinery.status.clone(),
            };

            let send = serde_json::to_string(&refinery_data).unwrap() + "\n";
            self.open = self.stream.write(send.as_bytes()).is_ok();

            let recv = self.receive();
            refinery.give_recv(recv, refinery_data);

            if refinery.status == RefineryStatus::Refined {
                storage.take_item(ItemType::CrudeMinerals);
                storage.give_item(Item::new(ItemType::Iron));
                storage.give_item(Item::new(ItemType::Hydrogen));
                refinery.taken();
            }
        }

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