numbat 1.23.0

use std::num::NonZeroUsize;

use crate::markup as m;
use crate::resolver::ModulePathBorrowed;
use crate::span::Span;
use crate::{
    arithmetic::Exponent, decorator::Decorator, markup::Markup, number::Number, prefix::Prefix,
    pretty_print::PrettyPrint,
};
use compact_str::{CompactString, ToCompactString, format_compact};
use itertools::Itertools;
use num_traits::Signed;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum UnaryOperator {
    Factorial(NonZeroUsize),
    Negate,
    LogicalNeg,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum BinaryOperator {
    Add,
    Sub,
    Mul,
    Div,
    Power,
    ConvertTo,
    LessThan,
    GreaterThan,
    LessOrEqual,
    GreaterOrEqual,
    Equal,
    NotEqual,
    LogicalAnd,
    LogicalOr,
}

impl PrettyPrint for BinaryOperator {
    fn pretty_print(&self) -> Markup {
        use BinaryOperator::*;

        let operator = m::operator(match self {
            Add => "+",
            Sub => "-",
            Mul => "×",
            Div => "/",
            Power => "^",
            ConvertTo => "➞",
            LessThan => "<",
            GreaterThan => ">",
            LessOrEqual => "≤",
            GreaterOrEqual => "≥",
            Equal => "==",
            NotEqual => "≠",
            LogicalAnd => "&&",
            LogicalOr => "||",
        });

        match self {
            Power => operator,
            _ => m::space() + operator + m::space(),
        }
    }
}

#[derive(Debug, Clone, PartialEq)]
pub enum StringPart<'a> {
    Fixed(CompactString),
    Interpolation {
        span: Span,
        expr: Box<Expression<'a>>,
        format_specifiers: Option<&'a str>,
    },
}

#[derive(Debug, Clone, PartialEq)]
pub enum Expression<'a> {
    Scalar(Span, Number),
    Identifier(Span, &'a str),
    UnitIdentifier {
        span: Span,
        prefix: Prefix,
        name: CompactString,      // can't easily be made &'a str
        full_name: CompactString, // can't easily be made &'a str
    },
    TypedHole(Span),
    UnaryOperator {
        op: UnaryOperator,
        expr: Box<Expression<'a>>,
        span_op: Span,
    },
    BinaryOperator {
        op: BinaryOperator,
        lhs: Box<Expression<'a>>,
        rhs: Box<Expression<'a>>,
        span_op: Option<Span>, // not available for implicit multiplication and unicode exponents
    },
    FunctionCall {
        ident_span: Span,
        full_span: Span,
        callable: Box<Expression<'a>>,
        args: Vec<Expression<'a>>,
    },
    Boolean(Span, bool),
    String(Span, Vec<StringPart<'a>>),
    Condition {
        span: Span,
        condition: Box<Expression<'a>>,
        then_expr: Box<Expression<'a>>,
        else_expr: Box<Expression<'a>>,
    },
    InstantiateStruct {
        full_span: Span,
        ident_span: Span,
        name: &'a str,
        fields: Vec<(Span, &'a str, Expression<'a>)>,
    },
    AccessField {
        full_span: Span,
        ident_span: Span,
        expr: Box<Expression<'a>>,
        field_name: &'a str,
    },
    List(Span, Vec<Expression<'a>>),
}

impl Expression<'_> {
    pub fn full_span(&self) -> Span {
        match self {
            Expression::Scalar(span, _) => *span,
            Expression::Identifier(span, _) => *span,
            Expression::UnitIdentifier { span, .. } => *span,
            Expression::UnaryOperator {
                op: _,
                expr,
                span_op,
            } => span_op.extend(&expr.full_span()),
            Expression::BinaryOperator {
                op: _,
                lhs,
                rhs,
                span_op,
            } => {
                let mut span = lhs.full_span().extend(&rhs.full_span());
                if let Some(span_op) = span_op {
                    span = span.extend(span_op);
                }
                span
            }
            Expression::FunctionCall { full_span, .. } => *full_span,
            Expression::Boolean(span, _) => *span,
            Expression::Condition {
                span, else_expr, ..
            } => span.extend(&else_expr.full_span()),
            Expression::String(span, _) => *span,
            Expression::InstantiateStruct { full_span, .. } => *full_span,
            Expression::AccessField { full_span, .. } => *full_span,
            Expression::List(span, _) => *span,
            Expression::TypedHole(span) => *span,
        }
    }

    /// Check if this expression is a plain scalar (no negation or other operations).
    pub fn is_scalar(&self) -> bool {
        matches!(self, Expression::Scalar(_, _))
    }

    /// Check if this expression is an identifier.
    pub fn is_identifier(&self) -> bool {
        matches!(self, Expression::Identifier(_, _))
    }
}

#[cfg(test)]
macro_rules! scalar {
    ( $num:expr ) => {{ crate::ast::Expression::Scalar(Span::dummy(), Number::from_f64($num)) }};
}

#[cfg(test)]
macro_rules! identifier {
    ( $name:expr ) => {{ crate::ast::Expression::Identifier(Span::dummy(), $name.into()) }};
}

#[cfg(test)]
macro_rules! boolean {
    ( $name:expr ) => {{ crate::ast::Expression::Boolean(Span::dummy(), $name.into()) }};
}

#[cfg(test)]
macro_rules! logical_neg {
    ( $rhs:expr ) => {{
        crate::ast::Expression::UnaryOperator {
            op: UnaryOperator::LogicalNeg,
            expr: Box::new($rhs),
            span_op: Span::dummy(),
        }
    }};
}

#[cfg(test)]
macro_rules! negate {
    ( $rhs:expr ) => {{
        crate::ast::Expression::UnaryOperator {
            op: UnaryOperator::Negate,
            expr: Box::new($rhs),
            span_op: Span::dummy(),
        }
    }};
}

#[cfg(test)]
macro_rules! factorial {
    ( $lhs:expr, $order:expr ) => {{
        crate::ast::Expression::UnaryOperator {
            op: UnaryOperator::Factorial(NonZeroUsize::new($order).unwrap()),
            expr: Box::new($lhs),
            span_op: Span::dummy(),
        }
    }};
}

#[cfg(test)]
macro_rules! binop {
    ( $lhs:expr, $op:ident, $rhs: expr ) => {{
        crate::ast::Expression::BinaryOperator {
            op: BinaryOperator::$op,
            lhs: Box::new($lhs),
            rhs: Box::new($rhs),
            span_op: Some(Span::dummy()),
        }
    }};
}

#[cfg(test)]
macro_rules! conditional {
    ( $cond:expr, $lhs:expr, $rhs: expr ) => {{
        crate::ast::Expression::Condition {
            span: Span::dummy(),
            condition: Box::new($cond),
            then_expr: Box::new($lhs),
            else_expr: Box::new($rhs),
        }
    }};
}

#[cfg(test)]
macro_rules! struct_ {
    ( $name:ident, $( $field:ident : $val:expr ),* ) => {{
        crate::ast::Expression::InstantiateStruct {
            full_span: Span::dummy(),
            ident_span: Span::dummy(),
            name: stringify!($name),
            fields: vec![
                $((Span::dummy(), stringify!($field), $val)),*
            ]
        }
    }};
}

#[cfg(test)]
macro_rules! list {
    ( $( $val:expr ),* ) => {
        crate::ast::Expression::List(
             Span::dummy(),
            vec![$($val,)*],
        )
    };
}

#[cfg(test)]
pub(crate) use binop;
#[cfg(test)]
pub(crate) use boolean;
#[cfg(test)]
pub(crate) use conditional;
#[cfg(test)]
pub(crate) use factorial;
#[cfg(test)]
pub(crate) use identifier;
#[cfg(test)]
pub(crate) use list;
#[cfg(test)]
pub(crate) use logical_neg;
#[cfg(test)]
pub(crate) use negate;
#[cfg(test)]
pub(crate) use scalar;
#[cfg(test)]
pub(crate) use struct_;

#[derive(Debug, Clone, PartialEq)]
pub enum TypeAnnotation {
    TypeExpression(TypeExpression),
    Bool(Span),
    String(Span),
    DateTime(Span),
    Fn(Span, Vec<TypeAnnotation>, Box<TypeAnnotation>),
    List(Span, Box<TypeAnnotation>),
}

impl TypeAnnotation {
    pub fn full_span(&self) -> Span {
        match self {
            TypeAnnotation::TypeExpression(d) => d.full_span(),
            TypeAnnotation::Bool(span) => *span,
            TypeAnnotation::String(span) => *span,
            TypeAnnotation::DateTime(span) => *span,
            TypeAnnotation::Fn(span, _, _) => *span,
            TypeAnnotation::List(span, _) => *span,
        }
    }
}

impl PrettyPrint for TypeAnnotation {
    fn pretty_print(&self) -> Markup {
        match self {
            TypeAnnotation::TypeExpression(d) => d.pretty_print(),
            TypeAnnotation::Bool(_) => m::type_identifier("Bool"),
            TypeAnnotation::String(_) => m::type_identifier("String"),
            TypeAnnotation::DateTime(_) => m::type_identifier("DateTime"),
            TypeAnnotation::Fn(_, parameter_types, return_type) => {
                m::type_identifier("Fn")
                    + m::operator("[(")
                    + Itertools::intersperse(
                        parameter_types.iter().map(|t| t.pretty_print()),
                        m::operator(",") + m::space(),
                    )
                    .sum()
                    + m::operator(")")
                    + m::space()
                    + m::operator("->")
                    + m::space()
                    + return_type.pretty_print()
                    + m::operator("]")
            }
            TypeAnnotation::List(_, element_type) => {
                m::type_identifier("List")
                    + m::operator("<")
                    + element_type.pretty_print()
                    + m::operator(">")
            }
        }
    }
}

#[derive(Debug, Clone, PartialEq)]

pub enum TypeExpression {
    Unity(Span),
    /// A type identifier, optionally with type arguments like `SomeStruct<A, B>`
    TypeIdentifier(Span, CompactString, Vec<TypeAnnotation>),
    Multiply(Span, Box<TypeExpression>, Box<TypeExpression>),
    Divide(Span, Box<TypeExpression>, Box<TypeExpression>),
    Power(
        Option<Span>, // operator span, not available for unicode exponents
        Box<TypeExpression>,
        Span, // span for the exponent
        Exponent,
    ),
}

impl TypeExpression {
    pub fn full_span(&self) -> Span {
        match self {
            TypeExpression::Unity(s) => *s,
            TypeExpression::TypeIdentifier(s, _, _) => *s,
            TypeExpression::Multiply(span_op, lhs, rhs) => {
                span_op.extend(&lhs.full_span()).extend(&rhs.full_span())
            }
            TypeExpression::Divide(span_op, lhs, rhs) => {
                span_op.extend(&lhs.full_span()).extend(&rhs.full_span())
            }
            TypeExpression::Power(span_op, lhs, span_exponent, _exp) => match span_op {
                Some(span_op) => span_op.extend(&lhs.full_span()).extend(span_exponent),
                None => lhs.full_span().extend(span_exponent),
            },
        }
    }
}

fn with_parens(dexpr: &TypeExpression) -> Markup {
    match dexpr {
        expr @ (TypeExpression::Unity(..)
        | TypeExpression::TypeIdentifier(..)
        | TypeExpression::Power(..)) => expr.pretty_print(),
        expr @ (TypeExpression::Multiply(..) | TypeExpression::Divide(..)) => {
            m::operator("(") + expr.pretty_print() + m::operator(")")
        }
    }
}

impl PrettyPrint for TypeExpression {
    fn pretty_print(&self) -> Markup {
        match self {
            TypeExpression::Unity(_) => m::type_identifier("1"),
            TypeExpression::TypeIdentifier(_, ident, type_args) => {
                let mut markup = m::type_identifier(ident.to_compact_string());
                if !type_args.is_empty() {
                    markup += m::operator("<");
                    for (i, arg) in type_args.iter().enumerate() {
                        if i > 0 {
                            markup += m::operator(",") + m::space();
                        }
                        markup += arg.pretty_print();
                    }
                    markup += m::operator(">");
                }
                markup
            }
            TypeExpression::Multiply(_, lhs, rhs) => {
                lhs.pretty_print() + m::space() + m::operator("×") + m::space() + rhs.pretty_print()
            }
            TypeExpression::Divide(_, lhs, rhs) => {
                lhs.pretty_print() + m::space() + m::operator("/") + m::space() + with_parens(rhs)
            }
            TypeExpression::Power(_, lhs, _, exp) => {
                with_parens(lhs)
                    + m::operator("^")
                    + if exp.is_positive() {
                        m::value(format_compact!("{exp}"))
                    } else {
                        m::operator("(") + m::value(format_compact!("{exp}")) + m::operator(")")
                    }
            }
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum ProcedureKind {
    Print,
    Assert,
    AssertEq,
    Type,
}

impl ProcedureKind {
    pub(crate) fn name(&self) -> &'static str {
        match self {
            ProcedureKind::Print => "print",
            ProcedureKind::Assert => "assert",
            ProcedureKind::AssertEq => "assert_eq",
            ProcedureKind::Type => "type",
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum TypeParameterBound {
    Dim,
}

#[derive(Debug, Clone, PartialEq)]
pub struct DefineVariable<'a> {
    pub identifier_span: Span,
    pub identifier: &'a str,
    pub expr: Expression<'a>,
    pub type_annotation: Option<TypeAnnotation>,
    pub decorators: Vec<Decorator<'a>>,
}

#[derive(Debug, Clone, PartialEq)]
pub enum Statement<'a> {
    Expression(Expression<'a>),
    DefineVariable(DefineVariable<'a>),
    DefineFunction {
        fn_keyword_span: Span,
        function_name_span: Span,
        function_name: &'a str,
        type_parameters: Vec<(Span, &'a str, Option<TypeParameterBound>)>,
        /// Parameters, optionally with type annotations.
        parameters: Vec<(Span, &'a str, Option<TypeAnnotation>)>,
        /// Function body. If it is absent, the function is implemented via FFI
        body: Option<Expression<'a>>,
        /// Local variables
        local_variables: Vec<DefineVariable<'a>>,
        /// Optional annotated return type
        return_type_annotation: Option<TypeAnnotation>,
        decorators: Vec<Decorator<'a>>,
    },
    DefineDimension(Span, &'a str, Vec<TypeExpression>),
    DefineBaseUnit(Span, &'a str, Option<TypeExpression>, Vec<Decorator<'a>>),
    DefineDerivedUnit {
        identifier_span: Span,
        identifier: &'a str,
        expr: Expression<'a>,
        type_annotation_span: Option<Span>,
        type_annotation: Option<TypeAnnotation>,
        decorators: Vec<Decorator<'a>>,
    },
    ProcedureCall(Span, ProcedureKind, Vec<Expression<'a>>),
    ModuleImport(Span, ModulePathBorrowed<'a>),
    DefineStruct {
        struct_name_span: Span,
        struct_name: &'a str,
        type_parameters: Vec<(Span, &'a str, Option<TypeParameterBound>)>,
        fields: Vec<(Span, &'a str, TypeAnnotation)>,
    },
}

impl Statement<'_> {
    pub fn full_span(&self) -> Span {
        match self {
            Statement::Expression(expr) => expr.full_span(),
            Statement::DefineVariable(DefineVariable {
                identifier_span,
                expr,
                ..
            }) => identifier_span.extend(&expr.full_span()),
            Statement::DefineFunction {
                fn_keyword_span,
                body,
                return_type_annotation,
                ..
            } => {
                let mut span = *fn_keyword_span;
                if let Some(body) = body {
                    span = span.extend(&body.full_span());
                }
                if let Some(annotation) = return_type_annotation {
                    span = span.extend(&annotation.full_span());
                }
                span
            }
            Statement::DefineDimension(span, _, type_exprs) => {
                let mut full = *span;
                if let Some(last) = type_exprs.last() {
                    full = full.extend(&last.full_span());
                }
                full
            }
            Statement::DefineBaseUnit(span, _, type_expr, _) => {
                let mut full = *span;
                if let Some(te) = type_expr {
                    full = full.extend(&te.full_span());
                }
                full
            }
            Statement::DefineDerivedUnit {
                identifier_span,
                expr,
                type_annotation_span,
                ..
            } => {
                let mut span = identifier_span.extend(&expr.full_span());
                if let Some(ta_span) = type_annotation_span {
                    span = span.extend(ta_span);
                }
                span
            }
            Statement::ProcedureCall(span, _, args) => {
                let mut full = *span;
                if let Some(last) = args.last() {
                    full = full.extend(&last.full_span());
                }
                full
            }
            Statement::ModuleImport(span, _) => *span,
            Statement::DefineStruct {
                struct_name_span,
                fields,
                ..
            } => {
                let mut span = *struct_name_span;
                if let Some((last_span, _, annotation)) = fields.last() {
                    span = span.extend(last_span).extend(&annotation.full_span());
                }
                span
            }
        }
    }
}

#[cfg(test)]
pub trait ReplaceSpans {
    fn replace_spans(&self) -> Self;
}

#[cfg(test)]
impl ReplaceSpans for TypeAnnotation {
    fn replace_spans(&self) -> Self {
        match self {
            TypeAnnotation::TypeExpression(d) => TypeAnnotation::TypeExpression(d.replace_spans()),
            TypeAnnotation::Bool(_) => TypeAnnotation::Bool(Span::dummy()),
            TypeAnnotation::String(_) => TypeAnnotation::String(Span::dummy()),
            TypeAnnotation::DateTime(_) => TypeAnnotation::DateTime(Span::dummy()),
            TypeAnnotation::Fn(_, pt, rt) => {
                TypeAnnotation::Fn(Span::dummy(), pt.clone(), rt.clone())
            }
            TypeAnnotation::List(_, et) => {
                TypeAnnotation::List(Span::dummy(), Box::new(et.replace_spans()))
            }
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for TypeExpression {
    fn replace_spans(&self) -> Self {
        match self {
            TypeExpression::Unity(_) => TypeExpression::Unity(Span::dummy()),
            TypeExpression::TypeIdentifier(_, name, type_args) => TypeExpression::TypeIdentifier(
                Span::dummy(),
                name.clone(),
                type_args.iter().map(|a| a.replace_spans()).collect(),
            ),
            TypeExpression::Multiply(_, lhs, rhs) => TypeExpression::Multiply(
                Span::dummy(),
                Box::new(lhs.replace_spans()),
                Box::new(rhs.replace_spans()),
            ),
            TypeExpression::Divide(_, lhs, rhs) => TypeExpression::Divide(
                Span::dummy(),
                Box::new(lhs.replace_spans()),
                Box::new(rhs.replace_spans()),
            ),
            TypeExpression::Power(span_op, lhs, _, exp) => TypeExpression::Power(
                span_op.map(|_| Span::dummy()),
                Box::new(lhs.replace_spans()),
                Span::dummy(),
                *exp,
            ),
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for StringPart<'_> {
    fn replace_spans(&self) -> Self {
        match self {
            f @ StringPart::Fixed(_) => f.clone(),
            StringPart::Interpolation {
                expr,
                format_specifiers,
                span: _,
            } => StringPart::Interpolation {
                span: Span::dummy(),
                expr: Box::new(expr.replace_spans()),
                format_specifiers: *format_specifiers,
            },
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for Expression<'_> {
    fn replace_spans(&self) -> Self {
        match self {
            Expression::Scalar(_, name) => Expression::Scalar(Span::dummy(), *name),
            Expression::Identifier(_, name) => Expression::Identifier(Span::dummy(), name),
            Expression::UnitIdentifier {
                prefix,
                name,
                full_name,
                ..
            } => Expression::UnitIdentifier {
                span: Span::dummy(),
                prefix: *prefix,
                name: name.clone(),
                full_name: full_name.clone(),
            },
            Expression::UnaryOperator {
                op,
                expr,
                span_op: _,
            } => Expression::UnaryOperator {
                op: *op,
                expr: Box::new(expr.replace_spans()),
                span_op: Span::dummy(),
            },
            Expression::BinaryOperator {
                op,
                lhs,
                rhs,
                span_op: _,
            } => Expression::BinaryOperator {
                op: *op,
                lhs: Box::new(lhs.replace_spans()),
                rhs: Box::new(rhs.replace_spans()),
                span_op: Some(Span::dummy()),
            },
            Expression::FunctionCall { callable, args, .. } => Expression::FunctionCall {
                ident_span: Span::dummy(),
                full_span: Span::dummy(),
                callable: Box::new(callable.replace_spans()),
                args: args
                    .iter()
                    .map(|a: &Expression| a.replace_spans())
                    .collect(),
            },
            Expression::Boolean(_, val) => Expression::Boolean(Span::dummy(), *val),
            Expression::Condition {
                condition,
                then_expr,
                else_expr,
                ..
            } => Expression::Condition {
                span: Span::dummy(),
                condition: Box::new(condition.replace_spans()),
                then_expr: Box::new(then_expr.replace_spans()),
                else_expr: Box::new(else_expr.replace_spans()),
            },
            Expression::String(_, parts) => Expression::String(
                Span::dummy(),
                parts.iter().map(|p| p.replace_spans()).collect(),
            ),
            Expression::InstantiateStruct { name, fields, .. } => Expression::InstantiateStruct {
                full_span: Span::dummy(),
                ident_span: Span::dummy(),
                name,
                fields: fields
                    .iter()
                    .map(|(_, n, v)| (Span::dummy(), *n, v.replace_spans()))
                    .collect(),
            },
            Expression::AccessField {
                expr, field_name, ..
            } => Expression::AccessField {
                full_span: Span::dummy(),
                ident_span: Span::dummy(),
                expr: Box::new(expr.replace_spans()),
                field_name,
            },
            Expression::List(_, elements) => Expression::List(
                Span::dummy(),
                elements.iter().map(|e| e.replace_spans()).collect(),
            ),
            Expression::TypedHole(_) => Expression::TypedHole(Span::dummy()),
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for DefineVariable<'_> {
    fn replace_spans(&self) -> Self {
        Self {
            identifier_span: Span::dummy(),
            identifier: self.identifier,
            expr: self.expr.replace_spans(),
            type_annotation: self.type_annotation.as_ref().map(|t| t.replace_spans()),
            decorators: self.decorators.clone(),
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for Statement<'_> {
    fn replace_spans(&self) -> Self {
        match self {
            Statement::Expression(expr) => Statement::Expression(expr.replace_spans()),
            Statement::DefineVariable(variable) => {
                Statement::DefineVariable(variable.replace_spans())
            }
            Statement::DefineFunction {
                function_name,
                type_parameters,
                parameters,
                body,
                local_variables,
                return_type_annotation,
                decorators,
                ..
            } => Statement::DefineFunction {
                fn_keyword_span: Span::dummy(),
                function_name_span: Span::dummy(),
                function_name,
                type_parameters: type_parameters
                    .iter()
                    .map(|(_, name, bound)| (Span::dummy(), *name, bound.clone()))
                    .collect(),
                parameters: parameters
                    .iter()
                    .map(|(_, name, type_)| {
                        (
                            Span::dummy(),
                            *name,
                            type_.as_ref().map(|t| t.replace_spans()),
                        )
                    })
                    .collect(),
                body: body.clone().map(|b| b.replace_spans()),
                local_variables: local_variables
                    .iter()
                    .map(DefineVariable::replace_spans)
                    .collect(),
                return_type_annotation: return_type_annotation.as_ref().map(|t| t.replace_spans()),
                decorators: decorators.clone(),
            },
            Statement::DefineDimension(_, name, dexprs) => Statement::DefineDimension(
                Span::dummy(),
                name,
                dexprs.iter().map(|t| t.replace_spans()).collect(),
            ),
            Statement::DefineBaseUnit(_, name, type_, decorators) => Statement::DefineBaseUnit(
                Span::dummy(),
                name,
                type_.as_ref().map(|t| t.replace_spans()),
                decorators.clone(),
            ),
            Statement::DefineDerivedUnit {
                identifier_span: _,
                identifier,
                expr,
                type_annotation_span,
                type_annotation,
                decorators,
            } => Statement::DefineDerivedUnit {
                identifier_span: Span::dummy(),
                identifier,
                expr: expr.replace_spans(),
                type_annotation_span: type_annotation_span.map(|_| Span::dummy()),
                type_annotation: type_annotation.as_ref().map(|t| t.replace_spans()),
                decorators: decorators.clone(),
            },
            Statement::ProcedureCall(_, proc, args) => Statement::ProcedureCall(
                Span::dummy(),
                proc.clone(),
                args.iter().map(|a| a.replace_spans()).collect(),
            ),
            Statement::ModuleImport(_, module_path) => {
                Statement::ModuleImport(Span::dummy(), module_path.clone())
            }
            Statement::DefineStruct {
                struct_name,
                type_parameters,
                fields,
                ..
            } => Statement::DefineStruct {
                struct_name_span: Span::dummy(),
                struct_name,
                type_parameters: type_parameters
                    .iter()
                    .map(|(_, name, bound)| (Span::dummy(), *name, bound.clone()))
                    .collect(),
                fields: fields
                    .iter()
                    .map(|(_span, name, type_)| (Span::dummy(), *name, type_.replace_spans()))
                    .collect(),
            },
        }
    }
}

#[cfg(test)]
impl ReplaceSpans for Vec<Statement<'_>> {
    fn replace_spans(&self) -> Self {
        self.iter().map(|s| s.replace_spans()).collect()
    }
}