use bevy::{
    diagnostic::{EntityCountDiagnosticsPlugin, FrameTimeDiagnosticsPlugin, LogDiagnosticsPlugin},
    prelude::*,
    window::{PresentMode, Window, WindowPlugin},
    DefaultPlugins,
};
use bevy_app::PluginGroup;
use bevy_asset::AssetServer;
use bevy_color::palettes::basic::*;

use bevy_particle_systems::{
    CircleSegment, ColorOverTime, Curve, CurvePoint, EmitterShape, JitteredValue, ParticleSystem,
    ParticleSystemBundle, ParticleSystemPlugin, Playing,
};

fn main() {
    App::new()
        .insert_resource(ClearColor(Color::BLACK))
        .add_plugins((
            EntityCountDiagnosticsPlugin,
            FrameTimeDiagnosticsPlugin,
            LogDiagnosticsPlugin::default(),
        ))
        .add_plugins(DefaultPlugins.set(WindowPlugin {
            primary_window: Some(Window {
                present_mode: PresentMode::AutoNoVsync,
                ..default()
            }),
            ..default()
        }))
        .add_plugins(ParticleSystemPlugin) // <-- Add the plugin
        .add_systems(Startup, startup_system)
        .run();
}

fn startup_system(mut commands: Commands, asset_server: Res<AssetServer>) {
    commands.spawn(Camera2dBundle::default());

    commands
        .spawn(ParticleSystemBundle {
            particle_system: ParticleSystem {
                max_particles: 50_000,
                texture: asset_server.load("arrow.png").into(),
                spawn_rate_per_second: 10.0.into(),
                initial_speed: JitteredValue::jittered(70.0, -3.0..3.0),
                lifetime: JitteredValue::jittered(3.0, -2.0..2.0),
                color: ColorOverTime::Gradient(Curve::new(vec![
                    CurvePoint::new(PURPLE.into(), 0.0),
                    CurvePoint::new(RED.into(), 0.5),
                    CurvePoint::new(Color::srgba(0.0, 0.0, 1.0, 0.0), 1.0),
                ])),
                emitter_shape: EmitterShape::line(200.0, std::f32::consts::FRAC_PI_4),
                looping: true,
                rotate_to_movement_direction: true,
                initial_rotation: (-90.0_f32).to_radians().into(),
                system_duration_seconds: 10.0,
                max_distance: Some(300.0),
                scale: 0.07.into(),
                ..ParticleSystem::default()
            },
            transform: Transform::from_xyz(0.0, -200.0, 0.0),
            ..ParticleSystemBundle::default()
        })
        .insert(Playing);

    commands
        .spawn(ParticleSystemBundle {
            particle_system: ParticleSystem {
                max_particles: 50_000,
                texture: asset_server.load("arrow.png").into(),
                spawn_rate_per_second: 10.0.into(),
                initial_speed: JitteredValue::jittered(70.0, -3.0..3.0),
                lifetime: JitteredValue::jittered(3.0, -2.0..2.0),
                color: ColorOverTime::Gradient(Curve::new(vec![
                    CurvePoint::new(PURPLE.into(), 0.0),
                    CurvePoint::new(RED.into(), 0.5),
                    CurvePoint::new(Color::srgba(0.0, 0.0, 1.0, 0.0), 1.0),
                ])),
                emitter_shape: CircleSegment {
                    radius: 10.0.into(),
                    opening_angle: std::f32::consts::PI,
                    direction_angle: std::f32::consts::FRAC_PI_4,
                }
                .into(),
                looping: true,
                rotate_to_movement_direction: true,
                initial_rotation: (-90.0_f32).to_radians().into(),
                system_duration_seconds: 10.0,
                max_distance: Some(300.0),
                scale: 0.07.into(),
                ..ParticleSystem::default()
            },
            transform: Transform::from_xyz(0.0, 200.0, 0.0),
            ..ParticleSystemBundle::default()
        })
        .insert(Playing);
}