progress-pile/progress-pile-server/src/entity/progress_entry.rs

use async_graphql::*;
use chrono::Local;
use sea_orm::{entity::prelude::*, ActiveValue::Set};
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
#[sea_orm(table_name = "progress_entry")]
pub struct Model {
    #[sea_orm(primary_key, auto_increment = false)]
    pub id: Uuid,
    #[sea_orm(primary_key, auto_increment = false)]
    pub user_id: i32,
    #[sea_orm(indexed)]
    pub progress_category_id: Uuid,
    #[sea_orm(indexed)]
    pub created_at: DateTimeWithTimeZone,
    #[sea_orm(indexed)]
    pub updated_at: DateTimeWithTimeZone,
    #[sea_orm(indexed)]
    pub deleted_at: Option<DateTimeWithTimeZone>,
    #[sea_orm(indexed)]
    pub progressed_at: DateTimeWithTimeZone,
    pub quantity: i32,
    pub note: String,
}

#[derive(Copy, Clone, Debug, DeriveRelation, EnumIter)]
pub enum Relation {
    #[sea_orm(
        belongs_to = "super::progress_category::Entity",
        from = "Column::ProgressCategoryId",
        to = "super::progress_category::Column::Id"
    )]
    ProgressCategory,
    #[sea_orm(
        belongs_to = "super::user::Entity",
        from = "Column::UserId",
        to = "super::user::Column::Id"
    )]
    User,
}


impl Related<super::progress_category::Entity> for Entity {
    fn to() -> RelationDef {
        Relation::ProgressCategory.def()
    }
}

impl Related<super::user::Entity> for Entity {
    fn to() -> RelationDef {
        Relation::User.def()
    }
}

impl ActiveModelBehavior for ActiveModel {}

impl ActiveModel {
    pub fn new() -> Self {
        let timestamp: DateTimeWithTimeZone = Local::now().fixed_offset();
        Self{
            id: Set(Uuid::new_v4()),
            created_at: Set(timestamp),
            updated_at: Set(timestamp),
            progressed_at: Set(timestamp),
            quantity: Set(1),
            note: Set("".to_string()),
            ..Default::default()
        }
    }
}