naskya/napi-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat(napi): implement From/ToNapiValue for chrono::DateTime<Local> and chrono::DateTime<FixedOffset> (#1902)

Browse source
This commit is contained in:
sup39 2024-04-29 18:50:39 +09:00 committed by naskya
parent a54b759bb1
commit 6a9ae1322d
Signed by: naskya
GPG key ID: 712D413B3A9FED5C
1 changed files with 18 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
crates/napi/src/bindgen_runtime/js_values/date.rs
View file

@ -1,10 +1,10 @@
use std::{ptr, str::FromStr}; use std::{ptr, str::FromStr};
use chrono::{DateTime, NaiveDateTime, Utc}; use chrono::{DateTime, Local, LocalResult, NaiveDateTime, TimeZone};
use crate::{bindgen_prelude::*, check_status, sys, ValueType}; use crate::{bindgen_prelude::*, check_status, sys, ValueType};
impl TypeName for DateTime<Utc> { impl<Tz: TimeZone> TypeName for DateTime<Tz> {
fn type_name() -> &'static str { fn type_name() -> &'static str {
"DateTime" "DateTime"
} }
@ -14,7 +14,7 @@ impl TypeName for DateTime<Utc> {
} }
} }
impl ValidateNapiValue for DateTime<Utc> { impl<Tz: TimeZone> ValidateNapiValue for DateTime<Tz> {
unsafe fn validate(env: sys::napi_env, napi_val: sys::napi_value) -> Result<sys::napi_value> { unsafe fn validate(env: sys::napi_env, napi_val: sys::napi_value) -> Result<sys::napi_value> {
let mut is_date = false; let mut is_date = false;
check_status!(unsafe { sys::napi_is_date(env, napi_val, &mut is_date) })?; check_status!(unsafe { sys::napi_is_date(env, napi_val, &mut is_date) })?;
@ -120,39 +120,38 @@ impl FromNapiValue for NaiveDateTime {
} }
} }
impl ToNapiValue for DateTime<Utc> { impl<Tz: TimeZone> ToNapiValue for DateTime<Tz> {
unsafe fn to_napi_value(env: sys::napi_env, val: DateTime<Utc>) -> Result<sys::napi_value> { unsafe fn to_napi_value(env: sys::napi_env, val: DateTime<Tz>) -> Result<sys::napi_value> {
let mut ptr = std::ptr::null_mut(); let mut ptr = std::ptr::null_mut();
let millis_since_epoch_utc = val.timestamp_millis() as f64; let millis_since_epoch_utc = val.timestamp_millis() as f64;
check_status!( check_status!(
unsafe { sys::napi_create_date(env, millis_since_epoch_utc, &mut ptr) }, unsafe { sys::napi_create_date(env, millis_since_epoch_utc, &mut ptr) },
"Failed to convert rust type `DateTime<Utc>` into napi value", "Failed to convert rust type `DateTime` into napi value",
)?; )?;
Ok(ptr) Ok(ptr)
} }
} }
impl FromNapiValue for DateTime<Utc> { impl<Tz: TimeZone> FromNapiValue for DateTime<Tz>
where
DateTime<Tz>: From<DateTime<Local>>,
{
unsafe fn from_napi_value(env: sys::napi_env, napi_val: sys::napi_value) -> Result<Self> { unsafe fn from_napi_value(env: sys::napi_env, napi_val: sys::napi_value) -> Result<Self> {
let mut milliseconds_since_epoch_utc = 0.0; let mut milliseconds_since_epoch_utc = 0.0;
check_status!( check_status!(
unsafe { sys::napi_get_date_value(env, napi_val, &mut milliseconds_since_epoch_utc) }, unsafe { sys::napi_get_date_value(env, napi_val, &mut milliseconds_since_epoch_utc) },
"Failed to convert napi value into rust type `DateTime<Utc>`", "Failed to convert napi value into rust type `DateTime`",
)?; )?;
let milliseconds_since_epoch_utc = milliseconds_since_epoch_utc as i64; match Local.timestamp_millis_opt(milliseconds_since_epoch_utc as i64) {
let timestamp_seconds = milliseconds_since_epoch_utc / 1_000; LocalResult::Single(dt) => Ok(dt.into()),
let naive = DateTime::from_timestamp( _ => Err(Error::new(
timestamp_seconds, Status::DateExpected,
(milliseconds_since_epoch_utc % 1_000 * 1_000_000) as u32, "Found invalid date".to_owned(),
) )),
.ok_or_else(|| Error::new(Status::DateExpected, "Found invalid date".to_owned()))?; }
Ok(DateTime::<Utc>::from_naive_utc_and_offset(
naive.naive_utc(),
Utc,
))
} }
} }