-added crude refinery methods

3 files changed, 77 insertions, 2 deletions

diff --git a/src/server/connection.rs b/src/server/connection.rs
index fd76661..d8e0eef 100644
--- a/src/server/connection.rs
+++ b/src/server/connection.rs
@@ -45,9 +45,10 @@ impl ServerConnection {
 
     pub fn process(&mut self, mut masses : &mut HashMap<String, Mass>) {
         self.open = match self.module_type {
-            ModuleType::Dashboard => self.server_dashboard(&mut masses),
-            ModuleType::Engines => self.server_engines(&mut masses),
             ModuleType::Mining => self.server_mining(&mut masses),
+            ModuleType::Engines => self.server_engines(&mut masses),
+            ModuleType::Refinery => self.server_refinery(&mut masses),
+            ModuleType::Dashboard => self.server_dashboard(&mut masses),
             ModuleType::Navigation => self.server_navigation(&mut masses),
         };
     }
diff --git a/src/server/mod.rs b/src/server/mod.rs
index 444e487..11dfdce 100644
--- a/src/server/mod.rs
+++ b/src/server/mod.rs
@@ -1,5 +1,6 @@
 pub mod mining;
 pub mod engines;
+pub mod refinery;
 pub mod dashboard;
 pub mod navigation;
 pub mod connection;
diff --git a/src/server/refinery.rs b/src/server/refinery.rs
new file mode 100644
index 0000000..bb0627f
--- /dev/null
+++ b/src/server/refinery.rs
@@ -0,0 +1,73 @@
+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>) -> bool {
+        let mut ship = masses.remove(&self.name).unwrap();
+        let ship_clone = ship.clone();
+        let mut refine = false;
+        let mut connection_good = true;
+
+        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,
+            };
+
+            let send = serde_json::to_string(&refinery_data).unwrap() + "\n";
+            match self.stream.write(send.as_bytes()) {
+                Ok(_result) => (),
+                Err(_error) => connection_good = 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 {
+                            refinery.toggle();
+                        }
+                    },
+                    _ => {
+                        if result == 0 {
+                            connection_good = false;
+                        }
+                    },
+                }
+                Err(_error) => (),
+            }
+
+            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);
+        connection_good
+    }
+}