//! OpenGL renderer.

use std::collections::HashMap;
use std::ffi::{c_void, CString};
use std::ops::Deref;
use std::process::abort;

use cstr::cstr;
use ecs::actions::Actions;
use ecs::component::local::Local;
use ecs::query::options::{Not, With};
use ecs::sole::Single;
use ecs::system::{Into as _, System};
use ecs::{Component, Query};

use crate::camera::Camera;
use crate::color::Color;
use crate::data_types::dimens::Dimens;
use crate::draw_flags::{DrawFlags, NoDraw, PolygonModeConfig};
use crate::event::{Present as PresentEvent, Start as StartEvent};
use crate::lighting::{DirectionalLight, GlobalLight, PointLight};
use crate::material::{Flags as MaterialFlags, Material};
use crate::matrix::Matrix;
use crate::mesh::Mesh;
use crate::opengl::buffer::{Buffer, Usage as BufferUsage};
#[cfg(feature = "debug")]
use crate::opengl::debug::{MessageSeverity, MessageSource, MessageType};
use crate::opengl::shader::{
    Error as GlShaderError,
    Program as GlShaderProgram,
    Shader as GlShader,
};
use crate::opengl::texture::{
    set_active_texture_unit,
    Texture as GlTexture,
    TextureUnit,
};
use crate::opengl::vertex_array::{
    DataType as VertexArrayDataType,
    PrimitiveKind,
    VertexArray,
};
use crate::opengl::{clear_buffers, enable, BufferClearMask, Capability};
use crate::projection::{new_perspective_matrix, Projection};
use crate::shader::Program as ShaderProgram;
use crate::texture::{Id as TextureId, Texture};
use crate::transform::{Position, Scale};
use crate::util::NeverDrop;
use crate::vector::{Vec2, Vec3};
use crate::vertex::{AttributeComponentType, Vertex};
use crate::window::Window;

type RenderableEntity = (
    Mesh,
    ShaderProgram,
    Material,
    Option<MaterialFlags>,
    Option<Position>,
    Option<Scale>,
    Option<DrawFlags>,
    Option<GlObjects>,
);

#[derive(Debug, Default)]
pub struct Extension {}

impl ecs::extension::Extension for Extension
{
    fn collect(self, mut collector: ecs::extension::Collector<'_>)
    {
        collector.add_system(StartEvent, initialize);

        collector.add_system(
            PresentEvent,
            render
                .into_system()
                .initialize((GlobalGlObjects::default(),)),
        );
    }
}

fn initialize(window: Single<Window>)
{
    window
        .make_context_current()
        .expect("Failed to make window context current");

    gl::load_with(|symbol| match window.get_proc_address(symbol) {
        Ok(addr) => addr as *const c_void,
        Err(err) => {
            println!(
                "FATAL ERROR: Failed to get adress of OpenGL function {symbol}: {err}",
            );

            abort();
        }
    });

    #[cfg(feature = "debug")]
    initialize_debug();

    let window_size = window.size().expect("Failed to get window size");

    set_viewport(Vec2 { x: 0, y: 0 }, window_size);

    window.set_framebuffer_size_callback(|new_window_size| {
        set_viewport(Vec2::ZERO, new_window_size);
    });

    enable(Capability::DepthTest);
    enable(Capability::MultiSample);
}

fn render(
    query: Query<RenderableEntity, Not<With<NoDraw>>>,
    point_light_query: Query<(PointLight,)>,
    directional_lights: Query<(DirectionalLight,)>,
    camera_query: Query<(Camera, Position)>,
    window: Single<Window>,
    global_light: Single<GlobalLight>,
    mut gl_objects: Local<GlobalGlObjects>,
    mut actions: Actions,
)
{
    let Some((camera, camera_pos)) =
        camera_query.iter().find(|(camera, _)| camera.current)
    else {
        #[cfg(feature = "debug")]
        tracing::warn!("No current camera. Nothing will be rendered");
        return;
    };

    let point_lights = point_light_query
        .iter()
        .map(|(point_light,)| point_light)
        .collect::<Vec<_>>();

    let directional_lights = directional_lights.iter().collect::<Vec<_>>();

    let GlobalGlObjects {
        shader_programs: gl_shader_programs,
        textures: gl_textures,
    } = &mut *gl_objects;

    clear_buffers(BufferClearMask::COLOR | BufferClearMask::DEPTH);

    for (
        entity_index,
        (
            mesh,
            shader_program,
            material,
            material_flags,
            position,
            scale,
            draw_flags,
            gl_objects,
        ),
    ) in query.iter().enumerate()
    {
        let material_flags = material_flags
            .map(|material_flags| material_flags.clone())
            .unwrap_or_default();

        let shader_program = gl_shader_programs
            .entry(shader_program.u64_hash())
            .or_insert_with(|| create_gl_shader_program(&shader_program).unwrap());

        let new_gl_objects;

        let gl_objects = match gl_objects.as_deref() {
            Some(gl_objects) => gl_objects,
            None => {
                // TODO: Account for when meshes are changed
                let gl_objects = GlObjects::new(&*mesh);

                new_gl_objects = Some(gl_objects.clone());

                actions.add_components(
                    query.entity_uid(entity_index).unwrap(),
                    (gl_objects,),
                );

                &*new_gl_objects.unwrap()
            }
        };

        apply_transformation_matrices(
            Transformation {
                position: position.map(|pos| *pos).unwrap_or_default().position,
                scale: scale.map(|scale| *scale).unwrap_or_default().scale,
            },
            shader_program,
            &camera,
            &camera_pos,
            window.size().expect("Failed to get window size"),
        );

        apply_light(
            &material,
            &material_flags,
            &global_light,
            shader_program,
            point_lights.as_slice(),
            directional_lights
                .iter()
                .map(|(dir_light,)| &**dir_light)
                .collect::<Vec<_>>()
                .as_slice(),
            &camera_pos,
        );

        for texture in &material.textures {
            let gl_texture = gl_textures
                .entry(texture.id())
                .or_insert_with(|| create_gl_texture(texture));

            let texture_unit =
                TextureUnit::from_texture_id(texture.id()).unwrap_or_else(|| {
                    panic!("Texture id {} is a invalid texture unit", texture.id());
                });

            set_active_texture_unit(texture_unit);

            gl_texture.bind();
        }

        shader_program.activate();

        if let Some(draw_flags) = &draw_flags {
            crate::opengl::set_polygon_mode(
                draw_flags.polygon_mode_config.face,
                draw_flags.polygon_mode_config.mode,
            )
        }

        draw_mesh(&gl_objects);

        if draw_flags.is_some() {
            let default_polygon_mode_config = PolygonModeConfig::default();

            crate::opengl::set_polygon_mode(
                default_polygon_mode_config.face,
                default_polygon_mode_config.mode,
            )
        }
    }
}

#[derive(Debug, Default, Component)]
struct GlobalGlObjects
{
    shader_programs: HashMap<u64, GlShaderProgram>,
    textures: HashMap<TextureId, GlTexture>,
}

fn set_viewport(position: Vec2<u32>, size: Dimens<u32>)
{
    crate::opengl::set_viewport(position, size);
}

#[cfg(feature = "debug")]
fn initialize_debug()
{
    use crate::opengl::debug::{
        enable_debug_output,
        set_debug_message_callback,
        set_debug_message_control,
        MessageIdsAction,
    };

    enable_debug_output();

    set_debug_message_callback(opengl_debug_message_cb);

    set_debug_message_control(None, None, None, &[], MessageIdsAction::Disable);
}

fn draw_mesh(gl_objects: &GlObjects)
{
    gl_objects.vertex_arr.bind();

    if gl_objects.index_buffer.is_some() {
        VertexArray::draw_elements(PrimitiveKind::Triangles, 0, gl_objects.index_cnt);
    } else {
        VertexArray::draw_arrays(PrimitiveKind::Triangles, 0, 3);
    }
}

fn create_gl_texture(texture: &Texture) -> GlTexture
{
    let mut gl_texture = GlTexture::new();

    gl_texture.generate(
        *texture.dimensions(),
        texture.image().as_bytes(),
        texture.pixel_data_format(),
    );

    gl_texture.apply_properties(texture.properties());

    gl_texture
}

fn create_gl_shader_program(
    shader_program: &ShaderProgram,
) -> Result<GlShaderProgram, GlShaderError>
{
    let gl_shaders = shader_program
        .shaders()
        .iter()
        .map(|shader| {
            let gl_shader = GlShader::new(shader.kind());

            gl_shader.set_source(shader.source())?;
            gl_shader.compile()?;

            Ok(gl_shader)
        })
        .collect::<Result<Vec<_>, _>>()?;

    let gl_shader_program = GlShaderProgram::new();

    for gl_shader in &gl_shaders {
        gl_shader_program.attach(gl_shader);
    }

    gl_shader_program.link()?;

    Ok(gl_shader_program)
}

#[derive(Debug, Component)]
struct GlObjects
{
    /// Vertex and index buffer has to live as long as the vertex array
    _vertex_buffer: Buffer<Vertex>,
    index_buffer: Option<Buffer<u32>>,
    index_cnt: u32,

    vertex_arr: VertexArray,
}

impl GlObjects
{
    #[cfg_attr(feature = "debug", tracing::instrument(skip_all))]
    fn new(mesh: &Mesh) -> Self
    {
        #[cfg(feature = "debug")]
        tracing::trace!(
            "Creating vertex array, vertex buffer{}",
            if mesh.indices().is_some() {
                " and index buffer"
            } else {
                ""
            }
        );

        let mut vertex_arr = VertexArray::new();
        let mut vertex_buffer = Buffer::new();

        vertex_buffer.store(mesh.vertices(), BufferUsage::Static);

        vertex_arr.bind_vertex_buffer(0, &vertex_buffer, 0);

        let mut offset = 0u32;

        for attrib in Vertex::attrs() {
            vertex_arr.enable_attrib(attrib.index);

            vertex_arr.set_attrib_format(
                attrib.index,
                match attrib.component_type {
                    AttributeComponentType::Float => VertexArrayDataType::Float,
                },
                false,
                offset,
            );

            vertex_arr.set_attrib_vertex_buf_binding(attrib.index, 0);

            offset += attrib.component_size * attrib.component_cnt as u32;
        }

        if let Some(indices) = mesh.indices() {
            let mut index_buffer = Buffer::new();

            index_buffer.store(indices, BufferUsage::Static);

            vertex_arr.bind_element_buffer(&index_buffer);

            return Self {
                _vertex_buffer: vertex_buffer,
                index_buffer: Some(index_buffer),
                index_cnt: indices.len().try_into().unwrap(),
                vertex_arr,
            };
        }

        Self {
            _vertex_buffer: vertex_buffer,
            index_buffer: None,
            index_cnt: 0,
            vertex_arr,
        }
    }

    pub fn clone(&self) -> NeverDrop<Self>
    {
        NeverDrop::new(Self {
            // SAFETY: The vertex buffer will never become dropped (NeverDrop ensures it)
            _vertex_buffer: unsafe { self._vertex_buffer.clone_weak() },
            index_buffer: self
                .index_buffer
                .as_ref()
                // SAFETY: The index buffer will never become dropped (NeverDrop ensures
                // it)
                .map(|index_buffer| unsafe { index_buffer.clone_weak() }),
            index_cnt: self.index_cnt,
            // SAFETY: The vertex array will never become dropped (NeverDrop ensures it)
            vertex_arr: unsafe { self.vertex_arr.clone_unsafe() },
        })
    }
}

fn apply_transformation_matrices(
    transformation: Transformation,
    gl_shader_program: &mut GlShaderProgram,
    camera: &Camera,
    camera_pos: &Position,
    window_size: Dimens<u32>,
)
{
    gl_shader_program.set_uniform_matrix_4fv(
        cstr!("model"),
        &create_transformation_matrix(transformation),
    );

    let view = create_view(camera, camera_pos);

    gl_shader_program.set_uniform_matrix_4fv(cstr!("view"), &view);

    #[allow(clippy::cast_precision_loss)]
    let projection = match &camera.projection {
        Projection::Perspective(perspective) => new_perspective_matrix(
            perspective,
            window_size.width as f32 / window_size.height as f32,
        ),
    };

    gl_shader_program.set_uniform_matrix_4fv(cstr!("projection"), &projection);
}

fn apply_light<PointLightHolder>(
    material: &Material,
    material_flags: &MaterialFlags,
    global_light: &GlobalLight,
    gl_shader_program: &mut GlShaderProgram,
    point_lights: &[PointLightHolder],
    directional_lights: &[&DirectionalLight],
    camera_pos: &Position,
) where
    PointLightHolder: Deref<Target = PointLight>,
{
    debug_assert!(
        point_lights.len() < 64,
        "Shader cannot handle more than 64 point lights"
    );

    debug_assert!(
        directional_lights.len() < 64,
        "Shader cannot handle more than 64 directional lights"
    );

    for (dir_light_index, dir_light) in directional_lights.iter().enumerate() {
        gl_shader_program.set_uniform_vec_3fv(
            &create_light_uniform_name(
                "directional_lights",
                dir_light_index,
                "direction",
            ),
            &dir_light.direction,
        );

        set_light_phong_uniforms(
            gl_shader_program,
            "directional_lights",
            dir_light_index,
            *dir_light,
        );
    }

    gl_shader_program.set_uniform_1i(
        cstr!("directional_light_cnt"),
        directional_lights.len() as i32,
    );

    for (point_light_index, point_light) in point_lights.iter().enumerate() {
        gl_shader_program.set_uniform_vec_3fv(
            &create_light_uniform_name("point_lights", point_light_index, "position"),
            &point_light.position,
        );

        set_light_phong_uniforms(
            gl_shader_program,
            "point_lights",
            point_light_index,
            &**point_light,
        );

        set_light_attenuation_uniforms(
            gl_shader_program,
            "point_lights",
            point_light_index,
            point_light,
        );
    }

    gl_shader_program.set_uniform_1i(cstr!("point_light_cnt"), point_lights.len() as i32);

    gl_shader_program.set_uniform_vec_3fv(
        cstr!("material.ambient"),
        &if material_flags.use_ambient_color {
            material.ambient.clone()
        } else {
            global_light.ambient.clone().into()
        }
        .into(),
    );

    gl_shader_program
        .set_uniform_vec_3fv(cstr!("material.diffuse"), &material.diffuse.clone().into());

    #[allow(clippy::cast_possible_wrap)]
    gl_shader_program.set_uniform_vec_3fv(
        cstr!("material.specular"),
        &material.specular.clone().into(),
    );

    #[allow(clippy::cast_possible_wrap)]
    gl_shader_program.set_uniform_1i(
        cstr!("material.ambient_map"),
        material.ambient_map.into_inner() as i32,
    );

    #[allow(clippy::cast_possible_wrap)]
    gl_shader_program.set_uniform_1i(
        cstr!("material.diffuse_map"),
        material.diffuse_map.into_inner() as i32,
    );

    #[allow(clippy::cast_possible_wrap)]
    gl_shader_program.set_uniform_1i(
        cstr!("material.specular_map"),
        material.specular_map.into_inner() as i32,
    );

    gl_shader_program.set_uniform_1fv(cstr!("material.shininess"), material.shininess);

    gl_shader_program.set_uniform_vec_3fv(cstr!("view_pos"), &camera_pos.position);
}

fn set_light_attenuation_uniforms(
    gl_shader_program: &mut GlShaderProgram,
    light_array: &str,
    light_index: usize,
    light: &PointLight,
)
{
    gl_shader_program.set_uniform_1fv(
        &create_light_uniform_name(
            light_array,
            light_index,
            "attenuation_props.constant",
        ),
        light.attenuation_params.constant,
    );

    gl_shader_program.set_uniform_1fv(
        &create_light_uniform_name(light_array, light_index, "attenuation_props.linear"),
        light.attenuation_params.linear,
    );

    gl_shader_program.set_uniform_1fv(
        &create_light_uniform_name(
            light_array,
            light_index,
            "attenuation_props.quadratic",
        ),
        light.attenuation_params.quadratic,
    );
}

fn set_light_phong_uniforms(
    gl_shader_program: &mut GlShaderProgram,
    light_array: &str,
    light_index: usize,
    light: &impl Light,
)
{
    gl_shader_program.set_uniform_vec_3fv(
        &create_light_uniform_name(light_array, light_index, "phong.diffuse"),
        &light.diffuse().clone().into(),
    );

    gl_shader_program.set_uniform_vec_3fv(
        &create_light_uniform_name(light_array, light_index, "phong.specular"),
        &light.specular().clone().into(),
    );
}

trait Light
{
    fn diffuse(&self) -> &Color<f32>;
    fn specular(&self) -> &Color<f32>;
}

impl Light for PointLight
{
    fn diffuse(&self) -> &Color<f32>
    {
        &self.diffuse
    }

    fn specular(&self) -> &Color<f32>
    {
        &self.specular
    }
}

impl Light for DirectionalLight
{
    fn diffuse(&self) -> &Color<f32>
    {
        &self.diffuse
    }

    fn specular(&self) -> &Color<f32>
    {
        &self.specular
    }
}

fn create_light_uniform_name(
    light_array: &str,
    light_index: usize,
    light_field: &str,
) -> CString
{
    unsafe {
        CString::from_vec_with_nul_unchecked(
            format!("{light_array}[{light_index}].{light_field}\0").into(),
        )
    }
}

fn create_view(camera: &Camera, camera_pos: &Position) -> Matrix<f32, 4, 4>
{
    let mut view = Matrix::new();

    view.look_at(&camera_pos.position, &camera.target, &camera.global_up);

    view
}

#[cfg(feature = "debug")]
#[tracing::instrument(skip_all)]
fn opengl_debug_message_cb(
    source: MessageSource,
    ty: MessageType,
    id: u32,
    severity: MessageSeverity,
    message: &str,
)
{
    use std::backtrace::{Backtrace, BacktraceStatus};

    use tracing::{event, Level};

    macro_rules! create_event {
        ($level: expr) => {
            event!($level, ?source, ?ty, id, ?severity, message);
        };
    }

    if matches!(severity, MessageSeverity::Notification) {
        return;
    }

    match ty {
        MessageType::Error => {
            create_event!(Level::ERROR);

            let backtrace = Backtrace::capture();

            if matches!(backtrace.status(), BacktraceStatus::Captured) {
                event!(Level::TRACE, "{backtrace}");
            }
        }
        MessageType::Other => {
            create_event!(Level::INFO);
        }
        _ => {
            create_event!(Level::WARN);
        }
    };
}

#[derive(Debug)]
struct Transformation
{
    position: Vec3<f32>,
    scale: Vec3<f32>,
}

fn create_transformation_matrix(transformation: Transformation) -> Matrix<f32, 4, 4>
{
    let mut matrix = Matrix::new_identity();

    matrix.translate(&transformation.position);
    matrix.scale(&transformation.scale);

    matrix
}