49a20a303a
* initial migration * barebones profiles * Finish profiles * Add back file watcher * UI support progress * Finish most of cache * Fix options page * Fix forge, finish modrinth auth * Accounts, process cache * Run SQLX prepare * Finish * Run lint + actions * Fix version to be compat with windows * fix lint * actually fix lint * actually fix lint again
75 lines
2.0 KiB
Rust
75 lines
2.0 KiB
Rust
//! Authentication flow interface
|
|
|
|
use crate::state::{Credentials, MinecraftLoginFlow};
|
|
use crate::State;
|
|
|
|
#[tracing::instrument]
|
|
pub async fn begin_login() -> crate::Result<MinecraftLoginFlow> {
|
|
let state = State::get().await?;
|
|
|
|
crate::state::login_begin(&state.pool).await
|
|
}
|
|
|
|
#[tracing::instrument]
|
|
pub async fn finish_login(
|
|
code: &str,
|
|
flow: MinecraftLoginFlow,
|
|
) -> crate::Result<Credentials> {
|
|
let state = State::get().await?;
|
|
|
|
crate::state::login_finish(code, flow, &state.pool).await
|
|
}
|
|
|
|
#[tracing::instrument]
|
|
pub async fn get_default_user() -> crate::Result<Option<uuid::Uuid>> {
|
|
let state = State::get().await?;
|
|
let users = Credentials::get_active(&state.pool).await?;
|
|
Ok(users.map(|x| x.id))
|
|
}
|
|
|
|
#[tracing::instrument]
|
|
pub async fn set_default_user(user: uuid::Uuid) -> crate::Result<()> {
|
|
let state = State::get().await?;
|
|
let users = Credentials::get_all(&state.pool).await?;
|
|
let (_, mut user) = users.remove(&user).ok_or_else(|| {
|
|
crate::ErrorKind::OtherError(format!(
|
|
"Tried to get nonexistent user with ID {user}"
|
|
))
|
|
.as_error()
|
|
})?;
|
|
|
|
user.active = true;
|
|
user.upsert(&state.pool).await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Remove a user account from the database
|
|
#[tracing::instrument]
|
|
pub async fn remove_user(uuid: uuid::Uuid) -> crate::Result<()> {
|
|
let state = State::get().await?;
|
|
|
|
let users = Credentials::get_all(&state.pool).await?;
|
|
|
|
if let Some((uuid, user)) = users.remove(&uuid) {
|
|
Credentials::remove(uuid, &state.pool).await?;
|
|
|
|
if user.active {
|
|
if let Some((_, mut user)) = users.into_iter().next() {
|
|
user.active = true;
|
|
user.upsert(&state.pool).await?;
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Get a copy of the list of all user credentials
|
|
#[tracing::instrument]
|
|
pub async fn users() -> crate::Result<Vec<Credentials>> {
|
|
let state = State::get().await?;
|
|
let users = Credentials::get_all(&state.pool).await?;
|
|
Ok(users.into_iter().map(|x| x.1).collect())
|
|
}
|