From 1069173d834975ef2acf7067714b120254bad71c Mon Sep 17 00:00:00 2001
From: appcrashwin7 <xcv678@wp.pl>
Date: Fri, 4 Oct 2019 17:48:14 +0200
Subject: [PATCH] add give_exp command

---
 server/src/cmd.rs | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/server/src/cmd.rs b/server/src/cmd.rs
index 2a8e90491c..9889268cae 100644
--- a/server/src/cmd.rs
+++ b/server/src/cmd.rs
@@ -213,6 +213,13 @@ lazy_static! {
              false,
              handle_debug_column,
          ),
+         ChatCommand::new(
+             "give_exp",
+             "{} {}",
+             "/give_exp <playername> <amount> : Give experience to specified player",
+             true,
+             handle_exp,
+         ),
     ];
 }
 
@@ -916,3 +923,33 @@ spawn_rate {:?} "#,
             .notify(entity, ServerMsg::private(String::from(action.help_string)));
     }
 }
+
+fn handle_exp(server: &mut Server, entity: EcsEntity, args: String, action: &ChatCommand) {
+    let (a_alias, a_exp) = scan_fmt_some!(&args, action.arg_fmt, String, i64);
+    if let (Some(alias), Some(exp)) = (a_alias, a_exp) {
+        let ecs = server.state.ecs_mut();
+        let opt_player = (&ecs.entities(), &ecs.read_storage::<comp::Player>())
+            .join()
+            .find(|(_, player)| player.alias == alias)
+            .map(|(entity, _)| entity);
+
+        match opt_player {
+            Some(_alias) => {
+                if let Some(stats) = ecs.write_storage::<comp::Stats>().get_mut(entity) {
+                    stats.exp.change_by(exp);
+                } else {
+                    server.clients.notify(
+                        entity,
+                        ServerMsg::private(String::from("Something go wrong")),
+                    );
+                }
+            }
+            _ => {
+                server.clients.notify(
+                    entity,
+                    ServerMsg::private(format!("Player '{}' not found!", alias)),
+                );
+            }
+        }
+    }
+}