File size: 3,231 Bytes
a21c316 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 | use tauri::State;
use crate::modules::cloudflared::{CloudflaredConfig, CloudflaredManager, CloudflaredStatus};
use std::sync::Arc;
use tokio::sync::RwLock;
/// Cloudflared服务状态管理
#[derive(Clone)]
pub struct CloudflaredState {
pub manager: Arc<RwLock<Option<CloudflaredManager>>>,
}
impl CloudflaredState {
pub fn new() -> Self {
Self {
manager: Arc::new(RwLock::new(None)),
}
}
/// 确保管理器已初始化
pub async fn ensure_manager(&self) -> Result<(), String> {
let mut lock = self.manager.write().await;
if lock.is_none() {
let data_dir = crate::modules::account::get_data_dir()?;
*lock = Some(CloudflaredManager::new(&data_dir));
}
Ok(())
}
}
/// 检查cloudflared是否已安装
#[tauri::command]
pub async fn cloudflared_check(
state: State<'_, CloudflaredState>,
) -> Result<CloudflaredStatus, String> {
state.ensure_manager().await?;
let lock = state.manager.read().await;
if let Some(manager) = lock.as_ref() {
let (installed, version) = manager.check_installed().await;
Ok(CloudflaredStatus {
installed,
version,
running: false,
url: None,
error: None,
})
} else {
Err("Manager not initialized".to_string())
}
}
/// 安装cloudflared
#[tauri::command]
pub async fn cloudflared_install(
state: State<'_, CloudflaredState>,
) -> Result<CloudflaredStatus, String> {
state.ensure_manager().await?;
let lock = state.manager.read().await;
if let Some(manager) = lock.as_ref() {
manager.install().await
} else {
Err("Manager not initialized".to_string())
}
}
/// 启动cloudflared隧道
#[tauri::command]
pub async fn cloudflared_start(
state: State<'_, CloudflaredState>,
config: CloudflaredConfig,
) -> Result<CloudflaredStatus, String> {
state.ensure_manager().await?;
let lock = state.manager.read().await;
if let Some(manager) = lock.as_ref() {
manager.start(config).await
} else {
Err("Manager not initialized".to_string())
}
}
/// 停止cloudflared隧道
#[tauri::command]
pub async fn cloudflared_stop(
state: State<'_, CloudflaredState>,
) -> Result<CloudflaredStatus, String> {
state.ensure_manager().await?;
let lock = state.manager.read().await;
if let Some(manager) = lock.as_ref() {
manager.stop().await
} else {
Err("Manager not initialized".to_string())
}
}
/// 获取cloudflared状态
#[tauri::command]
pub async fn cloudflared_get_status(
state: State<'_, CloudflaredState>,
) -> Result<CloudflaredStatus, String> {
state.ensure_manager().await?;
let lock = state.manager.read().await;
if let Some(manager) = lock.as_ref() {
let (installed, version) = manager.check_installed().await;
let mut status = manager.get_status().await;
status.installed = installed;
status.version = version;
if !installed {
status.running = false;
status.url = None;
}
Ok(status)
} else {
Ok(CloudflaredStatus::default())
}
}
|