Complete Slint UI mimicking Java Swing frmMain
This commit is contained in:
272
src/app.rs
Normal file
272
src/app.rs
Normal file
@@ -0,0 +1,272 @@
|
||||
//! Application state and integration with Slint
|
||||
|
||||
use std::sync::Arc;
|
||||
use parking_lot::RwLock;
|
||||
use anyhow::Result;
|
||||
use serde::Deserialize;
|
||||
|
||||
use crate::protocol::client::ServerClient;
|
||||
use crate::models::{Controller, RaidArray, Disk, Event};
|
||||
|
||||
pub struct AppState {
|
||||
pub client: Arc<RwLock<Option<ServerClient>>>,
|
||||
pub controllers: Arc<RwLock<Vec<Controller>>>,
|
||||
pub raids: Arc<RwLock<Vec<RaidArray>>>,
|
||||
pub disks: Arc<RwLock<Vec<Disk>>>,
|
||||
pub events: Arc<RwLock<Vec<Event>>>,
|
||||
pub selected_controller: Arc<RwLock<Option<i32>>>,
|
||||
pub selected_level: Arc<RwLock<String>>,
|
||||
pub connected: Arc<RwLock<bool>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Pagination {
|
||||
pub page: i32,
|
||||
pub page_size: i32,
|
||||
pub total_count: i32,
|
||||
pub total_pages: i32,
|
||||
pub has_next: bool,
|
||||
pub has_prev: bool,
|
||||
}
|
||||
|
||||
impl<'de> Deserialize<'de> for Pagination {
|
||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||
where D: serde::Deserializer<'de> {
|
||||
#[derive(Deserialize)]
|
||||
struct Inner {
|
||||
page: i32,
|
||||
page_size: i32,
|
||||
total_count: i32,
|
||||
total_pages: i32,
|
||||
has_next: bool,
|
||||
has_prev: bool,
|
||||
}
|
||||
let inner = Inner::deserialize(deserializer)?;
|
||||
Ok(Pagination {
|
||||
page: inner.page,
|
||||
page_size: inner.page_size,
|
||||
total_count: inner.total_count,
|
||||
total_pages: inner.total_pages,
|
||||
has_next: inner.has_next,
|
||||
has_prev: inner.has_prev,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl AppState {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
client: Arc::new(RwLock::new(None)),
|
||||
controllers: Arc::new(RwLock::new(Vec::new())),
|
||||
raids: Arc::new(RwLock::new(Vec::new())),
|
||||
disks: Arc::new(RwLock::new(Vec::new())),
|
||||
events: Arc::new(RwLock::new(Vec::new())),
|
||||
selected_controller: Arc::new(RwLock::new(None)),
|
||||
selected_level: Arc::new(RwLock::new("all".to_string())),
|
||||
connected: Arc::new(RwLock::new(false)),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn connect(&self, addr: &str) -> Result<()> {
|
||||
tracing::info!("=== AppState::connect() called with addr: {} ===", addr);
|
||||
let client = ServerClient::connect(addr).await?;
|
||||
tracing::info!("=== ServerClient connected, storing in AppState ===");
|
||||
*self.client.write() = Some(client);
|
||||
*self.connected.write() = true;
|
||||
tracing::info!("=== AppState connected flag set to true ===");
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn refresh(&self) -> Result<()> {
|
||||
tracing::info!("=== refresh() called ===");
|
||||
|
||||
// Get client - clone it so we don't interfere with the stored one
|
||||
let client_opt = self.client.read().clone();
|
||||
tracing::info!("Got client option: {:?}", client_opt.is_some());
|
||||
|
||||
let client = match client_opt {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
|
||||
tracing::info!("Fetching controllers...");
|
||||
// Fetch all data
|
||||
let controllers_data = client.get_controllers().await?;
|
||||
tracing::info!("Got {} controllers from server", controllers_data.len());
|
||||
|
||||
let mut controllers = Vec::new();
|
||||
for v in controllers_data {
|
||||
if let Some(c) = Controller::from_json(&v) {
|
||||
controllers.push(c);
|
||||
}
|
||||
}
|
||||
tracing::info!("Parsed {} controllers", controllers.len());
|
||||
*self.controllers.write() = controllers;
|
||||
|
||||
tracing::info!("Fetching raids...");
|
||||
let raids_data = client.get_raids(None).await?;
|
||||
tracing::info!("Got {} raids from server", raids_data.len());
|
||||
|
||||
let mut raids = Vec::new();
|
||||
for v in raids_data {
|
||||
if let Some(r) = RaidArray::from_json(&v) {
|
||||
raids.push(r);
|
||||
}
|
||||
}
|
||||
tracing::info!("Parsed {} raids", raids.len());
|
||||
*self.raids.write() = raids;
|
||||
|
||||
tracing::info!("Fetching disks...");
|
||||
let disks_data = client.get_disks(None).await?;
|
||||
tracing::info!("Got {} disks from server", disks_data.len());
|
||||
let mut disks = Vec::new();
|
||||
for v in disks_data {
|
||||
if let Some(d) = Disk::from_json(&v) {
|
||||
disks.push(d);
|
||||
}
|
||||
}
|
||||
*self.disks.write() = disks;
|
||||
|
||||
let events_data = client.get_events(None, Some(50)).await?;
|
||||
let mut events = Vec::new();
|
||||
for v in events_data {
|
||||
if let Some(e) = Event::from_json(&v) {
|
||||
events.push(e);
|
||||
}
|
||||
}
|
||||
*self.events.write() = events;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn is_connected(&self) -> bool {
|
||||
*self.connected.read()
|
||||
}
|
||||
|
||||
pub fn get_controllers(&self) -> Vec<Controller> {
|
||||
self.controllers.read().clone()
|
||||
}
|
||||
|
||||
pub fn get_raids(&self) -> Vec<RaidArray> {
|
||||
self.raids.read().clone()
|
||||
}
|
||||
|
||||
pub fn get_disks(&self) -> Vec<Disk> {
|
||||
self.disks.read().clone()
|
||||
}
|
||||
|
||||
pub fn get_events(&self) -> Vec<Event> {
|
||||
self.events.read().clone()
|
||||
}
|
||||
|
||||
pub async fn load_events_page(&self, page: i32, page_size: i32) -> Result<Option<Pagination>> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
|
||||
// Get selected level filter
|
||||
let level = self.selected_level.read().clone();
|
||||
let level_filter = if level == "all" { None } else { Some(level) };
|
||||
|
||||
// Fetch events with pagination
|
||||
let response = client.get_events_paginated(page, page_size, level_filter).await?;
|
||||
|
||||
let events_data = response.events;
|
||||
let pagination = response.pagination;
|
||||
|
||||
let mut events = Vec::new();
|
||||
for v in events_data {
|
||||
if let Some(e) = Event::from_json(&v) {
|
||||
events.push(e);
|
||||
}
|
||||
}
|
||||
*self.events.write() = events;
|
||||
|
||||
Ok(Some(pagination))
|
||||
}
|
||||
|
||||
pub async fn set_selected_level(&self, level: String) -> Result<()> {
|
||||
*self.selected_level.write() = level;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// RAID Operations
|
||||
pub async fn create_raid(&self, name: &str, level: u32, disks: Vec<u32>) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.create_raid(name, level, disks).await
|
||||
}
|
||||
|
||||
pub async fn delete_raid(&self, raid_id: u32) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.delete_raid(raid_id).await
|
||||
}
|
||||
|
||||
pub async fn rebuild_raid(&self, raid_id: u32) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.rebuild_raid(raid_id).await
|
||||
}
|
||||
|
||||
pub async fn initialize_disk(&self, disk_id: u32) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.initialize_disk(disk_id).await
|
||||
}
|
||||
|
||||
pub async fn locate_disk(&self, disk_id: u32) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.locate_disk(disk_id).await
|
||||
}
|
||||
|
||||
pub async fn remove_disk(&self, disk_id: u32, force: bool) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.remove_disk(disk_id, force).await
|
||||
}
|
||||
|
||||
pub async fn commit_config(&self) -> Result<serde_json::Value> {
|
||||
let client = self.client.read().clone();
|
||||
let client = match client {
|
||||
Some(c) => c,
|
||||
None => return Err(anyhow::anyhow!("Not connected")),
|
||||
};
|
||||
client.commit_config().await
|
||||
}
|
||||
|
||||
pub async fn connect_internal(addr: &str) -> Result<()> {
|
||||
let client = crate::protocol::client::ServerClient::connect(addr).await?;
|
||||
let mut this = Self::new();
|
||||
*this.client.write() = Some(client);
|
||||
*this.connected.write() = true;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for AppState {
|
||||
fn default() -> Self {
|
||||
Self::new()
|
||||
}
|
||||
}
|
||||
9
src/lib.rs
Normal file
9
src/lib.rs
Normal file
@@ -0,0 +1,9 @@
|
||||
//! RAIDGuard X GUI Client
|
||||
//!
|
||||
//! A cross-platform GUI client using Slint and TCP connection to the GUI server.
|
||||
|
||||
pub mod protocol;
|
||||
pub mod models;
|
||||
pub mod app;
|
||||
|
||||
pub use app::AppState;
|
||||
467
src/main.rs
Normal file
467
src/main.rs
Normal file
@@ -0,0 +1,467 @@
|
||||
//! RAIDGuard X GUI Client - Main entry point
|
||||
|
||||
use anyhow::Result;
|
||||
use std::sync::{Arc, Mutex};
|
||||
use slint::{ComponentHandle, SharedString};
|
||||
use raidguard_x_gui_client::{AppState, models::{Controller, RaidArray, Disk, Event}};
|
||||
|
||||
slint::include_modules!();
|
||||
|
||||
fn to_s(s: &str) -> SharedString {
|
||||
s.into()
|
||||
}
|
||||
|
||||
fn update_ui_from_state(app: &AppWindow, state: &AppState) {
|
||||
let controllers = state.get_controllers();
|
||||
let raids = state.get_raids();
|
||||
let disks = state.get_disks();
|
||||
let connected = state.is_connected();
|
||||
let selected_idx = app.get_selected_controller_index() as usize;
|
||||
|
||||
tracing::info!("update_ui_from_state: {} controllers, {} raids, {} disks, selected={}", controllers.len(), raids.len(), disks.len(), selected_idx);
|
||||
|
||||
app.set_connected(connected);
|
||||
app.set_status_text(to_s(if connected { "Connected" } else { "Disconnected" }));
|
||||
app.set_controller_count(controllers.len() as i32);
|
||||
app.set_raid_count(raids.len() as i32);
|
||||
app.set_disk_count(disks.len() as i32);
|
||||
|
||||
tracing::info!("Setting counts: controller={}, raid={}, disk={}", controllers.len(), raids.len(), disks.len());
|
||||
|
||||
// Controller 1
|
||||
if let Some(c) = controllers.get(0) {
|
||||
tracing::info!("Setting controller 1: {} {} {}", c.hostname, c.ip, c.status);
|
||||
app.set_ctrl1_name(to_s(&c.hostname));
|
||||
app.set_ctrl1_ip(to_s(&c.ip));
|
||||
app.set_ctrl1_status(to_s(&c.status));
|
||||
app.set_ctrl1_model(to_s(&c.model));
|
||||
app.set_ctrl1_firmware(to_s(&c.firmware_version));
|
||||
} else {
|
||||
tracing::info!("No controller data!");
|
||||
}
|
||||
|
||||
// Controller 2
|
||||
if let Some(c) = controllers.get(1) {
|
||||
tracing::info!("Setting controller 2: {} {} {}", c.hostname, c.ip, c.status);
|
||||
app.set_ctrl2_name(to_s(&c.hostname));
|
||||
app.set_ctrl2_ip(to_s(&c.ip));
|
||||
app.set_ctrl2_status(to_s(&c.status));
|
||||
app.set_ctrl2_model(to_s(&c.model));
|
||||
app.set_ctrl2_firmware(to_s(&c.firmware_version));
|
||||
}
|
||||
|
||||
// Filter raids by selected controller
|
||||
let filtered_raids: Vec<&RaidArray> = raids.iter().filter(|r| r.controller_id as usize == selected_idx).collect();
|
||||
tracing::info!("Selected controller {} has {} raids", selected_idx, filtered_raids.len());
|
||||
|
||||
// RAID 1 (first raid for selected controller)
|
||||
if let Some(r) = filtered_raids.get(0) {
|
||||
tracing::info!("Setting RAID 1: {} {} {}", r.name, r.raid_level, r.status);
|
||||
app.set_raid1_name(to_s(&r.name));
|
||||
app.set_raid1_level(to_s(&r.raid_level));
|
||||
app.set_raid1_status(to_s(&r.status));
|
||||
app.set_raid1_capacity(format!("{:.1} TB", r.total_capacity_tb).into());
|
||||
app.set_raid1_usage(format!("{:.0}%", r.usage_percent()).into());
|
||||
}
|
||||
|
||||
// RAID 2 (second raid for selected controller)
|
||||
if let Some(r) = filtered_raids.get(1) {
|
||||
tracing::info!("Setting RAID 2: {} {} {}", r.name, r.raid_level, r.status);
|
||||
app.set_raid2_name(to_s(&r.name));
|
||||
app.set_raid2_level(to_s(&r.raid_level));
|
||||
app.set_raid2_status(to_s(&r.status));
|
||||
app.set_raid2_capacity(format!("{:.1} TB", r.total_capacity_tb).into());
|
||||
app.set_raid2_usage(format!("{:.0}%", r.usage_percent()).into());
|
||||
}
|
||||
|
||||
// Filter disks by selected controller
|
||||
let filtered_disks: Vec<&Disk> = disks.iter().filter(|d| d.controller_id as usize == selected_idx).collect();
|
||||
tracing::info!("Selected controller {} has {} disks", selected_idx, filtered_disks.len());
|
||||
|
||||
// Disks
|
||||
for (i, d) in filtered_disks.iter().enumerate() {
|
||||
let slot = format!("Enclosure {} Slot {}", d.enclosure, d.slot);
|
||||
let cap = format!("{:.1} TB", d.capacity_tb);
|
||||
tracing::info!("Setting disk {}: {} {} {}", i, slot, d.model, d.status);
|
||||
match i {
|
||||
0 => {
|
||||
app.set_disk1_loc(to_s(&slot));
|
||||
app.set_disk1_model(to_s(&d.model));
|
||||
app.set_disk1_status(to_s(&d.status));
|
||||
app.set_disk1_capacity(cap.into());
|
||||
}
|
||||
1 => {
|
||||
app.set_disk2_loc(to_s(&slot));
|
||||
app.set_disk2_model(to_s(&d.model));
|
||||
app.set_disk2_status(to_s(&d.status));
|
||||
app.set_disk2_capacity(cap.into());
|
||||
}
|
||||
2 => {
|
||||
app.set_disk3_loc(to_s(&slot));
|
||||
app.set_disk3_model(to_s(&d.model));
|
||||
app.set_disk3_status(to_s(&d.status));
|
||||
app.set_disk3_capacity(cap.into());
|
||||
}
|
||||
3 => {
|
||||
app.set_disk4_loc(to_s(&slot));
|
||||
app.set_disk4_model(to_s(&d.model));
|
||||
app.set_disk4_status(to_s(&d.status));
|
||||
app.set_disk4_capacity(cap.into());
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
|
||||
// Events
|
||||
let events = state.get_events();
|
||||
for (i, e) in events.iter().enumerate() {
|
||||
match i {
|
||||
0 => {
|
||||
app.set_evtime1(to_s(&e.formatted_time()));
|
||||
app.set_evlevel1(to_s(&e.level));
|
||||
app.set_evmsg1(to_s(&e.message));
|
||||
app.set_evsource1(to_s(&e.event_type));
|
||||
}
|
||||
1 => {
|
||||
app.set_evtime2(to_s(&e.formatted_time()));
|
||||
app.set_evlevel2(to_s(&e.level));
|
||||
app.set_evmsg2(to_s(&e.message));
|
||||
app.set_evsource2(to_s(&e.event_type));
|
||||
}
|
||||
2 => {
|
||||
app.set_evtime3(to_s(&e.formatted_time()));
|
||||
app.set_evlevel3(to_s(&e.level));
|
||||
app.set_evmsg3(to_s(&e.message));
|
||||
app.set_evsource3(to_s(&e.event_type));
|
||||
}
|
||||
3 => {
|
||||
app.set_evtime4(to_s(&e.formatted_time()));
|
||||
app.set_evlevel4(to_s(&e.level));
|
||||
app.set_evmsg4(to_s(&e.message));
|
||||
app.set_evsource4(to_s(&e.event_type));
|
||||
}
|
||||
4 => {
|
||||
app.set_evtime5(to_s(&e.formatted_time()));
|
||||
app.set_evlevel5(to_s(&e.level));
|
||||
app.set_evmsg5(to_s(&e.message));
|
||||
app.set_evsource5(to_s(&e.event_type));
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
std::panic::set_hook(Box::new(|panic_info| {
|
||||
eprintln!("PANIC: {}", panic_info);
|
||||
}));
|
||||
|
||||
let file = std::fs::OpenOptions::new()
|
||||
.create(true)
|
||||
.append(true)
|
||||
.open("/tmp/raidguard_client.log")
|
||||
.unwrap();
|
||||
|
||||
tracing_subscriber::fmt()
|
||||
.with_env_filter(
|
||||
tracing_subscriber::EnvFilter::from_default_env()
|
||||
.add_directive(tracing::Level::DEBUG.into()),
|
||||
)
|
||||
.with_ansi(false)
|
||||
.init();
|
||||
|
||||
// Also log to file directly
|
||||
let mut log_file = std::fs::OpenOptions::new()
|
||||
.create(true)
|
||||
.append(true)
|
||||
.open("/tmp/raidguard_client.log")
|
||||
.unwrap();
|
||||
use std::io::Write;
|
||||
writeln!(log_file, "=== RAIDGuard X GUI Client Started ===").unwrap();
|
||||
drop(log_file);
|
||||
|
||||
tracing::info!("=== RAIDGuard X GUI Client ===");
|
||||
|
||||
let app = AppWindow::new()?;
|
||||
app.show().ok();
|
||||
let app_state = Arc::new(Mutex::new(AppState::new()));
|
||||
|
||||
// Connect to server
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_connect_server(move || {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
|
||||
// Spawn async task
|
||||
std::thread::spawn(move || {
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
tracing::info!("=== Connect button clicked ===");
|
||||
|
||||
// First connect
|
||||
let connect_result = {
|
||||
let s = match state.lock() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return,
|
||||
};
|
||||
tracing::info!("Calling s.connect()...");
|
||||
s.connect("127.0.0.1:8923").await
|
||||
};
|
||||
|
||||
match connect_result {
|
||||
Ok(_) => {
|
||||
tracing::info!("=== Connected to server! ===");
|
||||
|
||||
// Wait a bit to avoid lock contention
|
||||
tokio::time::sleep(std::time::Duration::from_millis(100)).await;
|
||||
|
||||
// Refresh data
|
||||
{
|
||||
let s = match state.lock() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return,
|
||||
};
|
||||
tracing::info!("Calling s.refresh()...");
|
||||
let _ = s.refresh().await;
|
||||
}
|
||||
|
||||
tracing::info!("=== Refresh complete ===");
|
||||
|
||||
// Need to invoke UI update from the main thread
|
||||
let app_handle = app_handle.clone();
|
||||
let state = state.clone();
|
||||
slint::invoke_from_event_loop(move || {
|
||||
tracing::info!("=== Updating UI in main thread ===");
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
tracing::info!("=== Upgrading app handle ===");
|
||||
if let Ok(s) = state.lock() {
|
||||
tracing::info!("=== Got state lock ===");
|
||||
a.set_connected(true);
|
||||
a.set_status_text(to_s("Connected"));
|
||||
update_ui_from_state(&a, &s);
|
||||
tracing::info!("=== UI update complete ===");
|
||||
}
|
||||
}
|
||||
}).ok();
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::error!("Failed to connect: {}", e);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// Refresh data
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_refresh_data(move || {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
if s.is_connected() {
|
||||
if let Err(e) = s.refresh().await {
|
||||
tracing::error!("Failed to refresh: {}", e);
|
||||
}
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
update_ui_from_state(&a, &s);
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Toggle auto refresh
|
||||
app.on_toggle_auto_refresh(move || {
|
||||
tracing::debug!("Toggle auto refresh");
|
||||
});
|
||||
|
||||
// Load events page
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_load_events_page(move |page| {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.load_events_page(page, 10).await;
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
let events = s.get_events();
|
||||
for (i, e) in events.iter().enumerate() {
|
||||
match i {
|
||||
0 => { a.set_evtime1(to_s(&e.formatted_time())); a.set_evlevel1(to_s(&e.level)); a.set_evmsg1(to_s(&e.message)); }
|
||||
1 => { a.set_evtime2(to_s(&e.formatted_time())); a.set_evlevel2(to_s(&e.level)); a.set_evmsg2(to_s(&e.message)); }
|
||||
2 => { a.set_evtime3(to_s(&e.formatted_time())); a.set_evlevel3(to_s(&e.level)); a.set_evmsg3(to_s(&e.message)); }
|
||||
3 => { a.set_evtime4(to_s(&e.formatted_time())); a.set_evlevel4(to_s(&e.level)); a.set_evmsg4(to_s(&e.message)); }
|
||||
4 => { a.set_evtime5(to_s(&e.formatted_time())); a.set_evlevel5(to_s(&e.level)); a.set_evmsg5(to_s(&e.message)); }
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Show event details
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_show_event_details(move |index| {
|
||||
if let Ok(s) = state.lock() {
|
||||
let events = s.get_events();
|
||||
if let Some(event) = events.get(index as usize) {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_event_popup(true);
|
||||
a.set_event_detail_time(to_s(&event.formatted_time()));
|
||||
a.set_event_detail_level(to_s(&event.level));
|
||||
a.set_event_detail_message(to_s(&event.message));
|
||||
a.set_event_detail_source(to_s(&event.event_type));
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
let app_handle = app.as_weak();
|
||||
app.on_close_event_details(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_event_popup(false);
|
||||
}
|
||||
});
|
||||
|
||||
// Create RAID dialog
|
||||
let app_handle = app.as_weak();
|
||||
app.on_open_create_raid_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_create_raid_dialog(true);
|
||||
a.set_create_raid_name(to_s(""));
|
||||
a.set_create_raid_status(to_s(""));
|
||||
a.set_create_raid_disk1(to_s(""));
|
||||
a.set_create_raid_disk2(to_s(""));
|
||||
a.set_create_raid_disk3(to_s(""));
|
||||
a.set_create_raid_disk4(to_s(""));
|
||||
}
|
||||
});
|
||||
|
||||
let app_handle = app.as_weak();
|
||||
app.on_close_create_raid_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_create_raid_dialog(false);
|
||||
}
|
||||
});
|
||||
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_confirm_create_raid(move || {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.commit_config().await;
|
||||
let _ = s.refresh().await;
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_create_raid_dialog(false);
|
||||
update_ui_from_state(&a, &s);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Delete RAID dialog
|
||||
let app_handle = app.as_weak();
|
||||
app.on_open_delete_raid_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_delete_raid_dialog(true);
|
||||
a.set_delete_raid_name(to_s("RAID-1"));
|
||||
a.set_delete_raid_status(to_s(""));
|
||||
}
|
||||
});
|
||||
|
||||
let app_handle = app.as_weak();
|
||||
app.on_close_delete_raid_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_delete_raid_dialog(false);
|
||||
}
|
||||
});
|
||||
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_confirm_delete_raid(move || {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.refresh().await;
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_delete_raid_dialog(false);
|
||||
update_ui_from_state(&a, &s);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Rebuild RAID
|
||||
let state = app_state.clone();
|
||||
app.on_rebuild_raid(move || {
|
||||
let state = state.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.rebuild_raid(1).await;
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Commit config
|
||||
let state = app_state.clone();
|
||||
app.on_commit_config(move || {
|
||||
let state = state.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.commit_config().await;
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// Disk operation dialog
|
||||
let app_handle = app.as_weak();
|
||||
app.on_open_disk_operation_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_disk_dialog(true);
|
||||
a.set_selected_disk(1);
|
||||
a.set_disk_operation_status(to_s(""));
|
||||
}
|
||||
});
|
||||
|
||||
let app_handle = app.as_weak();
|
||||
app.on_close_disk_operation_dialog(move || {
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_disk_dialog(false);
|
||||
}
|
||||
});
|
||||
|
||||
let state = app_state.clone();
|
||||
let app_handle = app.as_weak();
|
||||
app.on_confirm_disk_operation(move || {
|
||||
let state = state.clone();
|
||||
let app_handle = app_handle.clone();
|
||||
let rt = tokio::runtime::Runtime::new().unwrap();
|
||||
rt.block_on(async move {
|
||||
if let Ok(s) = state.lock() {
|
||||
let _ = s.refresh().await;
|
||||
if let Some(a) = app_handle.upgrade() {
|
||||
a.set_show_disk_dialog(false);
|
||||
update_ui_from_state(&a, &s);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
app.run()?;
|
||||
Ok(())
|
||||
}
|
||||
145
src/models/mod.rs
Normal file
145
src/models/mod.rs
Normal file
@@ -0,0 +1,145 @@
|
||||
//! Data models for RAIDGuard X
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Controller {
|
||||
pub id: i32,
|
||||
pub ip: String,
|
||||
pub hostname: String,
|
||||
pub serial_number: String,
|
||||
pub model: String,
|
||||
pub firmware_version: String,
|
||||
pub status: String,
|
||||
pub enclosures: i32,
|
||||
pub total_disks: i32,
|
||||
}
|
||||
|
||||
impl Controller {
|
||||
pub fn from_json(value: &serde_json::Value) -> Option<Self> {
|
||||
Some(Self {
|
||||
id: value["id"].as_i64()? as i32,
|
||||
ip: value["ip"].as_str()?.to_string(),
|
||||
hostname: value["hostname"].as_str()?.to_string(),
|
||||
serial_number: value["serial_number"].as_str()?.to_string(),
|
||||
model: value["model"].as_str()?.to_string(),
|
||||
firmware_version: value["firmware_version"].as_str()?.to_string(),
|
||||
status: value["status"].as_str()?.to_string(),
|
||||
enclosures: value["enclosures"].as_i64().unwrap_or(0) as i32,
|
||||
total_disks: value["total_disks"].as_i64().unwrap_or(0) as i32,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn is_online(&self) -> bool {
|
||||
self.status == "online"
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct RaidArray {
|
||||
pub id: i32,
|
||||
pub controller_id: i32,
|
||||
pub name: String,
|
||||
pub raid_level: String,
|
||||
pub status: String,
|
||||
pub total_capacity_tb: f64,
|
||||
pub used_capacity_tb: f64,
|
||||
pub free_capacity_tb: f64,
|
||||
pub disk_count: i32,
|
||||
pub rebuild_progress: Option<i32>,
|
||||
}
|
||||
|
||||
impl RaidArray {
|
||||
pub fn from_json(value: &serde_json::Value) -> Option<Self> {
|
||||
Some(Self {
|
||||
id: value["id"].as_i64()? as i32,
|
||||
controller_id: value["controller_id"].as_i64().unwrap_or(0) as i32,
|
||||
name: value["name"].as_str()?.to_string(),
|
||||
raid_level: value["raid_level"].as_str()?.to_string(),
|
||||
status: value["status"].as_str()?.to_string(),
|
||||
total_capacity_tb: value["total_capacity_tb"].as_f64().unwrap_or(0.0),
|
||||
used_capacity_tb: value["used_capacity_tb"].as_f64().unwrap_or(0.0),
|
||||
free_capacity_tb: value["free_capacity_tb"].as_f64().unwrap_or(0.0),
|
||||
disk_count: value["disk_count"].as_i64().unwrap_or(0) as i32,
|
||||
rebuild_progress: value["rebuild_progress"].as_i64().map(|v| v as i32),
|
||||
})
|
||||
}
|
||||
|
||||
pub fn usage_percent(&self) -> f64 {
|
||||
if self.total_capacity_tb > 0.0 {
|
||||
(self.used_capacity_tb / self.total_capacity_tb) * 100.0
|
||||
} else {
|
||||
0.0
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Disk {
|
||||
pub slot: i32,
|
||||
pub enclosure: i32,
|
||||
pub serial_number: String,
|
||||
pub model: String,
|
||||
pub vendor: String,
|
||||
pub capacity_tb: f64,
|
||||
pub status: String,
|
||||
pub disk_type: String,
|
||||
pub temperature: i32,
|
||||
pub array_id: Option<i32>,
|
||||
pub controller_id: i32,
|
||||
}
|
||||
|
||||
impl Disk {
|
||||
pub fn from_json(value: &serde_json::Value) -> Option<Self> {
|
||||
Some(Self {
|
||||
slot: value["slot"].as_i64().unwrap_or(0) as i32,
|
||||
enclosure: value["enclosure"].as_i64().unwrap_or(0) as i32,
|
||||
serial_number: value["serial_number"].as_str().unwrap_or("").to_string(),
|
||||
model: value["model"].as_str().unwrap_or("").to_string(),
|
||||
vendor: value["vendor"].as_str().unwrap_or("").to_string(),
|
||||
capacity_tb: value["capacity_tb"].as_f64().unwrap_or(0.0),
|
||||
status: value["status"].as_str().unwrap_or("").to_string(),
|
||||
disk_type: value["disk_type"].as_str().unwrap_or("").to_string(),
|
||||
temperature: value["temperature"].as_i64().unwrap_or(0) as i32,
|
||||
array_id: value["array_id"].as_i64().map(|v| v as i32),
|
||||
controller_id: value["controller_id"].as_i64().unwrap_or(0) as i32,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn is_online(&self) -> bool {
|
||||
self.status == "online"
|
||||
}
|
||||
|
||||
pub fn is_failed(&self) -> bool {
|
||||
self.status == "failed"
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Event {
|
||||
pub id: i32,
|
||||
pub controller_id: i32,
|
||||
pub timestamp: i64,
|
||||
pub level: String,
|
||||
pub event_type: String,
|
||||
pub message: String,
|
||||
}
|
||||
|
||||
impl Event {
|
||||
pub fn from_json(value: &serde_json::Value) -> Option<Self> {
|
||||
Some(Self {
|
||||
id: value["id"].as_i64().unwrap_or(0) as i32,
|
||||
controller_id: value["controller_id"].as_i64().unwrap_or(0) as i32,
|
||||
timestamp: value["timestamp"].as_i64().unwrap_or(0),
|
||||
level: value["level"].as_str().unwrap_or("info").to_string(),
|
||||
event_type: value["event_type"].as_str().unwrap_or("").to_string(),
|
||||
message: value["message"].as_str().unwrap_or("").to_string(),
|
||||
})
|
||||
}
|
||||
|
||||
pub fn formatted_time(&self) -> String {
|
||||
use chrono::{DateTime, Utc};
|
||||
let dt = DateTime::from_timestamp(self.timestamp, 0).unwrap_or_else(|| Utc::now());
|
||||
dt.format("%Y-%m-%d %H:%M:%S").to_string()
|
||||
}
|
||||
}
|
||||
275
src/protocol/client.rs
Normal file
275
src/protocol/client.rs
Normal file
@@ -0,0 +1,275 @@
|
||||
//! TCP Client for communicating with GUI Server
|
||||
|
||||
use tokio::net::TcpStream;
|
||||
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||
use tokio::sync::Mutex;
|
||||
use anyhow::Result;
|
||||
use std::sync::Arc;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::protocol::{Request, Response};
|
||||
use crate::app::Pagination;
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct EventsResponse {
|
||||
pub events: Vec<serde_json::Value>,
|
||||
pub pagination: Pagination,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct ServerClient {
|
||||
stream: Arc<Mutex<Option<TcpStream>>>,
|
||||
}
|
||||
|
||||
impl ServerClient {
|
||||
pub async fn connect(addr: &str) -> Result<Self> {
|
||||
tracing::info!("Connecting to server at {}...", addr);
|
||||
let stream = TcpStream::connect(addr).await?;
|
||||
stream.set_nodelay(true)?;
|
||||
tracing::info!("Connected to server");
|
||||
|
||||
Ok(Self {
|
||||
stream: Arc::new(Mutex::new(Some(stream))),
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn send_request(&self, request: &Request) -> Result<Response> {
|
||||
let mut stream_guard = self.stream.lock().await;
|
||||
let stream = stream_guard.as_mut().ok_or_else(|| anyhow::anyhow!("Not connected"))?;
|
||||
|
||||
// Serialize and send request
|
||||
let json = serde_json::to_string(request)?;
|
||||
let mut send_buf = json.into_bytes();
|
||||
send_buf.push(b'\n');
|
||||
|
||||
stream.write_all(&send_buf).await?;
|
||||
|
||||
// Read response (read until newline)
|
||||
let mut response_buf = Vec::new();
|
||||
let mut byte = [0u8; 1];
|
||||
loop {
|
||||
let n = stream.read(&mut byte).await?;
|
||||
if n == 0 {
|
||||
break;
|
||||
}
|
||||
if byte[0] == b'\n' {
|
||||
break;
|
||||
}
|
||||
response_buf.push(byte[0]);
|
||||
}
|
||||
let response_line = String::from_utf8(response_buf)?;
|
||||
|
||||
let response: Response = serde_json::from_str(&response_line.trim())?;
|
||||
Ok(response)
|
||||
}
|
||||
|
||||
pub async fn get_controllers(&self) -> Result<Vec<serde_json::Value>> {
|
||||
let request = Request::new("get_controllers");
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
let data = response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))?;
|
||||
let controllers = data["controllers"]
|
||||
.as_array()
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
|
||||
Ok(controllers)
|
||||
}
|
||||
|
||||
pub async fn get_raids(&self, controller_id: Option<i32>) -> Result<Vec<serde_json::Value>> {
|
||||
let params = match controller_id {
|
||||
Some(id) => serde_json::json!({ "controller_id": id }),
|
||||
None => serde_json::json!({}),
|
||||
};
|
||||
let request = Request::with_params("get_raids", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
let data = response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))?;
|
||||
let raids = data["raids"]
|
||||
.as_array()
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
|
||||
Ok(raids)
|
||||
}
|
||||
|
||||
pub async fn get_disks(&self, controller_id: Option<i32>) -> Result<Vec<serde_json::Value>> {
|
||||
let params = match controller_id {
|
||||
Some(id) => serde_json::json!({ "controller_id": id }),
|
||||
None => serde_json::json!({}),
|
||||
};
|
||||
let request = Request::with_params("get_disks", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
let data = response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))?;
|
||||
let disks = data["disks"]
|
||||
.as_array()
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
|
||||
Ok(disks)
|
||||
}
|
||||
|
||||
pub async fn get_events(&self, controller_id: Option<i32>, limit: Option<usize>) -> Result<Vec<serde_json::Value>> {
|
||||
let mut params = serde_json::json!({});
|
||||
if let Some(id) = controller_id {
|
||||
params["controller_id"] = serde_json::json!(id);
|
||||
}
|
||||
if let Some(l) = limit {
|
||||
params["limit"] = serde_json::json!(l);
|
||||
}
|
||||
|
||||
let request = Request::with_params("get_events", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
let data = response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))?;
|
||||
let events = data["events"]
|
||||
.as_array()
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
|
||||
Ok(events)
|
||||
}
|
||||
|
||||
pub async fn get_events_paginated(&self, page: i32, page_size: i32, level: Option<String>) -> Result<EventsResponse> {
|
||||
let mut params = serde_json::json!({
|
||||
"page": page,
|
||||
"page_size": page_size
|
||||
});
|
||||
if let Some(level) = level {
|
||||
params["level"] = serde_json::json!(level);
|
||||
}
|
||||
|
||||
let request = Request::with_params("get_events", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
let data = response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))?;
|
||||
let events_response: EventsResponse = serde_json::from_value(data)?;
|
||||
|
||||
Ok(events_response)
|
||||
}
|
||||
|
||||
pub async fn ping(&self) -> Result<bool> {
|
||||
let request = Request::new("ping");
|
||||
let response = self.send_request(&request).await?;
|
||||
Ok(response.is_success())
|
||||
}
|
||||
|
||||
pub fn is_connected(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
// RAID Operations
|
||||
pub async fn create_raid(&self, name: &str, level: u32, disks: Vec<u32>) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({
|
||||
"name": name,
|
||||
"level": level,
|
||||
"disks": disks
|
||||
});
|
||||
let request = Request::with_params("create_raid", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn delete_raid(&self, raid_id: u32) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({ "raid_id": raid_id });
|
||||
let request = Request::with_params("delete_raid", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn rebuild_raid(&self, raid_id: u32) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({ "raid_id": raid_id });
|
||||
let request = Request::with_params("rebuild_raid", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn initialize_disk(&self, disk_id: u32) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({ "disk_id": disk_id });
|
||||
let request = Request::with_params("initialize_disk", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn locate_disk(&self, disk_id: u32) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({ "disk_id": disk_id });
|
||||
let request = Request::with_params("locate_disk", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn remove_disk(&self, disk_id: u32, force: bool) -> Result<serde_json::Value> {
|
||||
let params = serde_json::json!({ "disk_id": disk_id, "force": force });
|
||||
let request = Request::with_params("remove_disk", params);
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
|
||||
pub async fn commit_config(&self) -> Result<serde_json::Value> {
|
||||
let request = Request::new("commit_config");
|
||||
let response = self.send_request(&request).await?;
|
||||
|
||||
if !response.is_success() {
|
||||
return Err(anyhow::anyhow!("Error: {:?}", response.error));
|
||||
}
|
||||
|
||||
response.data.ok_or_else(|| anyhow::anyhow!("No data in response"))
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for ServerClient {
|
||||
fn drop(&mut self) {
|
||||
tracing::info!("ServerClient dropped");
|
||||
}
|
||||
}
|
||||
69
src/protocol/messages.rs
Normal file
69
src/protocol/messages.rs
Normal file
@@ -0,0 +1,69 @@
|
||||
//! Protocol messages - shared with server
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Request {
|
||||
pub id: String,
|
||||
#[serde(rename = "type")]
|
||||
pub msg_type: String,
|
||||
pub action: String,
|
||||
#[serde(default)]
|
||||
pub params: serde_json::Value,
|
||||
}
|
||||
|
||||
impl Request {
|
||||
pub fn new(action: impl Into<String>) -> Self {
|
||||
Self {
|
||||
id: uuid::Uuid::new_v4().to_string(),
|
||||
msg_type: "request".to_string(),
|
||||
action: action.into(),
|
||||
params: serde_json::json!({}),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn with_params(action: impl Into<String>, params: serde_json::Value) -> Self {
|
||||
Self {
|
||||
id: uuid::Uuid::new_v4().to_string(),
|
||||
msg_type: "request".to_string(),
|
||||
action: action.into(),
|
||||
params,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Response {
|
||||
pub id: String,
|
||||
#[serde(rename = "type")]
|
||||
pub msg_type: String,
|
||||
pub status: String,
|
||||
#[serde(default)]
|
||||
pub data: Option<serde_json::Value>,
|
||||
#[serde(default)]
|
||||
pub error: Option<String>,
|
||||
}
|
||||
|
||||
impl Response {
|
||||
pub fn is_success(&self) -> bool {
|
||||
self.status == "success"
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Push {
|
||||
#[serde(rename = "type")]
|
||||
pub msg_type: String,
|
||||
pub event: String,
|
||||
#[serde(default)]
|
||||
pub data: Option<serde_json::Value>,
|
||||
}
|
||||
|
||||
pub mod actions {
|
||||
pub const GET_CONTROLLERS: &str = "get_controllers";
|
||||
pub const GET_CONTROLLER_INFO: &str = "get_controller_info";
|
||||
pub const GET_RAIDS: &str = "get_raids";
|
||||
pub const GET_DISKS: &str = "get_disks";
|
||||
pub const GET_EVENTS: &str = "get_events";
|
||||
pub const PING: &str = "ping";
|
||||
}
|
||||
7
src/protocol/mod.rs
Normal file
7
src/protocol/mod.rs
Normal file
@@ -0,0 +1,7 @@
|
||||
//! Client protocol module
|
||||
|
||||
pub mod client;
|
||||
pub mod messages;
|
||||
|
||||
pub use client::*;
|
||||
pub use messages::*;
|
||||
Reference in New Issue
Block a user