use rustfs_notify::arn::TargetID;
use rustfs_notify::global::notification_system;
use rustfs_notify::{init_logger, BucketNotificationConfig, Event, EventName, LogLevel, NotificationError};
use std::time::Duration;
use tracing::info;

#[tokio::main]
async fn main() -> Result<(), NotificationError> {
    init_logger(LogLevel::Debug);

    let system = notification_system();

    // --- Initial configuration (Webhook and MQTT) ---
    let mut config = rustfs_notify::Config::new();
    let current_root = rustfs_utils::dirs::get_project_root().expect("failed to get project root");
    println!("Current project root: {}", current_root.display());
    // Webhook target configuration
    let mut webhook_kvs = rustfs_notify::KVS::new();
    webhook_kvs.set("enable", "on");
    webhook_kvs.set("endpoint", "http://127.0.0.1:3020/webhook");
    webhook_kvs.set("auth_token", "secret-token");
    // webhook_kvs.set("queue_dir", "/tmp/data/webhook");
    webhook_kvs.set(
        "queue_dir",
        current_root
            .clone()
            .join("../../deploy/logs/notify/webhook")
            .to_str()
            .unwrap()
            .to_string(),
    );
    webhook_kvs.set("queue_limit", "10000");
    let mut webhook_targets = std::collections::HashMap::new();
    webhook_targets.insert("1".to_string(), webhook_kvs);
    config.insert("notify_webhook".to_string(), webhook_targets);

    // MQTT target configuration
    let mut mqtt_kvs = rustfs_notify::KVS::new();
    mqtt_kvs.set("enable", "on");
    mqtt_kvs.set("broker", "mqtt://localhost:1883");
    mqtt_kvs.set("topic", "rustfs/events");
    mqtt_kvs.set("qos", "1"); // AtLeastOnce
    mqtt_kvs.set("username", "test");
    mqtt_kvs.set("password", "123456");
    // webhook_kvs.set("queue_dir", "/tmp/data/mqtt");
    mqtt_kvs.set(
        "queue_dir",
        current_root
            .join("../../deploy/logs/notify/mqtt")
            .to_str()
            .unwrap()
            .to_string(),
    );
    mqtt_kvs.set("queue_limit", "10000");

    let mut mqtt_targets = std::collections::HashMap::new();
    mqtt_targets.insert("1".to_string(), mqtt_kvs);
    config.insert("notify_mqtt".to_string(), mqtt_targets);

    // Load the configuration and initialize the system
    *system.config.write().await = config;
    system.init().await?;
    info!("✅ System initialized with Webhook and MQTT targets.");

    // --- Query the currently active Target ---
    let active_targets = system.get_active_targets().await;
    info!("\n---> Currently active targets: {:?}", active_targets);
    assert_eq!(active_targets.len(), 2);

    tokio::time::sleep(Duration::from_secs(1)).await;

    // --- Exactly delete a Target (e.g. MQTT) ---
    info!("\n---> Removing MQTT target...");
    let mqtt_target_id = TargetID::new("1".to_string(), "mqtt".to_string());
    system.remove_target(&mqtt_target_id, "notify_mqtt").await?;
    info!("✅ MQTT target removed.");

    // --- Query the activity's Target again ---
    let active_targets_after_removal = system.get_active_targets().await;
    info!("\n---> Active targets after removal: {:?}", active_targets_after_removal);
    assert_eq!(active_targets_after_removal.len(), 1);
    assert_eq!(active_targets_after_removal[0].id, "1".to_string());

    // --- Send events for verification ---
    // Configure a rule to point to the Webhook and deleted MQTT
    let mut bucket_config = BucketNotificationConfig::new("us-east-1");
    bucket_config.add_rule(
        &[EventName::ObjectCreatedPut],
        "*".to_string(),
        TargetID::new("1".to_string(), "webhook".to_string()),
    );
    bucket_config.add_rule(
        &[EventName::ObjectCreatedPut],
        "*".to_string(),
        TargetID::new("1".to_string(), "mqtt".to_string()), // This rule will match, but the Target cannot be found
    );
    system.load_bucket_notification_config("my-bucket", &bucket_config).await?;

    info!("\n---> Sending an event...");
    let event = Event::new_test_event("my-bucket", "document.pdf", EventName::ObjectCreatedPut);
    system
        .send_event("my-bucket", "s3:ObjectCreated:Put", "document.pdf", event)
        .await;
    info!("✅ Event sent. Only the Webhook target should receive it. Check logs for warnings about the missing MQTT target.");

    tokio::time::sleep(Duration::from_secs(2)).await;

    info!("\nDemo completed successfully");
    Ok(())
}