90 lines
2.8 KiB
Rust
90 lines
2.8 KiB
Rust
mod agent;
|
|
#[cfg(target_os = "windows")]
|
|
mod rustdesk;
|
|
|
|
use agent::{collect_inventory_plain, collect_profile, AgentRuntime, MachineProfile};
|
|
use tauri_plugin_store::Builder as StorePluginBuilder;
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct RustdeskProvisioningResult {
|
|
pub id: String,
|
|
pub password: String,
|
|
pub installed_version: Option<String>,
|
|
pub updated: bool,
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn collect_machine_profile() -> Result<MachineProfile, String> {
|
|
collect_profile().map_err(|error| error.to_string())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn collect_machine_inventory() -> Result<serde_json::Value, String> {
|
|
Ok(collect_inventory_plain())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn start_machine_agent(
|
|
state: tauri::State<AgentRuntime>,
|
|
base_url: String,
|
|
token: String,
|
|
status: Option<String>,
|
|
interval_seconds: Option<u64>,
|
|
) -> Result<(), String> {
|
|
state
|
|
.start_heartbeat(base_url, token, status, interval_seconds)
|
|
.map_err(|error| error.to_string())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn stop_machine_agent(state: tauri::State<AgentRuntime>) -> Result<(), String> {
|
|
state.stop();
|
|
Ok(())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn open_devtools(window: tauri::WebviewWindow) -> Result<(), String> {
|
|
window.open_devtools();
|
|
Ok(())
|
|
}
|
|
|
|
#[tauri::command]
|
|
async fn provision_rustdesk(machine_id: Option<String>) -> Result<RustdeskProvisioningResult, String> {
|
|
let machine_id = machine_id
|
|
.filter(|value| !value.trim().is_empty())
|
|
.ok_or_else(|| "Informe o identificador da máquina para provisionar o RustDesk.".to_string())?;
|
|
tauri::async_runtime::spawn_blocking(move || run_rustdesk_provision(machine_id))
|
|
.await
|
|
.map_err(|error| error.to_string())?
|
|
}
|
|
|
|
#[cfg(target_os = "windows")]
|
|
fn run_rustdesk_provision(machine_id: String) -> Result<RustdeskProvisioningResult, String> {
|
|
rustdesk::provision(&machine_id).map_err(|error| error.to_string())
|
|
}
|
|
|
|
#[cfg(not(target_os = "windows"))]
|
|
fn run_rustdesk_provision(_machine_id: String) -> Result<RustdeskProvisioningResult, String> {
|
|
Err("Provisionamento automático do RustDesk está disponível apenas no Windows.".to_string())
|
|
}
|
|
|
|
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
|
pub fn run() {
|
|
tauri::Builder::default()
|
|
.manage(AgentRuntime::new())
|
|
.plugin(tauri_plugin_opener::init())
|
|
.plugin(StorePluginBuilder::default().build())
|
|
.plugin(tauri_plugin_updater::Builder::new().build())
|
|
.plugin(tauri_plugin_process::init())
|
|
.invoke_handler(tauri::generate_handler![
|
|
collect_machine_profile,
|
|
collect_machine_inventory,
|
|
start_machine_agent,
|
|
stop_machine_agent,
|
|
open_devtools,
|
|
provision_rustdesk
|
|
])
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
}
|