use bevy::prelude::*;

use crate::{bike::CyberBikeBody, input::InputState, ui::setup_ui};

// 85 degrees in radians
const MAX_PITCH: f32 = 1.48353;

#[derive(Clone, Copy, Eq, PartialEq, Debug, Hash, Component, States, Default)]
enum CyberCameras {
    #[default]
    Hero,
    Debug,
}

#[derive(Debug, Resource)]
pub struct DebugCamOffset {
    pub rot: f32,
    pub dist: f32,
    pub alt: f32,
}

impl Default for DebugCamOffset {
    fn default() -> Self {
        DebugCamOffset {
            rot: 60.0,
            dist: 10.0,
            alt: 4.0,
        }
    }
}

impl CyberCameras {
    fn next(self) -> Self {
        match self {
            CyberCameras::Debug => CyberCameras::Hero,
            CyberCameras::Hero => CyberCameras::Debug,
        }
    }
}

fn setup_cybercams(mut commands: Commands, asset_server: Res<AssetServer>) {
    let hero_projection = PerspectiveProjection {
        fov: std::f32::consts::FRAC_PI_3,
        ..Default::default()
    };

    let id = commands
        .spawn((
            Camera3d::default(),
            bevy::render::camera::Projection::Perspective(hero_projection),
        ))
        .insert((CyberCameras::Hero, Msaa::Sample4))
        .id();

    commands
        .spawn(Camera3d::default())
        .insert((CyberCameras::Debug, Msaa::Sample4));

    setup_ui(commands, asset_server, id);
}

fn follow_cyberbike(
    mut query: ParamSet<(
        // 0: the bike
        Query<&Transform, With<CyberBikeBody>>,
        // 1: the cameras
        Query<(&mut Transform, &CyberCameras)>,
    )>,
    input: Res<InputState>,
    offset: Res<DebugCamOffset>,
) {
    let bike_xform = *query.p0().single();
    let up = bike_xform.translation.normalize();

    for (mut cam_xform, cam_type) in query.p1().iter_mut() {
        match *cam_type {
            CyberCameras::Hero => {
                let look_at = bike_xform.translation + (bike_xform.forward() * 500.0);
                let cam_pos = bike_xform.translation + (bike_xform.back() * 0.1) + (up * 0.8);

                cam_xform.translation = cam_pos;
                cam_xform.look_at(look_at, up);

                // handle input pitch
                let angle = input.pitch.powi(3) * MAX_PITCH;
                let axis = cam_xform.right();
                cam_xform.rotate(Quat::from_axis_angle(*axis, angle));
            }
            CyberCameras::Debug => {
                let mut ncx = Transform::from_translation(bike_xform.translation);
                ncx.rotate(Quat::from_axis_angle(up, offset.rot.to_radians()));
                ncx.translation += ncx.forward() * offset.dist;
                ncx.translation += ncx.up() * offset.alt;
                *cam_xform = ncx;
                cam_xform.look_at(bike_xform.translation, up);
            }
        }
    }
}

fn update_active_camera(
    state: Res<State<CyberCameras>>,
    mut cameras: Query<(Entity, &mut Camera, &CyberCameras)>,
    mut target: Query<&mut TargetCamera>,
) {
    let mut target = target.single_mut();
    // find the camera with the current state, set it as the ActiveCamera
    cameras.iter_mut().for_each(|(ent, mut cam, cyber)| {
        if cyber.eq(state.get()) {
            cam.is_active = true;
            *target = TargetCamera(ent);
        } else {
            cam.is_active = false;
        }
    });
}

fn cycle_cam_state(
    state: Res<State<CyberCameras>>,
    mut next: ResMut<NextState<CyberCameras>>,
    mut keys: ResMut<ButtonInput<KeyCode>>,
) {
    if keys.just_pressed(KeyCode::KeyD) {
        let new_state = state.get().next();
        info!("{:?}", new_state);
        next.set(new_state);
        keys.reset(KeyCode::KeyD);
    }
}

pub struct CyberCamPlugin;

impl Plugin for CyberCamPlugin {
    fn build(&self, app: &mut bevy::prelude::App) {
        common(app);
    }
}

fn common(app: &mut bevy::prelude::App) {
    app.insert_resource(DebugCamOffset::default())
        .add_systems(Startup, setup_cybercams)
        .init_state::<CyberCameras>()
        .add_systems(Update, (cycle_cam_state, update_active_camera))
        .add_systems(
            PostUpdate,
            follow_cyberbike
                .after(avian3d::schedule::PhysicsSet::Sync)
                .before(TransformSystem::TransformPropagate),
        );
}