121 lines
5.5 KiB
Rust
121 lines
5.5 KiB
Rust
use std::path::Path;
|
|
use rhai::{Scope};
|
|
use heromodels::models::flow::{Flow, FlowStep, SignatureRequirement};
|
|
use engine::{create_heromodels_engine, eval_file};
|
|
use engine::mock_db::{create_mock_db, seed_mock_db};
|
|
use heromodels_core::Model;
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
println!("Flow Rhai Example");
|
|
println!("=================");
|
|
|
|
// Create a mock database
|
|
let db = create_mock_db();
|
|
|
|
// Seed the database with initial data
|
|
seed_mock_db(db.clone());
|
|
|
|
// Create the Rhai engine with all modules registered
|
|
let engine = create_heromodels_engine(db.clone());
|
|
|
|
// Get the path to the script
|
|
let script_path = Path::new(file!())
|
|
.parent()
|
|
.unwrap()
|
|
.join("flow_script.rhai");
|
|
|
|
println!("\nRunning script: {}", script_path.display());
|
|
println!("---------------------");
|
|
|
|
// Run the script
|
|
match eval_file(&engine, &script_path.to_string_lossy()) {
|
|
Ok(result) => {
|
|
if !result.is_unit() {
|
|
println!("\nScript returned: {:?}", result);
|
|
}
|
|
println!("\nScript executed successfully!");
|
|
},
|
|
Err(err) => {
|
|
eprintln!("\nError running script: {}", err);
|
|
return Err(Box::new(std::io::Error::new(std::io::ErrorKind::Other, err.to_string())));
|
|
}
|
|
}
|
|
|
|
// Demonstrate direct Rust interaction with the Rhai-exposed flow functionality
|
|
println!("\nDirect Rust interaction with Rhai-exposed flow functionality");
|
|
println!("----------------------------------------------------------");
|
|
|
|
// Create a new scope
|
|
let mut scope = Scope::new();
|
|
|
|
// Create a new flow using the Rhai function
|
|
let result = engine.eval::<Flow>("new_flow(0, \"Direct Rust Flow\")");
|
|
match result {
|
|
Ok(mut flow) => {
|
|
println!("Created flow from Rust: {} (ID: {})", flow.name, flow.get_id());
|
|
|
|
// Set flow status using the builder pattern
|
|
flow = flow.status("active".to_string());
|
|
println!("Set flow status to: {}", flow.status);
|
|
|
|
// Create a new flow step using the Rhai function
|
|
let result = engine.eval::<FlowStep>("new_flow_step(0, 1)");
|
|
|
|
match result {
|
|
Ok(mut step) => {
|
|
println!("Created flow step from Rust: Step Order {} (ID: {})",
|
|
step.step_order, step.get_id());
|
|
|
|
// Set step description
|
|
step = step.description("Direct Rust Step".to_string());
|
|
println!("Set step description to: {}",
|
|
step.description.clone().unwrap_or_else(|| "None".to_string()));
|
|
|
|
// Create a signature requirement using the Rhai function
|
|
let result = engine.eval::<SignatureRequirement>(
|
|
"new_signature_requirement(0, 1, \"Direct Rust Signer\", \"Please sign this document\")"
|
|
);
|
|
|
|
match result {
|
|
Ok(req) => {
|
|
println!("Created signature requirement from Rust: Public Key {} (ID: {})",
|
|
req.public_key, req.get_id());
|
|
|
|
// Add the step to the flow using the builder pattern
|
|
flow = flow.add_step(step);
|
|
println!("Added step to flow. Flow now has {} steps", flow.steps.len());
|
|
|
|
// Save the flow to the database using the Rhai function
|
|
let save_flow_script = "fn save_it(f) { return db::save_flow(f); }";
|
|
let save_flow_ast = engine.compile(save_flow_script).unwrap();
|
|
let result = engine.call_fn::<Flow>(&mut scope, &save_flow_ast, "save_it", (flow,));
|
|
match result {
|
|
Ok(saved_flow) => {
|
|
println!("Saved flow to database with ID: {}", saved_flow.get_id());
|
|
},
|
|
Err(err) => eprintln!("Error saving flow: {}", err),
|
|
}
|
|
|
|
// Save the signature requirement to the database using the Rhai function
|
|
let save_req_script = "fn save_it(r) { return db::save_signature_requirement(r); }";
|
|
let save_req_ast = engine.compile(save_req_script).unwrap();
|
|
let result = engine.call_fn::<SignatureRequirement>(&mut scope, &save_req_ast, "save_it", (req,));
|
|
match result {
|
|
Ok(saved_req) => {
|
|
println!("Saved signature requirement to database with ID: {}", saved_req.get_id());
|
|
},
|
|
Err(err) => eprintln!("Error saving signature requirement: {}", err),
|
|
}
|
|
},
|
|
Err(err) => eprintln!("Error creating signature requirement: {}", err),
|
|
}
|
|
},
|
|
Err(err) => eprintln!("Error creating flow step: {}", err),
|
|
}
|
|
},
|
|
Err(err) => eprintln!("Error creating flow: {}", err),
|
|
}
|
|
|
|
Ok(())
|
|
}
|