nyanoblog/src/data/account/pg.rs

use sqlx::PgPool;

use crate::core::*;
use crate::model::account::{Account, NewAccount};

pub struct PgAccountDataSource {
    pool: PgPool,
}

impl PgAccountDataSource {
    pub fn new(pool: PgPool) -> PgAccountDataSource {
        PgAccountDataSource { pool }
    }

    pub async fn by_id<I>(&self, id: I) -> Result<Account>
    where
        I: Into<Id> + Send,
    {
        let id: Id = id.into();
        let account: Account = sqlx::query_as("SELECT * FROM accounts WHERE id = $1")
            .bind(id)
            .fetch_one(&self.pool)
            .await?;
        Ok(account)
    }

    pub async fn create<T>(&self, new: T) -> Result<Account>
    where
        T: Into<NewAccount> + Send,
    {
        let new = new.into();
        let account: Account = sqlx::query_as(
            "INSERT INTO accounts (iri, name, domain, display_name)
            VALUES ($1, $2, $3, $4)
            RETURNING *",
        )
        .bind(new.iri)
        .bind(new.name)
        .bind(new.domain)
        .bind(new.display_name)
        .fetch_one(&self.pool)
        .await?;
        Ok(account)
    }
}