use bevy::prelude::*;
use bevy_render::extract_component::{ExtractComponent, ExtractComponentPlugin};
use bevy_render::sync_world::RenderEntity;
use bevy_render::{Extract, RenderApp};

fn main() {
    App::new()
        .add_plugins(DefaultPlugins)
        .add_plugins((
            TestExtractPlugin,
            ExtractComponentPlugin::<ComponentA>::default(),
            ExtractComponentPlugin::<ComponentB>::default(),
        ))
        .add_systems(Startup, setup)
        .add_systems(Update, update)
        .run();
}

#[derive(Component, ExtractComponent, Clone)]
struct ComponentA;
#[derive(Component, ExtractComponent, Clone)]
struct ComponentB;

/// set up a simple 3D scene
fn setup(mut commands: Commands) {
    commands.spawn((ComponentA, ComponentB));
}

fn update(mut commands: Commands, mut query: Query<(Entity), With<ComponentB>>) {
    for (entity) in query.iter() {
        commands.entity(entity).remove::<ComponentA>();
        commands.entity(entity).insert(ComponentA);
    }
}

struct TestExtractPlugin;

impl Plugin for TestExtractPlugin {
    fn build(&self, app: &mut App) {
        let Some(render_app) = app.get_sub_app_mut(RenderApp) else {
            return;
        };

        render_app
            .add_systems(ExtractSchedule, extract);
    }
}

fn extract(query: Extract<Query<RenderEntity, With<ComponentB>>>) {
    for (entity) in query.iter() {
        println!("Render entity id: {:?}", entity);
    }
}