xenon-codegen 0.1.0

Codegen and AST for the Xenon programming language
Documentation 
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
112
113
114
115
116
117
118
119
120
121

use core::fmt;

use crate::{Visibility, attribute::Attribute, statement::Statement, r#type::Type};

#[derive(Debug, Clone, Default)]
pub struct Function {
    pub r#async: bool,
    pub attrs: Vec<Attribute>,
    pub visibility: Visibility,
    pub name: String,
    pub arguments: Vec<Argument>,
    pub returns: Type,
    pub body: Statement,
}
impl Function {
    pub fn new(name: String, returns: Type, body: Statement) -> Function {
        Function {
            r#async: false,
            attrs: vec![],
            visibility: Visibility::Private,
            name,
            arguments: vec![],
            returns,
            body,
        }
    }

    pub fn is_valid(&self) -> bool {
        for i in 0..self.attrs.len() {
            if !self.attrs[i].is_valid() {
                return false;
            }
        }
        if self.name.is_empty() {
            return false;
        }
        for i in 0..self.arguments.len() {
            if !self.arguments[i].is_valid() {
                return false;
            }
        }
        if !self.returns.is_valid() {
            return false;
        }
        if !self.body.is_valid() {
            return false;
        }

        true
    }
}

impl fmt::Display for Function {
    fn fmt(&self, fmt: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        for i in 0..self.attrs.len() {
            match writeln!(fmt, "{}", self.attrs[i]) {
                Ok(_) => (),
                Err(e) => return Err(e),
            }
        }
        if self.r#async {
            match write!(fmt, "async") {
                Ok(_) => (),
                Err(e) => return Err(e),
            }
        }
        match write!(fmt, "{} fn {}(", self.visibility, self.name) {
            Ok(_) => (),
            Err(e) => return Err(e),
        }
        if !self.arguments.is_empty() {
            match write!(fmt, "{}", self.arguments[0]) {
                Ok(_) => (),
                Err(e) => return Err(e),
            }
            for i in 1..self.arguments.len() {
                match write!(fmt, ", {}", self.arguments[i]) {
                    Ok(_) => (),
                    Err(e) => return Err(e),
                }
            }
        }
        match write!(fmt, ") -> {}", self.returns) {
            Ok(_) => (),
            Err(e) => return Err(e),
        }
        match write!(fmt, " {}", self.body) {
            Ok(_) => (),
            Err(e) => return Err(e),
        }

        Ok(())
    }
}

#[derive(Debug, Clone, Default)]
pub struct Argument {
    pub name: String,
    pub r#type: Type,
}
impl Argument {
    pub fn new(nm: String, ty: Type) -> Argument {
        Argument {
            name: nm,
            r#type: ty,
        }
    }

    pub fn is_valid(&self) -> bool {
        !self.name.is_empty() && self.r#type.is_valid()
    }
}
impl fmt::Display for Argument {
    fn fmt(&self, fmt: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match write!(fmt, "{}: {}", self.name, self.r#type) {
            Ok(_) => (),
            Err(e) => return Err(e),
        }
        Ok(())
    }
}