leo_passes/code_generation/
visit_type.rs

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
// Copyright (C) 2019-2025 Provable 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 {
        // If the type is a record, handle it separately.
        if let Type::Composite(composite) = type_ {
            let this_program_name = self.program_id.unwrap().name.name;
            let program_name = composite.program.unwrap_or(this_program_name);
            if self.symbol_table.lookup_record(Location::new(program_name, composite.id.name)).is_some() {
                if program_name == this_program_name {
                    return format!("{}.record", composite.id.name);
                } else {
                    return format!("{}.aleo/{}.record", program_name, composite.id.name);
                }
            }
        }

        if let Mode::None = visibility {
            Self::visit_type(type_)
        } else {
            format!("{}.{visibility}", Self::visit_type(type_))
        }
    }
}