rhailib/examples/access_control/main.rs

use rhai_client::RhaiClientBuilder;
use rhailib_worker::spawn_rhai_worker;
use std::time::Duration;
use tempfile::Builder;
use tokio::sync::mpsc;

const ALICE_ID: &str = "alice_pk";
const BOB_ID: &str = "bob_pk";
const CHARLIE_ID: &str = "charlie_pk";
const CIRCLE_ID: &str = "circle_pk";
const REDIS_URL: &str = "redis://127.0.0.1/";

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();

    // Create a temporary directory for the database
    let temp_dir = Builder::new().prefix("rhai-example").tempdir()?;
    let db_path = temp_dir.path().to_str().unwrap().to_string();

    // 1. Create a Rhai engine and register custom functionality
    let engine = rhailib_engine::create_heromodels_engine();

    // 2. Spawn the Rhai worker
    let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
    let worker_handle = tokio::spawn(spawn_rhai_worker(
        ALICE_ID.to_string(),
        db_path.clone(),
        engine,
        REDIS_URL.to_string(),
        shutdown_rx,
        false, // use_sentinel
    ));
    
    log::info!("Rhai worker spawned for circle: {}", ALICE_ID);

    // Give the worker a moment to start up
    tokio::time::sleep(Duration::from_secs(1)).await;

    // Alice populates her rhai worker
    let client_alice = RhaiClientBuilder::new()
        .redis_url(REDIS_URL)
        .caller_id(ALICE_ID)
        .build()
        .unwrap();

    client_alice.new_play_request()
        .recipient_id(&ALICE_ID)
        .script_path("examples/access_control/alice.rhai")
        .timeout(Duration::from_secs(10))   
        .await_response().await.unwrap();
    
    log::info!("Alice's database populated.");

    // Bob queries Alice's rhai worker
    let client_bob = RhaiClientBuilder::new()
        .redis_url(REDIS_URL)
        .caller_id(BOB_ID)
        .build()
        .unwrap();
    
    client_bob.new_play_request()
        .recipient_id(&ALICE_ID)
        .script_path("examples/access_control/bob.rhai")
        .timeout(Duration::from_secs(10))   
        .await_response().await.unwrap();
    
    log::info!("Bob's query to Alice's database completed.");

    // Charlie queries Alice's rhai worker
    let client_charlie = RhaiClientBuilder::new()
        .redis_url(REDIS_URL)
        .caller_id(CHARLIE_ID)
        .build()
        .unwrap();
    
    client_charlie.new_play_request()
        .recipient_id(&ALICE_ID)
        .script_path("examples/access_control/charlie.rhai")
        .timeout(Duration::from_secs(10))   
        .await_response().await.unwrap();
    
    log::info!("Charlie's query to Alice's database completed.");

    
    // Spawn the Rhai worker for Alice's and Charlie's circle
    let engine = rhailib_engine::create_heromodels_engine();
    let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
    let worker_handle = tokio::spawn(spawn_rhai_worker(
        CIRCLE_ID.to_string(),
        db_path.clone(),
        engine,
        REDIS_URL.to_string(),
        shutdown_rx,
        false, // use_sentinel
    ));

    // Alice populates the rhai worker of their circle with Charlie.
    let client_circle = RhaiClientBuilder::new()
        .redis_url(REDIS_URL)
        .caller_id(CIRCLE_ID)
        .build()
        .unwrap();

    client_circle.new_play_request()
        .recipient_id(&CIRCLE_ID)
        .script_path("examples/access_control/circle.rhai")
        .timeout(Duration::from_secs(10))   
        .await_response().await.unwrap();

    log::info!("Circles's database populated.");

// Give the worker a moment to start up
tokio::time::sleep(Duration::from_secs(1)).await;

// Alice queries the rhai worker of their circle with Charlie.
client_alice.new_play_request()
    .recipient_id(&CIRCLE_ID)
    .script_path("examples/access_control/alice.rhai")
    .timeout(Duration::from_secs(10))   
    .await_response().await.unwrap();

log::info!("Bob's query to Alice's database completed.");

// Charlie queries Alice's rhai worker
let client_charlie = RhaiClientBuilder::new()
    .redis_url(REDIS_URL)
    .caller_id(CHARLIE_ID)
    .build()
    .unwrap();

client_charlie.new_play_request()
    .recipient_id(&ALICE_ID)
    .script_path("examples/access_control/charlie.rhai")
    .timeout(Duration::from_secs(10))   
    .await_response().await.unwrap();

log::info!("Charlie's query to Alice's database completed.");


    // 5. Shutdown the worker (optional, could also let it run until program exits)
    log::info!("Signaling worker to shutdown...");
    let _ = shutdown_tx.send(()).await;
    if let Err(e) = worker_handle.await {
        log::error!("Worker task panicked or encountered an error: {:?}", e);
    }
    log::info!("Worker shutdown complete.");

    Ok(())
}