You've already forked AstralRinth
93 lines
2.5 KiB
Rust
93 lines
2.5 KiB
Rust
//! Authentication flow interface
|
|
|
|
use crate::State;
|
|
use crate::state::{Credentials, MinecraftLoginFlow};
|
|
|
|
#[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 offline_auth(
|
|
name: &str
|
|
) -> crate::Result<Credentials> {
|
|
let state = State::get().await?;
|
|
crate::state::offline_auth(name, &state.pool).await
|
|
}
|
|
|
|
#[tracing::instrument]
|
|
pub async fn elyby_auth(
|
|
uuid: uuid::Uuid,
|
|
login: &str,
|
|
access_token: &str
|
|
) -> crate::Result<Credentials> {
|
|
let state = State::get().await?;
|
|
crate::state::elyby_auth(uuid, login, access_token, &state.pool).await
|
|
}
|
|
|
|
#[tracing::instrument]
|
|
pub async fn get_default_user() -> crate::Result<Option<uuid::Uuid>> {
|
|
let state = State::get().await?;
|
|
let user = Credentials::get_active(&state.pool).await?;
|
|
Ok(user.map(|user| user.offline_profile.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())
|
|
}
|