use bevy::color::palettes;
use bevy::prelude::*;
use bevy_mod_billboard::prelude::*;
use bevy_mod_billboard::{BillboardLockAxis, BillboardLockAxisBundle};

fn main() {
    App::new()
        .add_plugins(DefaultPlugins)
        .add_plugins(BillboardPlugin)
        .add_systems(Startup, (setup_billboard, setup_scene))
        .add_systems(Update, rotate_camera)
        .run();
}

fn setup_billboard(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    mut meshes: ResMut<Assets<Mesh>>,
) {
    let image_handle = asset_server.load("tree.png");
    commands.spawn(BillboardLockAxisBundle {
        billboard_bundle: BillboardTextureBundle {
            transform: Transform::from_translation(Vec3::new(2.0, 2.0, 0.0)),
            texture: BillboardTextureHandle(image_handle.clone()),
            mesh: BillboardMeshHandle(meshes.add(Rectangle::from_size(Vec2::new(2.0, 4.0)))),
            ..default()
        },
        lock_axis: BillboardLockAxis {
            y_axis: true,
            ..Default::default()
        },
    });
    commands.spawn(BillboardTextureBundle {
        transform: Transform::from_translation(Vec3::new(-2.0, 2.0, 0.0)),
        texture: BillboardTextureHandle(image_handle),
        mesh: BillboardMeshHandle(meshes.add(Rectangle::from_size(Vec2::new(2.0, 4.0)))),
        ..default()
    });
}

// Important bits are above, the code below is for camera, reference plane and rotation

#[derive(Component)]
pub struct CameraHolder;

fn setup_scene(
    mut commands: Commands,
    mut meshes: ResMut<Assets<Mesh>>,
    mut materials: ResMut<Assets<StandardMaterial>>,
) {
    commands.spawn(PbrBundle {
        transform: Transform::from_scale(Vec3::splat(3.0)),
        mesh: meshes.add(Plane3d::new(Vec3::Y, Vec2::splat(1.5))),
        material: materials.add(Color::Srgba(palettes::css::SILVER)),
        ..default()
    });

    commands
        .spawn((CameraHolder, Transform::IDENTITY, GlobalTransform::IDENTITY))
        .with_children(|parent| {
            parent.spawn(Camera3dBundle {
                transform: Transform::from_translation(Vec3::new(0., 15., 2.))
                    .looking_at(Vec3::ZERO, Vec3::Y),
                ..default()
            });
        });
}

fn rotate_camera(mut camera: Query<&mut Transform, With<CameraHolder>>, time: Res<Time>) {
    let mut camera = camera.single_mut();

    camera.rotate_y(time.delta_seconds());
}