Move login state to the ECS (#213)

* use packet handlers code for login custom_query

* initial broken implementation for ecs-only login

* fixes

* run Update schedule 60 times per second and delete code related to run_schedule_sender

* fix tests

* fix online-mode

* reply to query packets in a separate system and make it easier for plugins to disable individual replies

* remove unused imports

1 files changed, 148 insertions, 167 deletions

diff --git a/azalea-client/src/plugins/packet/game/mod.rs b/azalea-client/src/plugins/packet/game/mod.rs
index 8d896e65..60531d3b 100644
--- a/azalea-client/src/plugins/packet/game/mod.rs
+++ b/azalea-client/src/plugins/packet/game/mod.rs
@@ -32,171 +32,150 @@ use crate::{
     },
     movement::{KnockbackEvent, KnockbackType},
     packet::as_system,
-    raw_connection::RawConnection,
 };
 
-pub fn process_packet_events(ecs: &mut World) {
-    let mut events_owned = Vec::<(Entity, Arc<ClientboundGamePacket>)>::new();
-
-    {
-        let mut system_state = SystemState::<EventReader<ReceivePacketEvent>>::new(ecs);
-        let mut events = system_state.get_mut(ecs);
-        for ReceivePacketEvent {
-            entity: player_entity,
-            packet,
-        } in events.read()
-        {
-            // we do this so `ecs` isn't borrowed for the whole loop
-            events_owned.push((*player_entity, packet.clone()));
-        }
-    }
-
-    for (player_entity, packet) in events_owned {
-        let mut handler = GamePacketHandler {
-            player: player_entity,
-            ecs,
-        };
-
-        // the order of these doesn't matter, that's decided by the protocol library
-        declare_packet_handlers!(
-            ClientboundGamePacket,
-            packet.as_ref(),
-            handler,
-            [
-                login,
-                set_chunk_cache_radius,
-                chunk_batch_start,
-                chunk_batch_finished,
-                custom_payload,
-                change_difficulty,
-                commands,
-                player_abilities,
-                set_cursor_item,
-                update_tags,
-                disconnect,
-                update_recipes,
-                entity_event,
-                player_position,
-                player_info_update,
-                player_info_remove,
-                set_chunk_cache_center,
-                chunks_biomes,
-                light_update,
-                level_chunk_with_light,
-                add_entity,
-                set_entity_data,
-                update_attributes,
-                set_entity_motion,
-                set_entity_link,
-                initialize_border,
-                set_time,
-                set_default_spawn_position,
-                set_health,
-                set_experience,
-                teleport_entity,
-                update_advancements,
-                rotate_head,
-                move_entity_pos,
-                move_entity_pos_rot,
-                move_entity_rot,
-                keep_alive,
-                remove_entities,
-                player_chat,
-                system_chat,
-                disguised_chat,
-                sound,
-                level_event,
-                block_update,
-                animate,
-                section_blocks_update,
-                game_event,
-                level_particles,
-                server_data,
-                set_equipment,
-                update_mob_effect,
-                award_stats,
-                block_changed_ack,
-                block_destruction,
-                block_entity_data,
-                block_event,
-                boss_event,
-                command_suggestions,
-                container_set_content,
-                container_set_data,
-                container_set_slot,
-                container_close,
-                cooldown,
-                custom_chat_completions,
-                delete_chat,
-                explode,
-                forget_level_chunk,
-                horse_screen_open,
-                map_item_data,
-                merchant_offers,
-                move_vehicle,
-                open_book,
-                open_screen,
-                open_sign_editor,
-                ping,
-                place_ghost_recipe,
-                player_combat_end,
-                player_combat_enter,
-                player_combat_kill,
-                player_look_at,
-                remove_mob_effect,
-                resource_pack_push,
-                resource_pack_pop,
-                respawn,
-                start_configuration,
-                entity_position_sync,
-                select_advancements_tab,
-                set_action_bar_text,
-                set_border_center,
-                set_border_lerp_size,
-                set_border_size,
-                set_border_warning_delay,
-                set_border_warning_distance,
-                set_camera,
-                set_display_objective,
-                set_objective,
-                set_passengers,
-                set_player_team,
-                set_score,
-                set_simulation_distance,
-                set_subtitle_text,
-                set_title_text,
-                set_titles_animation,
-                clear_titles,
-                sound_entity,
-                stop_sound,
-                tab_list,
-                tag_query,
-                take_item_entity,
-                bundle_delimiter,
-                damage_event,
-                hurt_animation,
-                ticking_state,
-                ticking_step,
-                reset_score,
-                cookie_request,
-                debug_sample,
-                pong_response,
-                store_cookie,
-                transfer,
-                move_minecart_along_track,
-                set_held_slot,
-                set_player_inventory,
-                projectile_power,
-                custom_report_details,
-                server_links,
-                player_rotation,
-                recipe_book_add,
-                recipe_book_remove,
-                recipe_book_settings,
-                test_instance_block_status,
-            ]
-        );
-    }
+pub fn process_packet(ecs: &mut World, player: Entity, packet: &ClientboundGamePacket) {
+    let mut handler = GamePacketHandler { player, ecs };
+
+    // the order of these doesn't matter, that's decided by the protocol library
+    declare_packet_handlers!(
+        ClientboundGamePacket,
+        packet,
+        handler,
+        [
+            login,
+            set_chunk_cache_radius,
+            chunk_batch_start,
+            chunk_batch_finished,
+            custom_payload,
+            change_difficulty,
+            commands,
+            player_abilities,
+            set_cursor_item,
+            update_tags,
+            disconnect,
+            update_recipes,
+            entity_event,
+            player_position,
+            player_info_update,
+            player_info_remove,
+            set_chunk_cache_center,
+            chunks_biomes,
+            light_update,
+            level_chunk_with_light,
+            add_entity,
+            set_entity_data,
+            update_attributes,
+            set_entity_motion,
+            set_entity_link,
+            initialize_border,
+            set_time,
+            set_default_spawn_position,
+            set_health,
+            set_experience,
+            teleport_entity,
+            update_advancements,
+            rotate_head,
+            move_entity_pos,
+            move_entity_pos_rot,
+            move_entity_rot,
+            keep_alive,
+            remove_entities,
+            player_chat,
+            system_chat,
+            disguised_chat,
+            sound,
+            level_event,
+            block_update,
+            animate,
+            section_blocks_update,
+            game_event,
+            level_particles,
+            server_data,
+            set_equipment,
+            update_mob_effect,
+            award_stats,
+            block_changed_ack,
+            block_destruction,
+            block_entity_data,
+            block_event,
+            boss_event,
+            command_suggestions,
+            container_set_content,
+            container_set_data,
+            container_set_slot,
+            container_close,
+            cooldown,
+            custom_chat_completions,
+            delete_chat,
+            explode,
+            forget_level_chunk,
+            horse_screen_open,
+            map_item_data,
+            merchant_offers,
+            move_vehicle,
+            open_book,
+            open_screen,
+            open_sign_editor,
+            ping,
+            place_ghost_recipe,
+            player_combat_end,
+            player_combat_enter,
+            player_combat_kill,
+            player_look_at,
+            remove_mob_effect,
+            resource_pack_push,
+            resource_pack_pop,
+            respawn,
+            start_configuration,
+            entity_position_sync,
+            select_advancements_tab,
+            set_action_bar_text,
+            set_border_center,
+            set_border_lerp_size,
+            set_border_size,
+            set_border_warning_delay,
+            set_border_warning_distance,
+            set_camera,
+            set_display_objective,
+            set_objective,
+            set_passengers,
+            set_player_team,
+            set_score,
+            set_simulation_distance,
+            set_subtitle_text,
+            set_title_text,
+            set_titles_animation,
+            clear_titles,
+            sound_entity,
+            stop_sound,
+            tab_list,
+            tag_query,
+            take_item_entity,
+            bundle_delimiter,
+            damage_event,
+            hurt_animation,
+            ticking_state,
+            ticking_step,
+            reset_score,
+            cookie_request,
+            debug_sample,
+            pong_response,
+            store_cookie,
+            transfer,
+            move_minecart_along_track,
+            set_held_slot,
+            set_player_inventory,
+            projectile_power,
+            custom_report_details,
+            server_links,
+            player_rotation,
+            recipe_book_add,
+            recipe_book_remove,
+            recipe_book_settings,
+            test_instance_block_status,
+        ]
+    );
 }
 
 pub struct GamePacketHandler<'a> {
@@ -342,7 +321,7 @@ impl GamePacketHandler<'_> {
                     client_information
                 );
                 commands.trigger(SendPacketEvent::new(self.player,
-                    azalea_protocol::packets::game::s_client_information::ServerboundClientInformation { information: client_information.clone() },
+                    azalea_protocol::packets::game::s_client_information::ServerboundClientInformation { client_information: client_information.clone() },
                 ));
             },
         );
@@ -1506,9 +1485,11 @@ impl GamePacketHandler<'_> {
     pub fn start_configuration(&mut self, _p: &ClientboundStartConfiguration) {
         debug!("Got start configuration packet");
 
-        as_system::<(Query<&RawConnection>, Commands)>(self.ecs, |(query, mut commands)| {
-            let raw_conn = query.get(self.player).unwrap();
-            let _ = raw_conn.write_packet(ServerboundConfigurationAcknowledged);
+        as_system::<Commands>(self.ecs, |mut commands| {
+            commands.trigger(SendPacketEvent::new(
+                self.player,
+                ServerboundConfigurationAcknowledged,
+            ));
 
             commands
                 .entity(self.player)