1 files changed, 132 insertions, 0 deletions
diff --git a/client/src/world/mod.rs b/client/src/world/mod.rs
new file mode 100644
index 0000000..ae985a6
--- /dev/null
+++ b/client/src/world/mod.rs
@@ -0,0 +1,132 @@
+use gamenet::{
+    enums::{Emote, Team, Weapon},
+    SnapObj,
+};
+
+use self::map::Map;
+use crate::client::{helper::get_map_path, ClientMesgOut};
+use std::{collections::BTreeMap, fs::File};
+
+pub mod map;
+
+pub use gamenet::enums;
+
+#[derive(Debug)]
+pub struct Tee {
+    pub local: bool,
+    pub latency: i32,
+    pub score: i32,
+
+    pub team: Team,
+    pub weapon: Weapon,
+    pub armor: i32,
+    pub ammo: i32,
+    pub emote: Emote,
+    pub attack_tick: i32,
+
+    pub tick: i32,
+    pub angle: i32,
+    pub x: i32,
+    pub y: i32,
+    pub vel_x: i32,
+    pub vel_y: i32,
+    pub hook_x: i32,
+    pub hook_y: i32,
+    pub hook_dx: i32,
+    pub hook_dy: i32,
+    pub hook_player: i32,
+    pub hook_state: i32,
+}
+
+impl Default for Tee {
+    fn default() -> Self {
+        Self {
+            x: Default::default(),
+            y: Default::default(),
+            local: false,
+            team: Team::Spectators,
+            latency: Default::default(),
+            score: Default::default(),
+            weapon: Weapon::Shotgun,
+            armor: Default::default(),
+            ammo: Default::default(),
+            attack_tick: Default::default(),
+            emote: Emote::Normal,
+            tick: Default::default(),
+            angle: Default::default(),
+            vel_x: Default::default(),
+            vel_y: Default::default(),
+            hook_x: Default::default(),
+            hook_y: Default::default(),
+            hook_dx: Default::default(),
+            hook_dy: Default::default(),
+            hook_player: Default::default(),
+            hook_state: Default::default(),
+        }
+    }
+}
+
+pub struct World {
+    pub map: Map,
+    pub tees: BTreeMap<u16, Tee>,
+}
+
+impl World {
+    pub fn new() -> Self {
+        Self {
+            map: Map::empty(),
+            tees: BTreeMap::new(),
+        }
+    }
+
+    pub fn update(&mut self, m: &ClientMesgOut) {
+        match m {
+            ClientMesgOut::MapChange { name, crc } => {
+                let file = File::open(get_map_path(name.as_str(), *crc)).unwrap();
+                self.map = Map::load(file).unwrap();
+            }
+            ClientMesgOut::Snaps(s) => {
+                self.tees.clear();
+                for (id, o) in s {
+                    match o {
+                        SnapObj::ClientInfo(_o) => {
+                            // TODO
+                        }
+                        SnapObj::PlayerInfo(o) => {
+                            let e = self.tees.entry(*id).or_default();
+                            e.local = o.local == 1;
+                            e.team = o.team;
+                            e.latency = o.latency;
+                            e.score = o.score;
+                        }
+                        SnapObj::Character(c) => {
+                            let e = self.tees.entry(*id).or_default();
+                            e.ammo = c.ammo_count;
+                            e.weapon = c.weapon;
+                            e.emote = c.emote;
+                            e.attack_tick = c.attack_tick;
+
+                            e.x = c.character_core.x;
+                            e.y = c.character_core.y;
+                            e.vel_x = c.character_core.vel_x;
+                            e.vel_y = c.character_core.vel_y;
+
+                            e.tick = c.character_core.tick;
+                            e.hook_x = c.character_core.hook_x;
+                            e.hook_y = c.character_core.hook_y;
+                            e.hook_player = c.character_core.hooked_player;
+                            e.hook_dx = c.character_core.hook_dx;
+                            e.hook_dy = c.character_core.hook_dy;
+                            e.hook_state = c.character_core.hook_state;
+                        }
+                        _ => (),
+                    }
+                }
+            }
+        }
+    }
+
+    pub fn local_tee(&self) -> Option<&Tee> {
+        self.tees.values().find(|e| e.local)
+    }
+}