use std::any::{Any, TypeId};
use std::ops::{Deref, DerefMut};

use crate::component::Component;
use crate::system::{ComponentRefMut, Param as SystemParam, System};
use crate::WorldData;

/// Holds a component which is local to a single system.
#[derive(Debug)]
pub struct Local<'world, LocalComponent: Component>
{
    local_component: ComponentRefMut<'world, LocalComponent>,
}

unsafe impl<'world, LocalComponent> SystemParam<'world> for Local<'world, LocalComponent>
where
    LocalComponent: Component,
{
    type Flags = ();
    type Input = LocalComponent;

    fn initialize<SystemImpl>(
        system: &mut impl System<'world, SystemImpl>,
        input: Self::Input,
    )
    {
        system.set_local_component(input);
    }

    fn new<SystemImpl>(
        system: &'world impl System<'world, SystemImpl>,
        _world_data: &'world WorldData,
    ) -> Self
    {
        let local_component = system
            .get_local_component_mut::<LocalComponent>()
            .expect("Local component is uninitialized");

        Self { local_component }
    }

    fn is_compatible<Other: SystemParam<'world>>() -> bool
    {
        let other_comparable = Other::get_comparable();

        let Some(other_type_id) = other_comparable.downcast_ref::<TypeId>() else {
            return true;
        };

        TypeId::of::<LocalComponent>() != *other_type_id
    }

    fn get_comparable() -> Box<dyn Any>
    {
        Box::new(TypeId::of::<LocalComponent>())
    }
}

impl<'world, LocalComponent> Deref for Local<'world, LocalComponent>
where
    LocalComponent: Component,
{
    type Target = LocalComponent;

    fn deref(&self) -> &Self::Target
    {
        &self.local_component
    }
}

impl<'world, LocalComponent> DerefMut for Local<'world, LocalComponent>
where
    LocalComponent: Component,
{
    fn deref_mut(&mut self) -> &mut Self::Target
    {
        &mut self.local_component
    }
}