siujamo
53 lines
1.6 KiB
Rust
53 lines
1.6 KiB
Rust
mod config;
|
|
mod db;
|
|
mod validation;
|
|
|
|
use anyhow::Result;
|
|
use dialoguer::{Input, Password, theme::ColorfulTheme};
|
|
use dotenvy::dotenv;
|
|
|
|
use crate::db::RegistrationInput;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
dotenv().ok();
|
|
|
|
let config = config::AppConfig::from_env()?;
|
|
let pool = db::connect(&config).await?;
|
|
|
|
println!("Delta Force Guide 用户注册");
|
|
println!("请依次输入用户名、邮箱和密码。\n");
|
|
|
|
let registration = prompt_registration()?;
|
|
let user_id = db::register_local_user(&pool, registration).await?;
|
|
|
|
println!("\n注册成功，user_id = {user_id}");
|
|
Ok(())
|
|
}
|
|
|
|
fn prompt_registration() -> Result<RegistrationInput> {
|
|
let theme = ColorfulTheme::default();
|
|
|
|
let username = Input::<String>::with_theme(&theme)
|
|
.with_prompt("用户名")
|
|
.validate_with(|input: &String| validation::validate_username(input).map_err(str::to_owned))
|
|
.interact_text()?;
|
|
|
|
let email = Input::<String>::with_theme(&theme)
|
|
.with_prompt("电子邮箱")
|
|
.validate_with(|input: &String| validation::validate_email(input).map_err(str::to_owned))
|
|
.interact_text()?;
|
|
|
|
let password = Password::with_theme(&theme)
|
|
.with_prompt("密码")
|
|
.with_confirmation("再次输入密码", "两次输入的密码不一致")
|
|
.validate_with(|input: &String| validation::validate_password(input).map_err(str::to_owned))
|
|
.interact()?;
|
|
|
|
Ok(RegistrationInput {
|
|
username: validation::normalize_username(&username)?,
|
|
email: validation::normalize_email(&email)?,
|
|
password,
|
|
})
|
|
}
|