pub mod ui;
pub mod input;
pub mod plugins;
pub mod components;

use bevy::input_focus::InputFocus;
use crate::client::colors;
use crate::prelude::*;
use crate::shared::config::part::PartConfig;
use crate::shared::config::ship_editor::ShipEditorConfig;
use crate::ship_editor::components::{GhostCamera, MainCamera, Part, PlayerPartRequest, ShipEditorConfigHolder, SpawnPartRequest, GHOST_RENDER_LAYER, MAIN_RENDER_LAYER};
use crate::ship_editor::input::input_plugin;
use crate::ship_editor::ui::{ui_plugin, PendingPart};

pub struct ShipEditorPlugin;

impl Plugin for ShipEditorPlugin {
    fn build(&self, app: &mut App) {
        app
            .init_resource::<InputFocus>()
            .add_systems(Startup, setup)
            .add_systems(Update, (spawn_player_part_request, spawn_parts))
            .add_plugins(input_plugin)
            .add_plugins(ui_plugin);
    }
}

fn setup(
    mut commands: Commands,
    mut meshes: ResMut<Assets<Mesh>>,
    mut materials: ResMut<Assets<ColorMaterial>>,
) {
    commands.insert_resource(ClearColor(colors::BASE));
    commands.spawn((
        Camera2d,
        Camera {
            order: 0,
            ..Default::default()
        },
        Transform::from_xyz(0.0, 0.0, 0.0),
        MainCamera,
        IsDefaultUiCamera,
        MAIN_RENDER_LAYER,
    ));
    commands.spawn((
        Camera2d::default(),
        Camera {
            order: 1,
            ..Default::default()
        },
        GhostCamera,
        GHOST_RENDER_LAYER,
    ));
    let rectangle = meshes.add(Rectangle::new(50.0, 50.0));
    commands.spawn((
        PlayerPartRequest,
        Transform::from_xyz(0.0, 0.0, 0.0),
    ));
}

fn spawn_player_part_request(
    player_part_request_query: Query<Entity, With<PlayerPartRequest>>,
    ship_editor_config_holder: Res<ShipEditorConfigHolder>,
    ship_editor_config: Res<Assets<ShipEditorConfig>>,
    mut commands: Commands,
    asset_server: Res<AssetServer>,
) {
    let Some(handle) = ship_editor_config_holder.handle.clone() else { return };
    if let Some(strong_ship_editor_config) = ship_editor_config.get(&handle) {
        let player_part = &strong_ship_editor_config.general.player_part;
        for entity in player_part_request_query {
            commands.entity(entity)
                .insert(SpawnPartRequest(asset_server.load(format!("config/parts/{}.part.toml", player_part.to_lowercase()))))
                .remove::<PlayerPartRequest>();
        }
    }
}
fn spawn_parts(
    spawn_part_query: Query<(Entity, &SpawnPartRequest)>,
    part_config: Res<Assets<PartConfig>>,
    asset_server: Res<AssetServer>,
    mut commands: Commands,
) {
    for (entity, request) in spawn_part_query.iter() {
        if let Some(strong_part_config) = part_config.get(&request.0) {
            let mut sprite = Sprite::from_image(asset_server.load(strong_part_config.part.sprite_connected.clone()));
            sprite.custom_size = Some(vec2(strong_part_config.physics.width as f32, strong_part_config.physics.height as f32));

            commands.entity(entity)
                .insert(sprite)
                .insert(Part(strong_part_config.clone()))
                .remove::<SpawnPartRequest>();
        }
    }
}