use rhai::{Engine, EvalAltResult};
use rhai_client::RhaiClientBuilder;
use rhailib_engine::create_heromodels_engine;
use rhailib_worker::spawn_rhai_worker;
use std::{fs, path::Path, time::Duration};
use tokio::sync::mpsc;
use uuid::Uuid;

const ALICE_ID: &str = "alice_pk";
const BOB_ID: &str = "bob_pk";
const CHARLIE_ID: &str = "charlie_pk";
const REDIS_URL: &str = "redis://127.0.0.1/";
const DB_DIRECTORY: &str = "./db";

#[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();
    
    // 1. Create a Rhai engine and register custom functionality
    let mut 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_DIRECTORY.to_string(),
        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/end_to_end/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/end_to_end/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/end_to_end/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(())
}