candidate.rs

use chrono::{Local, NaiveDateTime};
use sqlx::{Result, SqliteExecutor, SqlitePool};

#[tracing::instrument(skip(ids, pool))]
pub async fn add(ids: impl Iterator<Item = &str>, pool: &SqlitePool) -> Result<()> {
    let mut transaction = pool.begin().await?;

    for id in ids {
        sqlx::query!("INSERT INTO candidates (id) VALUES($1)", id)
            .execute(&mut *transaction)
            .await?;
    }

    transaction.commit().await
}

pub struct Candidate {
    pub id: String,
    pub verification_time: Option<NaiveDateTime>,
}

#[tracing::instrument(skip(executor))]
pub async fn get(id: &str, executor: impl SqliteExecutor<'_>) -> Result<Option<Candidate>> {
    sqlx::query_as!(
        Candidate,
        r#"
            SELECT id, verification_time as "verification_time: NaiveDateTime"
            FROM candidates
            WHERE id = $1 COLLATE NOCASE
        "#,
        id
    )
    .fetch_optional(executor)
    .await
}

#[tracing::instrument(skip(executor))]
pub async fn verify(id: &str, executor: impl SqliteExecutor<'_>) -> Result<()> {
    let now = Local::now().naive_local();

    sqlx::query!(
        r#"
            UPDATE candidates
            SET verification_time = $2
            WHERE id = $1 COLLATE NOCASE
        "#,
        id,
        now
    )
    .execute(executor)
    .await?;

    Ok(())
}

#[tracing::instrument(skip(executor))]
pub async fn delete(id: &str, executor: impl SqliteExecutor<'_>) -> Result<()> {
    sqlx::query!("DELETE FROM candidates WHERE id = $1 COLLATE NOCASE", id)
        .execute(executor)
        .await?;

    Ok(())
}