feat(web-service): allow custom access token with persisted port and localized start errors

- Persist user-supplied access token and last-used port in app_metadata, falling back to defaults when unset
- Atomically guard concurrent starts via compare_exchange with RAII rollback of the running flag
- Wrap token and port persistence in a single SeaORM transaction to prevent partial writes
- Classify bind errors (port in use, permission denied, address unavailable, invalid address) into stable i18n keys
- Localize start-failure messages across all 10 supported languages
This commit is contained in:
xintaofei
2026-04-18 10:18:34 +08:00
parent 32b4c88582
commit fd10494128
16 changed files with 427 additions and 67 deletions

View File

@@ -1,13 +1,13 @@
use chrono::Utc;
use sea_orm::sea_query::OnConflict;
use sea_orm::DatabaseConnection;
use sea_orm::{ConnectionTrait, DatabaseConnection};
use sea_orm::{ActiveValue::NotSet, ColumnTrait, EntityTrait, QueryFilter, Set};
use crate::db::entities::app_metadata;
use crate::db::error::DbError;
pub async fn upsert_value(
conn: &DatabaseConnection,
pub async fn upsert_value<C: ConnectionTrait>(
conn: &C,
key: &str,
value: &str,
) -> Result<(), DbError> {

View File

@@ -449,6 +449,7 @@ mod tauri_app {
web::start_web_server,
web::stop_web_server,
web::get_web_server_status,
web::get_web_service_config,
])
.build(tauri::generate_context!())
.expect("error while building tauri application")

View File

@@ -18,6 +18,7 @@ pub async fn get_web_server_status(
pub struct StartWebServerParams {
pub port: Option<u16>,
pub host: Option<String>,
pub token: Option<String>,
}
pub async fn start_web_server(

View File

@@ -9,10 +9,16 @@ use std::path::PathBuf;
use std::sync::atomic::{AtomicU16, Ordering};
use std::sync::{Arc, Mutex};
use sea_orm::{DatabaseConnection, TransactionError, TransactionTrait};
use serde::Serialize;
use crate::app_error::{AppCommandError, AppErrorCode};
use crate::app_state::AppState;
use crate::db::service::app_metadata_service;
const WEB_SERVICE_TOKEN_KEY: &str = "web_service_token";
const WEB_SERVICE_PORT_KEY: &str = "web_service_port";
pub const DEFAULT_WEB_SERVICE_PORT: u16 = 3080;
pub struct WebServerState {
handle: Mutex<Option<tokio::task::JoinHandle<()>>>,
@@ -50,6 +56,120 @@ pub fn generate_random_token() -> String {
uuid::Uuid::new_v4().to_string().replace('-', "")
}
/// Resolve the token to use when starting the Web server:
/// 1. use the explicit override if non-empty;
/// 2. fall back to the persisted value in `AppMetadata`;
/// 3. otherwise generate a fresh random token.
async fn resolve_web_service_token(
conn: &DatabaseConnection,
override_token: Option<String>,
) -> Result<String, AppCommandError> {
let trimmed_override = override_token
.map(|s| s.trim().to_string())
.filter(|s| !s.is_empty());
if let Some(value) = trimmed_override {
return Ok(value);
}
match app_metadata_service::get_value(conn, WEB_SERVICE_TOKEN_KEY)
.await
.map_err(AppCommandError::from)?
{
Some(saved) if !saved.trim().is_empty() => Ok(saved),
_ => Ok(generate_random_token()),
}
}
async fn resolve_web_service_port(
conn: &DatabaseConnection,
override_port: Option<u16>,
) -> Result<u16, AppCommandError> {
if let Some(port) = override_port {
return Ok(port);
}
let saved = app_metadata_service::get_value(conn, WEB_SERVICE_PORT_KEY)
.await
.map_err(AppCommandError::from)?;
let port = saved
.as_deref()
.and_then(|s| s.trim().parse::<u16>().ok())
.unwrap_or(DEFAULT_WEB_SERVICE_PORT);
Ok(port)
}
/// Persist token and port atomically so a partial failure cannot leave
/// `app_metadata` in a mixed old/new state.
async fn persist_web_service_config(
conn: &DatabaseConnection,
token: &str,
port: u16,
) -> Result<(), AppCommandError> {
// Own the values so the inner future is 'static (required by transaction).
let token_owned = token.to_string();
let port_str = port.to_string();
conn.transaction::<_, (), AppCommandError>(move |txn| {
Box::pin(async move {
app_metadata_service::upsert_value(txn, WEB_SERVICE_TOKEN_KEY, &token_owned)
.await
.map_err(AppCommandError::from)?;
app_metadata_service::upsert_value(txn, WEB_SERVICE_PORT_KEY, &port_str)
.await
.map_err(AppCommandError::from)?;
Ok(())
})
})
.await
.map_err(|e: TransactionError<AppCommandError>| match e {
TransactionError::Connection(db) => AppCommandError::new(
AppErrorCode::DatabaseError,
"Database transaction failed",
)
.with_detail(db.to_string()),
TransactionError::Transaction(inner) => inner,
})
}
#[derive(Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct WebServiceConfig {
pub token: Option<String>,
pub port: Option<u16>,
}
pub async fn load_web_service_config(
conn: &DatabaseConnection,
) -> Result<WebServiceConfig, AppCommandError> {
let token = app_metadata_service::get_value(conn, WEB_SERVICE_TOKEN_KEY)
.await
.map_err(AppCommandError::from)?;
let port = app_metadata_service::get_value(conn, WEB_SERVICE_PORT_KEY)
.await
.map_err(AppCommandError::from)?
.as_deref()
.and_then(|s| s.trim().parse::<u16>().ok());
Ok(WebServiceConfig { token, port })
}
/// Stable i18n-key prefixes — the frontend maps these to localized text.
const ERR_ALREADY_RUNNING: &str = "web_server.already_running";
const ERR_INVALID_ADDRESS: &str = "web_server.invalid_address";
const ERR_PORT_IN_USE: &str = "web_server.port_in_use";
const ERR_PERMISSION_DENIED: &str = "web_server.permission_denied";
const ERR_ADDRESS_UNAVAILABLE: &str = "web_server.address_unavailable";
const ERR_BIND_FAILED: &str = "web_server.bind_failed";
fn classify_bind_error(err: std::io::Error) -> AppCommandError {
use std::io::ErrorKind;
let (code, key) = match err.kind() {
ErrorKind::AddrInUse => (AppErrorCode::AlreadyExists, ERR_PORT_IN_USE),
ErrorKind::PermissionDenied => (AppErrorCode::PermissionDenied, ERR_PERMISSION_DENIED),
ErrorKind::AddrNotAvailable => (AppErrorCode::InvalidInput, ERR_ADDRESS_UNAVAILABLE),
_ => (AppErrorCode::IoError, ERR_BIND_FAILED),
};
AppCommandError::new(code, key).with_detail(err.to_string())
}
#[cfg(feature = "tauri-runtime")]
pub(crate) fn find_static_dir_tauri(app: &tauri::AppHandle) -> PathBuf {
use tauri::Manager;
@@ -112,6 +232,27 @@ pub fn find_static_dir_standalone(explicit: Option<&str>) -> PathBuf {
find_static_dir_fallback()
}
/// RAII guard that resets `running` back to `false` on drop unless disarmed.
/// Used to guarantee the flag is released on any error during start.
struct RunningGuard<'a> {
running: &'a std::sync::atomic::AtomicBool,
armed: bool,
}
impl<'a> RunningGuard<'a> {
fn disarm(&mut self) {
self.armed = false;
}
}
impl<'a> Drop for RunningGuard<'a> {
fn drop(&mut self) {
if self.armed {
self.running.store(false, Ordering::Release);
}
}
}
pub fn get_local_addresses(port: u16) -> Vec<String> {
let mut addrs = vec![format!("http://127.0.0.1:{}", port)];
// Try to get LAN IPs
@@ -134,30 +275,38 @@ pub(crate) async fn do_start_web_server_with_state(
static_dir: PathBuf,
port: Option<u16>,
host: Option<String>,
token: Option<String>,
) -> Result<WebServerInfo, AppCommandError> {
let ws = &app_state.web_server_state;
if ws.running.load(Ordering::Relaxed) {
return Err(AppCommandError::new(
AppErrorCode::AlreadyExists,
"Web server is already running",
));
}
let port = port.unwrap_or(3080);
// Atomically claim the running flag; concurrent starts see AlreadyExists.
ws.running
.compare_exchange(false, true, Ordering::AcqRel, Ordering::Acquire)
.map_err(|_| AppCommandError::new(AppErrorCode::AlreadyExists, ERR_ALREADY_RUNNING))?;
let mut guard = RunningGuard {
running: &ws.running,
armed: true,
};
let port = resolve_web_service_port(&app_state.db.conn, port).await?;
let host = host.unwrap_or_else(|| "0.0.0.0".to_string());
let token = generate_random_token();
let router = router::build_router(app_state.clone(), token.clone(), static_dir);
let token = resolve_web_service_token(&app_state.db.conn, token).await?;
let addr: SocketAddr = format!("{}:{}", host, port)
.parse()
.map_err(|e: std::net::AddrParseError| {
AppCommandError::invalid_input("Invalid host/port").with_detail(e.to_string())
AppCommandError::new(AppErrorCode::InvalidInput, ERR_INVALID_ADDRESS)
.with_detail(e.to_string())
})?;
let listener = tokio::net::TcpListener::bind(addr).await.map_err(|e| {
AppCommandError::io_error("Failed to bind address").with_detail(e.to_string())
})?;
let listener = tokio::net::TcpListener::bind(addr)
.await
.map_err(classify_bind_error)?;
// Persist only after a successful bind so a failed attempt doesn't overwrite saved state.
persist_web_service_config(&app_state.db.conn, &token, port).await?;
let router = router::build_router(app_state.clone(), token.clone(), static_dir);
let actual_port = listener.local_addr().map(|a| a.port()).unwrap_or(port);
eprintln!("[WEB] Starting web server on {}", addr);
@@ -171,7 +320,8 @@ pub(crate) async fn do_start_web_server_with_state(
*ws.handle.lock().unwrap() = Some(handle);
ws.port.store(actual_port, Ordering::Relaxed);
*ws.token.lock().unwrap() = token.clone();
ws.running.store(true, Ordering::Relaxed);
// running already true from compare_exchange; disarm guard so it doesn't flip back.
guard.disarm();
let addresses = get_local_addresses(actual_port);
Ok(WebServerInfo {
@@ -214,6 +364,7 @@ pub async fn start_web_server(
state: tauri::State<'_, WebServerState>,
port: Option<u16>,
host: Option<String>,
token: Option<String>,
) -> Result<WebServerInfo, AppCommandError> {
// In Tauri mode, we still need to start via the legacy path because
// the full AppState isn't easily available from tauri::State here.
@@ -221,16 +372,34 @@ pub async fn start_web_server(
use tauri::Manager;
let ws = &*state;
if ws.running.load(Ordering::Relaxed) {
return Err(AppCommandError::new(
AppErrorCode::AlreadyExists,
"Web server is already running",
));
}
let port_val = port.unwrap_or(3080);
// Atomically claim the running flag; concurrent starts see AlreadyExists.
ws.running
.compare_exchange(false, true, Ordering::AcqRel, Ordering::Acquire)
.map_err(|_| AppCommandError::new(AppErrorCode::AlreadyExists, ERR_ALREADY_RUNNING))?;
let mut guard = RunningGuard {
running: &ws.running,
armed: true,
};
let db = app.state::<crate::db::AppDatabase>();
let port_val = resolve_web_service_port(&db.conn, port).await?;
let host_val = host.unwrap_or_else(|| "0.0.0.0".to_string());
let token = generate_random_token();
let token = resolve_web_service_token(&db.conn, token).await?;
let addr: SocketAddr = format!("{}:{}", host_val, port_val)
.parse()
.map_err(|e: std::net::AddrParseError| {
AppCommandError::new(AppErrorCode::InvalidInput, ERR_INVALID_ADDRESS)
.with_detail(e.to_string())
})?;
let listener = tokio::net::TcpListener::bind(addr)
.await
.map_err(classify_bind_error)?;
// Persist only after a successful bind so a failed attempt doesn't overwrite saved state.
persist_web_service_config(&db.conn, &token, port_val).await?;
let static_dir = find_static_dir_tauri(&app);
@@ -250,16 +419,6 @@ pub async fn start_web_server(
let router = router::build_router(app_state, token.clone(), static_dir);
let addr: SocketAddr = format!("{}:{}", host_val, port_val)
.parse()
.map_err(|e: std::net::AddrParseError| {
AppCommandError::invalid_input("Invalid host/port").with_detail(e.to_string())
})?;
let listener = tokio::net::TcpListener::bind(addr).await.map_err(|e| {
AppCommandError::io_error("Failed to bind address").with_detail(e.to_string())
})?;
let actual_port = listener.local_addr().map(|a| a.port()).unwrap_or(port_val);
eprintln!("[WEB] Starting web server on {}", addr);
@@ -272,7 +431,8 @@ pub async fn start_web_server(
*ws.handle.lock().unwrap() = Some(handle);
ws.port.store(actual_port, Ordering::Relaxed);
*ws.token.lock().unwrap() = token.clone();
ws.running.store(true, Ordering::Relaxed);
// running already true from compare_exchange; disarm guard so it doesn't flip back.
guard.disarm();
let addresses = get_local_addresses(actual_port);
Ok(WebServerInfo {
@@ -298,3 +458,11 @@ pub async fn get_web_server_status(
) -> Result<Option<WebServerInfo>, AppCommandError> {
Ok(do_get_web_server_status(&state))
}
#[cfg(feature = "tauri-runtime")]
#[tauri::command]
pub async fn get_web_service_config(
db: tauri::State<'_, crate::db::AppDatabase>,
) -> Result<WebServiceConfig, AppCommandError> {
load_web_service_config(&db.conn).await
}