use std::collections::HashMap;

use super::dsl::*;
use rsprocess::{graph, label, set, system, translator};
use rsprocess::translator::PrintableWithTranslator;

// ----------------------------------------------------------------------------
//                       Specific Assert Implementation
// ----------------------------------------------------------------------------

/// Module that has all types and structures for bisimilarity relabeler.
pub mod useful_types_edge_relabeler {
    macro_rules! export_types {
	( $( $x:ident ),* ) => {
	    $(
		pub type $x = super::super::dsl::$x<super::EdgeRelablerInput>;
	    )*
	};
    }

    macro_rules! export_types_no_parameter {
	( $( $x:ident ),* ) => {
	    $(
		pub type $x = super::super::dsl::$x;
	    )*
	};
    }

    export_types!(Assert, Tree, Variable, Expression, Range);

    export_types_no_parameter!(
        Unary,
        QualifierRestricted,
        QualifierLabel,
        QualifierSystem,
        QualifierEdge,
        QualifierNode,
        Qualifier,
        Binary,
        AssertReturnValue
    );

    pub type Special = super::EdgeRelablerInput;
}

// Implementation for graph labeling in bisimulation.

#[derive(Copy, Clone, PartialEq, Eq, Hash)]
pub enum EdgeRelablerInput {
    Label,
    Edge,
}

#[derive(Clone)]
enum EdgeRelablerInputValues {
    Label(label::Label),
    Edge(petgraph::graph::EdgeIndex),
}

impl SpecialVariables<EdgeRelablerInputValues> for EdgeRelablerInput {
    fn type_of(&self) -> AssertionTypes {
        match self {
            | Self::Edge => AssertionTypes::Edge,
            | Self::Label => AssertionTypes::Label,
        }
    }

    fn type_qualified(&self, q: &Qualifier) -> Result<AssertionTypes, String> {
        match (self, q) {
            | (Self::Label, Qualifier::Label(_))
            | (Self::Label, Qualifier::Restricted(_)) =>
                Ok(AssertionTypes::Set),
            | (s, q) =>
                Err(format!("Wrong use of qualifier {q:?} on variable {s:?}.")),
        }
    }

    fn new_context(
        input: HashMap<Self, EdgeRelablerInputValues>,
    ) -> HashMap<Self, AssertReturnValue> {
        input
            .iter()
            .map(|(key, value)| match value {
                | EdgeRelablerInputValues::Edge(e) =>
                    (*key, AssertReturnValue::Edge(*e)),
                | EdgeRelablerInputValues::Label(l) =>
                    (*key, AssertReturnValue::Label(l.clone())),
            })
            .collect::<HashMap<Self, AssertReturnValue>>()
    }

    fn correct_type(&self, other: &AssertReturnValue) -> bool {
        match (self, other) {
            | (Self::Edge, AssertReturnValue::Edge(_))
            | (Self::Label, AssertReturnValue::Label(_)) => true,
            | (_, _) => false,
        }
    }
}

impl std::fmt::Debug for EdgeRelablerInput {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            | Self::Label => write!(f, "label"),
            | Self::Edge => write!(f, "edge"),
        }
    }
}

impl PrintableWithTranslator for EdgeRelablerInput {
    fn print(
        &self,
        f: &mut std::fmt::Formatter,
        _translator: &translator::Translator,
    ) -> std::fmt::Result {
        write!(f, "{self:?}")
    }
}

impl Assert<EdgeRelablerInput> {
    pub fn typecheck(&self) -> Result<(), String> {
        let mut context = TypeContext::new();
        let ty = typecheck(&self.tree, &mut context)?;
        match ty {
            | AssertionTypes::Boolean
            | AssertionTypes::Integer
            | AssertionTypes::String
            | AssertionTypes::Label
            | AssertionTypes::Set
            | AssertionTypes::Element
            | AssertionTypes::Edge
            | AssertionTypes::Node
            | AssertionTypes::System
            | AssertionTypes::Context => Ok(()),
            | AssertionTypes::NoType =>
                Err("No return type, at least one return statement required."
                    .into()),
            | AssertionTypes::RangeInteger
            | AssertionTypes::RangeSet
            | AssertionTypes::RangeNeighbours =>
                Err(format!("Returned type {ty:?} is not a valid return type.")),
        }
    }

    pub fn execute(
        &self,
        graph: &graph::SystemGraph,
        edge: &<graph::SystemGraph as petgraph::visit::GraphBase>::EdgeId,
        translator: &mut translator::Translator,
    ) -> Result<AssertReturnValue, String> {
        let label = graph
            .edge_weight(*edge)
            .ok_or("Missing edge {{debug: {edge:?}}}")?;

        let mut input_vals = HashMap::new();
        input_vals.insert(
            EdgeRelablerInput::Edge,
            EdgeRelablerInputValues::Edge(*edge),
        );
        input_vals.insert(
            EdgeRelablerInput::Label,
            EdgeRelablerInputValues::Label(label.clone()),
        );

        let mut context = Context::new(input_vals);
        if let Some(v) = execute(&self.tree, &mut context, translator, graph)? {
            Ok(v)
        } else {
            Err("No value returned.".into())
        }
    }
}

// -----------------------------------------------------------------------------
// Implementation for node grouping.
// -----------------------------------------------------------------------------

/// Module that has all types and structures for bisimilarity relabeler.
pub mod useful_types_node_relabeler {
    macro_rules! export_types {
	( $( $x:ident ),* ) => {
	    $(
		pub type $x = super::super::dsl::$x<super::NodeRelablerInput>;
	    )*
	};
    }

    macro_rules! export_types_no_parameter {
	( $( $x:ident ),* ) => {
	    $(
		pub type $x = super::super::dsl::$x;
	    )*
	};
    }

    export_types!(Assert, Tree, Variable, Expression, Range);

    export_types_no_parameter!(
        Unary,
        QualifierRestricted,
        QualifierLabel,
        QualifierSystem,
        QualifierEdge,
        QualifierNode,
        Qualifier,
        Binary,
        AssertReturnValue
    );

    pub type Special = super::NodeRelablerInput;
}

#[derive(Copy, Clone, PartialEq, Eq, Hash)]
pub enum NodeRelablerInput {
    Entities,
    Node,
}

#[derive(Clone)]
enum NodeRelablerInputValues {
    Entities(set::Set),
    Node(petgraph::graph::NodeIndex),
}

impl SpecialVariables<NodeRelablerInputValues> for NodeRelablerInput {
    fn type_of(&self) -> AssertionTypes {
        match self {
            | Self::Entities => AssertionTypes::Set,
            | Self::Node => AssertionTypes::Node,
        }
    }

    fn type_qualified(&self, q: &Qualifier) -> Result<AssertionTypes, String> {
        match (self, q) {
            | (Self::Node, Qualifier::Node(QualifierNode::System)) =>
                Ok(AssertionTypes::System),
            | (Self::Node, Qualifier::Node(QualifierNode::Neighbours)) =>
                Ok(AssertionTypes::RangeNeighbours),
            | (s, q) =>
                Err(format!("Wrong use of qualifier {q:?} on variable {s:?}.")),
        }
    }

    fn new_context(
        input: HashMap<Self, NodeRelablerInputValues>,
    ) -> HashMap<Self, AssertReturnValue> {
        input
            .iter()
            .map(|(key, value)| match value {
                | NodeRelablerInputValues::Entities(e) =>
                    (*key, AssertReturnValue::Set(e.clone())),
                | NodeRelablerInputValues::Node(n) =>
                    (*key, AssertReturnValue::Node(*n)),
            })
            .collect::<HashMap<Self, AssertReturnValue>>()
    }

    fn correct_type(&self, other: &AssertReturnValue) -> bool {
        match (self, other) {
            | (Self::Entities, AssertReturnValue::Set(_))
            | (Self::Node, AssertReturnValue::Node(_)) => true,
            | (_, _) => false,
        }
    }
}

impl std::fmt::Debug for NodeRelablerInput {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            | Self::Entities => write!(f, "entities"),
            | Self::Node => write!(f, "node"),
        }
    }
}

impl PrintableWithTranslator for NodeRelablerInput {
    fn print(
        &self,
        f: &mut std::fmt::Formatter,
        _translator: &translator::Translator,
    ) -> std::fmt::Result {
        write!(f, "{self:?}")
    }
}

impl Assert<NodeRelablerInput> {
    pub fn typecheck(&self) -> Result<(), String> {
        let mut context = TypeContext::new();
        let ty = typecheck(&self.tree, &mut context)?;
        match ty {
            | AssertionTypes::Boolean
            | AssertionTypes::Integer
            | AssertionTypes::String
            | AssertionTypes::Label
            | AssertionTypes::Set
            | AssertionTypes::Element
            | AssertionTypes::Edge
            | AssertionTypes::Node
            | AssertionTypes::System
            | AssertionTypes::Context => Ok(()),
            | AssertionTypes::NoType =>
                Err("No return type, at least one return statement required."
                    .into()),
            | AssertionTypes::RangeInteger
            | AssertionTypes::RangeSet
            | AssertionTypes::RangeNeighbours =>
                Err(format!("Returned type {ty:?} is not a valid return type.")),
        }
    }

    pub fn execute(
        &self,
        graph: &graph::SystemGraph,
        node: &<graph::SystemGraph as petgraph::visit::GraphBase>::NodeId,
        translator: &mut translator::Translator,
    ) -> Result<AssertReturnValue, String> {
        let system::System {
            available_entities: entities,
            ..
        } = graph
            .node_weight(*node)
            .ok_or("Missing node {{debug: {node:?}}}")?;

        let mut input_vals = HashMap::new();
        input_vals.insert(
            NodeRelablerInput::Entities,
            NodeRelablerInputValues::Entities(entities.clone()),
        );
        input_vals.insert(
            NodeRelablerInput::Node,
            NodeRelablerInputValues::Node(*node),
        );

        let mut context = Context::new(input_vals);
        if let Some(v) = execute(&self.tree, &mut context, translator, graph)? {
            Ok(v)
        } else {
            Err("No value returned.".into())
        }
    }
}