1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use std::hash::{Hash, Hasher};

use sway_types::{Ident, Named, Span, Spanned};

use crate::{
    decl_engine::{DeclMapping, ReplaceDecls},
    engine_threading::*,
    language::{ty::*, CallPath, Visibility},
    transform,
    type_system::*,
};

#[derive(Clone, Debug)]
pub struct TyConstantDecl {
    pub call_path: CallPath,
    pub value: Option<TyExpression>,
    pub visibility: Visibility,
    pub is_configurable: bool,
    pub attributes: transform::AttributesMap,
    pub return_type: TypeId,
    pub type_ascription: TypeArgument,
    pub span: Span,
    pub implementing_type: Option<TyDecl>,
}

impl EqWithEngines for TyConstantDecl {}
impl PartialEqWithEngines for TyConstantDecl {
    fn eq(&self, other: &Self, engines: Engines<'_>) -> bool {
        let type_engine = engines.te();
        self.call_path == other.call_path
            && self.value.eq(&other.value, engines)
            && self.visibility == other.visibility
            && self.type_ascription.eq(&other.type_ascription, engines)
            && self.is_configurable == other.is_configurable
            && type_engine
                .get(self.return_type)
                .eq(&type_engine.get(other.return_type), engines)
            && match (&self.implementing_type, &other.implementing_type) {
                (Some(self_), Some(other)) => self_.eq(other, engines),
                _ => false,
            }
    }
}

impl HashWithEngines for TyConstantDecl {
    fn hash<H: Hasher>(&self, state: &mut H, engines: Engines<'_>) {
        let type_engine = engines.te();
        let TyConstantDecl {
            call_path,
            value,
            visibility,
            return_type,
            type_ascription,
            is_configurable,
            implementing_type,
            // these fields are not hashed because they aren't relevant/a
            // reliable source of obj v. obj distinction
            attributes: _,
            span: _,
        } = self;
        call_path.hash(state);
        value.hash(state, engines);
        visibility.hash(state);
        type_engine.get(*return_type).hash(state, engines);
        type_ascription.hash(state, engines);
        is_configurable.hash(state);
        if let Some(implementing_type) = implementing_type {
            (*implementing_type).hash(state, engines);
        }
    }
}

impl Named for TyConstantDecl {
    fn name(&self) -> &Ident {
        &self.call_path.suffix
    }
}

impl Spanned for TyConstantDecl {
    fn span(&self) -> Span {
        self.span.clone()
    }
}

impl SubstTypes for TyConstantDecl {
    fn subst_inner(&mut self, type_mapping: &TypeSubstMap, engines: Engines<'_>) {
        self.return_type.subst(type_mapping, engines);
        self.type_ascription.subst(type_mapping, engines);
        if let Some(expr) = &mut self.value {
            expr.subst(type_mapping, engines);
        }
    }
}

impl ReplaceSelfType for TyConstantDecl {
    fn replace_self_type(&mut self, engines: Engines<'_>, self_type: TypeId) {
        self.return_type.replace_self_type(engines, self_type);
        self.type_ascription.replace_self_type(engines, self_type);
        if let Some(expr) = &mut self.value {
            expr.replace_self_type(engines, self_type);
        }
    }
}

impl ReplaceDecls for TyConstantDecl {
    fn replace_decls_inner(&mut self, decl_mapping: &DeclMapping, engines: Engines<'_>) {
        if let Some(expr) = &mut self.value {
            expr.replace_decls(decl_mapping, engines);
        }
    }
}