[dependencies]
turso = { version = "=0.8.0-pre.11", default-features = false }

use golem_rust::{agent_definition, agent_implementation};
 
#[agent_definition(ephemeral)]
pub trait TursoAgent {
    fn new(name: String) -> Self;
    async fn round_trip(&self, value: String) -> String;
}
 
struct TursoAgentImpl;
 
#[agent_implementation]
impl TursoAgent for TursoAgentImpl {
    fn new(_name: String) -> Self {
        Self
    }
 
    async fn round_trip(&self, value: String) -> String {
        let database = turso::Builder::new_local(":memory:")
            .with_io(turso::IoBackend::Memory)
            .build()
            .await
            .expect("failed to create in-memory Turso database");
        let connection = database.connect().expect("failed to connect to Turso");
 
        connection
            .execute(
                "CREATE TABLE messages (id INTEGER PRIMARY KEY, value TEXT NOT NULL)",
                (),
            )
            .await
            .expect("failed to create table");
        connection
            .execute("INSERT INTO messages (value) VALUES (?1)", [value])
            .await
            .expect("failed to insert row");
 
        let mut rows = connection
            .query("SELECT value FROM messages WHERE id = 1", ())
            .await
            .expect("failed to query row");
        let row = rows
            .next()
            .await
            .expect("failed to read row")
            .expect("query returned no row");
        row.get(0).expect("failed to decode value")
    }
}