#[cfg(feature = "alloc")]
mod owned;
use core::{cmp::Ordering, convert::TryFrom, fmt, ops, str};
use crate::{
common::Hms6ColonStr,
error::{Error, ErrorKind},
rfc3339::SecfracStr,
};
#[cfg(feature = "alloc")]
pub use self::owned::PartialTimeString;
const PARTIAL_TIME_LEN_MIN: usize = 8;
fn validate_bytes(s: &[u8]) -> Result<(), Error> {
let (hms, dotfrac) = match s.len().cmp(&PARTIAL_TIME_LEN_MIN) {
Ordering::Greater => s.split_at(PARTIAL_TIME_LEN_MIN),
Ordering::Less => return Err(ErrorKind::TooShort.into()),
Ordering::Equal => return Hms6ColonStr::from_bytes(s).map(|_| ()),
};
debug_assert!(
!dotfrac.is_empty(),
"If `dotfrac` component is available, it should be non-empty string"
);
Hms6ColonStr::from_bytes(hms)?;
SecfracStr::from_bytes(dotfrac)?;
Ok(())
}
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[repr(transparent)]
#[allow(clippy::derive_hash_xor_eq)]
#[allow(unknown_lints, clippy::derive_ord_xor_partial_ord)]
pub struct PartialTimeStr([u8]);
impl PartialTimeStr {
#[inline]
#[must_use]
pub(crate) unsafe fn from_bytes_maybe_unchecked(s: &[u8]) -> &Self {
debug_assert_ok!(validate_bytes(s));
&*(s as *const [u8] as *const Self)
}
#[inline]
#[must_use]
pub(crate) unsafe fn from_bytes_maybe_unchecked_mut(s: &mut [u8]) -> &mut Self {
debug_assert_ok!(validate_bytes(s));
&mut *(s as *mut [u8] as *mut Self)
}
#[inline]
#[must_use]
unsafe fn from_str_maybe_unchecked_mut(s: &mut str) -> &mut Self {
Self::from_bytes_maybe_unchecked_mut(s.as_bytes_mut())
}
#[inline]
#[allow(clippy::should_implement_trait)]
pub fn from_str(s: &str) -> Result<&Self, Error> {
TryFrom::try_from(s)
}
#[inline]
pub fn from_mut_str(s: &mut str) -> Result<&mut Self, Error> {
TryFrom::try_from(s)
}
#[inline]
pub fn from_bytes(s: &[u8]) -> Result<&Self, Error> {
TryFrom::try_from(s)
}
#[inline]
pub fn from_bytes_mut(s: &mut [u8]) -> Result<&mut Self, Error> {
TryFrom::try_from(s)
}
#[inline]
#[must_use]
pub fn as_str(&self) -> &str {
unsafe {
debug_assert_safe_version_ok!(str::from_utf8(&self.0));
str::from_utf8_unchecked(&self.0)
}
}
#[inline]
#[must_use]
pub fn as_bytes(&self) -> &[u8] {
&self.0
}
#[inline]
#[must_use]
pub fn hms(&self) -> &Hms6ColonStr {
unsafe {
debug_assert_safe_version_ok!(Hms6ColonStr::from_bytes(
&self.0[..PARTIAL_TIME_LEN_MIN]
));
Hms6ColonStr::from_bytes_maybe_unchecked(self.0.get_unchecked(..PARTIAL_TIME_LEN_MIN))
}
}
#[inline]
#[must_use]
pub fn hms_mut(&mut self) -> &mut Hms6ColonStr {
unsafe {
debug_assert_ok!(Hms6ColonStr::from_bytes(&self.0[..PARTIAL_TIME_LEN_MIN]));
Hms6ColonStr::from_bytes_maybe_unchecked_mut(
self.0.get_unchecked_mut(..PARTIAL_TIME_LEN_MIN),
)
}
}
#[inline]
#[must_use]
pub fn secfrac(&self) -> Option<&SecfracStr> {
self.0.get(PARTIAL_TIME_LEN_MIN..).map(|v| {
unsafe {
debug_assert_safe_version_ok!(SecfracStr::from_bytes(v));
SecfracStr::from_bytes_maybe_unchecked(v)
}
})
}
#[inline]
#[must_use]
pub fn secfrac_mut(&mut self) -> Option<&mut SecfracStr> {
unsafe {
debug_assert_safe_version_ok!(self
.0
.get_mut(PARTIAL_TIME_LEN_MIN..)
.map(|v| SecfracStr::from_bytes(v))
.transpose());
self.0
.get_mut(PARTIAL_TIME_LEN_MIN..)
.map(|v| SecfracStr::from_bytes_maybe_unchecked_mut(v))
}
}
#[cfg(feature = "chrono04")]
#[cfg_attr(docsrs, doc(cfg(feature = "chrono04")))]
#[deprecated(since = "0.2.2", note = "renamed to `to_chrono04_naive_time`")]
#[inline]
#[must_use]
pub fn to_chrono_naive_time(&self) -> chrono04::NaiveTime {
self.to_chrono04_naive_time()
}
#[cfg(feature = "chrono04")]
#[cfg_attr(docsrs, doc(cfg(feature = "chrono04")))]
pub fn to_chrono04_naive_time(&self) -> chrono04::NaiveTime {
use chrono04::Timelike;
let hms: chrono04::NaiveTime = self.hms().into();
debug_assert!(hms.nanosecond() <= 1_000_000_000);
let secfrac: u32 = self
.secfrac()
.map_or(0, |secfrac| secfrac.digits().nanoseconds());
debug_assert!(secfrac < 1_000_000_000);
hms.with_nanosecond(hms.nanosecond() + secfrac)
.expect("Should never fail: `hms.nanoseconds() + secfrac` is less than 2 seconds")
}
#[cfg(feature = "time03")]
#[cfg_attr(docsrs, doc(cfg(feature = "time03")))]
pub fn to_time03_time(&self) -> time03::Time {
let hms = self.hms();
let hour = hms.hour();
let minute = hms.minute();
let second = hms.second().min(59);
let nanosecond = self
.secfrac()
.map_or(0, |secfrac| secfrac.digits().nanoseconds());
debug_assert!(nanosecond < 1_000_000_000);
time03::Time::from_hms_nano(hour, minute, second, nanosecond)
.expect("[validity] valid time and no leap seconds")
}
}
impl AsRef<[u8]> for PartialTimeStr {
#[inline]
fn as_ref(&self) -> &[u8] {
&self.0
}
}
impl AsRef<str> for PartialTimeStr {
#[inline]
fn as_ref(&self) -> &str {
self.as_str()
}
}
impl AsRef<PartialTimeStr> for PartialTimeStr {
#[inline]
fn as_ref(&self) -> &PartialTimeStr {
self
}
}
impl AsMut<PartialTimeStr> for PartialTimeStr {
#[inline]
fn as_mut(&mut self) -> &mut PartialTimeStr {
self
}
}
impl<'a> From<&'a PartialTimeStr> for &'a str {
#[inline]
fn from(v: &'a PartialTimeStr) -> Self {
v.as_str()
}
}
impl<'a> TryFrom<&'a [u8]> for &'a PartialTimeStr {
type Error = Error;
#[inline]
fn try_from(v: &'a [u8]) -> Result<Self, Self::Error> {
validate_bytes(v)?;
Ok(unsafe {
PartialTimeStr::from_bytes_maybe_unchecked(v)
})
}
}
impl<'a> TryFrom<&'a mut [u8]> for &'a mut PartialTimeStr {
type Error = Error;
#[inline]
fn try_from(v: &'a mut [u8]) -> Result<Self, Self::Error> {
validate_bytes(v)?;
Ok(unsafe {
PartialTimeStr::from_bytes_maybe_unchecked_mut(v)
})
}
}
impl<'a> TryFrom<&'a str> for &'a PartialTimeStr {
type Error = Error;
#[inline]
fn try_from(v: &'a str) -> Result<Self, Self::Error> {
Self::try_from(v.as_bytes())
}
}
impl<'a> TryFrom<&'a mut str> for &'a mut PartialTimeStr {
type Error = Error;
#[inline]
fn try_from(v: &'a mut str) -> Result<Self, Self::Error> {
validate_bytes(v.as_bytes())?;
Ok(unsafe {
PartialTimeStr::from_str_maybe_unchecked_mut(v)
})
}
}
impl fmt::Display for PartialTimeStr {
#[inline]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.as_str().fmt(f)
}
}
impl ops::Deref for PartialTimeStr {
type Target = str;
#[inline]
fn deref(&self) -> &Self::Target {
self.as_str()
}
}
impl_cmp_symmetric!(str, PartialTimeStr, &PartialTimeStr);
impl_cmp_symmetric!([u8], PartialTimeStr, [u8]);
impl_cmp_symmetric!([u8], PartialTimeStr, &[u8]);
impl_cmp_symmetric!([u8], &PartialTimeStr, [u8]);
impl_cmp_symmetric!(str, PartialTimeStr, str);
impl_cmp_symmetric!(str, PartialTimeStr, &str);
impl_cmp_symmetric!(str, &PartialTimeStr, str);
#[cfg(feature = "serde")]
#[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
impl serde::Serialize for PartialTimeStr {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serializer.serialize_str(self.as_str())
}
}
#[cfg(feature = "serde")]
mod serde_ {
use super::*;
use serde::de::{Deserialize, Deserializer, Visitor};
struct StrVisitor;
impl<'de> Visitor<'de> for StrVisitor {
type Value = &'de PartialTimeStr;
#[inline]
fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("RFC 3339 partial-time string")
}
#[inline]
fn visit_borrowed_bytes<E>(self, v: &'de [u8]) -> Result<Self::Value, E>
where
E: serde::de::Error,
{
Self::Value::try_from(v).map_err(E::custom)
}
#[inline]
fn visit_borrowed_str<E>(self, v: &'de str) -> Result<Self::Value, E>
where
E: serde::de::Error,
{
Self::Value::try_from(v).map_err(E::custom)
}
}
impl<'de> Deserialize<'de> for &'de PartialTimeStr {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_any(StrVisitor)
}
}
}
#[cfg(test)]
mod tests {
#[cfg(feature = "serde")]
use super::*;
use super::validate_bytes as s_validate;
#[cfg(feature = "serde")]
use serde_test::{assert_de_tokens, assert_tokens, Token};
#[test]
fn validate_bytes() {
assert!(s_validate(b"00:00:00").is_ok());
assert!(s_validate(b"00:00:00.0000").is_ok());
assert!(s_validate(b"00:00:00.00000000000").is_ok());
assert!(s_validate(b"12:34:56.7890").is_ok());
assert!(s_validate(b"23:59:60").is_ok());
assert!(s_validate(b"23:59:60.9999").is_ok());
assert!(s_validate(b"23:59:60.01234567890").is_ok());
assert!(s_validate(b"23:59:60.99999999999").is_ok());
}
#[cfg(feature = "serde")]
#[test]
fn ser_de_str() {
let raw: &'static str = "12:34:56.7890";
assert_tokens(
&PartialTimeStr::from_str(raw).unwrap(),
&[Token::BorrowedStr(raw)],
);
}
#[cfg(feature = "serde")]
#[test]
fn de_bytes_slice() {
let raw: &'static [u8; 13] = b"12:34:56.7890";
assert_de_tokens(
&PartialTimeStr::from_bytes(raw).unwrap(),
&[Token::BorrowedBytes(raw)],
);
}
}