use chrono::{DateTime, Utc, NaiveDate};
use serde::{Deserialize, Serialize};
use sqlx::FromRow;
use uuid::Uuid;
use super::MaintenanceScheduleStatus;
use super::AuditMetadata;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(transparent)]
pub struct MaintenanceScheduleId(pub Uuid);
impl MaintenanceScheduleId {
pub fn new(id: Uuid) -> Self { Self(id) }
pub fn generate() -> Self { Self(Uuid::new_v4()) }
pub fn into_inner(self) -> Uuid { self.0 }
}
impl std::fmt::Display for MaintenanceScheduleId {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.0)
}
}
impl std::str::FromStr for MaintenanceScheduleId {
type Err = uuid::Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
Ok(Self(Uuid::parse_str(s)?))
}
}
impl From<Uuid> for MaintenanceScheduleId {
fn from(id: Uuid) -> Self { Self(id) }
}
impl From<MaintenanceScheduleId> for Uuid {
fn from(id: MaintenanceScheduleId) -> Self { id.0 }
}
impl AsRef<Uuid> for MaintenanceScheduleId {
fn as_ref(&self) -> &Uuid { &self.0 }
}
impl std::ops::Deref for MaintenanceScheduleId {
type Target = Uuid;
fn deref(&self) -> &Self::Target { &self.0 }
}
#[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
pub struct MaintenanceSchedule {
pub id: Uuid,
pub asset_id: Uuid,
pub name: String,
pub interval_days: i32,
pub next_due_date: NaiveDate,
pub status: MaintenanceScheduleStatus,
#[serde(default)]
#[sqlx(json)]
pub metadata: AuditMetadata,
}
impl MaintenanceSchedule {
pub fn builder() -> MaintenanceScheduleBuilder {
<MaintenanceScheduleBuilder as Default>::default()
}
pub fn new(asset_id: Uuid, name: String, interval_days: i32, next_due_date: NaiveDate, status: MaintenanceScheduleStatus) -> Self {
Self {
id: Uuid::new_v4(),
asset_id,
name,
interval_days,
next_due_date,
status,
metadata: AuditMetadata::default(),
}
}
pub fn id(&self) -> &Uuid {
&self.id
}
pub fn typed_id(&self) -> MaintenanceScheduleId {
MaintenanceScheduleId(self.id)
}
pub fn created_at(&self) -> Option<&DateTime<Utc>> {
self.metadata.created_at.as_ref()
}
pub fn updated_at(&self) -> Option<&DateTime<Utc>> {
self.metadata.updated_at.as_ref()
}
pub fn is_deleted(&self) -> bool {
self.metadata.deleted_at.is_some()
}
pub fn is_active(&self) -> bool {
self.metadata.deleted_at.is_none()
}
pub fn deleted_at(&self) -> Option<&DateTime<Utc>> {
self.metadata.deleted_at.as_ref()
}
pub fn created_by(&self) -> Option<&Uuid> {
self.metadata.created_by.as_ref()
}
pub fn updated_by(&self) -> Option<&Uuid> {
self.metadata.updated_by.as_ref()
}
pub fn deleted_by(&self) -> Option<&Uuid> {
self.metadata.deleted_by.as_ref()
}
pub fn status(&self) -> &MaintenanceScheduleStatus {
&self.status
}
pub fn apply_patch(&mut self, fields: std::collections::HashMap<String, serde_json::Value>) {
for (key, value) in fields {
match key.as_str() {
"asset_id" => {
if let Ok(v) = serde_json::from_value(value) { self.asset_id = v; }
}
"name" => {
if let Ok(v) = serde_json::from_value(value) { self.name = v; }
}
"interval_days" => {
if let Ok(v) = serde_json::from_value(value) { self.interval_days = v; }
}
"next_due_date" => {
if let Ok(v) = serde_json::from_value(value) { self.next_due_date = v; }
}
"status" => {
if let Ok(v) = serde_json::from_value(value) { self.status = v; }
}
_ => {} }
}
}
}
impl super::Entity for MaintenanceSchedule {
type Id = Uuid;
fn entity_id(&self) -> &Self::Id {
&self.id
}
fn entity_type() -> &'static str {
"MaintenanceSchedule"
}
}
impl backbone_core::PersistentEntity for MaintenanceSchedule {
fn entity_id(&self) -> String {
self.id.to_string()
}
fn set_entity_id(&mut self, id: String) {
if let Ok(uuid) = uuid::Uuid::parse_str(&id) {
self.id = uuid;
}
}
fn created_at(&self) -> Option<chrono::DateTime<chrono::Utc>> {
self.metadata.created_at
}
fn set_created_at(&mut self, ts: chrono::DateTime<chrono::Utc>) {
self.metadata.created_at = Some(ts);
}
fn updated_at(&self) -> Option<chrono::DateTime<chrono::Utc>> {
self.metadata.updated_at
}
fn set_updated_at(&mut self, ts: chrono::DateTime<chrono::Utc>) {
self.metadata.updated_at = Some(ts);
}
fn deleted_at(&self) -> Option<chrono::DateTime<chrono::Utc>> {
self.metadata.deleted_at
}
fn set_deleted_at(&mut self, ts: Option<chrono::DateTime<chrono::Utc>>) {
self.metadata.deleted_at = ts;
}
}
impl backbone_orm::EntityRepoMeta for MaintenanceSchedule {
fn column_types() -> std::collections::HashMap<String, String> {
let mut m = std::collections::HashMap::new();
m.insert("id".to_string(), "uuid".to_string());
m.insert("asset_id".to_string(), "uuid".to_string());
m.insert("status".to_string(), "maintenance_schedule_status".to_string());
m.insert("next_due_date".to_string(), "date".to_string());
m
}
fn search_fields() -> &'static [&'static str] {
&["name"]
}
}
#[derive(Debug, Clone, Default)]
pub struct MaintenanceScheduleBuilder {
asset_id: Option<Uuid>,
name: Option<String>,
interval_days: Option<i32>,
next_due_date: Option<NaiveDate>,
status: Option<MaintenanceScheduleStatus>,
}
impl MaintenanceScheduleBuilder {
pub fn asset_id(mut self, value: Uuid) -> Self {
self.asset_id = Some(value);
self
}
pub fn name(mut self, value: String) -> Self {
self.name = Some(value);
self
}
pub fn interval_days(mut self, value: i32) -> Self {
self.interval_days = Some(value);
self
}
pub fn next_due_date(mut self, value: NaiveDate) -> Self {
self.next_due_date = Some(value);
self
}
pub fn status(mut self, value: MaintenanceScheduleStatus) -> Self {
self.status = Some(value);
self
}
pub fn build(self) -> Result<MaintenanceSchedule, String> {
let asset_id = self.asset_id.ok_or_else(|| "asset_id is required".to_string())?;
let name = self.name.ok_or_else(|| "name is required".to_string())?;
let interval_days = self.interval_days.ok_or_else(|| "interval_days is required".to_string())?;
let next_due_date = self.next_due_date.ok_or_else(|| "next_due_date is required".to_string())?;
Ok(MaintenanceSchedule {
id: Uuid::new_v4(),
asset_id,
name,
interval_days,
next_due_date,
status: self.status.unwrap_or_default(),
metadata: AuditMetadata::default(),
})
}
}