feat(oca-bundle): add support for flagged attributes

THCLab · Jan 11, 2024 · f19ba4d · f19ba4d
1 parent 01f1fdc
commit f19ba4d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/oca-bundle/src/build.rs b/oca-bundle/src/build.rs
@@ -101,7 +101,6 @@ pub fn from_ast(
                 }
             }
             Err(mut err) => {
-                println!("Error applying command: {:?}", err);
                 errors.extend(err.iter_mut().map(|e| Error::FromASTError {
                     line_number: command_meta.line_number,
                     raw_line: command_meta.raw_line.clone(),
@@ -515,6 +514,7 @@ mod tests {
             object_kind: ast::ObjectKind::CaptureBase(CaptureContent {
                 attributes: Some(attributes),
                 properties: None,
+                flagged_attributes: None,
             }),
         });
 
@@ -676,11 +676,14 @@ mod tests {
             "passed".to_string(),
             ast::NestedAttrType::Value(AttributeType::Boolean),
         );
+
+        let flagged_attributes = vec!["d".to_string(), "i".to_string()];
         commands.push(ast::Command {
             kind: ast::CommandType::Add,
             object_kind: ast::ObjectKind::CaptureBase(ast::CaptureContent {
                 attributes: Some(attributes),
                 properties: None,
+                flagged_attributes: Some(flagged_attributes.clone()),
             }),
         });
 

diff --git a/oca-bundle/src/state/oca.rs b/oca-bundle/src/state/oca.rs
@@ -891,6 +891,7 @@ impl OCABundle {
                 // TODO find out if we can use indexmap in capture base to simplify stuff
                 attributes: Some(self.capture_base.attributes.clone().into_iter().collect()),
                 properties,
+                flagged_attributes: None,
             }),
         };
         ast.commands.push(command);