Line data    Source code

       1              : #[cfg(feature = "testing")]
       2              : use camino::Utf8PathBuf;
       3              : use serde::{Deserialize, Serialize};
       4              : 
       5           12 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
       6              : pub enum Root {
       7              :     V1(V1),
       8              : }
       9           12 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
      10              : pub enum V1 {
      11              :     InProgress(InProgress),
      12              :     Done(Done),
      13              : }
      14              : 
      15           12 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
      16              : #[serde(transparent)]
      17              : pub struct IdempotencyKey(String);
      18              : 
      19              : impl IdempotencyKey {
      20           12 :     pub fn new(s: String) -> Self {
      21           12 :         Self(s)
      22           12 :     }
      23              : }
      24              : 
      25            0 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
      26              : pub struct InProgress {
      27              :     pub idempotency_key: IdempotencyKey,
      28              :     pub location: Location,
      29              :     pub started_at: chrono::NaiveDateTime,
      30              : }
      31              : 
      32           36 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
      33              : pub struct Done {
      34              :     pub idempotency_key: IdempotencyKey,
      35              :     pub started_at: chrono::NaiveDateTime,
      36              :     pub finished_at: chrono::NaiveDateTime,
      37              : }
      38              : 
      39            0 : #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
      40              : pub enum Location {
      41              :     #[cfg(feature = "testing")]
      42              :     LocalFs { path: Utf8PathBuf },
      43              :     AwsS3 {
      44              :         region: String,
      45              :         bucket: String,
      46              :         key: String,
      47              :     },
      48              : }
      49              : 
      50              : impl Root {
      51            0 :     pub fn is_done(&self) -> bool {
      52            0 :         match self {
      53            0 :             Root::V1(v1) => match v1 {
      54            0 :                 V1::Done(_) => true,
      55            0 :                 V1::InProgress(_) => false,
      56              :             },
      57              :         }
      58            0 :     }
      59            0 :     pub fn idempotency_key(&self) -> &IdempotencyKey {
      60            0 :         match self {
      61            0 :             Root::V1(v1) => match v1 {
      62            0 :                 V1::InProgress(in_progress) => &in_progress.idempotency_key,
      63            0 :                 V1::Done(done) => &done.idempotency_key,
      64              :             },
      65              :         }
      66            0 :     }
      67              : }