taler-rust

timestamp.rs (6830B)

---

 /*
   This file is part of TALER
   Copyright (C) 2024, 2025, 2026 Taler Systems SA
 
   TALER is free software; you can redistribute it and/or modify it under the
   terms of the GNU Affero General Public License as published by the Free Software
   Foundation; either version 3, or (at your option) any later version.
 
   TALER is distributed in the hope that it will be useful, but WITHOUT ANY
   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
   A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more details.
 
   You should have received a copy of the GNU Affero General Public License along with
   TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
 */
 
 use std::{fmt::Display, ops::Add, str::FromStr, time::Duration};
 
 use jiff::{Timestamp, civil::Time, tz::TimeZone};
 use serde::{Deserialize, Deserializer, Serialize, Serializer, de::Error, ser::SerializeStruct}; // codespell:ignore
 use serde_json::Value;
 
 /// <https://docs.taler.net/core/api-common.html#tsref-type-Timestamp>
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
 pub enum TalerTimestamp {
     Never,
     Timestamp(Timestamp),
 }
 
 impl FromStr for TalerTimestamp {
     type Err = anyhow::Error;
 
     fn from_str(s: &str) -> Result<Self, Self::Err> {
         if s == "never" {
             return Ok(Self::Never);
         }
         let s: i64 = s.parse()?;
 
         Ok(Self::Timestamp(jiff::Timestamp::from_second(s)?))
     }
 }
 
 #[derive(Serialize, Deserialize)]
 struct TimestampImpl {
     t_s: Value,
 }
 
 impl<'de> Deserialize<'de> for TalerTimestamp {
     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
     where
         D: Deserializer<'de>,
     {
         let tmp = TimestampImpl::deserialize(deserializer)?;
         match tmp.t_s {
             Value::Number(s) => {
                 if let Some(since_epoch_s) = s.as_i64() {
                     jiff::Timestamp::from_second(since_epoch_s)
                         .map(Self::Timestamp)
                         .map_err(Error::custom)
                 } else {
                     Err(Error::custom("Expected epoch time"))
                 }
             }
             Value::String(str) if str == "never" => Ok(Self::Never),
             _ => Err(Error::custom("Expected epoch time or 'never'")),
         }
     }
 }
 
 impl Serialize for TalerTimestamp {
     fn serialize<S>(&self, se: S) -> Result<S::Ok, S::Error>
     where
         S: Serializer,
     {
         let mut se_struct = se.serialize_struct("Timestamp", 1)?;
         match self {
             TalerTimestamp::Never => se_struct.serialize_field("t_s", "never")?,
             TalerTimestamp::Timestamp(timestamp) => {
                 se_struct.serialize_field("t_s", &timestamp.as_second())?
             }
         }
         se_struct.end()
     }
 }
 
 impl Display for TalerTimestamp {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
             TalerTimestamp::Never => f.write_str("never"),
             TalerTimestamp::Timestamp(timestamp) => timestamp.fmt(f),
         }
     }
 }
 
 impl From<jiff::Timestamp> for TalerTimestamp {
     fn from(time: jiff::Timestamp) -> Self {
         Self::Timestamp(time)
     }
 }
 
 impl From<jiff::civil::Date> for TalerTimestamp {
     fn from(date: jiff::civil::Date) -> Self {
         date.to_datetime(Time::midnight())
             .to_zoned(TimeZone::UTC)
             .unwrap()
             .timestamp()
             .into()
     }
 }
 
 impl Add<jiff::Span> for TalerTimestamp {
     type Output = Self;
 
     fn add(self, rhs: jiff::Span) -> Self::Output {
         match self {
             TalerTimestamp::Never => TalerTimestamp::Never,
             TalerTimestamp::Timestamp(timestamp) => TalerTimestamp::Timestamp(timestamp + rhs),
         }
     }
 }
 
 impl sqlx::Type<sqlx::Postgres> for TalerTimestamp {
     fn type_info() -> <sqlx::Postgres as sqlx::Database>::TypeInfo {
         Option::<i64>::type_info()
     }
 }
 
 impl<'q> sqlx::Encode<'q, sqlx::Postgres> for TalerTimestamp {
     fn encode_by_ref(
         &self,
         buf: &mut <sqlx::Postgres as sqlx::Database>::ArgumentBuffer<'q>,
     ) -> Result<sqlx::encode::IsNull, sqlx::error::BoxDynError> {
         match self {
             TalerTimestamp::Never => None,
             TalerTimestamp::Timestamp(timestamp) => Some(timestamp.as_microsecond()),
         }
         .encode_by_ref(buf)
     }
 }
 
 impl<'r> sqlx::Decode<'r, sqlx::Postgres> for TalerTimestamp {
     fn decode(
         value: <sqlx::Postgres as sqlx::Database>::ValueRef<'r>,
     ) -> Result<Self, sqlx::error::BoxDynError> {
         let micros = Option::<i64>::decode(value)?;
         Ok(match micros {
             Some(micros) => Self::Timestamp(Timestamp::from_microsecond(micros)?),
             None => Self::Never,
         })
     }
 }
 
 /// <https://docs.taler.net/core/api-common.html#tsref-type-RelativeTime>
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
 pub enum RelativeTime {
     Forever,
     Duration(Duration),
 }
 
 impl FromStr for RelativeTime {
     type Err = anyhow::Error;
 
     fn from_str(s: &str) -> Result<Self, Self::Err> {
         if s == "forever" {
             return Ok(Self::Forever);
         }
         let micros: u64 = s.parse()?;
 
         Ok(Self::Duration(Duration::from_micros(micros)))
     }
 }
 
 #[derive(Serialize, Deserialize)]
 struct RelativeTimeImpl {
     d_us: Value,
 }
 
 impl<'de> Deserialize<'de> for RelativeTime {
     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
     where
         D: Deserializer<'de>,
     {
         let tmp = RelativeTimeImpl::deserialize(deserializer)?;
         match tmp.d_us {
             Value::Number(s) => {
                 if let Some(micros) = s.as_u64() {
                     Ok(Self::Duration(Duration::from_micros(micros)))
                 } else {
                     Err(Error::custom("Expected microseconds"))
                 }
             }
             Value::String(str) if str == "forever" => Ok(Self::Forever),
             _ => Err(Error::custom("Expected epoch time or 'forever'")),
         }
     }
 }
 
 impl Serialize for RelativeTime {
     fn serialize<S>(&self, se: S) -> Result<S::Ok, S::Error>
     where
         S: Serializer,
     {
         let mut se_struct = se.serialize_struct("RelativeTime", 1)?;
         match self {
             RelativeTime::Forever => se_struct.serialize_field("d_us", "forever")?,
             RelativeTime::Duration(duration) => {
                 se_struct.serialize_field("d_us", &duration.as_micros())?
             }
         }
         se_struct.end()
     }
 }
 
 impl Display for RelativeTime {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
             RelativeTime::Forever => f.write_str("forever"),
             RelativeTime::Duration(duration) => write!(f, "{duration:?}"),
         }
     }
 }