1 files changed, 16 insertions, 4 deletions
diff --git a/src/client/cli/cmd/edit.rs b/src/client/cli/cmd/edit.rs
index 7a02959..1cfb530 100644
--- a/src/client/cli/cmd/edit.rs
+++ b/src/client/cli/cmd/edit.rs
@@ -2,8 +2,9 @@
 
 use super::Command;
 use super::{CmdParseError, FromArgs};
-use crate::client::map::MapData;
 use crate::client::Editor;
+use crate::net::Cargo;
+use crate::world::World;
 use std::path::PathBuf;
 
 /// Command to load a file from the disk and replace the current editor contents with it's info.
@@ -25,8 +26,8 @@ impl FromArgs for Edit {
 
 impl Command for Edit {
     fn process(&self, editor: &mut Editor) -> Result<String, String> {
-        let data = match MapData::load_from_file(&self.file) {
-            Ok(data) => data,
+        let world = match World::load_from_file(&self.file) {
+            Ok(world) => world,
             Err(err) => {
                 return Err(format!(
                     "Unable to read file: {:?}, reason: {:?}",
@@ -35,7 +36,18 @@ impl Command for Edit {
             }
         };
 
-        editor.map_mut().set_data(data);
+        // Clear all data from the world, afterwards add all components from the file.
+        editor.server().send(Cargo::ClearAll);
+        for (_, icon) in world.icons().iter() {
+            editor.server().send(Cargo::AddIcon(icon.clone()));
+        }
+        for (_, room) in world.rooms().iter() {
+            editor.server().send(Cargo::AddRoom(room.clone()));
+        }
+        for (_, wall) in world.walls().iter() {
+            editor.server().send(Cargo::AddWall(wall.clone()));
+        }
+
         Ok(format!("Map data from {:?} loaded.", &self.file))
     }
 }