editor: save

src/editor.rs

@@ -21,6 +21,7 @@ impl Plugin for EditorPlugin {
 			.add_system_set(
 				SystemSet::on_update(AppState::Editor)
 					.with_system(move_system)
+					.with_system(input_control_system)
 					.with_system(follow_ends_system),
 			);
 	}
@@ -32,10 +33,15 @@ struct DragEndEvent(Entity);
 
 // Resources
 
+struct CharacterList(Vec<Entity>);
+
 // Components
 
 #[derive(Component)]
-struct Platform(Entity, Entity);
+struct Platform;
+
+#[derive(Component)]
+struct Ends(Entity, Entity);
 
 #[derive(Component)]
 struct Draggable;
@@ -43,12 +49,20 @@ struct Draggable;
 #[derive(Component)]
 struct End(Entity);
 
+#[derive(Component)]
+struct CharacterColor(Color);
+
+#[derive(Component)]
+struct Size(Vec2);
+
 // Bundles
 
 #[derive(Bundle)]
 struct PlatformBundle {
 	#[bundle]
 	mesh: ColorMesh2dBundle,
+	size: Size,
+	platform: Platform,
 }
 
 #[derive(Bundle)]
@@ -65,6 +79,7 @@ struct PlatformEndBundle {
 struct CharacterBundle {
 	#[bundle]
 	mesh: ColorMesh2dBundle,
+	color: CharacterColor,
 	#[bundle]
 	pickable: PickableBundle,
 	draggable: Draggable,
@@ -88,9 +103,11 @@ fn spawn_platform(
 				transform,
 				..default()
 			},
+			size: Size(size),
+			platform: Platform,
 		})
 		.id();
-	let ends = Platform(
+	let ends = Ends(
 		commands
 			.spawn_bundle(PlatformEndBundle {
 				mesh: ColorMesh2dBundle {
@@ -148,7 +165,7 @@ fn spawn_character(
 	transform: Transform,
 	color: Color,
 	index: usize,
-) {
+) -> Entity {
 	let font = asset_server.get_handle("UacariLegacy-Thin.ttf");
 	commands
 		.spawn_bundle(CharacterBundle {
@@ -163,6 +180,7 @@ fn spawn_character(
 				transform,
 				..default()
 			},
+			color: CharacterColor(color),
 			pickable: PickableBundle::default(),
 			draggable: Draggable,
 		})
@@ -180,10 +198,11 @@ fn spawn_character(
 				transform: Transform::from_xyz(0., 0., 1.),
 				..Default::default()
 			});
-		});
+		})
+		.id()
 }
 
-pub fn spawn_stored_level(
+fn spawn_stored_level(
 	commands: &mut Commands,
 	meshes: &mut ResMut<Assets<Mesh>>,
 	materials: &mut ResMut<Assets<ColorMaterial>>,
@@ -201,8 +220,9 @@ pub fn spawn_stored_level(
 		);
 	}
 
+	let mut character_list = Vec::new();
 	for (i, character) in stored_level.characters.iter().enumerate() {
-		spawn_character(
+		character_list.push(spawn_character(
 			commands,
 			meshes,
 			materials,
@@ -210,7 +230,57 @@ pub fn spawn_stored_level(
 			Transform::from_xyz(character.pos.x, character.pos.y, 0.),
 			character.color.into(),
 			i,
-		);
+		));
+	}
+	commands.insert_resource(CharacterList(character_list));
+}
+
+fn save_level(
+	level_id: &Res<crate::game::FirstLevel>,
+	stored_levels_assets: &mut ResMut<Assets<StoredLevels>>,
+	stored_levels_handle: &Res<Handle<StoredLevels>>,
+	character_list: &Res<CharacterList>,
+	character_query: &Query<(&Transform, &CharacterColor), Without<Platform>>,
+	platform_query: &Query<(&Transform, &Size), With<Platform>>,
+) {
+	let stored_levels = stored_levels_assets.get_mut(stored_levels_handle).unwrap();
+	if let Some(stored_level) = stored_levels.levels.get_mut(level_id.0 .0 as usize) {
+		stored_level.platforms.clear();
+		for (transform, size) in platform_query.iter() {
+			stored_level.platforms.push(StoredPlatform {
+				pos: Vec2 {
+					x: transform.translation.x,
+					y: transform.translation.y,
+				},
+				size: size.0,
+			})
+		}
+
+		stored_level.characters.clear();
+		for entity in character_list.0.iter() {
+			if let Ok((transform, color)) = character_query.get(*entity) {
+				stored_level.characters.push(StoredCharacter {
+					pos: Vec2 {
+						x: transform.translation.x,
+						y: transform.translation.y,
+					},
+					color: color.0.as_rgba_f32().into(),
+				})
+			}
+		}
+	} else {
+		return;
+	}
+	match std::fs::OpenOptions::new()
+		.write(true)
+		.truncate(true)
+		.open("assets/game.levels.json")
+	{
+		Ok(mut file) => {
+			serde_json::to_writer_pretty(&mut file, stored_levels).unwrap();
+			println!("Saved!");
+		}
+		Err(e) => eprintln!("Error writing levels file: {:?}", e),
 	}
 }
 
@@ -246,6 +316,29 @@ fn setup(
 	}
 }
 
+fn input_control_system(
+	keyboard_input: Res<Input<KeyCode>>,
+	level_id: Res<crate::game::FirstLevel>,
+	mut stored_levels_assets: ResMut<Assets<StoredLevels>>,
+	stored_levels_handle: Res<Handle<StoredLevels>>,
+	character_list: Res<CharacterList>,
+	character_query: Query<(&Transform, &CharacterColor), Without<Platform>>,
+	platform_query: Query<(&Transform, &Size), With<Platform>>,
+) {
+	if keyboard_input.just_released(KeyCode::S)
+		&& (keyboard_input.pressed(KeyCode::LControl) || keyboard_input.pressed(KeyCode::RControl))
+	{
+		save_level(
+			&level_id,
+			&mut stored_levels_assets,
+			&stored_levels_handle,
+			&character_list,
+			&character_query,
+			&platform_query,
+		);
+	}
+}
+
 fn move_system(
 	keyboard_input: Res<Input<KeyCode>>,
 	mut camera_query: Query<&mut Transform, (With<Camera>, Without<Draggable>)>,
@@ -297,22 +390,24 @@ fn move_system(
 
 fn follow_ends_system(
 	mut meshes: ResMut<Assets<Mesh>>,
-	mut platform_query: Query<(&mut Transform, &mut Mesh2dHandle, &Platform)>,
-	end_query: Query<&Transform, Without<Platform>>,
+	mut follower_query: Query<(&mut Transform, &mut Mesh2dHandle, &mut Size, &Ends)>,
+	end_query: Query<&Transform, Without<Ends>>,
 	mut drag_end_event: EventReader<DragEndEvent>,
 ) {
 	for DragEndEvent(entity) in drag_end_event.iter() {
-		if let Ok((mut transform, mut mesh, Platform(end1, end2))) = platform_query.get_mut(*entity)
+		if let Ok((mut transform, mut mesh, mut size, Ends(end1, end2))) =
+			follower_query.get_mut(*entity)
 		{
 			if let (Ok(end1), Ok(end2)) = (end_query.get(*end1), end_query.get(*end2)) {
 				transform.translation.x = (end1.translation.x + end2.translation.x) / 2.;
 				transform.translation.y = (end1.translation.y + end2.translation.y) / 2.;
+				size.0 = Vec2 {
+					x: (end2.translation.x - end1.translation.x).abs(),
+					y: (end2.translation.y - end1.translation.y).abs(),
+				};
 				*mesh = meshes
 					.add(Mesh::from(Quad {
-						size: Vec2 {
-							x: (end2.translation.x - end1.translation.x).abs(),
-							y: (end2.translation.y - end1.translation.y).abs(),
-						},
+						size: size.0,
 						flip: false,
 					}))
 					.into();