1 files changed, 125 insertions, 0 deletions
diff --git a/lib/sellershut-core/src/google/helpers.rs b/lib/sellershut-core/src/google/helpers.rs
new file mode 100644
index 0000000..2d75655
--- /dev/null
+++ b/lib/sellershut-core/src/google/helpers.rs
@@ -0,0 +1,125 @@
+use crate::google::protobuf::Timestamp;
+
+#[derive(serde::Serialize, serde::Deserialize)]
+#[serde(untagged)]
+/// Date utility
+#[derive(Clone, Debug)]
+pub enum DateItem {
+    /// string
+    String(String),
+    /// ts
+    Timestamp { seconds: i64, nanos: i32 },
+}
+
+impl TryFrom<DateItem> for Timestamp {
+    type Error = time::Error;
+
+    fn try_from(value: DateItem) -> Result<Self, Self::Error> {
+        match value {
+            DateItem::String(ref string) => <Timestamp as std::str::FromStr>::from_str(string),
+            DateItem::Timestamp { seconds, nanos } => Ok(Self { seconds, nanos }),
+        }
+    }
+}
+
+impl From<time::OffsetDateTime> for Timestamp {
+    fn from(dt: time::OffsetDateTime) -> Self {
+        Timestamp {
+            seconds: dt.unix_timestamp(),
+            nanos: dt.nanosecond() as i32,
+        }
+    }
+}
+
+impl From<Timestamp> for String {
+    fn from(value: Timestamp) -> Self {
+        let odt = time::OffsetDateTime::try_from(value).expect("invalid date");
+        odt.format(&time::format_description::well_known::Rfc3339)
+            .expect("format is not rfc3339")
+    }
+}
+
+impl std::str::FromStr for Timestamp {
+    type Err = time::Error;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        let timestamp =
+            time::OffsetDateTime::parse(s, &time::format_description::well_known::Rfc3339)?;
+
+        Ok(Timestamp::from(timestamp))
+    }
+}
+
+impl TryFrom<String> for Timestamp {
+    type Error = time::Error;
+
+    fn try_from(value: String) -> Result<Self, Self::Error> {
+        <Timestamp as std::str::FromStr>::from_str(&value)
+    }
+}
+
+impl TryFrom<Timestamp> for time::OffsetDateTime {
+    type Error = time::Error;
+
+    fn try_from(value: Timestamp) -> Result<Self, Self::Error> {
+        let dt = time::OffsetDateTime::from_unix_timestamp(value.seconds)?;
+
+        Ok(dt.replace_nanosecond(value.nanos as u32)?)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use time::{Duration, OffsetDateTime};
+
+    #[test]
+    fn test_offsetdatetime_to_timestamp() {
+        let now = OffsetDateTime::now_utc();
+        let timestamp: Timestamp = now.into();
+
+        assert_eq!(timestamp.seconds, now.unix_timestamp());
+        assert_eq!(timestamp.nanos, now.nanosecond() as i32);
+    }
+
+    #[test]
+    fn test_timestamp_to_offsetdatetime() {
+        let now = OffsetDateTime::now_utc();
+        let timestamp: Timestamp = now.into();
+        let dt: OffsetDateTime = timestamp.try_into().unwrap();
+
+        assert_eq!(dt, now);
+    }
+
+    #[test]
+    fn test_timestamp_to_offsetdatetime_with_nanos() {
+        let now = OffsetDateTime::now_utc();
+        let nanos = 123456789;
+        let dt = now + Duration::nanoseconds(nanos);
+        let timestamp: Timestamp = dt.into();
+        let dt_from_timestamp: OffsetDateTime = timestamp.try_into().unwrap();
+
+        assert_eq!(dt_from_timestamp, dt);
+    }
+
+    #[test]
+    fn test_timestamp_to_offsetdatetime_with_negative_nanos() {
+        let now = OffsetDateTime::now_utc();
+        let nanos = -123456789;
+        let dt = now + Duration::nanoseconds(nanos);
+        let timestamp: Timestamp = dt.into();
+        let dt_from_timestamp: OffsetDateTime = timestamp.try_into().unwrap();
+
+        assert_eq!(dt_from_timestamp, dt);
+    }
+
+    #[test]
+    fn test_timestamp_to_offsetdatetime_invalid_seconds() {
+        let timestamp = Timestamp {
+            seconds: i64::MIN,
+            nanos: 0,
+        };
+        let result: Result<OffsetDateTime, time::Error> = timestamp.try_into();
+        assert!(result.is_err());
+    }
+}