use std::time::SystemTime;

use pliantdb::{
    core::{
        connection::{Connection, ServerConnection},
        schema::{Collection, CollectionName, InvalidNameError, Schematic},
        Error,
    },
    local::{config::Configuration, Storage},
};
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
struct Message {
    pub timestamp: SystemTime,
    pub contents: String,
}

impl Collection for Message {
    fn collection_name() -> Result<CollectionName, InvalidNameError> {
        CollectionName::new("khonsulabs", "messages")
    }

    fn define_views(_schema: &mut Schematic) -> Result<(), Error> {
        Ok(())
    }
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let storage =
        Storage::open_local("basic.pliantdb", &Configuration::default())
            .await?;
    // Before you can create a database, you must register the schema you're
    // wanting to use.
    storage.register_schema::<Message>().await?;
    storage.create_database::<Message>("messages").await?;
    let messages = storage.database::<Message>("messages").await?;
    storage
        .create_database::<Message>("private-messages")
        .await?;
    let private_messages =
        storage.database::<Message>("private-messages").await?;

    insert_a_message(&messages, "Hello, World!").await?;
    insert_a_message(&private_messages, "Hey!").await?;

    Ok(())
}

// ANCHOR: reusable-code
async fn insert_a_message<C: Connection>(
    connection: &C,
    value: &str,
) -> anyhow::Result<()> {
    connection
        .collection::<Message>()
        .push(&Message {
            contents: String::from(value),
            timestamp: SystemTime::now(),
        })
        .await?;
    Ok(())
}
// ANCHOR_END: reusable-code