summaryrefslogtreecommitdiff
path: root/src/server/mining.rs
blob: 06f241859ff108fc58042298930baa8aeebf4293 (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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
extern crate serde_json;

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

use math::distance;
use mass::{Mass, MassType};
use module::{Module, ModuleType};
use server::connection::ServerConnection;

#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct MiningData {
    pub has_astroid_target  : bool,
    pub is_within_range     : bool,
    pub range               : f64,
    pub status              : bool,
}

impl ServerConnection {
    pub fn server_mining(&mut self, masses : &mut HashMap<String, Mass>) -> bool {
        let masses_clone = masses.clone();
        let ship = masses.get_mut(&self.name).unwrap();
        let ship_clone = ship.clone();


        if let MassType::Ship{ref mut modules, ..} = ship.mass_type {
            let mining_data = get_mining_data(ship_clone, modules, masses_clone);

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

            if let ModuleType::Mining{ref mut status, ..} = modules.get_mut("Mining").unwrap().module_type {
                let mut recv = String::new();
                match self.buff_r.read_line(&mut recv) {
                    Ok(result) => match recv.as_bytes() {
                        b"F\n" => {
                            if mining_data.is_within_range {
                                *status = !*status;
                            }
                        },
                        _ => {
                            if result == 0 {
                                return false
                            }
                        },
                    }
                    Err(_error) => (),
                }
            }
        }
        true
    }
}

fn get_mining_data(ship : Mass, modules : &mut HashMap<String, Module>, masses_clone : HashMap<String, Mass>) -> MiningData {
    let mut mining_range = 0.0;
    let mut mining_status = false;
    if let ModuleType::Mining{ref range, ref status, ..} = modules.get("Mining").unwrap().module_type {
        mining_range = *range;
        mining_status = *status;
    }

    let mut has_astroid_target = false;
    let mut is_within_range = false;
    if let ModuleType::Navigation{ref target_name, ..} = modules.get("Navigation").unwrap().module_type {
        match target_name.clone() {
            Some(name) => {
                let target = masses_clone.get(&name);
                has_astroid_target = match target {
                    Some(target) => match target.mass_type {
                        MassType::Astroid{..} => true,
                        _ => false,
                    },
                    None => false,
                };
                is_within_range = match has_astroid_target {
                    true => match target {
                        Some(target) => mining_range > distance(ship.position, target.position),
                        _ => false,
                    }
                    _ => false,
                };
            }
            _ => (),
        }
    }

    MiningData {
        has_astroid_target  : has_astroid_target,
        is_within_range     : is_within_range,
        range               : mining_range,
        status              : mining_status,
    }
}