\

#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
use bevy::diagnostic::{FrameTimeDiagnosticsPlugin, LogDiagnosticsPlugin};
use bevy::prelude::*;
use bevy::sprite::Material2dPlugin;
use bevy::window::{PresentMode, WindowTheme};

use bevy_egui::EguiPlugin;

mod audio_capture;
mod audio_processing;
mod cfg;
mod materials;
mod ui;
mod visualization;

use crate::audio_capture::{audio_capture_startup_system, AudioReceiver};
use crate::audio_processing::{audio_event_system, AudioVisualizerState};
use crate::cfg::*;
use crate::materials::{BarEntity, BarMaterial};
use crate::materials::{CircleSplitEntity, CircleSplitMaterial};
use crate::materials::{PolygonEntity, PolygonMaterial};
use crate::materials::{StringEntity, StringMaterial};
use crate::materials::{WaveEntity, WaveMaterial};
use crate::ui::{Colors, UIPlugin};
use crate::visualization::{
    spawn_visualization, visualization_toggle_system, window_resized_event, VisualizationType,
};
use cpal::traits::{DeviceTrait, HostTrait, StreamTrait};
use cpal::available_hosts;

const ARRAY_UNIFORM_SIZE: usize = 16;
const NUM_BUCKETS: usize = ARRAY_UNIFORM_SIZE * 4;

#[derive(Resource)]
pub struct CfgResource(MyConfig);

#[derive(Resource)]
#[derive(Default)]
pub struct GUIToggle {
    pub active: bool,
}



fn main() {
    let config = match confy::load("bevy_audioviz", "config") {
        Ok(config) => config,
        Err(_) => MyConfig::default(),
    };
    println!(
        "Config file location: {:#?}",
        confy::get_configuration_file_path("bevy_audioviz", "config").unwrap()
    );
    println!("{:?}", config);
    list_available_hosts();
    list_audio_devices();

    App::new()
        .add_plugins((DefaultPlugins.set(WindowPlugin {
            primary_window: Some(Window {
                title: "Audio Visualization".into(),
                //resolution: (500., 300.).into(),
                present_mode: PresentMode::AutoVsync,
                // Tells wasm to resize the window according to the available canvas
                fit_canvas_to_parent: true,
                // Tells wasm not to override default event handling, like F5, Ctrl+R etc.
                prevent_default_event_handling: false,
                window_theme: Some(WindowTheme::Dark),
                ..default()
            }),
            ..default()
        }),))
        .add_plugins(EguiPlugin)
        .add_plugins(LogDiagnosticsPlugin::default())
        .add_plugins(FrameTimeDiagnosticsPlugin)
        .add_plugins(UIPlugin)
        .insert_resource(AudioVisualizerState::new(NUM_BUCKETS))
        .insert_resource(CfgResource(config))
        .insert_resource(GUIToggle::default())
        .insert_resource(Colors::default())
        .init_resource::<AudioReceiver>() // Initialize the `AudioReceiver` resource.
        .init_resource::<VisualizationType>()
        .add_systems(Startup, setup)
        .add_systems(Update, spawn_visualization)
        .add_systems(Update, visualization_toggle_system)
        .add_systems(Update, window_resized_event.after(spawn_visualization))
        .add_systems(Update, audio_capture_startup_system)
        .add_systems(
            Update,
            audio_event_system
                .after(audio_capture_startup_system)
                .before(visualization_toggle_system)
                .before(spawn_visualization),
        )
        .add_systems(Update, toggle_vsync)
        .add_systems(Update, toggle_gui)
        .init_resource::<BarEntity>()
        .init_resource::<StringEntity>()
        .init_resource::<CircleSplitEntity>()
        .init_resource::<PolygonEntity>()
        .init_resource::<WaveEntity>()
        .add_plugins(Material2dPlugin::<BarMaterial>::default())
        .add_plugins(Material2dPlugin::<StringMaterial>::default())
        .add_plugins(Material2dPlugin::<CircleSplitMaterial>::default())
        .add_plugins(Material2dPlugin::<PolygonMaterial>::default())
        .add_plugins(Material2dPlugin::<WaveMaterial>::default())
        .run();
}

fn setup(mut commands: Commands) {
    commands.spawn(Camera2dBundle::default());
}
/// This system toggles the vsync mode when pressing the button V.
/// You'll see fps increase displayed in the console.
fn toggle_vsync(input: Res<Input<KeyCode>>, mut windows: Query<&mut Window>) {
    if input.just_pressed(KeyCode::V) {
        let mut window = windows.single_mut();

        window.present_mode = if matches!(window.present_mode, PresentMode::AutoVsync) {
            PresentMode::AutoNoVsync
        } else {
            PresentMode::AutoVsync
        };
        info!("PRESENT_MODE: {:?}", window.present_mode);
    }
}

fn list_available_hosts() {
    println!("Available hosts:");
    for host_id in available_hosts() {
        println!("{:?}", host_id);
    }
}

 
fn list_audio_devices() {
    let host = cpal::default_host();
    let input_devices = host.input_devices().unwrap();
    let output_devices = host.output_devices().unwrap();

    println!("Input Devices:");
    for device in input_devices {
        println!("{}", device.name().unwrap());
    }

    println!("\nOutput Devices:");
    for device in output_devices {
        println!("{}", device.name().unwrap());
    }
}

fn toggle_gui(input: Res<Input<KeyCode>>, mut toggle: ResMut<GUIToggle>) {
    if input.just_pressed(KeyCode::G) {
        toggle.active = !toggle.active;
        info!("GUI Toggled: {}", toggle.active);
    }
}