121 lines
3.6 KiB
Rust
121 lines
3.6 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use uuid::Uuid;
|
|
use yewdux::{prelude::*, storage};
|
|
|
|
use plate_tool_lib::plate::*;
|
|
use plate_tool_lib::plate_instances::PlateInstance;
|
|
use plate_tool_lib::transfer::Transfer;
|
|
|
|
#[derive(Debug, Default, Clone, PartialEq, Serialize, Deserialize, Store)]
|
|
#[store(storage = "session")]
|
|
#[non_exhaustive]
|
|
pub struct CurrentTransfer {
|
|
pub transfer: Transfer,
|
|
}
|
|
|
|
#[derive(PartialEq, Clone, Copy, Serialize, Deserialize)]
|
|
pub struct Preferences {
|
|
pub in_transfer_hashes: bool,
|
|
}
|
|
|
|
impl Default for Preferences {
|
|
fn default() -> Self {
|
|
Self { in_transfer_hashes: true }
|
|
}
|
|
}
|
|
|
|
#[derive(Default, PartialEq, Clone, Serialize, Deserialize)]
|
|
#[non_exhaustive]
|
|
pub struct MainState {
|
|
pub source_plates: Vec<PlateInstance>,
|
|
pub destination_plates: Vec<PlateInstance>,
|
|
pub transfers: Vec<Transfer>,
|
|
pub selected_source_plate: Uuid,
|
|
pub selected_dest_plate: Uuid,
|
|
pub selected_transfer: Uuid,
|
|
|
|
#[serde(default)]
|
|
pub preferences: Preferences,
|
|
}
|
|
|
|
impl Store for MainState {
|
|
fn new() -> Self {
|
|
init_listener(storage::StorageListener::<Self>::new(storage::Area::Local));
|
|
|
|
storage::load(storage::Area::Local)
|
|
.expect("Unable to load state")
|
|
.unwrap_or_default()
|
|
}
|
|
|
|
fn should_notify(&self, old: &Self) -> bool {
|
|
self != old
|
|
}
|
|
}
|
|
|
|
impl MainState {
|
|
fn purge_transfers(&mut self) {
|
|
// Removes any transfers for which the associated plates are gone
|
|
self.transfers.retain(|tr| {
|
|
self.source_plates
|
|
.iter()
|
|
.any(|spi| spi.get_uuid() == tr.source_id)
|
|
&& self
|
|
.destination_plates
|
|
.iter()
|
|
.any(|dpi| dpi.get_uuid() == tr.dest_id)
|
|
});
|
|
}
|
|
|
|
pub fn add_source_plate(&mut self, plate: PlateInstance) {
|
|
assert!(plate.plate.plate_type == PlateType::Source);
|
|
self.source_plates.push(plate);
|
|
}
|
|
pub fn add_dest_plate(&mut self, plate: PlateInstance) {
|
|
assert!(plate.plate.plate_type == PlateType::Destination);
|
|
self.destination_plates.push(plate);
|
|
}
|
|
pub fn del_plate(&mut self, id: Uuid) {
|
|
if let Some(index) = self
|
|
.source_plates
|
|
.iter()
|
|
.position(|spi| spi.get_uuid() == id)
|
|
{
|
|
self.source_plates.swap_remove(index);
|
|
self.purge_transfers();
|
|
}
|
|
if let Some(index) = self
|
|
.destination_plates
|
|
.iter()
|
|
.position(|dpi| dpi.get_uuid() == id)
|
|
{
|
|
self.destination_plates.swap_remove(index);
|
|
self.purge_transfers();
|
|
}
|
|
}
|
|
pub fn rename_plate(&mut self, id: Uuid, new_name: &str) {
|
|
if let Some(index) = self
|
|
.source_plates
|
|
.iter()
|
|
.position(|spi| spi.get_uuid() == id)
|
|
{
|
|
self.source_plates[index].change_name(new_name.to_string());
|
|
}
|
|
if let Some(index) = self
|
|
.destination_plates
|
|
.iter()
|
|
.position(|dpi| dpi.get_uuid() == id)
|
|
{
|
|
self.destination_plates[index].change_name(new_name.to_string());
|
|
}
|
|
}
|
|
|
|
pub fn change_format(&mut self, id: Uuid, new_format: &PlateFormat) {
|
|
if let Some(index) = self.source_plates.iter().position(|spi| spi.get_uuid() == id) {
|
|
self.source_plates[index].change_format(new_format);
|
|
}
|
|
if let Some(index) = self.destination_plates.iter().position(|dpi| dpi.get_uuid() == id) {
|
|
self.destination_plates[index].change_format(new_format);
|
|
}
|
|
}
|
|
}
|