1 files changed, 167 insertions, 0 deletions
diff --git a/src/entities.rs b/src/entities.rs
new file mode 100644
index 0000000..145741c
--- /dev/null
+++ b/src/entities.rs
@@ -0,0 +1,167 @@
+use std::cmp;
+
+use crate::world::{Point, Direction};
+use crate::tiling::TileType;
+
+pub trait Entity {
+    fn info(&self) -> String;
+    fn place(&mut self, location: Point);
+}
+
+#[derive(Clone)]
+pub struct Character {
+    pub name: String,
+    pub class: String,
+    pub health: i32,
+    max_health: i32,
+    attack: i32,
+    dodge: i32,
+    luck: i32,
+    xp: i32,
+    location: Point,
+    tile_type: TileType
+}
+
+pub trait Enemy {
+    fn new(
+        class: String,
+        health: i32,
+        attack: i32,
+        dodge: i32,
+        luck: i32,
+        location: Point
+    ) -> Self;
+
+    fn set_tile_type(&mut self, tile_type: TileType);
+}
+
+pub trait Player {
+    fn new(
+        name: String,
+        class: String,
+        health: i32,
+        attack: i32,
+        dodge: i32,
+        luck: i32,
+        location: Point
+    ) -> Self;
+    fn damage(&mut self, damage_amount: i32);
+    fn heal(&mut self, heal_amount: i32);
+    fn attack(&self) -> i32;
+    fn dodge(&self) -> i32;
+    fn stats(&self) -> String;
+    fn walk(&mut self, dir: Direction);
+}
+
+impl Entity for Character {
+    fn place(&mut self, location: Point) {
+        self.location = location;
+    }
+
+    fn info(&self) -> String {
+        format!(
+            "{} \thp: {} attack: {} dodge: {} luck: {}",
+            self.class, self.health, self.attack, self.dodge, self.luck
+        )
+    }
+}
+
+impl Enemy for Character {
+    fn new(
+        class: String,
+        health: i32,
+        attack: i32,
+        dodge: i32,
+        luck: i32,
+        location: Point
+    ) -> Character {
+        Character {
+            name: class.clone(),
+            class: class.clone(),
+            max_health: health,
+            health,
+            attack,
+            dodge,
+            luck,
+            xp: 0,
+            location: location,
+            tile_type: TileType::Character
+        }
+    }
+
+    fn set_tile_type(&mut self, tile_type: TileType) {
+        self.tile_type = tile_type
+    }
+}
+
+impl Player for Character {
+    fn new(
+        name: String,
+        class: String,
+        health: i32,
+        attack: i32,
+        dodge: i32,
+        luck: i32,
+        location: Point
+    ) -> Character {
+        Character {
+            name: name,
+            class: class,
+            max_health: health,
+            health: health,
+            attack: attack,
+            dodge: dodge,
+            luck: luck,
+            xp: 0,
+            location: location,
+            tile_type: TileType::Player
+        }
+    }
+
+    fn walk(&mut self, dir: Direction) {
+        match dir {
+            Direction::North => { (); },
+            Direction::South => { (); },
+            Direction::East => { (); },
+            Direction::West => { (); }
+        }
+    }
+
+    fn damage(&mut self, damage_amount: i32) {
+        self.health = cmp::max(0, self.health - damage_amount);
+        self.xp += 2;
+    }
+
+    fn heal(&mut self, heal_amount: i32) {
+        if (self.health) <= self.max_health {
+            self.health = cmp::min(self.health + heal_amount, self.max_health);
+            self.xp += 1;
+        }
+    }
+
+    fn attack(&self) -> i32 {
+        self.xp + self.attack + self.luck / 2
+    }
+
+    fn dodge(&self) -> i32 {
+        self.xp + self.dodge + self.luck / 2
+    }
+
+    fn stats(&self) -> String {
+        format!(
+            "{} - hp: {}/{} attack: {} dodge: {} luck: {} experience: {}",
+            self.class, self.health, self.max_health, self.attack, self.dodge, self.luck, self.xp
+        )
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    fn test_attack() {
+        let bob: Character = Enemy::new("Rogue".to_string(), 1, 4, 1, 4, (0, 0));
+
+        assert_eq!(bob.attack(), 6);
+    }
+}