use crate::data_storage::DataStorage;
use oca_ast::ast::{self, RefValue};
use oca_bundle::state::oca::{OCABox, OCABundle};
use said::version::Encode;
pub fn build_oca(
db: Box<dyn DataStorage>,
commands: Vec<ast::Command>,
) -> Result<OCABundle, String> {
let mut base: Option<OCABox> = None;
for command in commands {
if let ast::CommandType::From = command.kind {
let said = match command.clone().object_kind.oca_bundle_content() {
Some(oca_bundle_content) => match oca_bundle_content.clone().said {
ast::ReferenceAttrType::Reference(refs) => match refs {
RefValue::Said(said) => said.to_string(),
RefValue::Name(_) => return Err("Not implemented".to_string()),
},
},
None => return Err("Missing bundle content".to_string()),
};
let oca_bundle_str = match db.get(&format!("oca.{}", said))? {
Some(oca_bundle_str) => String::from_utf8(oca_bundle_str).unwrap(),
None => return Err("OCA not found".to_string()),
};
let oca_b = serde_json::from_str::<OCABundle>(&oca_bundle_str).unwrap();
base = Some(oca_b.into());
} else {
let result = oca_bundle::build::apply_command(base.clone(), command.clone());
let mut oca_box: OCABox = OCABox::new();
match result {
Ok(oca) => {
oca_box = oca;
}
Err(errors) => {
println!("{:?}", errors);
}
};
let oca_bundle = oca_box.generate_bundle();
let command_str = serde_json::to_string(&command).unwrap();
let mut input: Vec<u8> = vec![];
match base {
Some(ref mut base) => {
let base_said = base.generate_bundle().said.unwrap().to_string();
input.push(base_said.as_bytes().len().try_into().unwrap());
input.extend(base_said.as_bytes());
}
None => {
input.push(0);
}
}
input.push(command_str.as_bytes().len().try_into().unwrap());
input.extend(command_str.as_bytes());
db.insert(
&format!("oca.{}.operation", oca_bundle.clone().said.unwrap()),
&input,
)?;
db.insert(
&format!("oca.{}", oca_bundle.clone().said.unwrap()),
&oca_bundle.encode().unwrap(),
)?;
base = Some(oca_box);
}
}
Ok(base.unwrap().generate_bundle())
}
#[cfg(test)]
mod tests {
use std::str::FromStr;
use super::*;
use crate::data_storage::{DataStorage, SledDataStorage};
use indexmap::IndexMap;
use oca_ast::ast::{BundleContent, CaptureContent, Content};
use said::SelfAddressingIdentifier;
#[test]
#[ignore]
fn test_ocafile_build() {
let mut commands = vec![];
let mut attributes = IndexMap::new();
attributes.insert(
"d".to_string(),
ast::NestedAttrType::Value(ast::AttributeType::Text),
);
attributes.insert(
"i".to_string(),
ast::NestedAttrType::Value(ast::AttributeType::Text),
);
attributes.insert(
"passed".to_string(),
ast::NestedAttrType::Value(ast::AttributeType::Boolean),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::CaptureBase(CaptureContent {
attributes: Some(attributes),
properties: None,
flagged_attributes: None,
}),
});
let mut properties = IndexMap::new();
properties.insert(
"lang".to_string(),
ast::NestedValue::Value("en".to_string()),
);
properties.insert(
"name".to_string(),
ast::NestedValue::Value("Entrance credential".to_string()),
);
properties.insert(
"description".to_string(),
ast::NestedValue::Value("Entrance credential".to_string()),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::Overlay(
ast::OverlayType::Meta,
Content {
attributes: None,
properties: Some(properties),
},
),
});
let mut attributes = IndexMap::new();
attributes.insert(
"d".to_string(),
ast::NestedValue::Value("Schema digest".to_string()),
);
attributes.insert(
"i".to_string(),
ast::NestedValue::Value("Credential Issuee".to_string()),
);
attributes.insert(
"passed".to_string(),
ast::NestedValue::Value("Passed".to_string()),
);
let mut properties = IndexMap::new();
properties.insert(
"lang".to_string(),
ast::NestedValue::Value("en".to_string()),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::Overlay(
ast::OverlayType::Label,
Content {
attributes: Some(attributes),
properties: Some(properties),
},
),
});
let mut attributes = IndexMap::new();
attributes.insert(
"d".to_string(),
ast::NestedValue::Value("Schema digest".to_string()),
);
attributes.insert(
"i".to_string(),
ast::NestedValue::Value("Credential Issuee".to_string()),
);
attributes.insert(
"passed".to_string(),
ast::NestedValue::Value("Enables or disables passing".to_string()),
);
let mut properties = IndexMap::new();
properties.insert(
"lang".to_string(),
ast::NestedValue::Value("en".to_string()),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::Overlay(
ast::OverlayType::Information,
Content {
attributes: Some(attributes),
properties: Some(properties),
},
),
});
let mut attributes = IndexMap::new();
attributes.insert(
"d".to_string(),
ast::NestedValue::Value("utf-8".to_string()),
);
attributes.insert(
"i".to_string(),
ast::NestedValue::Value("utf-8".to_string()),
);
attributes.insert(
"passed".to_string(),
ast::NestedValue::Value("utf-8".to_string()),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::Overlay(
ast::OverlayType::CharacterEncoding,
Content {
attributes: Some(attributes),
properties: None,
},
),
});
let mut attributes = IndexMap::new();
attributes.insert("d".to_string(), ast::NestedValue::Value("M".to_string()));
attributes.insert("i".to_string(), ast::NestedValue::Value("M".to_string()));
attributes.insert(
"passed".to_string(),
ast::NestedValue::Value("M".to_string()),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::Overlay(
ast::OverlayType::Conformance,
Content {
attributes: Some(attributes),
properties: None,
},
),
});
let db = SledDataStorage::open("db_test");
let oca = build_oca(Box::new(db), commands);
let db_read = SledDataStorage::open("db_test");
let op = db_read
.get(&format!("oca.{}.operation", oca.unwrap().said.unwrap()))
.unwrap();
println!("{:?}", String::from_utf8_lossy(&op.unwrap()));
}
#[test]
#[ignore]
fn test_ocafile_build_from() {
let mut commands = vec![];
let said =
SelfAddressingIdentifier::from_str("EF5ERATRBBN_ewEo9buQbznirhBmvrSSC0O2GIR4Gbfs")
.unwrap();
let reference = ast::ReferenceAttrType::Reference(RefValue::Said(said));
commands.push(ast::Command {
kind: ast::CommandType::From,
object_kind: ast::ObjectKind::OCABundle(BundleContent { said: reference }),
});
let mut attributes = IndexMap::new();
attributes.insert(
"new".to_string(),
ast::NestedAttrType::Value(ast::AttributeType::Text),
);
commands.push(ast::Command {
kind: ast::CommandType::Add,
object_kind: ast::ObjectKind::CaptureBase(CaptureContent {
attributes: Some(attributes),
properties: None,
flagged_attributes: None,
}),
});
let db = SledDataStorage::open("db_test");
let oca = build_oca(Box::new(db), commands);
println!(
"{:?}",
String::from_utf8(oca.clone().unwrap().encode().unwrap())
);
let db_read = SledDataStorage::open("db_test");
let op = db_read
.get(&format!("oca.{}.operation", oca.unwrap().said.unwrap()))
.unwrap();
println!("{:?}", String::from_utf8_lossy(&op.unwrap()));
}
}