77 lines
2.2 KiB
Rust
77 lines
2.2 KiB
Rust
use crate::error::{handle_error, Error};
|
|
use sqlx::sqlite::SqliteQueryResult;
|
|
use sqlx::types::chrono::{DateTime, Utc};
|
|
use sqlx::{FromRow, SqliteExecutor};
|
|
|
|
#[derive(FromRow)]
|
|
pub struct AuthorizationCodes {
|
|
// Info
|
|
pub code: String,
|
|
pub app: String,
|
|
pub user: String,
|
|
|
|
// Timings
|
|
pub created_at: DateTime<Utc>,
|
|
pub expires_at: DateTime<Utc>,
|
|
pub used_at: Option<DateTime<Utc>>,
|
|
}
|
|
|
|
impl AuthorizationCodes {
|
|
pub async fn get_all(conn: impl SqliteExecutor<'_>) -> Result<Vec<Self>, Error> {
|
|
sqlx::query_file_as!(Self, "queries/authorization_codes/get_all.sql")
|
|
.fetch_all(conn)
|
|
.await
|
|
.map_err(handle_error)
|
|
}
|
|
|
|
pub async fn insert(
|
|
conn: impl SqliteExecutor<'_>,
|
|
code: &str,
|
|
app: &str,
|
|
user: &str,
|
|
expires_at: i64,
|
|
) -> Result<Option<()>, Error> {
|
|
let query: SqliteQueryResult = sqlx::query_file!(
|
|
"queries/authorization_codes/insert.sql",
|
|
code,
|
|
app,
|
|
user,
|
|
expires_at
|
|
)
|
|
.execute(conn)
|
|
.await
|
|
.map_err(handle_error)?;
|
|
|
|
Ok((query.rows_affected() == 1).then_some(()))
|
|
}
|
|
|
|
pub async fn get_one(conn: impl SqliteExecutor<'_>, code: &str) -> Result<Option<Self>, Error> {
|
|
sqlx::query_file_as!(Self, "queries/authorization_codes/get_one.sql", code)
|
|
.fetch_optional(conn)
|
|
.await
|
|
.map_err(handle_error)
|
|
}
|
|
|
|
pub async fn use_code(conn: impl SqliteExecutor<'_>, code: &str) -> Result<Option<()>, Error> {
|
|
let query: SqliteQueryResult =
|
|
sqlx::query_file!("queries/authorization_codes/use_code.sql", code)
|
|
.execute(conn)
|
|
.await
|
|
.map_err(handle_error)?;
|
|
|
|
Ok((query.rows_affected() == 1).then_some(()))
|
|
}
|
|
|
|
pub async fn use_all_for_user(
|
|
conn: impl SqliteExecutor<'_>,
|
|
user: &str,
|
|
) -> Result<Option<()>, Error> {
|
|
let query: SqliteQueryResult =
|
|
sqlx::query_file!("queries/authorization_codes/use_all_for_user.sql", user)
|
|
.execute(conn)
|
|
.await
|
|
.map_err(handle_error)?;
|
|
|
|
Ok((query.rows_affected() >= 1).then_some(()))
|
|
}
|
|
}
|