shape-ast 0.1.8

AST types and Pest grammar for the Shape 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
122
123
124
125
126
127
128
129
130
131
132
133
134
135

//! Literal types for Shape AST

use rust_decimal::Decimal;
use serde::{Deserialize, Serialize};

use crate::data::Timeframe;
use crate::int_width::IntWidth;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
pub enum InterpolationMode {
    Braces,
    Dollar,
    Hash,
}

impl InterpolationMode {
    pub fn prefix(self) -> &'static str {
        match self {
            InterpolationMode::Braces => "f",
            InterpolationMode::Dollar => "f$",
            InterpolationMode::Hash => "f#",
        }
    }

    pub fn sigil(self) -> Option<char> {
        match self {
            InterpolationMode::Braces => None,
            InterpolationMode::Dollar => Some('$'),
            InterpolationMode::Hash => Some('#'),
        }
    }
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub enum Literal {
    Int(i64),
    /// Unsigned integer literal > i64::MAX (e.g., 18446744073709551615u64)
    UInt(u64),
    /// Explicitly width-typed integer literal (e.g., 42i8, 100u16)
    TypedInt(i64, IntWidth),
    Number(f64),
    /// Decimal type for exact arithmetic (finance, currency)
    Decimal(Decimal),
    String(String),
    /// Unicode scalar value char literal (`'a'`, `'\n'`, `'\u{1F600}'`)
    Char(char),
    /// Formatted string literal (`f"..."`, `f$"..."`, `f#"..."` + triple variants)
    FormattedString {
        value: String,
        mode: InterpolationMode,
    },
    /// Content string literal (`c"..."`, `c$"..."`, `c#"..."` + triple variants)
    ContentString {
        value: String,
        mode: InterpolationMode,
    },
    Bool(bool),
    /// Option::None literal (replaces null)
    None,
    /// Unit literal `()`
    Unit,
    Timeframe(Timeframe),
}

impl std::fmt::Display for Literal {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Literal::Int(i) => write!(f, "{}", i),
            Literal::UInt(u) => write!(f, "{}u64", u),
            Literal::TypedInt(v, w) => write!(f, "{}{}", v, w),
            Literal::Number(n) => {
                if n.fract() == 0.0 {
                    write!(f, "{}", *n as i64)
                } else {
                    write!(f, "{}", n)
                }
            }
            Literal::Decimal(d) => write!(f, "{}D", d),
            Literal::String(s) => write!(f, "\"{}\"", s),
            Literal::Char(c) => write!(f, "'{}'", c.escape_default()),
            Literal::FormattedString { value, mode } => write!(f, "{}\"{}\"", mode.prefix(), value),
            Literal::ContentString { value, mode } => {
                let prefix = match mode {
                    InterpolationMode::Braces => "c",
                    InterpolationMode::Dollar => "c$",
                    InterpolationMode::Hash => "c#",
                };
                write!(f, "{}\"{}\"", prefix, value)
            }
            Literal::Bool(b) => write!(f, "{}", b),
            Literal::None => write!(f, "None"),
            Literal::Unit => write!(f, "()"),
            Literal::Timeframe(tf) => write!(f, "{}", tf),
        }
    }
}

impl Literal {
    /// Convert literal to a JSON value
    pub fn to_json_value(&self) -> serde_json::Value {
        match self {
            Literal::Int(i) => serde_json::json!(*i),
            Literal::UInt(u) => serde_json::json!(*u),
            Literal::TypedInt(v, _) => serde_json::json!(*v),
            Literal::Number(n) => serde_json::json!(*n),
            Literal::Decimal(d) => serde_json::json!(d.to_string()),
            Literal::String(s) => serde_json::json!(s),
            Literal::Char(c) => serde_json::json!(c.to_string()),
            Literal::FormattedString { value, .. } => serde_json::json!(value),
            Literal::ContentString { value, .. } => serde_json::json!(value),
            Literal::Bool(b) => serde_json::json!(*b),
            Literal::None => serde_json::Value::Null,
            Literal::Unit => serde_json::Value::Null,
            Literal::Timeframe(t) => serde_json::json!(t.to_string()),
        }
    }
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct Duration {
    pub value: f64,
    pub unit: DurationUnit,
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub enum DurationUnit {
    Seconds,
    Minutes,
    Hours,
    Days,
    Weeks,
    Months,
    Years,
    Samples,
}