summaryrefslogtreecommitdiff
path: root/src/mining.rs
blob: ecc83c516de56cf65e05ab098d21464bfdb63102 (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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
use std::io::{BufReader, BufRead};
use std::net::TcpStream;
use std::io::{stdout, Read, Write};
use termion::raw::IntoRawMode;
use termion::async_stdin;

extern crate serde_json;
extern crate termion;

use ship::Ship;
use math::distance;
use mass::{Mass, MassType};
use connection::Connection;

#[derive(Serialize, Deserialize, Debug, Clone)]
struct ServerData {
    has_astroid_target  : bool,
    is_within_range     : bool,
    mining_range        : f64,
    mining_status       : bool,
}

pub fn client_mining(mut stream : TcpStream, mut buff_r : BufReader<TcpStream>) {
    let stdout = stdout();
    let mut stdout = stdout.lock().into_raw_mode().unwrap();
    let mut stdin = async_stdin().bytes();

    loop {
        let mut recv = String::new();
        buff_r.read_line(&mut recv).unwrap();
        let data : ServerData = serde_json::from_str(&recv.replace("\n", "")).unwrap();

        write!(stdout, "{}", termion::clear::All).unwrap();

        match data.has_astroid_target {
            true => match data.is_within_range {
                true => write!(stdout, "{}Press F to begin mining.", termion::cursor::Goto(1,1)).unwrap(),
                false => write!(stdout, "{}Astroid must be within range of {}.", termion::cursor::Goto(1,1), data.mining_range).unwrap(),
            },
            false => write!(stdout, "{}Ship has no astroid targeted.", termion::cursor::Goto(1,1)).unwrap(),
        }

        match stdin.next() {
            Some(c) => {
                let c = c.unwrap();
                let mut send = String::new();
                send.push(c as char);
                if send.as_bytes() == b"q" {
                    break;
                }
                send.push_str("\n");
                stream.write(send.as_bytes()).unwrap();
            }
            None => ()
        }

        stdout.flush().unwrap();
    }
}

impl Connection {
    pub fn server_mining(&mut self, masses : &mut Vec<Box<Mass>>) -> bool {
        let m = masses.to_vec();
        let mass = masses.into_iter().find(|ship| ship.name() == &self.name).unwrap();
        let ship = mass.downcast_mut::<Ship>().unwrap();
        let target = match ship.recv_target() {
            Some(name) => m.iter().find(|target| target.name() == &name),
            None => None,
        };

        let has_astroid_target = match target {
            Some(target) => match target.recv_mass_type() {
                MassType::Ship => false,
                MassType::Astroid => true,
            },
            None => false,
        };

        let is_within_range = match has_astroid_target {
            true => match target {
                Some(target) => match ship.recv_mining_range() > distance(ship.position(), target.position()) {
                    true => true,
                    false => false,
                },
                None => false,
            }
            false => false,
        };

        let send = serde_json::to_string(&ServerData {
                                            has_astroid_target  : has_astroid_target,
                                            is_within_range     : is_within_range,
                                            mining_range        : ship.recv_mining_range(),
                                            mining_status       : ship.recv_mining_status(),
                                         }).unwrap() + "\n";

        match self.stream.write(send.as_bytes()) {
            Ok(_result) => (),
            Err(_error) => return false,
        }

        let mut recv = String::new();
        match self.buff_r.read_line(&mut recv) {
            Ok(result) => match recv.as_bytes() {
                b"F\n" => {
                    if is_within_range {
                        match ship.recv_mining_status() {
                            true => ship.stop_mining(),
                            false => ship.start_mining(),
                        }
                    }
                },
                _ => {
                    if result == 0 {
                        return false
                    }
                },
            }
            Err(_error) => (),
        }

        true
    }
}