bitis_lib/lib_impl/
compiler.rs

1use std::collections::HashMap;
2use std::process::{abort};
3use askama::Template;
4use logos::{Lexer, Logos, Span};
5use regex::Regex;
6use stringcase::Caser;
7
8
9// ************************************************************************
10// ************************************************************************
11fn integer_bit_size(bit_size: &u8) -> u8 {
12    match bit_size {
13        0..=16 => 16,
14        // 0..=8 => 8,
15        // 9..=16 => 16,
16        17..=32 => 32,
17        _ => 64
18    }
19}
20// ************************************************************************
21#[derive(Debug, Clone )]
22#[allow(unused)]
23pub struct AttributeEx {
24    base: Attribute,
25    rust_type_str: String,
26    // rust_pyo3_str: String,
27    base_type_str: String,
28    is_py_wrapped: bool, is_msg: bool, is_enum: bool, is_oo: bool, add_val: bool,
29}
30#[derive(Debug, Clone )]
31pub struct MessageR {
32    pub name: String,
33    // pub version_info: VersionInfo,
34    pub comment: Option<String>,
35    pub parent: Option<String>,
36    pub attributes: Vec<AttributeEx>,
37}
38#[derive(Debug, Clone )]
39#[allow(dead_code)]
40pub struct OneOfInfoR {
41    msg_name: String,
42    name: String,
43    dyn_bits: u8,
44    attributes: Vec<AttributeEx>,
45    default_attrib_name: String,
46}
47fn to_rust_attribute(attribute: &Attribute, msg_names: &Vec<String>) -> AttributeEx {
48    let (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val) = {
49        let mut is_py_wrapped = false;
50        let mut is_enum = false;
51        let mut is_msg = false;
52        let mut is_oo = false;
53        let mut add_val = false;
54
55        let (rtype, base_type) = match &attribute.specific_details {
56            AttributeDetails::AttributeSimple(a) => {
57                match a {
58                    SimpleType::NoTypeSetYet => {
59                        println!("Unexpected unspecified attribute type");
60                        abort()
61                    },
62                    SimpleType::Bool => { ("bool".to_string(), "bool".to_string()) },
63                    SimpleType::UIntFixed(b) => {
64                        add_val = true;
65                        let base = format!("u{}", integer_bit_size(&b));
66                        (format!("IntWithGivenBitSize<{}, {}>", base.clone(), b), base) },
67                    SimpleType::IntFixed(b) => {
68                        add_val = true;
69                        let base = format!("i{}", integer_bit_size(&b));
70                        (format!("IntWithGivenBitSize<{}, {}>", base.clone(), b), base) },
71                    SimpleType::UIntDyn(b) => {
72                        add_val = true;
73                        let base = format!("u{}", integer_bit_size(&b.0));
74                        (format!("DynInteger<{}, {}, {}>", base.clone(), b.0, b.1), base) },
75                    SimpleType::IntDyn(b) => {
76                        add_val = true;
77                        let base = format!("i{}", integer_bit_size(&b.0));
78                        (format!("DynInteger<{}, {}, {}>", base.clone(), b.0, b.1), base) },
79                    SimpleType::Float => {
80                        add_val = true;
81                        let base = "f32".to_string();
82                        (base.clone(), base)
83                    },
84                    SimpleType::Double => {
85                        let base = "f64".to_string();
86                        (base.clone(), base) },
87                    SimpleType::FixedPrecision(fpp) => {
88                        add_val = true;
89                        (format!("FixPrecisionMinMax<{}, {}, {}>", fpp.bits, fpp.min_val, fpp.max_val), "f64".to_string())
90                    },
91                    SimpleType::Binary(b) => {
92                        add_val = true;
93                        (format!("Binary<{}>", b), "Vec<u8>".to_string()) },
94                    SimpleType::AString(b) => {
95                        add_val = true;
96                        (format!("BitisAString<{}>", b), "String".to_string()) },
97                }
98            }
99            AttributeDetails::AttributeEnumOrMsg(em) => {
100                is_py_wrapped = true;
101                is_msg = msg_names.contains(&em);
102                is_enum = !is_msg.clone();
103                (em.clone(), em.clone()) }
104            AttributeDetails::AttributeOneOf(ooi) => {
105                is_py_wrapped=true; is_oo = true;
106                (ooi.name.clone(), ooi.name.clone()) }
107        };
108        (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo,  add_val)
109    };
110    AttributeEx{base: attribute.clone(), rust_type_str: rtype, base_type_str: base_type,
111        is_py_wrapped, is_msg, is_enum, is_oo, add_val }
112}
113
114pub fn to_rust_messages(msgs: &Vec<Message>) -> Vec<MessageR> {
115    let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
116
117    msgs.iter().map(|msg| {
118        let attrs_rust: Vec<_> = msg.attributes.iter().map(|attribute| {
119            to_rust_attribute(attribute, &msgs_names) }).collect();
120        MessageR{name: msg.name.clone(), comment: msg.comment.clone(), parent: msg.parent.clone(),
121            attributes: attrs_rust}
122    }).collect()
123}
124pub fn to_rust_oneofs(oos: &Vec<(String, OneOfInfo)>, msgs: &Vec<Message>) -> HashMap<String, OneOfInfoR> {
125    let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
126
127    oos.iter().map(|(msg_name, oo)| {
128        let attrs_rust: Vec<_> = oo.attributes.iter().map(|attribute| {
129            to_rust_attribute(attribute, &msgs_names) }).collect();
130        (oo.name.clone(), OneOfInfoR{msg_name: msg_name.clone(), name: oo.name.clone(), dyn_bits: oo.dyn_bits,
131            attributes: attrs_rust, default_attrib_name: oo.default_attrib_name.clone()})
132    }).collect()
133}
134
135fn to_cpp_attribute(attribute: &Attribute, msg_names: &Vec<String>) -> AttributeEx {
136    let (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val) = {
137        let mut is_py_wrapped = false;
138        let mut is_enum = false;
139        let mut is_msg = false;
140        let mut is_oo = false;
141        let mut add_val = false;
142
143        let (rtype, base_type) = match &attribute.specific_details {
144            AttributeDetails::AttributeSimple(a) => {
145                match a {
146                    SimpleType::NoTypeSetYet => {
147                        println!("Unexpected unspecified attribute type");
148                        abort()
149                    },
150                    SimpleType::Bool => { ("BitisBool".to_string(), "bool".to_string()) }
151                    SimpleType::UIntFixed(b) => {
152                        add_val = true;
153                        let base = format!("uint{}_t", integer_bit_size(&b));
154                        (format!("IntgralWithGivenBitSize<{}, {}>", base.clone(), b), base) }
155                    SimpleType::IntFixed(b) => {
156                        add_val = true;
157                        let base = format!("int{}_t", integer_bit_size(&b));
158                        (format!("IntgralWithGivenBitSize<{}, {}>", base.clone(), b), base) }
159                    SimpleType::UIntDyn(b) => {
160                        add_val = true;
161                        let base = format!("uint{}_t", integer_bit_size(&b.0));
162                        (format!("DynInteger<{}, {}, {}>", base.clone(), b.0, b.1), base) }
163                    SimpleType::IntDyn(b) => {
164                        add_val = true;
165                        let base = format!("int{}_t", integer_bit_size(&b.0));
166                        (format!("DynInteger<{}, {}, {}>", base.clone(), b.0, b.1), base) }
167                    SimpleType::Float => {
168                        add_val = true;
169                        let base = "float".to_string();
170                        (format!("BitisFloatingPoint<{}>", base.clone()), base)
171                    }
172                    SimpleType::Double => {
173                        let base = "double".to_string();
174                        (format!("BitisFloatingPoint<{}>", base.clone()), base) },
175                    SimpleType::FixedPrecision(fpp) => {
176                        add_val = true;
177                        (format!("FixPrecisionMinMax<{}, {}, {}>", fpp.bits, fpp.min_val, fpp.max_val), "double".to_string())
178                    },
179                    SimpleType::Binary(b) => {
180                        add_val = true;
181                        (format!("Binary<{}>", b), "Vec<u8>".to_string()) },
182                    SimpleType::AString(b) => {
183                        add_val = true;
184                        (format!("BitisAString<{}>", b), "char *".to_string()) },
185                }
186            }
187            AttributeDetails::AttributeEnumOrMsg(em) => {
188                is_py_wrapped = true;
189                is_msg = msg_names.contains(&em);
190                is_enum = !is_msg.clone();
191                (em.clone(), em.clone()) }
192            AttributeDetails::AttributeOneOf(ooi) => {
193                is_py_wrapped=true; is_oo = true;
194                (ooi.name.clone(), ooi.name.clone()) }
195        };
196        (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo,  add_val)
197    };
198    AttributeEx{base: attribute.clone(), rust_type_str: rtype, base_type_str: base_type,
199        is_py_wrapped, is_msg, is_enum, is_oo, add_val }
200}
201pub fn to_cpp_messages(msgs: &Vec<Message>) -> Vec<MessageR> {
202    let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
203
204    msgs.iter().map(|msg| {
205        let attrs_rust: Vec<_> = msg.attributes.iter().map(|attribute| {
206            to_cpp_attribute(attribute, &msgs_names) }).collect();
207        MessageR{name: msg.name.clone(), comment: msg.comment.clone(), parent: msg.parent.clone(),
208            attributes: attrs_rust}
209    }).collect()
210}
211pub fn to_cpp_oneofs(oos: &Vec<(String, OneOfInfo)>, msgs: &Vec<Message>) -> HashMap<String, OneOfInfoR> {
212    let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
213
214    oos.iter().map(|(msg_name, oo)| {
215        let attrs_cpp: Vec<_> = oo.attributes.iter().map(|attribute| {
216            to_cpp_attribute(attribute, &msgs_names) }).collect();
217        (oo.name.clone(), OneOfInfoR{msg_name: msg_name.clone(), name: oo.name.clone(), dyn_bits: oo.dyn_bits.clone(),
218            attributes: attrs_cpp, default_attrib_name: oo.default_attrib_name.clone() })
219    }).collect()
220}
221
222#[derive(Clone, Debug)]
223pub struct JinjaData {
224    pub enums: Vec<Enum>,
225    pub msgs: Vec<MessageR>,
226    pub oos: HashMap<String, OneOfInfoR>,
227}
228
229#[derive(Template, Clone, Debug)]
230#[template(path = "data_objects.rs.jinja")]
231pub struct RustDataObjects {
232    pub d: JinjaData
233}
234#[derive(Template, Clone, Debug)]
235#[template(path = "pyclasses.py.rs.jinja")]
236pub struct RustPyDataObjects {
237    pub d: JinjaData
238}
239#[derive(Template, Clone, Debug)]
240#[template(path = "pylib.py.rs.jinja")]
241pub struct RustPyLib {
242    pub d: JinjaData,
243    pub lib_name: String
244}
245#[derive(Template, Clone, Debug)]
246#[template(path = "py_type_hints.pyi.jinja")]
247pub struct PyTypeHints {
248    pub d: JinjaData
249}
250#[derive(Template, Clone, Debug)]
251#[template(path = "data_objects.cpp.jinja")]
252pub struct CppDataObjects {
253    pub d: JinjaData,
254    pub object_order: Vec<String>,
255    pub bitis_header_lib_file_name: String,
256    pub bitis_version: String,
257}
258
259
260mod filters {
261    #[allow(dead_code)]
262    pub fn snake_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values) -> ::askama::Result<String> {
263        Ok(stringcase::snake_case(s.to_string().as_str()))
264    }
265    #[allow(dead_code)]
266    pub fn camel_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
267        Ok(stringcase::camel_case(s.to_string().as_str()))
268    }
269    #[allow(dead_code)]
270    pub fn pascal_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
271        Ok(stringcase::pascal_case(s.to_string().as_str()))
272    }
273    #[allow(dead_code)]
274    pub fn to_py_type<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
275        if ["u8", "u16", "u32", "u64", "i8", "i16", "i32", "i64"].contains(&s.to_string().as_str()) {
276            Ok("int".to_string()) }
277        else if ["f32", "f64"].contains(&s.to_string().as_str()) {
278            Ok("float".to_string())
279        }
280        else { Ok(s.to_string()) }
281    }
282}
283
284// ************************************************************************
285type Error = (String, Span);
286
287type Result<T> = std::result::Result<T, Error>;
288
289#[derive(Debug, Clone)]
290pub enum DynOrFixedType {
291    Dyn(u8),
292    Fixed(u8)
293}
294#[derive(Debug, PartialEq, Eq, Clone, Copy)]
295pub struct FixedPrecisionProperties {
296    bits: u8, min_val: i64, max_val: i64
297}
298#[derive(Debug, PartialEq, Eq, Clone, Copy)]
299pub enum SimpleType {
300    NoTypeSetYet,
301    Bool,
302    UIntFixed(u8), IntFixed(u8),
303    UIntDyn((u8,u8)), IntDyn((u8,u8)),
304    Float, Double,
305    FixedPrecision(FixedPrecisionProperties),
306    Binary(u8),
307    AString(u8)
308}
309/*impl SimpleType {
310    fn int_size(min_bits: u8) -> std::result::Result<u8, String> {
311        match min_bits {
312            1..=8 => Ok(8),
313            9..=16 => Ok(16),
314            17..=32 => Ok(32),
315            33..=64 => Ok(64),
316            34..=128 => Ok(128),
317            0 => Err("Bitsize of zero for integer is not allowed".into()),
318            _ => Err("Bitsize larger than 128  for integer are not allowed".into())
319        }
320    }
321    fn get_int_bits(self) -> std::result::Result<u8, String> {
322        match self {
323            SimpleType::UIntFixed(s) => Ok(SimpleType::int_size(s)?),
324            SimpleType::IntFixed(s) => Ok(SimpleType::int_size(s)?),
325            SimpleType::UIntDyn((s,_)) => Ok(SimpleType::int_size(s)?),
326            SimpleType::IntDyn((s,_)) => Ok(SimpleType::int_size(s)?),
327            SimpleType::FixedPrecision(fps) => Ok(SimpleType::int_size(fps.bits)?),
328            SimpleType::UFixedPrecision(fps) => Ok(SimpleType::int_size(fps.bits)?),
329            _ => Err("get_int_bits(): Only integers types allowed.".into())
330        }
331    }
332    fn get_int_bits_no_error(self) -> u8 {
333        match self.get_int_bits() {
334            Ok(bits) => bits,
335            Err(e) => { println!("Error: {}", e); abort(); }
336        }
337    }
338}*/
339
340#[derive(Debug, Clone )]
341pub struct OneOfInfo {
342    name: String,
343    dyn_bits: u8,
344    attributes: Vec<Attribute>,
345    default_attrib_name: String,
346}
347#[derive(Debug, Clone )]
348pub enum AttributeDetails {
349    AttributeSimple(SimpleType),
350    AttributeEnumOrMsg(String),
351    AttributeOneOf(OneOfInfo),
352}
353#[derive(Debug, Clone )]
354pub struct Attribute {
355    name: String,
356    comment: Option<String>,
357    is_repeated_and_size: Option<DynOrFixedType>,
358    is_optional: bool,
359    specific_details: AttributeDetails
360}
361// #[derive(Debug, Clone)]
362// pub enum VersionInfo {
363//     Version(u16),
364//     BaseWithAllowedVersion(u16),
365// }
366#[derive(Debug, Clone)]
367pub struct Message {
368    pub name: String,
369    // pub version_info: VersionInfo,
370    pub comment: Option<String>,
371    pub parent: Option<String>,
372    pub attributes: Vec<Attribute>,
373}
374
375/// Enum information for bitis. The ids are always DynInteger with a given bit size.
376#[derive(Debug, Clone)]
377pub struct Enum {
378    pub name: String,
379    // pub version_info: VersionInfo,
380    pub comment: Option<String>,
381    pub bit_size: u8,
382    pub values: Vec<String>,
383    pub default: String
384}
385
386pub fn get_suffix_number(lex: &mut Lexer<Token>) -> Option<u8> {
387    let slice = lex.slice();
388    let re = Regex::new(r".*_d?([0-9]+)$").unwrap();
389    let num_str = re.captures(slice)?.get(1)?;
390    num_str.as_str().parse().ok()
391}
392pub fn get_d_suffix_numbers(lex: &mut Lexer<Token>) -> Option<(u8,u8)> {
393    let slice = lex.slice();
394    let re = Regex::new(r".*_([0-9]+)d([0-9]+)$").unwrap();
395    let first_num_str = re.captures(slice)?.get(1)?.as_str().parse().ok()?;
396    let second_num_str = re.captures(slice)?.get(2)?.as_str().parse().ok()?;
397    Some((first_num_str, second_num_str))
398}
399pub fn get_fp_properties_number(lex: &mut Lexer<Token>) -> Option<FixedPrecisionProperties> {
400    let slice = lex.slice();
401    let re = Regex::new(r"fp_([0-9]+)\[ *(-?[0-9]+) *, *(-?[0-9]+) *]").unwrap();
402    let bits = re.captures(slice)?.get(1)?.as_str().parse::<u8>().ok()?;
403    let min_val = re.captures(slice)?.get(2)?.as_str().parse::<i64>().ok()?;
404    let max_val = re.captures(slice)?.get(3)?.as_str().parse::<i64>().ok()?;
405    Some(FixedPrecisionProperties {bits, min_val, max_val})
406}
407/*pub fn get_dyn_or_fixed_from_args(lex: &mut Lexer<Token>) -> Option<DynOrFixedType> {
408    let slice = lex.slice();
409    let re = Regex::new(r" *(dyn|fixed) *, *([0-9]+)").unwrap();
410    let type_str = re.captures(slice)?.get(1)?.as_str();
411    let bits = re.captures(slice)?.get(2)?.as_str().parse::<u8>().ok()?;
412    if type_str == "dyn" {
413        Some(DynOrFixedType::Dyn(bits))
414    }
415    else {
416        Some(DynOrFixedType::Fixed(bits))
417    }
418}*/
419pub fn get_enum_bit_size(lex: &mut Lexer<Token>) -> Option<u8> {
420    let slice = lex.slice();
421    let re = Regex::new(r"\( *([0-9]+) *\)").unwrap();
422    let bits = re.captures(slice)?.get(1)?.as_str().parse::<u8>().ok()?;
423    Some(bits)
424}
425pub fn get_version(lex: &mut Lexer<Token>) -> Option<u16> {
426    let slice = lex.slice();
427    let re = Regex::new(r"\[.* +(v[0-9]+) *]").unwrap();
428    let ver_str = re.captures(slice)?.get(1)?.as_str();
429    Some(ver_str.parse::<u16>().ok()?)
430}
431
432#[derive(Debug, Logos)]
433#[logos(skip r"[ \t\r\n\f]+")]
434#[logos(extras = u16)]
435#[allow(dead_code)]
436pub enum Token{
437    #[regex(r"//[^\n]*\n?", priority=40)] Comment,
438    #[regex(r"//\|[^\n]*\n?", |lex| lex.slice()[3..].to_owned(), priority=41)] SpecificComment(String),
439    #[token("msg", priority=20)] Msg,
440    #[token("enum", priority=20)] Enum,
441    #[token("oneof", priority=20)] OneOf,
442    #[token("{")] CBraceOpen,
443    #[token("}")] CBraceClose,
444    #[token("(")] BraceOpen,
445    #[token(")")] BraceClose,
446    #[token(":")] Colon,
447    #[token(";")] SemiColon,
448    #[token(",")] Comma,
449    #[token("*")] Star,
450    // #[token("fixed", priority=20)] FixedFlag,
451    // #[token("dyn", priority=20)] DynFlag,
452    #[regex(r"\[ *base +use +starting +with +v[0-9]+ *\]", get_version, priority=35)] MsgBaseInfoToken(u16),
453    #[regex(r"\[ *version +v[0-9]+ *\]", get_version, priority=35)] MsgVersionToken(u16),
454    // #[regex(r"\[ *base +use +starting +with +v[0-9]+ *\]", get_version, priority=35)] MsgVersionToken((MsgVersion, u16)),
455    #[regex("[0-9]+", |lex| lex.slice().parse::<isize>().unwrap(), priority=1)] IntegerVal(isize),
456    #[regex(r"-?(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?",
457        |lex| lex.slice().parse::<f64>().unwrap(), priority=2)] Number(f64),
458    #[token("bool", priority=30)] Bool,
459    #[token("msg_size_type", priority=30)] MsgSizeType,
460    #[regex(r"uint_[0-9]+", get_suffix_number, priority=30)] UIntFixed(u8),
461    #[regex(r"int_[0-9]+", get_suffix_number, priority=30)] IntFixed(u8),
462    #[regex(r"uint_[0-9]+d[0-9]+", get_d_suffix_numbers, priority=31)] UIntDyn((u8,u8)),
463    #[regex(r"int_[0-9]+d[0-9]+", get_d_suffix_numbers, priority=31)] IntDyn((u8,u8)),
464    #[token("float", priority=30)] Float,
465    #[token("double", priority=30)] Double,
466    #[regex(r"astr_d[0-9]+", get_suffix_number, priority=31)] AStr(u8),
467    #[regex(r"fp_[0-9]+\[ *-?[0-9]+ *, *-?[0-9]+ *]", get_fp_properties_number, priority=30)] FixedPoint(FixedPrecisionProperties),
468    // #[regex(r"ufp_[0-9]+\[ *-?[0-9]+ *, *-?[0-9]+ *]", get_fp_properties_number, priority=30)] UFixedPoint(FixedPrecisionProperties),
469    #[token("binary_d[0-9]+", get_suffix_number, priority=30)] Binary(u8),
470    #[regex(r"repeated_dyn_[0-9]+", get_suffix_number, priority=30)] RepeatedDyn(u8),
471    #[regex(r"repeated_fixed_[0-9]+", get_suffix_number, priority=30)] RepeatedFixed(u8),
472    #[token("optional", priority=30)] Optional,
473    #[regex(r"[A-Za-z][A-Za-z0-9_-]+", |lex| lex.slice().to_owned(), priority=11)] StringVal(String),
474    #[token("false", |_| false, priority=20)]
475    #[token("true", |_| true, priority=20)] BoolVal(bool),
476    #[regex(r"\( *([0-9]+) *\)", get_enum_bit_size, priority=40)] EnumDynSize(u8),
477}
478
479#[derive(Debug, Clone)]
480pub enum Value {
481    /// null.
482    Message(Message),
483    /// Enum.
484    Enum(Enum)
485}
486
487macro_rules! parse_one_token {
488    ($token_enum: path, $lexer: expr, $error_msg_or_empty: expr) => {
489        loop {
490            let rv = $lexer.next();
491            if let Some(token) = rv {
492                match token {
493                    Ok($token_enum) => {
494                        break Ok(Ok(()));
495                    },
496                    Ok(Token::Comment) => (),
497                    _ => {
498                        if let Some(err_str) = $error_msg_or_empty {
499                            break Err((format!("{err_str}\nToken: {token:?}").to_owned(), $lexer.span()));
500                        }
501                        else {
502                            break Ok(Err($lexer.span()));
503                        }
504                    }
505                }
506            }
507            else {
508                break Err((format!("Unexpected end or text {rv:?}").to_owned(), $lexer.span()));
509            }
510        }
511    }
512}
513macro_rules! parse_one_token_with_arg {
514    ($token_enum: path, $lexer: expr, $error_msg_or_empty: expr) => {
515        loop {
516            let rv = $lexer.next();
517            if let Some(token) = rv {
518                match token {
519                    Ok($token_enum(s)) => {
520                        break Ok(Ok(s));
521                    },
522                    Ok(Token::Comment) => (),
523                    _ => {
524                        if let Some(err_str) = $error_msg_or_empty {
525                            break Err((format!("{}\nFound token: {:?}.",
526                                err_str, token).to_owned(), $lexer.span()));
527                        }
528                        else {
529                            break Ok(Err($lexer.span()));
530                        }
531                    }
532                }
533            }
534            else {
535                break Err((format!("Unexpected end or text {rv:?}").to_owned(), $lexer.span()));
536            }
537        }
538    }
539}
540
541pub fn parse_root(lexer: &mut Lexer<'_, Token>) -> Result<Vec<Value>> {
542    let mut list: Vec<Value> = Vec::new();
543    let mut specific_comment: Option<String> = None;
544    loop {
545        if let Some(token) = lexer.next() {
546            let rv = match token {
547                Ok(Token::Msg) => Some(parse_msg(lexer, specific_comment.clone())),
548                Ok(Token::Enum) => Some(parse_enum(lexer, specific_comment.clone())),
549                Ok(Token::Comment) => None,
550                Ok(Token::SpecificComment(s)) => {
551                    specific_comment = Some(s); None },
552                _ => Some(Err((format!("Unexpected token {:?}", token).to_owned(), lexer.span()))),
553            };
554            match rv {
555                None => (),
556                Some(Ok(value)) => { list.push(value); specific_comment = None; },
557                Some(Err(err)) => return Err(err)
558            }
559        }
560        else { break; }
561    }
562    Ok(list)
563}
564
565pub fn parse_msg(lexer: &mut Lexer<'_, Token>, comment_for_msg: Option<String>) -> Result<Value> {
566    let mut attributes = Vec::new();
567
568    let name = match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name but received:"))? {
569        Ok(s) => s,
570        Err(s) => { return Err(("Code should not be reached".into(), s)); }
571    };
572
573    // let version_info = if(lexer.extras == 0) {
574    //     if let Some(token) = lexer.next() {
575    //         match token {
576    //             Ok(Token::MsgVersionToken(v)) => VersionInfo::Version(v),
577    //             Ok(Token::MsgBaseInfoToken(v)) => VersionInfo::BaseWithAllowedVersion(v),
578    //             Ok(_) => { return Err((format!("Unexpected token {:?} for message '{}' when expecting version info", token, name)
579    //                                    .to_owned(), lexer.span())); }
580    //             Err(_) => { return Err((format!("Error: Syntax error for message '{}'", name).to_owned(), lexer.span())); }
581    //         }
582    //     } else { return Err(("Unexpectedly did not find version information".to_owned(), lexer.span())); }
583    // }
584    // else { VersionInfo::Version(lexer.extras) };
585
586    let parent = {
587        let has_parent; let p;
588        if let Some(token) = lexer.next() {
589            match token {
590                Ok(Token::Colon) => has_parent = true,
591                Ok(Token::CBraceOpen) => has_parent = false,
592                _ => { return Err((format!("Unexpected text for msg '{name}'.").into(), lexer.span())) },
593            }
594            if has_parent {
595                match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name."))? {
596                    Ok(s) => p = Some(s),
597                    Err(s) => { return Err((format!("For msg '{name} colon found but no parent name").into(), s)); }
598                };
599                parse_one_token!(Token::CBraceOpen, lexer, Some(format!("Expected curly bracket open for msg '{name}'")))?.unwrap();
600            }
601            else {
602                p = None
603            }
604        }
605        else { return Err(("Unexpected end of file".into(), lexer.span())); }
606        p
607    };
608
609    loop {
610        if let Some(token) = lexer.next() {
611            match token {
612                Ok(Token::CBraceClose) => break,
613                Ok(Token::Comment) => (),
614                Ok(ctoken) => match parse_attribute(ctoken, lexer, name.clone(), false) {
615                    Ok(a) => { attributes.push(a); },
616                    Err(e) => { return Err(e); }
617                },
618                _ => { return Err((format!("Error: Unexpected text found for msg '{name}'.").into(), lexer.span())) },
619            };
620        }
621        else { return Err(("Unexpected end of file".into(), lexer.span())); }
622    }
623
624    Ok(Value::Message(Message{name, /*version_info,*/ comment: comment_for_msg, parent, attributes}))
625}
626
627pub fn parse_attribute(last_token: Token, lexer: &mut Lexer<'_, Token>,
628                       parent_name: String, attributes_for_oneof: bool) -> Result<Attribute> {
629    let mut is_optional = false;
630    let mut is_repeated_and_size: Option<DynOrFixedType> = None;
631    let mut attr_type = SimpleType::NoTypeSetYet;
632    let mut ctoken = last_token;
633    let mut enum_or_msg_str = None;
634    let mut oneof_infos = None;
635    let lexer_span_start = lexer.span();
636    let mut specific_comment: Option<String> = None;
637
638    loop {
639        match ctoken {
640            Token::SpecificComment(s) => {
641                specific_comment = Some(s); () },
642            Token::Optional if is_repeated_and_size.is_some() =>
643                return Err(("Error: Optional and repeated not allowed together".to_owned(), lexer.span())),
644            Token::RepeatedFixed(_) | Token::RepeatedDyn(_) if is_optional =>
645                return Err(("Error: Optional and repeated are not allowed together".to_owned(), lexer.span())),
646
647            Token::Optional | Token::RepeatedDyn(_) | Token::RepeatedFixed(_) if attributes_for_oneof =>
648                return Err(("Error: Optional and repeated are not allowed in oneof".to_owned(), lexer.span())),
649
650            Token::Optional => is_optional = true,
651            Token::RepeatedDyn(b) => is_repeated_and_size = Some(DynOrFixedType::Dyn(b)),
652            Token::RepeatedFixed(b) => is_repeated_and_size = Some(DynOrFixedType::Fixed(b)),
653            Token::Bool => { attr_type = SimpleType::Bool; break; },
654            Token::AStr(s) => {
655                attr_type = SimpleType::AString(s); break; },
656            Token::UIntFixed(s) => { attr_type = SimpleType::UIntFixed(s); break; },
657            Token::UIntDyn((m,s)) if m < s =>
658                return Err(("Error: Unsigned dyn integer bit size of integer type must be bigger than the bit size of the package".to_owned(), lexer.span())),
659            Token::UIntDyn((m,_)) if (m & 3) != 0 =>
660                return Err(("Error: Unsigned dyn integer bit size of integer type must be a multiple of 8".to_owned(), lexer.span())),
661            Token::UIntDyn((m,s)) => { attr_type = SimpleType::UIntDyn((m, s)); break; },
662            Token::IntFixed(s) => { attr_type = SimpleType::IntFixed(s); break; },
663            Token::IntDyn((m,s)) if m < s =>
664                return Err(("Error: Unsigned dyn integer bit size of integer type must be bigger than the bit size of the package".to_owned(), lexer.span())),
665            Token::IntDyn((m,_)) if (m & 3) != 0 =>
666                return Err(("Error: Unsigned dyn integer bit size of integer type must be a multiple of 8".to_owned(), lexer.span())),
667            Token::IntDyn((m,s)) => {
668                attr_type = SimpleType::IntDyn((m,s)); break;
669            },
670            //Token::String => { attr_type = SimpleType::String; break; },
671            Token::Float => { attr_type = SimpleType::Float; break; },
672            Token::Double => { attr_type = SimpleType::Double; break; },
673            Token::FixedPoint(s) => { attr_type = SimpleType::FixedPrecision(s); break; },
674            Token::Binary(b) => { attr_type = SimpleType::Binary(b); break; },
675            Token::StringVal(s) => { enum_or_msg_str = Some(s); break; }
676            Token::OneOf => {
677                oneof_infos = match parse_oneof(lexer, parent_name.clone(), specific_comment.clone(),
678                                                is_repeated_and_size.clone(), is_optional.clone()) {
679                    Ok(oo) => Some(oo),
680                    Err(s) => { return Err(s); }
681                };
682                break;
683            }
684            _ => { return Err((format!("Error: Expected attribute type or modifier (got {ctoken:?}) when parsing msg or oneof '{parent_name}'")
685                                   .to_owned(), lexer.span())); }
686        }
687        if let Some(token) = lexer.next() {
688            match token {
689                Ok(t) => ctoken = t,
690                Err(_) => { return Err((format!("Error: Unexpected text found for msg '{parent_name}'.").to_owned(), lexer.span())); }
691            }
692        } else {
693            return Err(("Unexpected end of file".to_string(), lexer.span()));
694        }
695    }
696
697    let mut name= "".to_owned();
698    if oneof_infos.is_none() {
699        name = parse_one_token_with_arg!(
700            Token::StringVal, lexer, Some(format!("Error: Expected attribute name for msg '{parent_name}' (type: {attr_type:?}/{enum_or_msg_str:?})")))?.unwrap();
701
702        parse_one_token!(Token::SemiColon, lexer, Some(format!(
703            "Error: Expected semicolon to end line of attribute '{name}' of msg or oneof '{parent_name}'")))?.unwrap();
704    }
705    let num_of_set_types_or_opts = vec![(attr_type != SimpleType::NoTypeSetYet), enum_or_msg_str.is_some(), oneof_infos.is_some()]
706        .iter().map(|&x| if x { 1_u8 } else { 0_u8 }).sum::<u8>();
707    if num_of_set_types_or_opts > 1 {
708        let mut span = lexer_span_start.clone();
709        span.end = lexer.span().end;
710        return Err(("Error: Attribute contains inconsistent optional, simple types and messages or Enums".to_string(), span));
711    }
712
713    if let Some(oo) = oneof_infos {
714        Ok(oo)
715    }
716    else if let Some(t) = enum_or_msg_str {
717        Ok(Attribute{name, comment: specific_comment,
718            is_repeated_and_size: is_repeated_and_size, is_optional,
719            specific_details: AttributeDetails::AttributeEnumOrMsg(t)})
720    }
721    else {
722        Ok(Attribute{name, comment: specific_comment,
723            is_repeated_and_size: is_repeated_and_size, is_optional,
724            specific_details: AttributeDetails::AttributeSimple(attr_type)})
725    }
726}
727
728pub fn parse_oneof(lexer: &mut Lexer<'_, Token>, parent_name: String, comment: Option<String>,
729                   is_repeated_and_size: Option<DynOrFixedType>, is_optional: bool) -> Result<Attribute> {
730    let oo_name = parse_one_token_with_arg!(
731            Token::StringVal, lexer, Some(format!("Error: Expected name for oneof in parent '{parent_name}'")))?.unwrap();
732
733    let bit_size = match parse_one_token_with_arg!(Token::EnumDynSize, lexer, Some("Expected oneof properties for dyn size, e.g. (4)."))? {
734        Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
735    };
736
737    parse_one_token!(Token::CBraceOpen, lexer, Some("Error: Expected open curly bracket to enclose oneof elements"))?.unwrap();
738
739    let mut oo_attribs = Vec::new();
740    let mut is_default = false; let mut default_name = None;
741    loop {
742        if let Some(token) = lexer.next() {
743            match token {
744                Ok(Token::CBraceClose) => break,
745                Ok(Token::Star) => { is_default = true; },
746                Ok(last_token) => {
747                    let oo_attr = match parse_attribute(last_token, lexer, oo_name.clone(), true) {
748                        Ok(o) => o,
749                        Err(s) => return Err(s),
750                    };
751                    oo_attribs.push(oo_attr.clone());
752
753                    if is_default {
754                        if default_name.is_some() {
755                            return Err((format!("Error: Multiple attributes of one-of '{}' (in '{}') are marked as default.",
756                                                oo_name, parent_name), lexer.span())); }
757                        default_name = Some(oo_attr.name); is_default = false;
758                    }
759                }
760                Err(_) => { return Err((format!("Error: Unexpected text when decoding oneof ({token:?})").to_owned(), lexer.span())); },
761            }
762        }
763    }
764
765    if default_name.is_none() {
766        return Err((format!("Error: No default name in oneof elements for '{}' in '{}'",
767                            oo_name, parent_name), lexer.span()));
768    }
769
770    Ok(Attribute{name: oo_name.clone(), comment,
771        is_repeated_and_size: is_repeated_and_size, is_optional,
772        specific_details: AttributeDetails::AttributeOneOf(OneOfInfo{
773            name: format!("OO_{}_{}", parent_name.to_pascal_case(), oo_name.to_pascal_case()),
774            dyn_bits: bit_size, attributes: oo_attribs,
775            default_attrib_name: default_name.unwrap(),
776        })})
777}
778
779pub fn parse_enum(lexer: &mut Lexer<'_, Token>, comment: Option<String>) -> Result<Value> {
780    let name = match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name but received."))? {
781        Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
782    };
783
784    let bit_size = match parse_one_token_with_arg!(Token::EnumDynSize, lexer, Some("Expected enum properties for dyn size, e.g. (4)."))? {
785        Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
786    };
787
788    // let version_info = if(lexer.extras == 0) {
789    //     if let Some(token) = lexer.next() {
790    //         match token {
791    //             Ok(Token::MsgVersionToken(v)) => VersionInfo::Version(v),
792    //             Ok(Token::MsgBaseInfoToken(v)) => VersionInfo::BaseWithAllowedVersion(v),
793    //             Ok(_) => { return Err((format!("Unexpected token {:?} for enum '{}' when expecting version info", token, name)
794    //                                        .to_owned(), lexer.span())); }
795    //             Err(_) => { return Err((format!("Error: Syntax error for enum '{}'", name).to_owned(), lexer.span())); }
796    //         }
797    //     } else { return Err(("Unexpectedly did not find version information".to_owned(), lexer.span())); }
798    // }
799    // else { VersionInfo::Version(lexer.extras) };
800
801    parse_one_token!(Token::CBraceOpen, lexer, Some(format!("Expected open curly bracket for enum '{name}'")))?.unwrap();
802
803    let mut values = Vec::new();
804    let (mut value_found, mut is_default) = (false, false);
805    let mut default_val = None;
806    loop {
807        if let Some(token) = lexer.next() {
808            match token {
809                Ok(Token::CBraceClose) => break,
810                Ok(Token::StringVal(s)) => {
811                    values.push(s.clone()); value_found=true;
812                    if is_default { default_val = Some(s); }
813                    is_default = false;
814                },
815                Ok(Token::Comma) => {
816                    if !value_found { return Err(("Error: found comma but no enum value.".to_string(), lexer.span())) }
817                    value_found=false },
818                Ok(Token::Comment) => (),
819                Ok(Token::Star) => {
820                    if default_val.is_some() || value_found {
821                        return Err(("Error: found default value marker (*) but default already set or marker not preceding value.".to_string(), lexer.span()))}
822                    is_default = true;
823                }
824                _ => { return Err((format!("Error: Unexpected text found for enum '{name}'."), lexer.span())) },
825            }
826        } else { return Err(("Unexpected end of file".into(), lexer.span())); }
827    }
828    if default_val.is_none() { return Err(("Error: missing default value marker.".to_string(), lexer.span())); }
829
830    Ok(Value::Enum(Enum{name, comment, bit_size, values, default: default_val.unwrap() }))
831}
832
833
834/*pub fn validate_bitis(parsed_bitis: &Vec<Value>) -> Option<String> {
835    let enum_types = parsed_bitis.iter().filter_map(|x|
836        match x { Value::Enum(ev) => Some(ev.name.clone()), _ => None }).collect::<Vec<String>>();
837    let msg_types = parsed_bitis.iter().filter_map(|x|
838        match x { Value::Message(msg) => Some(msg.name.clone()), _ => None }).collect::<Vec<String>>();
839
840    // ***
841    if let Some (err_str) = parsed_bitis.iter().find_map(|s| match s {
842        Value::Message(msg) => {
843            msg.attributes.iter().find_map(|a| match a.specific_details.clone() {
844                AttributeDetails::AttributeEnumOrMsg(eon) => {
845                    if !enum_types.contains(&eon) && !msg_types.contains(&eon) {
846                        Some(format!("Type or enum '{eon}' unknown"))
847                    }
848                    else { None }
849                },
850                _ => None
851            })
852        },
853        _ => None,
854    }) { return Some(err_str); };
855
856    // *** check msg versions
857    let msgs_with_version: Vec<_> = parsed_bitis.iter().filter_map(|s| match s {
858        Value::Message(msg) => {
859            Some(match msg.version {
860                MsgVersion::Fixed => format!("{}_fixed", msg.name),
861                // MsgVersion::VersionedMsg => format!("{}_versioned", msg.name),
862                // MsgVersion::Base => format!("{}_base", msg.name),
863                MsgVersion::Versioned(v) => format!("{}_v{}", msg.name, v)
864            })
865        },
866        _ => None,
867    }).collect();
868    match (1..msgs_with_version.len()).into_iter().find_map(|i| {
869        let s = &msgs_with_version[i - 1];
870        if msgs_with_version[i..].contains(s) { Some(s.clone()) } else { None }
871    })
872    {
873        Some(msg) => return Some(format!("Conflicting versions of {}.", msg)),
874        None => ()
875    };
876
877    let fixed_msgs: Vec<_> = parsed_bitis.iter().filter_map(|s| match s {
878        Value::Message(msg) => {
879            match msg.version.clone() {
880                MsgVersion::Fixed => Some(msg.name.clone()), _ => None }
881        }, _ => None,
882    }).collect();
883    if let Some (err_str) = parsed_bitis.iter().find_map(|s| match s {
884        Value::Message(msg) => {
885            match msg.version.clone() {
886                MsgVersion::Fixed => None,
887                _ => {
888                    if fixed_msgs.contains(&msg.name) {
889                        Some(format!("Multiple conflicting versions of {} (fixed and version).", msg.name))
890                    }
891                    else { None }
892                }
893            }
894        }, _ => None,
895    }) { return Some(err_str); };
896
897    // ***
898    // Check that only attributes were added
899    // parsed_bitis.iter().for_each(|s| match s {
900    //     Value::Message(msg) => {
901    //
902    //     },
903    //     Value::Enum(eon) => {
904    //
905    //     }
906    // })
907
908    None
909}
910*/
911// Struct that collects all bitis information
912#[derive(Debug, Clone)]
913pub struct Dependencies {
914    pub in_deps: Vec<String>,
915    pub out_deps: Vec<String>,
916}
917#[derive(Debug, Clone)]
918pub struct BitisProcessed {
919    pub max_version_number: u16,
920    pub msgs: Vec<Message>,
921    pub enums: Vec<Enum>,
922    pub oo_enums: Vec<(String, OneOfInfo)>,
923}
924
925/// This function prepares message and enums for rendering
926pub fn process_and_validate_bitis(parsed_bitis: &Vec<Value>) -> BitisProcessed {
927    /*let (processed_msgs, max_msg_version_number) = {
928        let msgs: Vec<_> = parsed_bitis.iter().filter_map(|v| {
929            match v { Value::Message(msg) => Some(msg.clone()), _ => None }
930        }).collect();
931
932        let max_version_number: u16 = msgs.iter().fold(0_u16, |cur_max, v| {
933            std::cmp::max(cur_max, match v.version_info.clone()
934            { VersionInfo::BaseWithAllowedVersion(_) => 0, VersionInfo::Version(v) => v })
935        });
936        println!("Max version number for msgs found: {}", max_version_number);
937
938        // ***
939        // sort msgs per versions
940        let msgs_per_version: HashMap<u16, HashMap<String, Message>> = (0..=max_version_number).map(|cver| {
941            let msgs: HashMap<String, Message> = msgs.iter().filter_map(|cmsg| {
942                match &cmsg.version_info {
943                    VersionInfo::BaseWithAllowedVersion(_) if cver==0  => Some(cmsg.clone()),
944                    VersionInfo::Version(msg_ver) => {
945                        if *msg_ver == 0 {
946                            println!("Error: Message '{}' has version zero which is not allowed", cmsg.name);
947                            abort();
948                        } else if *msg_ver == cver { Some(cmsg.clone()) }
949                        else { None }
950                    }, _ => None
951                } }
952            ).map(|msg| { (msg.name.clone(), msg) }).collect();
953            (cver, msgs)
954        }).collect();
955
956        // todo check that messages as attributes are used with the correct version
957
958        // todo check that messages have the same allowed_to_be_used_starting_with_version for all versions
959
960        // todo check that attributes for different versions are unique
961
962        // ***
963        let msg_names_and_ver_type: HashMap<_, _> = msgs.iter().map(|v| {
964            (v.name.clone(), v.version_info.clone()) }).collect();
965
966        let msg_version_to_use_per_version: HashMap<String, HashMap<u16, u16>> = {
967            let mut temp_msg_last_version: HashMap<String, u16> =
968                msgs.iter().enumerate().map(|(x1, x2)| { (x2.name.clone(), x1.clone() as u16) }).collect();
969
970            msgs.iter().map(|v| {
971                (1..=max_version_number).map(|cver| {
972
973                }
974            }
975        }
976
977
978        // let msg_base_with_version: Vec<_> = msgs.iter().filter_map(|msg| {
979        // });
980
981        let msgs_processed: Vec<Vec<Message>> = msgs_per_version.iter()
982            .map(|(&cver, cver_msgs) | {
983                let mut msgs_for_version: HashMap<String, Message> = HashMap::new();
984
985                println!("Processing V{} msgs: {:?}", cver, cver_msgs);
986                if cver == 0 {
987                    msg_names_and_ver_type.iter().for_each(|(mi_name, _)| {
988                        if !cver_msgs.contains_key(mi_name) {
989                            println!("Error: Message '{}' not found in base version. All messages must be declared in base.", mi_name);
990                            abort();
991                        }
992                    });
993                }
994                else {
995                    // add missing msg definitions for each version
996                    let new_msgs: HashMap<_, _> = msg_names_and_ver_type.iter().filter_map(|(mi_name, ver_type)| {
997                        // do it only for versioned msgs
998                        if let VersionInfo::BaseWithAllowedVersion(_) = ver_type { None } else {
999                            // check if
1000                            if !cver_msgs.contains_key(mi_name) {
1001                                println!("Generating empty version msg '{mi_name}'");
1002
1003                                let base_ver_msg = msgs_per_version.get(&0).unwrap().get(mi_name).unwrap();
1004
1005                                let name = format!("{}_DataV{}", mi_name, cver);
1006                                Some((name.clone(), Message { name, attributes: vec![], comment: Some("Automatically generated empty msg".to_string()),
1007                                    ..base_ver_msg.clone() }))
1008                            } else { None }
1009                        }
1010                    }).collect();
1011                    msgs_for_version.extend(new_msgs);
1012                }
1013
1014                cver_msgs.iter().for_each(|(_, cmsg)| {
1015                    if let VersionInfo::Version(msg_ver) = &cmsg.version_info {
1016                        match cver {
1017                            cver_iter if cver_iter >= *msg_ver => {
1018                                let cname = format!("{}_DataV{}", cmsg.name, cver_iter);
1019                                msgs_for_version.insert(cname.clone(), Message { name: cname, ..cmsg.clone() });
1020                            },
1021                            _ => ()
1022                        };
1023                    }
1024                    else {
1025                        let processed_attributes: Vec<_> = cmsg.attributes.iter().map(|attr| {
1026                            match &attr.specific_details {
1027                                AttributeDetails::AttributeSimple(_) => attr.clone(),
1028                                AttributeDetails::AttributeEnumOrMsg(at) => {
1029                                    Attribute{specific_details: AttributeEnumOrMsg(format!("{}_V{}", at, cver)), ..attr.clone()}
1030                                }
1031                                AttributeDetails::AttributeOneOf(_) => {
1032                                    Attribute{name: format!("{}_TODO", attr.name), ..attr.clone()}
1033                                }
1034                            }
1035                        }).collect();
1036                        let cname = format!("{}_BaseV{}", cmsg.name, cver);
1037                        msgs_for_version.insert(cname.clone(), Message { name: cname, attributes: processed_attributes, ..cmsg.clone() });
1038                    }
1039                });
1040                msgs_for_version.values().cloned().collect::<Vec<Message>>()
1041            }).collect();
1042
1043        let mut msg_processed_concat: Vec<_> = msgs_processed.concat();
1044        msg_processed_concat.sort_by_key(|msg| { msg.name.to_lowercase() });
1045
1046
1047        (msg_processed_concat, max_version_number)
1048    };*/
1049
1050    //
1051    let msgs: Vec<_> = parsed_bitis.iter().filter_map(|v| {
1052        match v { Value::Message(msg) => Some(msg.clone()), _ => None }
1053    }).collect();
1054    let enums: Vec<_> = parsed_bitis.iter().filter_map(|v| {
1055        match v { Value::Enum(enm) => Some(enm.clone()), _ => None }
1056    }).collect();
1057
1058    fn get_oneofs(msg_name: String, attrs: &Vec<Attribute>) -> Option<Vec<(String, OneOfInfo)>> {
1059        let direct_oos = attrs.iter().filter_map(|attr| {
1060            match &attr.specific_details {
1061                AttributeDetails::AttributeOneOf(oo) => Some(vec![(msg_name.clone(), oo.clone())]),
1062                _ => None
1063            }
1064        }).collect::<Vec<Vec<(String, OneOfInfo)>>>().concat();
1065
1066        let inner_oos = direct_oos.iter().filter_map(|(_, doo)| {
1067            get_oneofs(msg_name.clone(), &doo.attributes)
1068        }).collect::<Vec<Vec<_>>>().concat();
1069
1070        let all_oos = vec![direct_oos, inner_oos].concat();
1071        if all_oos.len() == 0 { None }
1072        else { Some(all_oos) }
1073    }
1074    let oo_enums: Vec<_> = msgs.iter().filter_map(|msg| {
1075        get_oneofs(msg.name.clone(), &msg.attributes)
1076    }).collect::<Vec<_>>().concat();
1077
1078    // println!("\noo_enums:\n{:?}\n", oo_enums);
1079
1080    { // Test msgs and enum
1081        let msg_names = msgs.iter().map(|msg| &msg.name).collect::<Vec<_>>();
1082        msg_names.iter().for_each(|name| {
1083            // println!("name: {}", name);
1084            if msg_names.iter().filter(|cname| **cname == *name).count() > 1 {
1085                println!("Error: Multiple instances of msg '{}' found.", name);
1086                abort()
1087            }
1088        });
1089        let enum_names = enums.iter().map(|enm| &enm.name).collect::<Vec<_>>();
1090        enum_names.iter().for_each(|name| {
1091            if enum_names.iter().filter(|cname| **cname == *name).count() > 1 {
1092                println!("Error: Multiple instances of enum '{}' found.", name); abort()
1093            }
1094        });
1095
1096        let enums_and_msg_names = [&msg_names[..], &enum_names[..]].concat();
1097
1098        // check that all attributes are defined
1099        msgs.iter().for_each(|msg| {
1100            for attr in msg.attributes.clone() {
1101                match attr.specific_details {
1102                    AttributeDetails::AttributeEnumOrMsg(enum_or_msg) => {
1103                        let em_found = enums_and_msg_names.contains(&&enum_or_msg);
1104                        if !em_found {
1105                            println!("!!! Error: Attribute '{}' type '{}' in message '{}' not defined.", attr.name, enum_or_msg, msg.name);
1106                            panic!("->Exiting");
1107                        }
1108                    },
1109                    _ => {}
1110                }
1111            }
1112        });
1113    }
1114
1115    BitisProcessed { max_version_number: 0, msgs, enums, oo_enums}
1116}
1117
1118pub fn dependencies_process(jd: JinjaData) -> Vec<String>{
1119    let mut dependencies = HashMap::new();
1120
1121    for msgs in jd.msgs.clone() {
1122        dependencies.insert(
1123            msgs.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1124    }
1125    for enm in jd.enums {
1126        dependencies.insert(
1127            enm.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1128    }
1129    for (_, oos) in jd.oos.clone() {
1130        dependencies.insert(
1131            oos.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1132    }
1133
1134    println!("{:?}", dependencies);
1135
1136    if dependencies.len() == 0 {
1137        println!("No dependencies found, skipping message and type analysis!");
1138        return Vec::new();
1139    }
1140
1141    // msgs
1142    for msgs in jd.msgs {
1143        for attr in msgs.attributes {
1144            if attr.is_enum || attr.is_msg || attr.is_oo {
1145                println!("attr '{}' -> attr.rust_type_str: {}", attr.base.name, attr.rust_type_str);
1146                println!("{:?}", attr);
1147                dependencies.get_mut(&attr.rust_type_str).unwrap().out_deps.push(msgs.name.clone());
1148                dependencies.get_mut(&msgs.name).unwrap().in_deps.push(attr.rust_type_str.clone());
1149            }
1150        }
1151    }
1152    for (_, msgs) in jd.oos {
1153        for attr in msgs.attributes {
1154            if attr.is_enum || attr.is_msg || attr.is_oo {
1155                dependencies.get_mut(&attr.rust_type_str).unwrap().out_deps.push(msgs.name.clone());
1156                dependencies.get_mut(&msgs.name).unwrap().in_deps.push(attr.rust_type_str.clone());
1157            }
1158        }
1159    }
1160    println!("{:#?}", dependencies.clone());
1161
1162    let mut object_order = Vec::new();
1163    while dependencies.len() > 0 {
1164        let mut cobjs: Vec<_> = dependencies.clone().iter().filter_map(|(obj_name, deps)| {
1165            if deps.in_deps.len() == 0 { Some(obj_name.clone()) }
1166            else { None }
1167        }).collect();
1168
1169        for co in cobjs.clone() {
1170            dependencies.remove(&co);
1171        }
1172        for co in cobjs.clone() {
1173            for (_, deps) in &mut dependencies {
1174                deps.in_deps.retain(|x| *x != co);
1175            }
1176        }
1177        object_order.append(&mut cobjs);
1178    }
1179    println!("{:?}", object_order);
1180
1181    object_order
1182}
1183// ***************************************************
1184
1185#[cfg(test)]
1186mod bitis_semantic {
1187    use rstest::rstest;
1188    use super::*;
1189
1190    #[rstest]
1191    fn msg_empty_msg() {
1192        let test_empty_msg = "msg Lala { }";
1193
1194        let mut lexer = Token::lexer(test_empty_msg);
1195        lexer.extras = 0;
1196
1197        let parsed_bitis = parse_root(&mut lexer);
1198        if let Err(s) = parsed_bitis.clone() {
1199            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1200        }
1201        assert_eq!(parsed_bitis.is_ok(), true);
1202
1203        let parsed_bitis = parsed_bitis.unwrap();
1204        assert_eq!(parsed_bitis.len(), 1);
1205
1206        assert!(if let Value::Message(_) = parsed_bitis[0].clone() { true } else { false });
1207
1208        if let Value::Message(msg) = parsed_bitis[0].clone() {
1209            assert_eq!(msg.name, "Lala".to_string());
1210        }
1211
1212        // let validate_result = validate_bitis(&parsed_bitis);
1213        // println!("validate_result: {:?}", validate_result);
1214
1215        let process_result = process_and_validate_bitis(&parsed_bitis);
1216        println!("process_result {:?}", process_result);
1217
1218        assert_eq!(process_result.msgs.len(), 1);
1219        assert_eq!(process_result.enums.len(), 0);
1220    }
1221
1222    #[rstest]
1223    fn msg_simple_msg() {
1224        let test_empty_msg = "msg Lala { uint_7 a1; }";
1225
1226        let mut lexer = Token::lexer(test_empty_msg);
1227        lexer.extras = 0;
1228
1229        let parsed_bitis = parse_root(&mut lexer);
1230        if let Err(s) = parsed_bitis.clone() {
1231            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1232        }
1233        assert_eq!(parsed_bitis.is_ok(), true);
1234
1235        let parsed_bitis = parsed_bitis.unwrap();
1236        assert_eq!(parsed_bitis.len(), 1);
1237
1238        if let Value::Message(msg) = parsed_bitis[0].clone() {
1239            assert_eq!(msg.attributes.len(), 1);
1240            assert_eq!(msg.attributes[0].name, "a1".to_string());
1241            if let AttributeDetails::AttributeSimple(s) = msg.attributes[0].specific_details.clone() {
1242                assert_eq!(s, SimpleType::UIntFixed(7));
1243            }
1244            else { assert!(false, "Attribute type must be AttributeSimple."); }
1245        }
1246        else { assert!(false, "Value must be a message."); }
1247    }
1248
1249    #[rstest]
1250    fn msg_simple_enum() {
1251        let test_empty_msg = "enum Lala(4) { one, two }";
1252
1253        let mut lexer = Token::lexer(test_empty_msg);
1254        lexer.extras = 0;
1255
1256        let parsed_bitis = parse_root(&mut lexer);
1257        if let Err(s) = parsed_bitis.clone() {
1258            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1259        }
1260        assert_eq!(parsed_bitis.is_ok(), true);
1261
1262        let parsed_bitis = parsed_bitis.unwrap();
1263        assert_eq!(parsed_bitis.len(), 1);
1264
1265        if let Value::Enum(enm) = parsed_bitis[0].clone() {
1266            assert_eq!(enm.values.len(), 2);
1267            assert_eq!(enm.values[0], "one".to_string());
1268            assert_eq!(enm.values[1], "two".to_string());
1269        }
1270        else { assert!(false, "Value must be a message."); }
1271    }
1272
1273
1274    /*#[rstest]
1275        #[case::float("float", SimpleType::Float)]
1276        #[case::uint_12("uint_12", SimpleType::UIntFixed(12))]
1277        #[case::uint_32d4("uint_32d4", SimpleType::UIntDyn((32,4)))]
1278        #[case::bool("bool", SimpleType::Bool)]
1279        fn msg_various_attribute_types(#[case] attr_type_str: String, #[case] attr_type: SimpleType) {
1280            let test_msg = format!("msg Lala [fixed] {{ {attr_type_str} attr; }}");
1281
1282            let mut lexer = Token::lexer(test_msg.as_str());
1283            lexer.extras = 0;
1284
1285            let parsed_bitis = parse_root(&mut lexer);
1286            assert_eq!(parsed_bitis.is_ok(), true);
1287
1288            let parsed_bitis = parsed_bitis.unwrap();
1289            assert_eq!(parsed_bitis.len(), 1);
1290
1291            assert!(if let Value::Message(_) = parsed_bitis[0].clone() { true } else { false });
1292
1293            if let Value::Message(msg) = parsed_bitis[0].clone() {
1294                assert_eq!(msg.name, "Lala".to_string());
1295
1296                assert_eq!(msg.attributes.len(), 1);
1297                assert_eq!(msg.attributes[0].name, "attr".to_string());
1298
1299                assert!(if let AttributeDetails::AttributeSimple(_) = msg.attributes[0].specific_details.clone() { true } else { false });
1300
1301                if let AttributeDetails::AttributeSimple(at) = msg.attributes[0].specific_details.clone() {
1302                    assert_eq!(at, attr_type);
1303                }
1304            }
1305
1306            let validate_result = validate_bitis(&parsed_bitis);
1307            println!("validate_result: {:?}", validate_result);
1308            assert!(validate_result.is_none());
1309        }*/
1310}
1311
1312#[cfg(test)]
1313mod bitis_generate_rust {
1314    use rstest::rstest;
1315    use super::*;
1316
1317    const HEADER: &str = "use bitis_lib::*;\n\n";
1318    const ENUMS_HEADER: &str = "// Enums\n";
1319    const OO_HEADER: &str = "// Enums for oneof\n";
1320    const MSG_HEADER: &str = "// Messages\n";
1321    const PER_ENUM_HEADER: &str = "#[derive(BiserdiEnum, Debug, Clone, PartialEq)]\n#[biserdi_enum_id_dynbits(3)]\n#[allow(nonstandard_style)]\n";
1322    const PER_OO_HEADER: &str = "#[derive(BiserdiOneOf, Debug, Clone, PartialEq)]\n#[biserdi_enum_id_dynbits(3)]\n#[allow(nonstandard_style)]\n";
1323    const PER_MSG_HEADER: &str = "#[derive(BiserdiMsg, Debug, Clone, PartialEq)]\n#[allow(nonstandard_style)]\n";
1324
1325    #[rstest]
1326    fn msg_empty_msg() {
1327        let test_empty_msg = "msg Lala { }";
1328
1329        let mut lexer = Token::lexer(test_empty_msg);
1330        lexer.extras = 0;
1331
1332        let parsed_bitis = parse_root(&mut lexer);
1333        if let Err(s) = parsed_bitis.clone() {
1334            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1335        }
1336        assert_eq!(parsed_bitis.is_ok(), true);
1337
1338        let parsed_bitis = parsed_bitis.unwrap();
1339        assert_eq!(parsed_bitis.len(), 1);
1340
1341        let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1342        let rdo = RustDataObjects{ d: JinjaData{
1343            enums: processed_bitis.enums,
1344            msgs: to_rust_messages(&processed_bitis.msgs),
1345            oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1346
1347        let rendered = rdo.render().unwrap();
1348        let lala_empty = "pub struct Lala {\n}\n";
1349        assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + "\n\n"  +
1350            MSG_HEADER + PER_MSG_HEADER +lala_empty).to_string());
1351    }
1352
1353    #[rstest]
1354    fn msg_simple_msg() {
1355        let test_empty_msg = "//| comment for Lala\nmsg Lala { int_5 a1; repeated_fixed_4 bool bool_array; }";
1356        println!("Input code:\n{}", test_empty_msg);
1357
1358        let mut lexer = Token::lexer(test_empty_msg);
1359        lexer.extras = 0;
1360
1361        let parsed_bitis = parse_root(&mut lexer);
1362        if let Err(s) = parsed_bitis.clone() {
1363            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1364        }
1365        assert_eq!(parsed_bitis.is_ok(), true);
1366
1367        let parsed_bitis = parsed_bitis.unwrap();
1368        assert_eq!(parsed_bitis.len(), 1);
1369
1370        let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1371        let rdo = RustDataObjects{ d: JinjaData{
1372            enums: processed_bitis.enums, msgs: to_rust_messages(&processed_bitis.msgs),
1373            oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1374
1375        let rendered = rdo.render().unwrap();
1376        let lala_commment = "/// comment for Lala\n";
1377        let lala_empty = "pub struct Lala {\n  pub a1: IntWithGivenBitSize<i8, 5>,\n  pub bool_array: FixedArray<bool,4>,\n}\n";
1378        println!("rendered:\n{}",rendered);
1379        assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + "\n\n" +
1380            MSG_HEADER + lala_commment + PER_MSG_HEADER +lala_empty).to_string());
1381    }
1382
1383    #[rstest]
1384    fn msg_simple_enum() {
1385        let test_enum_msg = "//| comment for Numbers\nenum Numbers(3) {\n  // Comment for One\n  One,\n  Two,\n  Three\n}";
1386        println!("Input code:\n{}", test_enum_msg);
1387
1388        let mut lexer = Token::lexer(test_enum_msg);
1389        lexer.extras = 0;
1390
1391        let parsed_bitis = parse_root(&mut lexer);
1392        if let Err(s) = parsed_bitis.clone() {
1393            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_enum_msg[s.1.clone()], s.1);
1394        }
1395        assert_eq!(parsed_bitis.is_ok(), true);
1396
1397        let parsed_bitis = parsed_bitis.unwrap();
1398        assert_eq!(parsed_bitis.len(), 1);
1399
1400        let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1401        let rdo = RustDataObjects{ d: JinjaData{ enums: processed_bitis.enums,
1402            msgs: to_rust_messages(&processed_bitis.msgs),
1403            oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1404
1405        let rendered = rdo.render().unwrap();
1406        let lala_commment = "/// comment for Numbers\n";
1407        let lala_enum = "pub enum Numbers {\n  One,\n  Two,\n  Three,\n}\n\n";
1408        println!("*rendered:\n{}",rendered);
1409        assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + lala_commment + PER_ENUM_HEADER + lala_enum + OO_HEADER +
1410            "\n\n" + MSG_HEADER ).to_string());
1411    }
1412
1413    #[rstest]
1414    fn msg_simple_oneof() {
1415        let test_enum_msg = "//| comment for Oneof\nmsg TestOO {\n  oneof oo_li(3) { uint_3 test1; float test2; }\n  bool b1;\n}";
1416        println!("Input code:\n{}", test_enum_msg);
1417
1418        let mut lexer = Token::lexer(test_enum_msg);
1419        lexer.extras = 0;
1420
1421        let parsed_bitis = parse_root(&mut lexer);
1422        if let Err(s) = parsed_bitis.clone() {
1423            panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_enum_msg[s.1.clone()], s.1);
1424        }
1425        assert_eq!(parsed_bitis.is_ok(), true);
1426
1427        let parsed_bitis = parsed_bitis.unwrap();
1428        assert_eq!(parsed_bitis.len(), 1);
1429
1430        let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1431        let rdo = RustDataObjects{ d: JinjaData{ enums: processed_bitis.enums,
1432            msgs: to_rust_messages(&processed_bitis.msgs),
1433            oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1434
1435        let rendered = rdo.render().unwrap();
1436        let testoo_commment = "/// comment for Oneof\n";
1437        let testoo_enum = "pub enum OO_TestOo_OoLi {\n  Test1(IntWithGivenBitSize<u8, 3>),\n  Test2(f32),\n}\n\n";
1438        let testoo_msg = "pub struct TestOo {\n  pub oo_li: OO_TestOo_OoLi,\n  pub b1: bool,\n}\n";
1439        println!("*rendered:\n{}",rendered);
1440        assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + PER_OO_HEADER
1441            + testoo_enum + MSG_HEADER + testoo_commment + PER_MSG_HEADER + testoo_msg).to_string());
1442    }
1443}
1444
1445#[cfg(test)]
1446mod bitis_compile {
1447    use std::fs;
1448    use std::path::Path;
1449    use rstest::rstest;
1450    use super::*;
1451
1452    fn compile(content: &str) -> BitisProcessed {
1453        let mut lexer = Token::lexer(content);
1454        lexer.extras = 0;
1455        println!("*** content:\n{}", content);
1456        let bitis_parsed = match parse_root(&mut lexer) {
1457            Ok(v) => v,
1458            Err(e) => {
1459                let (err_str, err_span) = e.clone();
1460                let content_err = &content[err_span];
1461                println!("Error: {}\n  -> Source: '{}'", err_str, content_err);
1462                abort()
1463            }
1464        };
1465        println!("** content:\n{:?}", bitis_parsed);
1466        process_and_validate_bitis(&bitis_parsed)
1467    }
1468    fn render(d: JinjaData) {
1469        let rdo = RustDataObjects{ d: d.clone() };
1470        let rendered_rust = rdo.render().unwrap();
1471        println!("*** rendered DO:\n{}", rendered_rust);
1472        fs::write(Path::new("./test_data/test_py/bitis/src/messages_test.rs"), rendered_rust).expect("Unable to write file");
1473
1474        let rdo = RustPyDataObjects{ d: d.clone() };
1475        let rendered_rust = rdo.render().unwrap();
1476        println!("*** rendered PyDO:\n{}", rendered_rust);
1477        fs::write(Path::new("./test_data/test_py/bitis/src/pyrust_test.rs"), rendered_rust).expect("Unable to write file");
1478
1479        let rdo = RustPyLib{ d: d.clone(), lib_name: "bitis_msgs".into() };
1480        let rendered_rust = rdo.render().unwrap();
1481        println!("*** rendered pyLib:\n{}", rendered_rust);
1482        fs::write(Path::new("./test_data/test_py/bitis/src/lib_test.rs"), rendered_rust).expect("Unable to write file");
1483
1484        let rdo = PyTypeHints{ d };
1485        let rendered_rust = rdo.render().unwrap();
1486        println!("*** rendered py_type_hints:\n{}", rendered_rust);
1487        fs::write(Path::new("./test_data/test_py/bitis/bitis_msgs/bitis_msgs.pyi"), rendered_rust).expect("Unable to write file");
1488    }
1489
1490    #[rstest]
1491    #[ignore]
1492    fn simple_rust_py() {
1493        let bitis_str = "msg ParamTestSimple { uint_4 param_1; bool param_2; }";
1494
1495        let bitis_processed_org = compile(bitis_str);
1496
1497        let bitis_processed = bitis_processed_org.clone();
1498        let d = JinjaData{
1499            enums: bitis_processed.enums,
1500            msgs: to_rust_messages(&bitis_processed.msgs),
1501            oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1502        };
1503        render(d);
1504    }
1505
1506    #[rstest]
1507    #[ignore]
1508    fn nested_rust_py() {
1509        let bitis_str = "msg Inner { uint_2 val; }\nmsg ParamTestWithInner { uint_4 param_1; bool param_2; Inner inner; }";
1510
1511        let bitis_processed_org = compile(bitis_str);
1512
1513        let bitis_processed = bitis_processed_org.clone();
1514
1515        let d = JinjaData{
1516            enums: bitis_processed.enums,
1517            msgs: to_rust_messages(&bitis_processed.msgs),
1518            oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1519        };
1520        render(d);
1521    }
1522    #[test]
1523    #[ignore]
1524    fn nested_and_enum_rust_py() {
1525        let bitis_str = [
1526            "enum Numbers(4) { one, two, three, four }\n/// Test comment for Inner\nmsg Inner { uint_3 val; Numbers num; }\n",
1527            "msg ParamTestWithInner { uint_4 param_1; bool param_2; Inner inner; } }"
1528        ].join("");
1529
1530        let bitis_processed_org = compile(bitis_str.as_str());
1531
1532        let bitis_processed = bitis_processed_org.clone();
1533
1534        let d = JinjaData{
1535            enums: bitis_processed.enums,
1536            msgs: to_rust_messages(&bitis_processed.msgs),
1537            oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1538        };
1539        render(d);
1540    }
1541    #[rstest]
1542    fn oneof_nested_and_enum_rust_py() {
1543        let bitis_str = [
1544            "//| Test comment for Enum\nenum Numbers(4) { one, two, three, four }\n\n//| Test comment for Inner\nmsg Inner { uint_3 val; Numbers num; }\n",
1545            "msg ParamTestWithInner { uint_4 param_1; bool param_2; oneof action(4) { Inner inner; uint_3 val; } }"
1546        ].join("");
1547
1548        let bitis_processed_org = compile(bitis_str.as_str());
1549
1550        let bitis_processed = bitis_processed_org.clone();
1551
1552        let d = JinjaData{
1553            enums: bitis_processed.enums,
1554            msgs: to_rust_messages(&bitis_processed.msgs),
1555            oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1556        };
1557        render(d);
1558    }
1559}
1560
1561#[cfg(test)]
1562mod bitis_serialization {
1563    // use std::fs;
1564    use rstest::rstest;
1565    use super::*;
1566
1567    //noinspection DuplicatedCode
1568    #[rstest]
1569    fn msg_simple_msg_compile() {
1570        let test_empty_msg = "msg Lala { repeated_fixed_10 bool data_bool; uint_4 data1_uint; uint_12 data2_uint; }";
1571
1572        let mut lexer = Token::lexer(test_empty_msg);
1573        lexer.extras = 0;
1574
1575        let parsed_bitis = parse_root(&mut lexer);
1576        assert_eq!(parsed_bitis.is_ok(), true);
1577
1578        let _parsed_bitis = parsed_bitis.unwrap();
1579
1580        // let rdo = RustDataObjects {
1581        //     enums: parsed_bitis.iter().filter_map(|x|
1582        //         match x {
1583        //             Value::Enum(ev) => Some((ev.name.clone(), ev.clone())),
1584        //             _ => None
1585        //         })
1586        //         .collect::<HashMap<_, _>>(),
1587        //     msgs: parsed_bitis.iter().filter_map(|x|
1588        //         match x {
1589        //             Value::Message(mv) => Some((mv.name.clone(), mv.clone())),
1590        //             _ => None
1591        //         })
1592        //         .collect::<HashMap<_, _>>(),
1593        // };
1594        //
1595        // let rendered = rdo.render().unwrap();
1596        //
1597        // let current_test_simple_code = String::from(std::str::from_utf8(&fs::read("test_data/test_simple_msg.rs")
1598        //     .expect("Unable to read test_simple_msg.rs file")).unwrap());
1599        // assert_eq!(current_test_simple_code, rendered);
1600        //
1601        // let validate_result = validate_bitis(&parsed_bitis);
1602        // println!("validate_result: {:?}", validate_result);
1603        // assert!(validate_result.is_none());
1604    }
1605}
1606
1607#[cfg(test)]
1608mod bitis_processing {
1609    use rstest::rstest;
1610    use crate::AttributeDetails::{AttributeEnumOrMsg, AttributeSimple};
1611    use super::*;
1612
1613    #[rstest]
1614    #[ignore]
1615    fn msg_base_and_v2() {
1616        let bitis_values = vec![
1617            Value::Message(Message{
1618                name: "TestMsg".to_string(),
1619                /*version_info: VersionInfo::BaseWithAllowedVersion(0),*/
1620                comment: Some("This is a test".to_string()),
1621                parent: None,
1622                attributes: vec![Attribute{name: "a1".to_string(), comment: None,
1623                    is_repeated: None, is_optional: false,
1624                    specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1625                }],
1626            }),
1627            Value::Message(Message{
1628                name: "TestMsg".to_string(),
1629                /*version_info: VersionInfo::Version(2),*/
1630                comment: Some("This is a test".to_string()),
1631                parent: None,
1632                attributes: vec![Attribute{name: "a2".to_string(), comment: None,
1633                    is_repeated: None, is_optional: false,
1634                    specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1635                }],
1636            })
1637        ];
1638        let pb = process_and_validate_bitis(&bitis_values);
1639
1640        assert_eq!(pb.max_version_number, 2);
1641        assert_eq!(pb.msgs.len(), 3);
1642
1643        assert_eq!(pb.msgs[0].name, "TestMsg_Base".to_string());
1644        assert_eq!(pb.msgs[1].name, "TestMsg_V1".to_string());
1645        assert_eq!(pb.msgs[2].name, "TestMsg_V2".to_string());
1646
1647        assert_eq!(pb.msgs[0].attributes.len(), 1);
1648        assert_eq!(pb.msgs[0].attributes.get(0).unwrap().name, "a1".to_string());
1649        assert_eq!(pb.msgs[1].attributes.len(), 0);
1650        assert_eq!(pb.msgs[2].attributes.len(), 1);
1651        assert_eq!(pb.msgs[2].attributes.get(0).unwrap().name, "a2".to_string());
1652    }
1653
1654    #[rstest]
1655    #[ignore]
1656    fn msg_base_and_v2_and_add_msg() {
1657        let bitis_values = vec![
1658            Value::Message(Message{
1659                name: "TestMsgInner".to_string(),
1660                /*version_info: VersionInfo::BaseWithAllowedVersion(0),*/
1661                comment: Some("This is a test2".to_string()),
1662                parent: None,
1663                attributes: vec![Attribute{name: "lala".to_string(), comment: None,
1664                    is_repeated: None, is_optional: false,
1665                    specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1666                }],
1667            }),
1668            Value::Message(Message{
1669                name: "TestMsgInner".to_string(),
1670                /*version_info: VersionInfo::Version(1),*/
1671                comment: Some("This is a test2".to_string()),
1672                parent: None,
1673                attributes: vec![
1674                    Attribute{name: "lala".to_string(), comment: None, is_repeated: None, is_optional: false,
1675                        specific_details: AttributeSimple(SimpleType::UIntFixed(4)),},
1676                    Attribute{name: "lala2".to_string(), comment: None, is_repeated: None, is_optional: false,
1677                        specific_details: AttributeSimple(SimpleType::UIntFixed(3)),},
1678                ],
1679            }),
1680            Value::Message(Message{
1681                name: "TestMsg".to_string(),
1682                /*version_info: VersionInfo::BaseWithAllowedVersion(0),*/
1683                comment: Some("This is a test".to_string()),
1684                parent: None,
1685                attributes: vec![
1686                    Attribute{ name: "a1".to_string(), comment: None, is_repeated: None, is_optional: false,
1687                        specific_details: AttributeSimple(SimpleType::UIntFixed(4)) },
1688                    Attribute{ name: "lala_use".to_string(), comment: None, is_repeated: None, is_optional: false,
1689                        specific_details: AttributeEnumOrMsg("TestMsgInner".to_string()) },
1690                ],
1691            }),
1692            Value::Message(Message{
1693                name: "TestMsg".to_string(),
1694                /*version_info: VersionInfo::Version(2),*/
1695                comment: Some("This isa test".to_string()),
1696                parent: None,
1697                attributes: vec![Attribute{name: "a2".to_string(), comment: None,
1698                    is_repeated: None, is_optional: false,
1699                    specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1700                }],
1701            }),
1702        ];
1703        let pb = process_and_validate_bitis(&bitis_values);
1704
1705        assert_eq!(pb.max_version_number, 2);
1706        assert_eq!(pb.msgs.len(), 4);
1707
1708/*        assert_eq!(pb.msgs[0].name, "TestMsg_Base".to_string());
1709        assert_eq!(pb.msgs[1].name, "TestMsg_V1".to_string());
1710        assert_eq!(pb.msgs[2].name, "TestMsg_V2".to_string());
1711        assert_eq!(pb.msgs[3].name, "TestMsgLala".to_string());
1712
1713        if let MsgVersion::Versioned(l) = pb.msgs[0].version { assert_eq!(l, 0); }
1714        assert_eq!(pb.msgs[0].attributes.len(), 1);
1715        assert_eq!(pb.msgs[0].attributes.get(0).unwrap().name, "a1".to_string());
1716        assert_eq!(pb.msgs[1].attributes.len(), 0);
1717        assert_eq!(pb.msgs[2].attributes.len(), 1);
1718        assert_eq!(pb.msgs[2].attributes.get(0).unwrap().name, "a2".to_string());
1719        if let MsgVersion::Fixed = pb.msgs[0].version { assert!(false) }
1720        assert_eq!(pb.msgs[3].attributes.len(), 1);
1721        assert_eq!(pb.msgs[3].attributes.get(0).unwrap().name, "lala".to_string());*/
1722    }
1723}
1724
bitis_lib/lib_impl/compiler.rs

bitis_lib/lib_impl/
compiler.rs
