1use std::collections::HashMap;
2use std::process::{abort};
3use askama::Template;
4use logos::{Lexer, Logos, Span};
5use regex::Regex;
6use stringcase::Caser;
7
8
9fn integer_bit_size(bit_size: &u8) -> u8 {
12 match bit_size {
13 0..=8 => 8,
14 9..=16 => 16,
15 17..=32 => 32,
16 _ => 64
17 }
18}
19#[derive(Debug, Clone )]
21#[allow(unused)]
22pub struct AttributeEx {
23 base: Attribute,
24 rust_type_str: String,
25 base_type_str: String,
27 is_py_wrapped: bool, is_msg: bool, is_enum: bool, is_oo: bool, add_val: bool,
28}
29#[derive(Debug, Clone )]
30pub struct MessageR {
31 pub name: String,
32 pub comment: Option<String>,
34 pub parent: Option<String>,
35 pub attributes: Vec<AttributeEx>,
36}
37#[derive(Debug, Clone )]
38#[allow(dead_code)]
39pub struct OneOfInfoR {
40 msg_name: String,
41 name: String,
42 dyn_bits: u8,
43 attributes: Vec<AttributeEx>,
44}
45fn to_rust_attribute(attribute: &Attribute, msg_names: &Vec<String>) -> AttributeEx {
46 let (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val) = {
47 let mut is_py_wrapped = false;
48 let mut is_enum = false;
49 let mut is_msg = false;
50 let mut is_oo = false;
51 let mut add_val = false;
52
53 let (rtype, base_type) = match &attribute.specific_details {
54 AttributeDetails::AttributeSimple(a) => {
55 match a {
56 SimpleType::NoTypeSetYet => {
57 println!("Unexpected unspecified attribute type");
58 abort()
59 },
60 SimpleType::Bool => { ("bool".to_string(), "bool".to_string()) }
61 SimpleType::UIntFixed(b) => {
62 add_val = true;
63 let base = format!("u{}", integer_bit_size(&b));
64 (format!("VarWithGivenBitSize<{}, {}>", base.clone(), b), base) }
65 SimpleType::IntFixed(b) => {
66 add_val = true;
67 let base = format!("i{}", integer_bit_size(&b));
68 (format!("VarWithGivenBitSize<{}, {}>", base.clone(), b), base) }
69 SimpleType::UIntDyn(b) => {
70 add_val = true;
71 let base = format!("u{}", integer_bit_size(&b.0));
72 (format!("DynInteger<{}, {}>", base.clone(), b.1), base) }
73 SimpleType::IntDyn(b) => {
74 add_val = true;
75 let base = format!("i{}", integer_bit_size(&b.0));
76 (format!("DynInteger<{}, {}>", base.clone(), b.1), base) }
77 SimpleType::Float => {
78 add_val = true;
79 let base = "f32".to_string();
80 (base.clone(), base)
81 }
82 SimpleType::Double => {
83 let base = "f64".to_string();
84 (base.clone(), base) }
85 SimpleType::FixedPrecision(fpp) => {
86 add_val = true;
87 (format!("FixPrecisionMinMax<{}, {}, {}>", fpp.bits, fpp.min_val, fpp.max_val), "f64".to_string())
88 }
89 SimpleType::Binary(b) => {
90 add_val = true;
91 (format!("Binary<{}>", b), "Vec<u8>".to_string()) }
92 }
93 }
94 AttributeDetails::AttributeEnumOrMsg(em) => {
95 is_py_wrapped = true;
96 is_msg = msg_names.contains(&em);
97 is_enum = !is_msg.clone();
98 (em.clone(), em.clone()) }
99 AttributeDetails::AttributeOneOf(ooi) => {
100 is_py_wrapped=true; is_oo = true;
101 (ooi.name.clone(), ooi.name.clone()) }
102 };
103 (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val)
104 };
105 AttributeEx{base: attribute.clone(), rust_type_str: rtype, base_type_str: base_type,
106 is_py_wrapped, is_msg, is_enum, is_oo, add_val }
107}
108
109pub fn to_rust_messages(msgs: &Vec<Message>) -> Vec<MessageR> {
110 let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
111
112 msgs.iter().map(|msg| {
113 let attrs_rust: Vec<_> = msg.attributes.iter().map(|attribute| {
114 to_rust_attribute(attribute, &msgs_names) }).collect();
115 MessageR{name: msg.name.clone(), comment: msg.comment.clone(), parent: msg.parent.clone(),
116 attributes: attrs_rust}
117 }).collect()
118}
119pub fn to_rust_oneofs(oos: &Vec<(String, OneOfInfo)>, msgs: &Vec<Message>) -> HashMap<String, OneOfInfoR> {
120 let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
121
122 oos.iter().map(|(msg_name, oo)| {
123 let attrs_rust: Vec<_> = oo.attributes.iter().map(|attribute| {
124 to_rust_attribute(attribute, &msgs_names) }).collect();
125 (oo.name.clone(), OneOfInfoR{msg_name: msg_name.clone(), name: oo.name.clone(), dyn_bits: oo.dyn_bits, attributes: attrs_rust})
126 }).collect()
127}
128
129fn to_cpp_attribute(attribute: &Attribute, msg_names: &Vec<String>) -> AttributeEx {
130 let (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val) = {
131 let mut is_py_wrapped = false;
132 let mut is_enum = false;
133 let mut is_msg = false;
134 let mut is_oo = false;
135 let mut add_val = false;
136
137 let (rtype, base_type) = match &attribute.specific_details {
138 AttributeDetails::AttributeSimple(a) => {
139 match a {
140 SimpleType::NoTypeSetYet => {
141 println!("Unexpected unspecified attribute type");
142 abort()
143 },
144 SimpleType::Bool => { ("BitisBool".to_string(), "bool".to_string()) }
145 SimpleType::UIntFixed(b) => {
146 add_val = true;
147 let base = format!("uint{}_t", integer_bit_size(&b));
148 (format!("IntgralWithGivenBitSize<{}, {}>", base.clone(), b), base) }
149 SimpleType::IntFixed(b) => {
150 add_val = true;
151 let base = format!("int{}_t", integer_bit_size(&b));
152 (format!("IntgralWithGivenBitSize<{}, {}>", base.clone(), b), base) }
153 SimpleType::UIntDyn(b) => {
154 add_val = true;
155 let base = format!("uint{}_t", integer_bit_size(&b.0));
156 (format!("DynInteger<{}, {}>", base.clone(), b.1), base) }
157 SimpleType::IntDyn(b) => {
158 add_val = true;
159 let base = format!("int{}_t", integer_bit_size(&b.0));
160 (format!("DynInteger<{}, {}>", base.clone(), b.1), base) }
161 SimpleType::Float => {
162 add_val = true;
163 let base = "float".to_string();
164 (base.clone(), base)
165 }
166 SimpleType::Double => {
167 let base = "double".to_string();
168 (base.clone(), base) }
169 SimpleType::FixedPrecision(fpp) => {
170 add_val = true;
171 (format!("FixPrecisionMinMax<{}, {}, {}>", fpp.bits, fpp.min_val, fpp.max_val), "double".to_string())
172 }
173 SimpleType::Binary(b) => {
174 add_val = true;
175 (format!("Binary<{}>", b), "Vec<u8>".to_string()) }
176 }
177 }
178 AttributeDetails::AttributeEnumOrMsg(em) => {
179 is_py_wrapped = true;
180 is_msg = msg_names.contains(&em);
181 is_enum = !is_msg.clone();
182 (em.clone(), em.clone()) }
183 AttributeDetails::AttributeOneOf(ooi) => {
184 is_py_wrapped=true; is_oo = true;
185 (ooi.name.clone(), ooi.name.clone()) }
186 };
187 (rtype, base_type, is_py_wrapped, is_msg, is_enum, is_oo, add_val)
188 };
189 AttributeEx{base: attribute.clone(), rust_type_str: rtype, base_type_str: base_type,
190 is_py_wrapped, is_msg, is_enum, is_oo, add_val }
191}
192pub fn to_cpp_messages(msgs: &Vec<Message>) -> Vec<MessageR> {
193 let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
194
195 msgs.iter().map(|msg| {
196 let attrs_rust: Vec<_> = msg.attributes.iter().map(|attribute| {
197 to_cpp_attribute(attribute, &msgs_names) }).collect();
198 MessageR{name: msg.name.clone(), comment: msg.comment.clone(), parent: msg.parent.clone(),
199 attributes: attrs_rust}
200 }).collect()
201}
202pub fn to_cpp_oneofs(oos: &Vec<(String, OneOfInfo)>, msgs: &Vec<Message>) -> HashMap<String, OneOfInfoR> {
203 let msgs_names: Vec<_> = msgs.iter().map(|m| {m.name.clone()}).collect();
204
205 oos.iter().map(|(msg_name, oo)| {
206 let attrs_cpp: Vec<_> = oo.attributes.iter().map(|attribute| {
207 to_cpp_attribute(attribute, &msgs_names) }).collect();
208 (oo.name.clone(), OneOfInfoR{msg_name: msg_name.clone(), name: oo.name.clone(), dyn_bits: oo.dyn_bits, attributes: attrs_cpp })
209 }).collect()
210}
211
212#[derive(Clone, Debug)]
213pub struct JinjaData {
214 pub enums: Vec<Enum>,
215 pub msgs: Vec<MessageR>,
216 pub oos: HashMap<String, OneOfInfoR>,
217}
218
219#[derive(Template, Clone, Debug)]
220#[template(path = "data_objects.rs.jinja")]
221pub struct RustDataObjects {
222 pub d: JinjaData
223}
224#[derive(Template, Clone, Debug)]
225#[template(path = "pyclasses.py.rs.jinja")]
226pub struct RustPyDataObjects {
227 pub d: JinjaData
228}
229#[derive(Template, Clone, Debug)]
230#[template(path = "pylib.py.rs.jinja")]
231pub struct RustPyLib {
232 pub d: JinjaData,
233 pub lib_name: String
234}
235#[derive(Template, Clone, Debug)]
236#[template(path = "py_type_hints.pyi.jinja")]
237pub struct PyTypeHints {
238 pub d: JinjaData
239}
240#[derive(Template, Clone, Debug)]
241#[template(path = "data_objects.cpp.jinja")]
242pub struct CppDataObjects {
243 pub d: JinjaData,
244 pub object_order: Vec<String>,
245}
246
247
248mod filters {
249 #[allow(dead_code)]
250 pub fn snake_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values) -> ::askama::Result<String> {
251 Ok(stringcase::snake_case(s.to_string().as_str()))
252 }
253 #[allow(dead_code)]
254 pub fn camel_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
255 Ok(stringcase::camel_case(s.to_string().as_str()))
256 }
257 #[allow(dead_code)]
258 pub fn pascal_case<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
259 Ok(stringcase::pascal_case(s.to_string().as_str()))
260 }
261 #[allow(dead_code)]
262 pub fn to_py_type<T: std::fmt::Display>(s: T, _: &dyn askama::Values,) -> ::askama::Result<String> {
263 if ["u8", "u16", "u32", "u64", "i8", "i16", "i32", "i64"].contains(&s.to_string().as_str()) {
264 Ok("int".to_string()) }
265 else if ["f32", "f64"].contains(&s.to_string().as_str()) {
266 Ok("float".to_string())
267 }
268 else { Ok(s.to_string()) }
269 }
270}
271
272type Error = (String, Span);
274
275type Result<T> = std::result::Result<T, Error>;
276
277#[derive(Debug, Clone)]
278pub enum DynOrFixedType {
279 Dyn(u8),
280 Fixed(u8)
281}
282#[derive(Debug, PartialEq, Eq, Clone, Copy)]
283pub struct FixedPrecisionProperties {
284 bits: u8, min_val: i64, max_val: i64
285}
286#[derive(Debug, PartialEq, Eq, Clone, Copy)]
287pub enum SimpleType {
288 NoTypeSetYet,
289 Bool,
290 UIntFixed(u8), IntFixed(u8),
291 UIntDyn((u8,u8)), IntDyn((u8,u8)),
292 Float, Double,
293 FixedPrecision(FixedPrecisionProperties),
294 Binary(u8),
295}
296#[derive(Debug, Clone )]
328pub struct OneOfInfo {
329 name: String,
330 dyn_bits: u8,
331 attributes: Vec<Attribute>,
332}
333#[derive(Debug, Clone )]
334pub enum AttributeDetails {
335 AttributeSimple(SimpleType),
336 AttributeEnumOrMsg(String),
337 AttributeOneOf(OneOfInfo),
338}
339#[derive(Debug, Clone )]
340pub struct Attribute {
341 name: String,
342 comment: Option<String>,
343 is_repeated_and_size: Option<DynOrFixedType>,
344 is_optional: bool,
345 specific_details: AttributeDetails
346}
347#[derive(Debug, Clone)]
353pub struct Message {
354 pub name: String,
355 pub comment: Option<String>,
357 pub parent: Option<String>,
358 pub attributes: Vec<Attribute>,
359}
360
361#[derive(Debug, Clone)]
363pub struct Enum {
364 pub name: String,
365 pub comment: Option<String>,
367 pub bit_size: u8,
368 pub values: Vec<String>,
369}
370
371pub fn get_suffix_number(lex: &mut Lexer<Token>) -> Option<u8> {
372 let slice = lex.slice();
373 let re = Regex::new(r".*_d?([0-9]+)$").unwrap();
374 let num_str = re.captures(slice)?.get(1)?;
375 num_str.as_str().parse().ok()
376}
377pub fn get_d_suffix_numbers(lex: &mut Lexer<Token>) -> Option<(u8,u8)> {
378 let slice = lex.slice();
379 let re = Regex::new(r".*_([0-9]+)d([0-9]+)$").unwrap();
380 let first_num_str = re.captures(slice)?.get(1)?.as_str().parse().ok()?;
381 let second_num_str = re.captures(slice)?.get(2)?.as_str().parse().ok()?;
382 Some((first_num_str, second_num_str))
383}
384pub fn get_fp_properties_number(lex: &mut Lexer<Token>) -> Option<FixedPrecisionProperties> {
385 let slice = lex.slice();
386 let re = Regex::new(r"fp_([0-9]+)\[ *(-?[0-9]+) *, *(-?[0-9]+) *]").unwrap();
387 let bits = re.captures(slice)?.get(1)?.as_str().parse::<u8>().ok()?;
388 let min_val = re.captures(slice)?.get(2)?.as_str().parse::<i64>().ok()?;
389 let max_val = re.captures(slice)?.get(3)?.as_str().parse::<i64>().ok()?;
390 Some(FixedPrecisionProperties {bits, min_val, max_val})
391}
392pub fn get_enum_bit_size(lex: &mut Lexer<Token>) -> Option<u8> {
405 let slice = lex.slice();
406 let re = Regex::new(r"\( *([0-9]+) *\)").unwrap();
407 let bits = re.captures(slice)?.get(1)?.as_str().parse::<u8>().ok()?;
408 Some(bits)
409}
410pub fn get_version(lex: &mut Lexer<Token>) -> Option<u16> {
411 let slice = lex.slice();
412 let re = Regex::new(r"\[.* +(v[0-9]+) *]").unwrap();
413 let ver_str = re.captures(slice)?.get(1)?.as_str();
414 Some(ver_str.parse::<u16>().ok()?)
415}
416
417#[derive(Debug, Logos)]
418#[logos(skip r"[ \t\r\n\f]+")]
419#[logos(extras = u16)]
420#[allow(dead_code)]
421pub enum Token{
422 #[regex(r"//[^\n]*\n?", priority=40)] Comment,
423 #[regex(r"//\|[^\n]*\n?", |lex| lex.slice()[3..].to_owned(), priority=41)] SpecificComment(String),
424 #[token("msg", priority=20)] Msg,
425 #[token("enum", priority=20)] Enum,
426 #[token("oneof", priority=20)] OneOf,
427 #[token("{")] CBraceOpen,
428 #[token("}")] CBraceClose,
429 #[token("(")] BraceOpen,
430 #[token(")")] BraceClose,
431 #[token(":")] Colon,
432 #[token(";")] SemiColon,
433 #[token(",")] Comma,
434 #[regex(r"\[ *base +use +starting +with +v[0-9]+ *\]", get_version, priority=35)] MsgBaseInfoToken(u16),
437 #[regex(r"\[ *version +v[0-9]+ *\]", get_version, priority=35)] MsgVersionToken(u16),
438 #[regex("[0-9]+", |lex| lex.slice().parse::<isize>().unwrap(), priority=1)] IntegerVal(isize),
440 #[regex(r"-?(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?",
441 |lex| lex.slice().parse::<f64>().unwrap(), priority=2)] Number(f64),
442 #[token("bool", priority=30)] Bool,
443 #[token("msg_size_type", priority=30)] MsgSizeType,
444 #[regex(r"uint_[0-9]+", get_suffix_number, priority=30)] UIntFixed(u8),
445 #[regex(r"int_[0-9]+", get_suffix_number, priority=30)] IntFixed(u8),
446 #[regex(r"uint_[0-9]+d[0-9]+", get_d_suffix_numbers, priority=31)] UIntDyn((u8,u8)),
447 #[regex(r"int_[0-9]+d[0-9]+", get_d_suffix_numbers, priority=31)] IntDyn((u8,u8)),
448 #[token("float", priority=30)] Float,
449 #[token("double", priority=30)] Double,
450 #[regex(r"fp_[0-9]+\[ *-?[0-9]+ *, *-?[0-9]+ *]", get_fp_properties_number, priority=30)] FixedPoint(FixedPrecisionProperties),
451 #[token("binary_d[0-9]+", get_suffix_number, priority=30)] Binary(u8),
454 #[regex(r"repeated_dyn_[0-9]+", get_suffix_number, priority=30)] RepeatedDyn(u8),
455 #[regex(r"repeated_fixed_[0-9]+", get_suffix_number, priority=30)] RepeatedFixed(u8),
456 #[token("optional", priority=30)] Optional,
457 #[regex(r"[A-Za-z][A-Za-z0-9_-]+", |lex| lex.slice().to_owned(), priority=11)] StringVal(String),
458 #[token("false", |_| false, priority=20)]
459 #[token("true", |_| true, priority=20)] BoolVal(bool),
460 #[regex(r"\( *([0-9]+) *\)", get_enum_bit_size, priority=40)] EnumDynSize(u8),
461}
462
463#[derive(Debug, Clone)]
464pub enum Value {
465 Message(Message),
467 Enum(Enum)
469}
470
471macro_rules! parse_one_token {
472 ($token_enum: path, $lexer: expr, $error_msg_or_empty: expr) => {
473 loop {
474 let rv = $lexer.next();
475 if let Some(token) = rv {
476 match token {
477 Ok($token_enum) => {
478 break Ok(Ok(()));
479 },
480 Ok(Token::Comment) => (),
481 _ => {
482 if let Some(err_str) = $error_msg_or_empty {
483 break Err((format!("{err_str}\nToken: {token:?}").to_owned(), $lexer.span()));
484 }
485 else {
486 break Ok(Err($lexer.span()));
487 }
488 }
489 }
490 }
491 else {
492 break Err((format!("Unexpected end or text {rv:?}").to_owned(), $lexer.span()));
493 }
494 }
495 }
496}
497macro_rules! parse_one_token_with_arg {
498 ($token_enum: path, $lexer: expr, $error_msg_or_empty: expr) => {
499 loop {
500 let rv = $lexer.next();
501 if let Some(token) = rv {
502 match token {
503 Ok($token_enum(s)) => {
504 break Ok(Ok(s));
505 },
506 Ok(Token::Comment) => (),
507 _ => {
508 if let Some(err_str) = $error_msg_or_empty {
509 break Err((format!("{}\nFound token: {:?}.",
510 err_str, token).to_owned(), $lexer.span()));
511 }
512 else {
513 break Ok(Err($lexer.span()));
514 }
515 }
516 }
517 }
518 else {
519 break Err((format!("Unexpected end or text {rv:?}").to_owned(), $lexer.span()));
520 }
521 }
522 }
523}
524
525pub fn parse_root(lexer: &mut Lexer<'_, Token>) -> Result<Vec<Value>> {
526 let mut list: Vec<Value> = Vec::new();
527 let mut specific_comment: Option<String> = None;
528 loop {
529 if let Some(token) = lexer.next() {
530 let rv = match token {
531 Ok(Token::Msg) => Some(parse_msg(lexer, specific_comment.clone())),
532 Ok(Token::Enum) => Some(parse_enum(lexer, specific_comment.clone())),
533 Ok(Token::Comment) => None,
534 Ok(Token::SpecificComment(s)) => {
535 specific_comment = Some(s); None },
536 _ => Some(Err((format!("Unexpected token {:?}", token).to_owned(), lexer.span()))),
537 };
538 match rv {
539 None => (),
540 Some(Ok(value)) => { list.push(value); specific_comment = None; },
541 Some(Err(err)) => return Err(err)
542 }
543 }
544 else { break; }
545 }
546 Ok(list)
547}
548
549pub fn parse_msg(lexer: &mut Lexer<'_, Token>, comment_for_msg: Option<String>) -> Result<Value> {
550 let mut attributes = Vec::new();
551
552 let name = match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name but received:"))? {
553 Ok(s) => s,
554 Err(s) => { return Err(("Code should not be reached".into(), s)); }
555 };
556
557 let parent = {
571 let has_parent; let p;
572 if let Some(token) = lexer.next() {
573 match token {
574 Ok(Token::Colon) => has_parent = true,
575 Ok(Token::CBraceOpen) => has_parent = false,
576 _ => { return Err((format!("Unexpected text for msg '{name}'.").into(), lexer.span())) },
577 }
578 if has_parent {
579 match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name."))? {
580 Ok(s) => p = Some(s),
581 Err(s) => { return Err((format!("For msg '{name} colon found but no parent name").into(), s)); }
582 };
583 parse_one_token!(Token::CBraceOpen, lexer, Some(format!("Expected curly bracket open for msg '{name}'")))?.unwrap();
584 }
585 else {
586 p = None
587 }
588 }
589 else { return Err(("Unexpected end of file".into(), lexer.span())); }
590 p
591 };
592
593 loop {
594 if let Some(token) = lexer.next() {
595 match token {
596 Ok(Token::CBraceClose) => break,
597 Ok(Token::Comment) => (),
598 Ok(ctoken) => match parse_attribute(ctoken, lexer, name.clone(), false) {
599 Ok(a) => { attributes.push(a); },
600 Err(e) => { return Err(e); }
601 },
602 _ => { return Err((format!("Error: Unexpected text found for msg '{name}'.").into(), lexer.span())) },
603 };
604 }
605 else { return Err(("Unexpected end of file".into(), lexer.span())); }
606 }
607
608 Ok(Value::Message(Message{name, comment: comment_for_msg, parent, attributes}))
609}
610
611pub fn parse_attribute(last_token: Token, lexer: &mut Lexer<'_, Token>,
612 parent_name: String, attributes_for_oneof: bool) -> Result<Attribute> {
613 let mut is_optional = false;
614 let mut is_repeated_and_size: Option<DynOrFixedType> = None;
615 let mut attr_type = SimpleType::NoTypeSetYet;
616 let mut ctoken = last_token;
617 let mut enum_or_msg_str = None;
618 let mut oneof_infos = None;
619 let lexer_span_start = lexer.span();
620 let mut specific_comment: Option<String> = None;
621
622 loop {
623 match ctoken {
624 Token::SpecificComment(s) => {
625 specific_comment = Some(s); () },
626 Token::Optional if is_repeated_and_size.is_some() =>
627 return Err(("Error: Optional and repeated not allowed together".to_owned(), lexer.span())),
628 Token::RepeatedFixed(_) | Token::RepeatedDyn(_) if is_optional =>
629 return Err(("Error: Optional and repeated are not allowed together".to_owned(), lexer.span())),
630
631 Token::Optional | Token::RepeatedDyn(_) | Token::RepeatedFixed(_) if attributes_for_oneof =>
632 return Err(("Error: Optional and repeated are not allowed in oneof".to_owned(), lexer.span())),
633
634 Token::Optional => is_optional = true,
635 Token::RepeatedDyn(b) => is_repeated_and_size = Some(DynOrFixedType::Dyn(b)),
636 Token::RepeatedFixed(b) => is_repeated_and_size = Some(DynOrFixedType::Fixed(b)),
637 Token::Bool => { attr_type = SimpleType::Bool; break; },
638 Token::UIntFixed(s) => { attr_type = SimpleType::UIntFixed(s); break; },
639 Token::UIntDyn((m,s)) if m < s =>
640 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())),
641 Token::UIntDyn((m,_)) if (m & 3) != 0 =>
642 return Err(("Error: Unsigned dyn integer bit size of integer type must be a multiple of 8".to_owned(), lexer.span())),
643 Token::UIntDyn((m,s)) => { attr_type = SimpleType::UIntDyn((m, s)); break; },
644 Token::IntFixed(s) => { attr_type = SimpleType::IntFixed(s); break; },
645 Token::IntDyn((m,s)) if m < s =>
646 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())),
647 Token::IntDyn((m,_)) if (m & 3) != 0 =>
648 return Err(("Error: Unsigned dyn integer bit size of integer type must be a multiple of 8".to_owned(), lexer.span())),
649 Token::IntDyn((m,s)) => {
650 attr_type = SimpleType::IntDyn((m,s)); break;
651 },
652 Token::Float => { attr_type = SimpleType::Float; break; },
654 Token::Double => { attr_type = SimpleType::Double; break; },
655 Token::FixedPoint(s) => { attr_type = SimpleType::FixedPrecision(s); break; },
656 Token::Binary(b) => { attr_type = SimpleType::Binary(b); break; },
657 Token::StringVal(s) => { enum_or_msg_str = Some(s); break; }
658 Token::OneOf => {
659 oneof_infos = match parse_oneof(lexer, parent_name.clone(), specific_comment.clone(),
660 is_repeated_and_size.clone(), is_optional.clone()) {
661 Ok(oo) => Some(oo),
662 Err(s) => { return Err(s); }
663 };
664 break;
665 }
666 _ => { return Err((format!("Error: Expected attribute type or modifier (got {ctoken:?}) when parsing msg or oneof '{parent_name}'")
667 .to_owned(), lexer.span())); }
668 }
669 if let Some(token) = lexer.next() {
670 match token {
671 Ok(t) => ctoken = t,
672 Err(_) => { return Err((format!("Error: Unexpected text found for msg '{parent_name}'.").to_owned(), lexer.span())); }
673 }
674 } else {
675 return Err(("Unexpected end of file".to_string(), lexer.span()));
676 }
677 }
678
679 let mut name= "".to_owned();
680 if oneof_infos.is_none() {
681 name = parse_one_token_with_arg!(
682 Token::StringVal, lexer, Some(format!("Error: Expected attribute name for msg '{parent_name}' (type: {attr_type:?}/{enum_or_msg_str:?})")))?.unwrap();
683
684 parse_one_token!(Token::SemiColon, lexer, Some(format!(
685 "Error: Expected semicolon to end line of attribute '{name}' of msg or oneof '{parent_name}'")))?.unwrap();
686 }
687 let num_of_set_types_or_opts = vec![(attr_type != SimpleType::NoTypeSetYet), enum_or_msg_str.is_some(), oneof_infos.is_some()]
688 .iter().map(|&x| if x { 1_u8 } else { 0_u8 }).sum::<u8>();
689 if num_of_set_types_or_opts > 1 {
690 let mut span = lexer_span_start.clone();
691 span.end = lexer.span().end;
692 return Err(("Error: Attribute contains inconsistent optional, simple types and messages or Enums".to_string(), span));
693 }
694
695 if let Some(oo) = oneof_infos {
696 Ok(oo)
697 }
698 else if let Some(t) = enum_or_msg_str {
699 Ok(Attribute{name, comment: specific_comment, is_repeated_and_size, is_optional,
700 specific_details: AttributeDetails::AttributeEnumOrMsg(t)})
701 }
702 else {
703 Ok(Attribute{name, comment: specific_comment, is_repeated_and_size, is_optional,
704 specific_details: AttributeDetails::AttributeSimple(attr_type)})
705 }
706}
707
708pub fn parse_oneof(lexer: &mut Lexer<'_, Token>, parent_name: String, comment: Option<String>,
709 is_repeated_and_size: Option<DynOrFixedType>, is_optional: bool) -> Result<Attribute> {
710 let oo_name = parse_one_token_with_arg!(
711 Token::StringVal, lexer, Some(format!("Error: Expected name for oneof in parent '{parent_name}'")))?.unwrap();
712
713 let bit_size = match parse_one_token_with_arg!(Token::EnumDynSize, lexer, Some("Expected oneof properties for dyn size, e.g. (4)."))? {
714 Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
715 };
716
717 parse_one_token!(Token::CBraceOpen, lexer, Some("Error: Expected open curly bracket to enclose oneof elements"))?.unwrap();
718
719 let mut oo_attribs = Vec::new();
720 loop {
721 if let Some(token) = lexer.next() {
722 match token {
723 Ok(Token::CBraceClose) => break,
724 Ok(last_token) => {
725 match parse_attribute(last_token, lexer, oo_name.clone(), true) {
726 Ok(o) => oo_attribs.push(o),
727 Err(s) => return Err(s),
728 }
729 }
730 Err(_) => { return Err((format!("Error: Unexpected text when decoding oneof ({token:?})").to_owned(), lexer.span())); },
731 }
732 }
733 }
734 Ok(Attribute{name: oo_name.clone(), comment, is_repeated_and_size, is_optional,
735 specific_details: AttributeDetails::AttributeOneOf(OneOfInfo{
736 name: format!("OO_{}_{}", parent_name.to_pascal_case(), oo_name.to_pascal_case()),
737 dyn_bits: bit_size, attributes: oo_attribs})})
738}
739
740pub fn parse_enum(lexer: &mut Lexer<'_, Token>, comment: Option<String>) -> Result<Value> {
741 let name = match parse_one_token_with_arg!(Token::StringVal, lexer, Some("Expected msg name but received."))? {
742 Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
743 };
744
745 let bit_size = match parse_one_token_with_arg!(Token::EnumDynSize, lexer, Some("Expected enum properties for dyn size, e.g. (4)."))? {
746 Ok(s) => s, Err(s) => { return Err(("Code should not be reached".into(), s)); }
747 };
748
749 parse_one_token!(Token::CBraceOpen, lexer, Some(format!("Expected open curly bracket for enum '{name}'")))?.unwrap();
763
764 let mut values = Vec::new();
765 loop {
766 if let Some(token) = lexer.next() {
767 match token {
768 Ok(Token::CBraceClose) => break,
769 Ok(Token::StringVal(s)) => values.push(s),
770 Ok(Token::Comma) | Ok(Token::Comment)=> (),
771 _ => { return Err((format!("Error: Unexpected text found for enum '{name}'.").into(), lexer.span())) },
772 }
773 } else { return Err(("Unexpected end of file".into(), lexer.span())); }
774 }
775
776 Ok(Value::Enum(Enum{name, comment, bit_size, values}))
777}
778
779
780#[derive(Debug, Clone)]
859pub struct Dependencies {
860 pub in_deps: Vec<String>,
861 pub out_deps: Vec<String>,
862}
863#[derive(Debug, Clone)]
864pub struct BitisProcessed {
865 pub max_version_number: u16,
866 pub msgs: Vec<Message>,
867 pub enums: Vec<Enum>,
868 pub oo_enums: Vec<(String, OneOfInfo)>,
869}
870
871pub fn process_and_validate_bitis(parsed_bitis: &Vec<Value>) -> BitisProcessed {
873 let msgs: Vec<_> = parsed_bitis.iter().filter_map(|v| {
998 match v { Value::Message(msg) => Some(msg.clone()), _ => None }
999 }).collect();
1000 let enums: Vec<_> = parsed_bitis.iter().filter_map(|v| {
1001 match v { Value::Enum(enm) => Some(enm.clone()), _ => None }
1002 }).collect();
1003
1004 fn get_oneofs(msg_name: String, attrs: &Vec<Attribute>) -> Option<Vec<(String, OneOfInfo)>> {
1005 let direct_oos = attrs.iter().filter_map(|attr| {
1006 match &attr.specific_details {
1007 AttributeDetails::AttributeOneOf(oo) => Some(vec![(msg_name.clone(), oo.clone())]),
1008 _ => None
1009 }
1010 }).collect::<Vec<Vec<(String, OneOfInfo)>>>().concat();
1011
1012 let inner_oos = direct_oos.iter().filter_map(|(_, doo)| {
1013 get_oneofs(msg_name.clone(), &doo.attributes)
1014 }).collect::<Vec<Vec<_>>>().concat();
1015
1016 let all_oos = vec![direct_oos, inner_oos].concat();
1017 if all_oos.len() == 0 { None }
1018 else { Some(all_oos) }
1019 }
1020 let oo_enums: Vec<_> = msgs.iter().filter_map(|msg| {
1021 get_oneofs(msg.name.clone(), &msg.attributes)
1022 }).collect::<Vec<_>>().concat();
1023
1024 { let msg_names = msgs.iter().map(|msg| &msg.name).collect::<Vec<_>>();
1028 msg_names.iter().for_each(|name| {
1029 if msg_names.iter().filter(|cname| **cname == *name).count() > 1 {
1031 println!("Error: Multiple instances of msg '{}' found.", name);
1032 abort()
1033 }
1034 });
1035 let enum_names = enums.iter().map(|enm| &enm.name).collect::<Vec<_>>();
1036 enum_names.iter().for_each(|name| {
1037 if enum_names.iter().filter(|cname| **cname == *name).count() > 1 {
1038 println!("Error: Multiple instances of enum '{}' found.", name); abort()
1039 }
1040 });
1041
1042 }
1045
1046
1047 BitisProcessed { max_version_number: 0, msgs, enums, oo_enums}
1048}
1049
1050pub fn dependencies_process(jd: JinjaData) -> Vec<String>{
1051 let mut dependencies = HashMap::new();
1052
1053 for msgs in jd.msgs.clone() {
1054 dependencies.insert(
1055 msgs.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1056 }
1057 for enm in jd.enums {
1058 dependencies.insert(
1059 enm.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1060 }
1061 for (_, oos) in jd.oos.clone() {
1062 dependencies.insert(
1063 oos.name.clone(), Dependencies{in_deps: vec![], out_deps: vec![]});
1064 }
1065
1066 for msgs in jd.msgs {
1068 for attr in msgs.attributes {
1069 if attr.is_enum || attr.is_msg || attr.is_oo {
1070 dependencies.get_mut(&attr.rust_type_str).unwrap().out_deps.push(msgs.name.clone());
1071 dependencies.get_mut(&msgs.name).unwrap().in_deps.push(attr.rust_type_str.clone());
1072 }
1073 }
1074 }
1075 for (_, msgs) in jd.oos {
1076 for attr in msgs.attributes {
1077 if attr.is_enum || attr.is_msg || attr.is_oo {
1078 dependencies.get_mut(&attr.rust_type_str).unwrap().out_deps.push(msgs.name.clone());
1079 dependencies.get_mut(&msgs.name).unwrap().in_deps.push(attr.rust_type_str.clone());
1080 }
1081 }
1082 }
1083 println!("{:#?}", dependencies.clone());
1084
1085 let mut object_order = Vec::new();
1086 while dependencies.len() > 0 {
1087 let mut cobjs: Vec<_> = dependencies.clone().iter().filter_map(|(obj_name, deps)| {
1088 if deps.in_deps.len() == 0 { Some(obj_name.clone()) }
1089 else { None }
1090 }).collect();
1091
1092 for co in cobjs.clone() {
1093 dependencies.remove(&co);
1094 }
1095 for co in cobjs.clone() {
1096 for (_, deps) in &mut dependencies {
1097 deps.in_deps.retain(|x| *x != co);
1098 }
1099 }
1100 object_order.append(&mut cobjs);
1101 }
1102 println!("{:?}", object_order);
1103
1104 object_order
1105}
1106#[cfg(test)]
1109mod bitis_semantic {
1110 use rstest::rstest;
1111 use super::*;
1112
1113 #[rstest]
1114 fn msg_empty_msg() {
1115 let test_empty_msg = "msg Lala { }";
1116
1117 let mut lexer = Token::lexer(test_empty_msg);
1118 lexer.extras = 0;
1119
1120 let parsed_bitis = parse_root(&mut lexer);
1121 if let Err(s) = parsed_bitis.clone() {
1122 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1123 }
1124 assert_eq!(parsed_bitis.is_ok(), true);
1125
1126 let parsed_bitis = parsed_bitis.unwrap();
1127 assert_eq!(parsed_bitis.len(), 1);
1128
1129 assert!(if let Value::Message(_) = parsed_bitis[0].clone() { true } else { false });
1130
1131 if let Value::Message(msg) = parsed_bitis[0].clone() {
1132 assert_eq!(msg.name, "Lala".to_string());
1133 }
1134
1135 let process_result = process_and_validate_bitis(&parsed_bitis);
1139 println!("process_result {:?}", process_result);
1140
1141 assert_eq!(process_result.msgs.len(), 1);
1142 assert_eq!(process_result.enums.len(), 0);
1143 }
1144
1145 #[rstest]
1146 fn msg_simple_msg() {
1147 let test_empty_msg = "msg Lala { uint_7 a1; }";
1148
1149 let mut lexer = Token::lexer(test_empty_msg);
1150 lexer.extras = 0;
1151
1152 let parsed_bitis = parse_root(&mut lexer);
1153 if let Err(s) = parsed_bitis.clone() {
1154 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1155 }
1156 assert_eq!(parsed_bitis.is_ok(), true);
1157
1158 let parsed_bitis = parsed_bitis.unwrap();
1159 assert_eq!(parsed_bitis.len(), 1);
1160
1161 if let Value::Message(msg) = parsed_bitis[0].clone() {
1162 assert_eq!(msg.attributes.len(), 1);
1163 assert_eq!(msg.attributes[0].name, "a1".to_string());
1164 if let AttributeDetails::AttributeSimple(s) = msg.attributes[0].specific_details.clone() {
1165 assert_eq!(s, SimpleType::UIntFixed(7));
1166 }
1167 else { assert!(false, "Attribute type must be AttributeSimple."); }
1168 }
1169 else { assert!(false, "Value must be a message."); }
1170 }
1171
1172 #[rstest]
1173 fn msg_simple_enum() {
1174 let test_empty_msg = "enum Lala(4) { one, two }";
1175
1176 let mut lexer = Token::lexer(test_empty_msg);
1177 lexer.extras = 0;
1178
1179 let parsed_bitis = parse_root(&mut lexer);
1180 if let Err(s) = parsed_bitis.clone() {
1181 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1182 }
1183 assert_eq!(parsed_bitis.is_ok(), true);
1184
1185 let parsed_bitis = parsed_bitis.unwrap();
1186 assert_eq!(parsed_bitis.len(), 1);
1187
1188 if let Value::Enum(enm) = parsed_bitis[0].clone() {
1189 assert_eq!(enm.values.len(), 2);
1190 assert_eq!(enm.values[0], "one".to_string());
1191 assert_eq!(enm.values[1], "two".to_string());
1192 }
1193 else { assert!(false, "Value must be a message."); }
1194 }
1195
1196
1197 }
1234
1235#[cfg(test)]
1236mod bitis_generate_rust {
1237 use rstest::rstest;
1238 use super::*;
1239
1240 const HEADER: &str = "use bitis_lib::*;\n\n";
1241 const ENUMS_HEADER: &str = "// Enums\n";
1242 const OO_HEADER: &str = "// Enums for oneof\n";
1243 const MSG_HEADER: &str = "// Messages\n";
1244 const PER_ENUM_HEADER: &str = "#[derive(BiserdiEnum, Debug, Clone, PartialEq)]\n#[biserdi_enum_id_dynbits(3)]\n#[allow(nonstandard_style)]\n";
1245 const PER_OO_HEADER: &str = "#[derive(BiserdiOneOf, Debug, Clone, PartialEq)]\n#[biserdi_enum_id_dynbits(3)]\n#[allow(nonstandard_style)]\n";
1246 const PER_MSG_HEADER: &str = "#[derive(BiserdiMsg, Debug, Clone, PartialEq)]\n#[allow(nonstandard_style)]\n";
1247
1248 #[rstest]
1249 fn msg_empty_msg() {
1250 let test_empty_msg = "msg Lala { }";
1251
1252 let mut lexer = Token::lexer(test_empty_msg);
1253 lexer.extras = 0;
1254
1255 let parsed_bitis = parse_root(&mut lexer);
1256 if let Err(s) = parsed_bitis.clone() {
1257 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1258 }
1259 assert_eq!(parsed_bitis.is_ok(), true);
1260
1261 let parsed_bitis = parsed_bitis.unwrap();
1262 assert_eq!(parsed_bitis.len(), 1);
1263
1264 let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1265 let rdo = RustDataObjects{ d: JinjaData{
1266 enums: processed_bitis.enums,
1267 msgs: to_rust_messages(&processed_bitis.msgs),
1268 oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1269
1270 let rendered = rdo.render().unwrap();
1271 let lala_empty = "pub struct Lala {\n}\n";
1272 assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + "\n\n" +
1273 MSG_HEADER + PER_MSG_HEADER +lala_empty).to_string());
1274 }
1275
1276 #[rstest]
1277 fn msg_simple_msg() {
1278 let test_empty_msg = "//| comment for Lala\nmsg Lala { int_5 a1; repeated_fixed_4 bool bool_array; }";
1279 println!("Input code:\n{}", test_empty_msg);
1280
1281 let mut lexer = Token::lexer(test_empty_msg);
1282 lexer.extras = 0;
1283
1284 let parsed_bitis = parse_root(&mut lexer);
1285 if let Err(s) = parsed_bitis.clone() {
1286 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_empty_msg[s.1.clone()], s.1);
1287 }
1288 assert_eq!(parsed_bitis.is_ok(), true);
1289
1290 let parsed_bitis = parsed_bitis.unwrap();
1291 assert_eq!(parsed_bitis.len(), 1);
1292
1293 let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1294 let rdo = RustDataObjects{ d: JinjaData{
1295 enums: processed_bitis.enums, msgs: to_rust_messages(&processed_bitis.msgs),
1296 oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1297
1298 let rendered = rdo.render().unwrap();
1299 let lala_commment = "/// comment for Lala\n";
1300 let lala_empty = "pub struct Lala {\n pub a1: VarWithGivenBitSize<i8, 5>,\n pub bool_array: FixedArray<bool,4>,\n}\n";
1301 println!("rendered:\n{}",rendered);
1302 assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + "\n\n" +
1303 MSG_HEADER + lala_commment + PER_MSG_HEADER +lala_empty).to_string());
1304 }
1305
1306 #[rstest]
1307 fn msg_simple_enum() {
1308 let test_enum_msg = "//| comment for Numbers\nenum Numbers(3) {\n // Comment for One\n One,\n Two,\n Three\n}";
1309 println!("Input code:\n{}", test_enum_msg);
1310
1311 let mut lexer = Token::lexer(test_enum_msg);
1312 lexer.extras = 0;
1313
1314 let parsed_bitis = parse_root(&mut lexer);
1315 if let Err(s) = parsed_bitis.clone() {
1316 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_enum_msg[s.1.clone()], s.1);
1317 }
1318 assert_eq!(parsed_bitis.is_ok(), true);
1319
1320 let parsed_bitis = parsed_bitis.unwrap();
1321 assert_eq!(parsed_bitis.len(), 1);
1322
1323 let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1324 let rdo = RustDataObjects{ d: JinjaData{ enums: processed_bitis.enums,
1325 msgs: to_rust_messages(&processed_bitis.msgs),
1326 oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1327
1328 let rendered = rdo.render().unwrap();
1329 let lala_commment = "/// comment for Numbers\n";
1330 let lala_enum = "pub enum Numbers {\n One,\n Two,\n Three,\n}\n\n";
1331 println!("*rendered:\n{}",rendered);
1332 assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + lala_commment + PER_ENUM_HEADER + lala_enum + OO_HEADER +
1333 "\n\n" + MSG_HEADER ).to_string());
1334 }
1335
1336 #[rstest]
1337 fn msg_simple_oneof() {
1338 let test_enum_msg = "//| comment for Oneof\nmsg TestOO {\n oneof oo_li(3) { uint_3 test1; float test2; }\n bool b1;\n}";
1339 println!("Input code:\n{}", test_enum_msg);
1340
1341 let mut lexer = Token::lexer(test_enum_msg);
1342 lexer.extras = 0;
1343
1344 let parsed_bitis = parse_root(&mut lexer);
1345 if let Err(s) = parsed_bitis.clone() {
1346 panic!("Error: {} ('{}' ,span: {:?})", s.0, &test_enum_msg[s.1.clone()], s.1);
1347 }
1348 assert_eq!(parsed_bitis.is_ok(), true);
1349
1350 let parsed_bitis = parsed_bitis.unwrap();
1351 assert_eq!(parsed_bitis.len(), 1);
1352
1353 let processed_bitis = process_and_validate_bitis(&parsed_bitis);
1354 let rdo = RustDataObjects{ d: JinjaData{ enums: processed_bitis.enums,
1355 msgs: to_rust_messages(&processed_bitis.msgs),
1356 oos: to_rust_oneofs(&processed_bitis.oo_enums, &processed_bitis.msgs) } };
1357
1358 let rendered = rdo.render().unwrap();
1359 let testoo_commment = "/// comment for Oneof\n";
1360 let testoo_enum = "pub enum OO_TestOo_OoLi {\n Test1(VarWithGivenBitSize<u8, 3>),\n Test2(f32),\n}\n\n";
1361 let testoo_msg = "pub struct TestOo {\n pub oo_li: OO_TestOo_OoLi,\n pub b1: bool,\n}\n";
1362 println!("*rendered:\n{}",rendered);
1363 assert_eq!(rendered, (HEADER.to_owned() + ENUMS_HEADER + "\n\n" + OO_HEADER + PER_OO_HEADER
1364 + testoo_enum + MSG_HEADER + testoo_commment + PER_MSG_HEADER + testoo_msg).to_string());
1365 }
1366}
1367
1368#[cfg(test)]
1369mod bitis_compile {
1370 use std::fs;
1371 use std::path::Path;
1372 use rstest::rstest;
1373 use super::*;
1374
1375 fn compile(content: &str) -> BitisProcessed {
1376 let mut lexer = Token::lexer(content);
1377 lexer.extras = 0;
1378 println!("*** content:\n{}", content);
1379 let bitis_parsed = match parse_root(&mut lexer) {
1380 Ok(v) => v,
1381 Err(e) => {
1382 let (err_str, err_span) = e.clone();
1383 let content_err = &content[err_span];
1384 println!("Error: {}\n -> Source: '{}'", err_str, content_err);
1385 abort()
1386 }
1387 };
1388 println!("** content:\n{:?}", bitis_parsed);
1389 process_and_validate_bitis(&bitis_parsed)
1390 }
1391 fn render(d: JinjaData) {
1392 let rdo = RustDataObjects{ d: d.clone() };
1393 let rendered_rust = rdo.render().unwrap();
1394 println!("*** rendered DO:\n{}", rendered_rust);
1395 fs::write(Path::new("./test_data/test_py/bitis/src/messages_test.rs"), rendered_rust).expect("Unable to write file");
1396
1397 let rdo = RustPyDataObjects{ d: d.clone() };
1398 let rendered_rust = rdo.render().unwrap();
1399 println!("*** rendered PyDO:\n{}", rendered_rust);
1400 fs::write(Path::new("./test_data/test_py/bitis/src/pyrust_test.rs"), rendered_rust).expect("Unable to write file");
1401
1402 let rdo = RustPyLib{ d: d.clone(), lib_name: "bitis_msgs".into() };
1403 let rendered_rust = rdo.render().unwrap();
1404 println!("*** rendered pyLib:\n{}", rendered_rust);
1405 fs::write(Path::new("./test_data/test_py/bitis/src/lib_test.rs"), rendered_rust).expect("Unable to write file");
1406
1407 let rdo = PyTypeHints{ d };
1408 let rendered_rust = rdo.render().unwrap();
1409 println!("*** rendered py_type_hints:\n{}", rendered_rust);
1410 fs::write(Path::new("./test_data/test_py/bitis/bitis_msgs/bitis_msgs.pyi"), rendered_rust).expect("Unable to write file");
1411 }
1412
1413 #[rstest]
1414 #[ignore]
1415 fn simple_rust_py() {
1416 let bitis_str = "msg ParamTestSimple { uint_4 param_1; bool param_2; }";
1417
1418 let bitis_processed_org = compile(bitis_str);
1419
1420 let bitis_processed = bitis_processed_org.clone();
1421 let d = JinjaData{
1422 enums: bitis_processed.enums,
1423 msgs: to_rust_messages(&bitis_processed.msgs),
1424 oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1425 };
1426 render(d);
1427 }
1428
1429 #[rstest]
1430 #[ignore]
1431 fn nested_rust_py() {
1432 let bitis_str = "msg Inner { uint_2 val; }\nmsg ParamTestWithInner { uint_4 param_1; bool param_2; Inner inner; }";
1433
1434 let bitis_processed_org = compile(bitis_str);
1435
1436 let bitis_processed = bitis_processed_org.clone();
1437
1438 let d = JinjaData{
1439 enums: bitis_processed.enums,
1440 msgs: to_rust_messages(&bitis_processed.msgs),
1441 oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1442 };
1443 render(d);
1444 }
1445 #[test]
1446 #[ignore]
1447 fn nested_and_enum_rust_py() {
1448 let bitis_str = [
1449 "enum Numbers(4) { one, two, three, four }\n/// Test comment for Inner\nmsg Inner { uint_3 val; Numbers num; }\n",
1450 "msg ParamTestWithInner { uint_4 param_1; bool param_2; Inner inner; } }"
1451 ].join("");
1452
1453 let bitis_processed_org = compile(bitis_str.as_str());
1454
1455 let bitis_processed = bitis_processed_org.clone();
1456
1457 let d = JinjaData{
1458 enums: bitis_processed.enums,
1459 msgs: to_rust_messages(&bitis_processed.msgs),
1460 oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1461 };
1462 render(d);
1463 }
1464 #[test]
1465 fn oneof_nested_and_enum_rust_py() {
1466 let bitis_str = [
1467 "//| 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",
1468 "msg ParamTestWithInner { uint_4 param_1; bool param_2; oneof action(4) { Inner inner; uint_3 val; } }"
1469 ].join("");
1470
1471 let bitis_processed_org = compile(bitis_str.as_str());
1472
1473 let bitis_processed = bitis_processed_org.clone();
1474
1475 let d = JinjaData{
1476 enums: bitis_processed.enums,
1477 msgs: to_rust_messages(&bitis_processed.msgs),
1478 oos: to_rust_oneofs(&bitis_processed.oo_enums, &bitis_processed.msgs)
1479 };
1480 render(d);
1481 }
1482}
1483
1484#[cfg(test)]
1485mod bitis_serialization {
1486 use rstest::rstest;
1488 use super::*;
1489
1490 #[rstest]
1492 fn msg_simple_msg_compile() {
1493 let test_empty_msg = "msg Lala { repeated_fixed_10 bool data_bool; uint_4 data1_uint; uint_12 data2_uint; }";
1494
1495 let mut lexer = Token::lexer(test_empty_msg);
1496 lexer.extras = 0;
1497
1498 let parsed_bitis = parse_root(&mut lexer);
1499 assert_eq!(parsed_bitis.is_ok(), true);
1500
1501 let _parsed_bitis = parsed_bitis.unwrap();
1502
1503 }
1528}
1529
1530#[cfg(test)]
1531mod bitis_processing {
1532 use rstest::rstest;
1533 use crate::AttributeDetails::{AttributeEnumOrMsg, AttributeSimple};
1534 use super::*;
1535
1536 #[rstest]
1537 #[ignore]
1538 fn msg_base_and_v2() {
1539 let bitis_values = vec![
1540 Value::Message(Message{
1541 name: "TestMsg".to_string(),
1542 comment: Some("This is a test".to_string()),
1544 parent: None,
1545 attributes: vec![Attribute{name: "a1".to_string(), comment: None,
1546 is_repeated_and_size: None, is_optional: false,
1547 specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1548 }],
1549 }),
1550 Value::Message(Message{
1551 name: "TestMsg".to_string(),
1552 comment: Some("This is a test".to_string()),
1554 parent: None,
1555 attributes: vec![Attribute{name: "a2".to_string(), comment: None,
1556 is_repeated_and_size: None, is_optional: false,
1557 specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1558 }],
1559 })
1560 ];
1561 let pb = process_and_validate_bitis(&bitis_values);
1562
1563 assert_eq!(pb.max_version_number, 2);
1564 assert_eq!(pb.msgs.len(), 3);
1565
1566 assert_eq!(pb.msgs[0].name, "TestMsg_Base".to_string());
1567 assert_eq!(pb.msgs[1].name, "TestMsg_V1".to_string());
1568 assert_eq!(pb.msgs[2].name, "TestMsg_V2".to_string());
1569
1570 assert_eq!(pb.msgs[0].attributes.len(), 1);
1571 assert_eq!(pb.msgs[0].attributes.get(0).unwrap().name, "a1".to_string());
1572 assert_eq!(pb.msgs[1].attributes.len(), 0);
1573 assert_eq!(pb.msgs[2].attributes.len(), 1);
1574 assert_eq!(pb.msgs[2].attributes.get(0).unwrap().name, "a2".to_string());
1575 }
1576
1577 #[rstest]
1578 #[ignore]
1579 fn msg_base_and_v2_and_add_msg() {
1580 let bitis_values = vec![
1581 Value::Message(Message{
1582 name: "TestMsgInner".to_string(),
1583 comment: Some("This is a test2".to_string()),
1585 parent: None,
1586 attributes: vec![Attribute{name: "lala".to_string(), comment: None,
1587 is_repeated_and_size: None, is_optional: false,
1588 specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1589 }],
1590 }),
1591 Value::Message(Message{
1592 name: "TestMsgInner".to_string(),
1593 comment: Some("This is a test2".to_string()),
1595 parent: None,
1596 attributes: vec![
1597 Attribute{name: "lala".to_string(), comment: None, is_repeated_and_size: None, is_optional: false,
1598 specific_details: AttributeSimple(SimpleType::UIntFixed(4)),},
1599 Attribute{name: "lala2".to_string(), comment: None, is_repeated_and_size: None, is_optional: false,
1600 specific_details: AttributeSimple(SimpleType::UIntFixed(3)),},
1601 ],
1602 }),
1603 Value::Message(Message{
1604 name: "TestMsg".to_string(),
1605 comment: Some("This is a test".to_string()),
1607 parent: None,
1608 attributes: vec![
1609 Attribute{ name: "a1".to_string(), comment: None, is_repeated_and_size: None, is_optional: false,
1610 specific_details: AttributeSimple(SimpleType::UIntFixed(4)) },
1611 Attribute{ name: "lala_use".to_string(), comment: None, is_repeated_and_size: None, is_optional: false,
1612 specific_details: AttributeEnumOrMsg("TestMsgInner".to_string()) },
1613 ],
1614 }),
1615 Value::Message(Message{
1616 name: "TestMsg".to_string(),
1617 comment: Some("This isa test".to_string()),
1619 parent: None,
1620 attributes: vec![Attribute{name: "a2".to_string(), comment: None,
1621 is_repeated_and_size: None, is_optional: false,
1622 specific_details: AttributeSimple(SimpleType::UIntFixed(4)),
1623 }],
1624 }),
1625 ];
1626 let pb = process_and_validate_bitis(&bitis_values);
1627
1628 assert_eq!(pb.max_version_number, 2);
1629 assert_eq!(pb.msgs.len(), 4);
1630
1631}
1646}
1647