use serde::de;
use serde::ser;
use crate::__constants::TIMESTAMP;
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Timestamp(i64);
impl Timestamp {
pub fn into_inner(self) -> i64 {
self.0
}
}
impl From<i64> for Timestamp {
fn from(val: i64) -> Self {
Self(val)
}
}
impl Timestamp {
pub fn from_milliseconds(milliseconds: i64) -> Self {
Self(milliseconds)
}
pub fn milliseconds(&self) -> i64 {
self.0
}
}
impl ser::Serialize for Timestamp {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serializer.serialize_newtype_struct(TIMESTAMP, &self.0)
}
}
struct Visitor {}
impl<'de> de::Visitor<'de> for Visitor {
type Value = Timestamp;
fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
formatter.write_str("struct Timestamp")
}
fn visit_i64<E>(self, v: i64) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(Timestamp::from(v))
}
}
impl<'de> de::Deserialize<'de> for Timestamp {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
deserializer.deserialize_newtype_struct(TIMESTAMP, Visitor {})
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "time")))]
#[cfg(feature = "time")]
impl From<time::OffsetDateTime> for Timestamp {
fn from(val: time::OffsetDateTime) -> Self {
Self((val.unix_timestamp_nanos() / 1_000_000) as i64)
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "time")))]
#[cfg(feature = "time")]
impl From<Timestamp> for time::OffsetDateTime {
fn from(val: Timestamp) -> Self {
let millis = val.0;
let duration = time::Duration::milliseconds(millis);
time::OffsetDateTime::UNIX_EPOCH + duration
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "time")))]
#[cfg(feature = "time")]
impl From<time::Duration> for Timestamp {
fn from(val: time::Duration) -> Self {
Self(val.whole_milliseconds() as i64)
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "time")))]
#[cfg(feature = "time")]
impl From<Timestamp> for time::Duration {
fn from(value: Timestamp) -> Self {
time::Duration::milliseconds(value.0)
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "chrono")))]
#[cfg(feature = "chrono")]
impl From<chrono::Duration> for Timestamp {
fn from(val: chrono::Duration) -> Self {
Self(val.num_milliseconds())
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "chrono")))]
#[cfg(feature = "chrono")]
impl From<Timestamp> for chrono::Duration {
fn from(value: Timestamp) -> Self {
chrono::Duration::milliseconds(value.0)
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "chrono")))]
#[cfg(feature = "chrono")]
impl From<chrono::DateTime<chrono::Utc>> for Timestamp {
fn from(val: chrono::DateTime<chrono::Utc>) -> Self {
Self(val.timestamp_millis())
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "chrono")))]
#[cfg(feature = "chrono")]
impl TryFrom<Timestamp> for chrono::DateTime<chrono::Utc> {
type Error = Timestamp;
fn try_from(value: Timestamp) -> Result<Self, Self::Error> {
let datetime =
chrono::DateTime::from_timestamp_millis(value.milliseconds()).ok_or(value)?;
Ok(chrono::DateTime::<chrono::Utc>::from_naive_utc_and_offset(
datetime.naive_utc(),
chrono::Utc,
))
}
}
#[cfg_attr(docsrs, doc(cfg(feature = "chrono")))]
#[cfg(feature = "chrono")]
impl From<Timestamp> for Option<chrono::DateTime<chrono::Utc>> {
fn from(value: Timestamp) -> Self {
let datetime = chrono::DateTime::from_timestamp_millis(value.milliseconds())
.ok_or(value)
.ok()?;
Some(chrono::DateTime::<chrono::Utc>::from_naive_utc_and_offset(
datetime.naive_utc(),
chrono::Utc,
))
}
}