use std::convert::Infallible;

use fj_math::Vector;

use crate::{
    geometry::{path::GlobalPath, surface::SurfaceGeometry},
    storage::{Handle, Store},
    validate::{
        CycleValidationError, FaceValidationError, HalfEdgeValidationError,
        SurfaceVertexValidationError, Validate, VertexValidationError,
    },
};

use super::{
    Curve, Cycle, Face, GlobalCurve, GlobalEdge, GlobalVertex, HalfEdge, Shell,
    Sketch, Solid, Surface, SurfaceVertex, Vertex,
};

/// The available object stores
///
/// # Implementation Note
///
/// The intention is to eventually manage all objects in here. Making this
/// happen is simply a case of putting in the required work. See [#1021].
///
/// [#1021]: https://github.com/hannobraun/Fornjot/issues/1021
#[derive(Debug, Default)]
pub struct Objects {
    /// Store for [`Curve`]s
    pub curves: Curves,

    /// Store for [`Cycle`]s
    pub cycles: Cycles,

    /// Store for [`Face`]s
    pub faces: Faces,

    /// Store for [`GlobalCurve`]s
    pub global_curves: GlobalCurves,

    /// Store for [`GlobalEdge`]s
    pub global_edges: GlobalEdges,

    /// Store for [`GlobalVertex`] objects
    pub global_vertices: GlobalVertices,

    /// Store for [`HalfEdge`]s
    pub half_edges: HalfEdges,

    /// Store for [`Shell`]s
    pub shells: Shells,

    /// Store for [`Sketch`]es
    pub sketches: Sketches,

    /// Store for [`Solid`]s
    pub solids: Solids,

    /// Store for [`SurfaceVertex`] objects
    pub surface_vertices: SurfaceVertices,

    /// Store for [`Surface`]s
    pub surfaces: Surfaces,

    /// Store for [`Vertex`] objects
    pub vertices: Vertices,
}

impl Objects {
    /// Construct a new instance of `Stores`
    pub fn new() -> Self {
        Self::default()
    }
}

/// Store for [`Curve`]s
#[derive(Debug, Default)]
pub struct Curves {
    store: Store<Curve>,
}

impl Curves {
    /// Insert a [`Curve`] into the store
    pub fn insert(&self, curve: Curve) -> Result<Handle<Curve>, Infallible> {
        curve.validate()?;
        Ok(self.store.insert(curve))
    }
}

/// Store for [`Cycle`]s
#[derive(Debug, Default)]
pub struct Cycles {
    store: Store<Cycle>,
}

impl Cycles {
    /// Insert a [`Cycle`] into the store
    pub fn insert(
        &self,
        cycle: Cycle,
    ) -> Result<Handle<Cycle>, CycleValidationError> {
        cycle.validate()?;
        Ok(self.store.insert(cycle))
    }
}

/// Store for [`Face`]s
#[derive(Debug, Default)]
pub struct Faces {
    store: Store<Face>,
}

impl Faces {
    /// Insert a [`Face`] into the store
    pub fn insert(
        &self,
        face: Face,
    ) -> Result<Handle<Face>, FaceValidationError> {
        face.validate()?;
        Ok(self.store.insert(face))
    }
}

/// Store for [`GlobalCurve`]s
#[derive(Debug, Default)]
pub struct GlobalCurves {
    store: Store<GlobalCurve>,
}

impl GlobalCurves {
    /// Insert a [`GlobalCurve`] into the store
    pub fn insert(
        &self,
        global_curve: GlobalCurve,
    ) -> Result<Handle<GlobalCurve>, Infallible> {
        global_curve.validate()?;
        Ok(self.store.insert(global_curve))
    }
}

/// Store for [`GlobalEdge`]s
#[derive(Debug, Default)]
pub struct GlobalEdges {
    store: Store<GlobalEdge>,
}

impl GlobalEdges {
    /// Insert a [`GlobalEdge`] into the store
    pub fn insert(
        &self,
        global_edge: GlobalEdge,
    ) -> Result<Handle<GlobalEdge>, Infallible> {
        global_edge.validate()?;
        Ok(self.store.insert(global_edge))
    }
}

/// Store for [`GlobalVertex`] objects
#[derive(Debug, Default)]
pub struct GlobalVertices {
    store: Store<GlobalVertex>,
}

impl GlobalVertices {
    /// Insert a [`GlobalVertex`] into the store
    pub fn insert(
        &self,
        global_vertex: GlobalVertex,
    ) -> Result<Handle<GlobalVertex>, Infallible> {
        global_vertex.validate()?;
        Ok(self.store.insert(global_vertex))
    }
}

/// Store for [`HalfEdge`]s
#[derive(Debug, Default)]
pub struct HalfEdges {
    store: Store<HalfEdge>,
}

impl HalfEdges {
    /// Insert a [`HalfEdge`] into the store
    pub fn insert(
        &self,
        half_edge: HalfEdge,
    ) -> Result<Handle<HalfEdge>, HalfEdgeValidationError> {
        half_edge.validate()?;
        Ok(self.store.insert(half_edge))
    }
}

/// Store for [`Shell`]s
#[derive(Debug, Default)]
pub struct Shells {
    store: Store<Shell>,
}

impl Shells {
    /// Insert a [`Shell`] into the store
    pub fn insert(&self, shell: Shell) -> Result<Handle<Shell>, Infallible> {
        shell.validate()?;
        Ok(self.store.insert(shell))
    }
}

/// Store for [`Sketch`]es
#[derive(Debug, Default)]
pub struct Sketches {
    store: Store<Sketch>,
}

impl Sketches {
    /// Insert a [`Sketch`] into the store
    pub fn insert(&self, sketch: Sketch) -> Result<Handle<Sketch>, Infallible> {
        sketch.validate()?;
        Ok(self.store.insert(sketch))
    }
}

/// Store for [`Solid`]s
#[derive(Debug, Default)]
pub struct Solids {
    store: Store<Solid>,
}

impl Solids {
    /// Insert a [`Solid`] into the store
    pub fn insert(&self, solid: Solid) -> Result<Handle<Solid>, Infallible> {
        solid.validate()?;
        Ok(self.store.insert(solid))
    }
}

/// Store for [`SurfaceVertex`] objects
#[derive(Debug, Default)]
pub struct SurfaceVertices {
    store: Store<SurfaceVertex>,
}

impl SurfaceVertices {
    /// Insert a [`SurfaceVertex`] into the store
    pub fn insert(
        &self,
        surface_vertex: SurfaceVertex,
    ) -> Result<Handle<SurfaceVertex>, SurfaceVertexValidationError> {
        surface_vertex.validate()?;
        Ok(self.store.insert(surface_vertex))
    }
}

/// Store for [`Surface`]s
#[derive(Debug)]
pub struct Surfaces {
    store: Store<Surface>,

    xy_plane: Handle<Surface>,
    xz_plane: Handle<Surface>,
    yz_plane: Handle<Surface>,
}

impl Surfaces {
    /// Insert a [`Surface`] into the store
    pub fn insert(
        &self,
        surface: Surface,
    ) -> Result<Handle<Surface>, Infallible> {
        surface.validate()?;
        Ok(self.store.insert(surface))
    }

    /// Access the xy-plane
    pub fn xy_plane(&self) -> Handle<Surface> {
        self.xy_plane.clone()
    }

    /// Access the xz-plane
    pub fn xz_plane(&self) -> Handle<Surface> {
        self.xz_plane.clone()
    }

    /// Access the yz-plane
    pub fn yz_plane(&self) -> Handle<Surface> {
        self.yz_plane.clone()
    }
}

impl Default for Surfaces {
    fn default() -> Self {
        let store = Store::new();

        let xy_plane = store.insert(Surface::new(SurfaceGeometry {
            u: GlobalPath::x_axis(),
            v: Vector::unit_y(),
        }));
        let xz_plane = store.insert(Surface::new(SurfaceGeometry {
            u: GlobalPath::x_axis(),
            v: Vector::unit_z(),
        }));
        let yz_plane = store.insert(Surface::new(SurfaceGeometry {
            u: GlobalPath::y_axis(),
            v: Vector::unit_z(),
        }));

        Self {
            store,
            xy_plane,
            xz_plane,
            yz_plane,
        }
    }
}

/// Store for [`Vertex`] objects
#[derive(Debug, Default)]
pub struct Vertices {
    store: Store<Vertex>,
}

impl Vertices {
    /// Insert a [`Vertex`] into the store
    pub fn insert(
        &self,
        vertex: Vertex,
    ) -> Result<Handle<Vertex>, VertexValidationError> {
        vertex.validate()?;
        Ok(self.store.insert(vertex))
    }
}