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

// Copyright (C) 2019-2024 Aleo Systems Inc.
// This file is part of the Leo library.

// The Leo library is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// The Leo library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with the Leo library. If not, see <https://www.gnu.org/licenses/>.

use crate::CodeGenerator;

use leo_ast::{Location, Mode, Type};

impl<'a> CodeGenerator<'a> {
    pub(crate) fn visit_type(input: &Type) -> String {
        match input {
            Type::Address
            | Type::Field
            | Type::Group
            | Type::Scalar
            | Type::Signature
            | Type::String
            | Type::Future(..)
            | Type::Composite(..)
            | Type::Identifier(..)
            | Type::Integer(..) => format!("{input}"),
            Type::Boolean => {
                // Leo calls this just `bool`, which isn't what we need.
                "boolean".into()
            }
            Type::Array(array_type) => {
                format!("[{}; {}u32]", Self::visit_type(array_type.element_type()), array_type.length())
            }
            Type::Mapping(_) => {
                unreachable!("Mapping types are not supported at this phase of compilation")
            }
            //Type::Struct(_) =>  unreachable!("Struct types should not be visited at this phase of compilation"),
            Type::Tuple(_) => {
                unreachable!("Tuple types should not be visited at this phase of compilation")
            }
            Type::Err => unreachable!("Error types should not exist at this phase of compilation"),
            Type::Unit => unreachable!("Unit types are not supported at this phase of compilation"),
        }
    }

    pub(crate) fn visit_type_with_visibility(&self, type_: &'a Type, visibility: Mode) -> String {
        match type_ {
            // When the type is a record
            Type::Composite(struct_)
                if self
                    .symbol_table
                    .lookup_struct(Location::from(struct_), self.program_id.map(|p| p.name.name))
                    .unwrap()
                    .is_record =>
            {
                if struct_.program == self.program_id.map(|p| p.name.name) || struct_.program.is_none() {
                    format!("{}.record", struct_.id.name)
                } else {
                    format!("{}.aleo/{}.record", struct_.program.unwrap(), struct_.id.name)
                }
            }
            _ => match visibility {
                Mode::None => Self::visit_type(type_),
                _ => format!("{}.{visibility}", Self::visit_type(type_)),
            },
        }
    }
}