signer-daemon 0.2.3

use sea_orm::{
  ActiveModelTrait, ActiveValue::Set, DatabaseTransaction, EntityTrait,
};

use crate::{
  SignerDaemon,
  entity::user,
  model::{deltaobject::user_do::UserDO, viewobject::UserVO},
};

use super::{CrdtType, crdt::CrdtDelta};

impl CrdtType for UserDO {
  type PrimaryKey = String;

  async fn put(
    &self,
    _daemon: &SignerDaemon,
    tx: &DatabaseTransaction,
  ) -> anyhow::Result<CrdtDelta<Self>> {
    let u = user::Entity::find_by_id(&self.pub_key).one(tx).await?;
    if let Some(val) = &u {
      if !self.has_changed(&val.clone().into()) {
        return Ok(CrdtDelta::Skip);
      }
    }

    let m = user::ActiveModel {
      pub_key: Set(self.pub_key.clone()),
      user_public: self.user_public.clone().into(),
      signed_user_public: self.signed_user_public.clone().into(),
    };

    match u {
      Some(val) => {
        let m = m.update(tx).await?;
        let user_do = UserDO::new(&m.into(), &val.into())?;
        Ok(CrdtDelta::Put(user_do))
      }
      None => {
        let m = m.insert(tx).await?;
        Ok(CrdtDelta::Del(m.pub_key))
      }
    }
  }

  async fn del(
    _daemon: &SignerDaemon,
    key: &String,
    tx: &DatabaseTransaction,
  ) -> anyhow::Result<CrdtDelta<Self>> {
    let m = user::Entity::find_by_id(key).one(tx).await?;
    match m {
      Some(val) => {
        let user_do = UserDO::from(UserVO::from(val.clone()));
        user::Entity::delete_by_id(&val.pub_key).exec(tx).await?;

        Ok(CrdtDelta::Put(user_do))
      }
      None => Ok(CrdtDelta::Skip),
    }
  }
}