use crate::camera;
use bevy::prelude::*;
use bevy::core_pipeline::Skybox;
use bevy::asset::LoadState;
use bevy::render::render_resource::{TextureViewDescriptor, TextureViewDimension};

#[derive(Resource)]
pub struct Cubemap {
    is_loaded: bool,
    image_handle: Handle<Image>,
}

const CUBEMAP_PATH: &str = "textures/stars_cubemap.png";

pub fn setup(
    mut commands: Commands,
    mut meshes: ResMut<Assets<Mesh>>,
    mut materials: ResMut<Assets<StandardMaterial>>,
    asset_server: Res<AssetServer>,
) {
    // Add skybox
    let skybox_handle = asset_server.load(CUBEMAP_PATH);
    commands.spawn((
        Camera3dBundle {
            transform: Transform::from_xyz(0.0, 0.0, 8.0).looking_at(Vec3::ZERO, Vec3::Y),
            ..default()
        },
        camera::CameraController::default(),
        Skybox {
            image: skybox_handle.clone(),
            brightness: 150.0,
        },
    ));
    commands.insert_resource(Cubemap {
        is_loaded: false,
        image_handle: skybox_handle,
    });

    // Add spheres
    let sphere_radius = 0.25;
    let sphere_handle = meshes.add(Sphere::new(sphere_radius));
    let white_handle = materials.add(StandardMaterial {
        base_color: Color::WHITE,
        perceptual_roughness: 1.0,
        ..default()
    });
    commands.spawn(PbrBundle {
        mesh: sphere_handle.clone(),
        material: white_handle.clone(),
        transform: Transform::from_xyz(
            0.0,
            sphere_radius,
            0.0,
        ),
        ..default()
    });
}

pub fn load_cubemap_asset(
    mut cubemap: ResMut<Cubemap>,
    asset_server: Res<AssetServer>,
) {
    cubemap.image_handle = asset_server.load(CUBEMAP_PATH);
    cubemap.is_loaded = false;
}

pub fn asset_loaded(
    asset_server: Res<AssetServer>,
    mut images: ResMut<Assets<Image>>,
    mut cubemap: ResMut<Cubemap>,
    mut skyboxes: Query<&mut Skybox>,
) {
    if !cubemap.is_loaded && asset_server.load_state(&cubemap.image_handle) == LoadState::Loaded {
        let image = images.get_mut(&cubemap.image_handle).unwrap();
        if image.texture_descriptor.array_layer_count() == 1 {
            image.reinterpret_stacked_2d_as_array(image.height() / image.width());
            image.texture_view_descriptor = Some(TextureViewDescriptor {
                dimension: Some(TextureViewDimension::Cube),
                ..default()
            });
        }

        for mut skybox in &mut skyboxes {
            skybox.image = cubemap.image_handle.clone();
        }

        cubemap.is_loaded = true;
    }
}