use heromodels::db::{Collection, Db};
use heromodels_core::{BaseModelData, Model};
use heromodels_derive::model;
use serde::{Deserialize, Serialize};

// Basic usage
#[derive(Debug, Clone, Serialize, Deserialize)]
#[model]
pub struct SimpleUser {
    pub base_data: BaseModelData,

    #[index]
    pub login: String,

    pub full_name: String,
}

// With customization options
#[derive(Debug, Clone, Serialize, Deserialize)]
#[model]
pub struct CustomUser {
    pub base_data: BaseModelData,

    #[index(name = "user_name")]
    pub login_name: String,

    #[index]
    pub is_active: bool,

    pub full_name: String,
}

fn main() {
    println!("Hero Models - Model Macro Example");
    println!("=================================");

    // Create a new DB instance, reset before every run
    let db_path = "/tmp/ourdb_model_macro_example";
    let db = heromodels::db::hero::OurDB::new(db_path, true).expect("Can create DB");

    // Example usage of the generated implementations
    println!("SimpleUser DB Prefix: {}", SimpleUser::db_prefix());
    println!("CustomUser DB Prefix: {}", CustomUser::db_prefix());

    let user = SimpleUser {
        base_data: BaseModelData::new(), // ID will be auto-generated by OurDB
        login: "johndoe".to_string(),
        full_name: "John Doe".to_string(),
    };

    let custom_user = CustomUser {
        base_data: BaseModelData::new(), // ID will be auto-generated by OurDB
        login_name: "janesmith".to_string(),
        is_active: true,
        full_name: "Jane Smith".to_string(),
    };

    println!("\nBefore saving - SimpleUser ID: {}", user.get_id());
    println!("Before saving - SimpleUser DB Keys: {:?}", user.db_keys());

    println!("\nBefore saving - CustomUser ID: {}", custom_user.get_id());
    println!(
        "Before saving - CustomUser DB Keys: {:?}",
        custom_user.db_keys()
    );

    // Save the models to the database
    let simple_collection = db
        .collection::<SimpleUser>()
        .expect("can open simple user collection");
    let custom_collection = db
        .collection::<CustomUser>()
        .expect("can open custom user collection");

    let (user_id, saved_user) = simple_collection.set(&user).expect("can save simple user");
    let (custom_user_id, saved_custom_user) = custom_collection
        .set(&custom_user)
        .expect("can save custom user");

    println!("\nAfter saving - SimpleUser ID: {}", saved_user.get_id());
    println!(
        "After saving - SimpleUser DB Keys: {:?}",
        saved_user.db_keys()
    );
    println!("Returned SimpleUser ID: {}", user_id);

    println!(
        "\nAfter saving - CustomUser ID: {}",
        saved_custom_user.get_id()
    );
    println!(
        "After saving - CustomUser DB Keys: {:?}",
        saved_custom_user.db_keys()
    );
    println!("Returned CustomUser ID: {}", custom_user_id);

    // Verify that the IDs were auto-generated
    assert_eq!(saved_user.get_id(), user_id);
    assert_ne!(saved_user.get_id(), 0);
    assert_eq!(saved_custom_user.get_id(), custom_user_id);
    assert_ne!(saved_custom_user.get_id(), 0);

    println!("\nExample finished. DB stored at {}", db_path);
    println!(
        "To clean up, you can manually delete the directory: {}",
        db_path
    );
}