elastic_types 0.20.10

/*!
Functions that are exported and used by `elastic_types_derive`.

This module is 'private' and should only be consumed by `elastic_types_derive`.
Its contents aren't subject to SemVer.
*/

use serde_json;
use serde::Serialize;
use chrono::{DateTime, Utc};
use chrono::format::{self, Parsed};

use private::field::{DocumentField, FieldMapping, FieldType};

pub use date::{DateFormat, DateValue, FormattedDate, ParseError};
pub use document::DocumentType;
pub use document::mapping::{DocumentMapping, PropertiesMapping};

pub use chrono::format::{Fixed, Item, Numeric, Pad};
pub use serde::ser::SerializeStruct;

/** Get the mapping for a field. */
#[inline]
pub fn mapping<TField, TMapping, TPivot>() -> TMapping
where
    TField: FieldType<TMapping, TPivot>,
    TMapping: FieldMapping<TPivot>,
    TPivot: Default,
{
    TMapping::default()
}

/** Serialise a field mapping as a field using the given serialiser. */
#[inline]
pub fn field_ser<S, TMapping, TPivot>(state: &mut S, field: &'static str, _: TMapping) -> Result<(), S::Error>
where
    S: SerializeStruct,
    TMapping: FieldMapping<TPivot>,
    TPivot: Default,
    DocumentField<TMapping, TPivot>: Serialize,
{
    state.serialize_field(field, &DocumentField::<TMapping, TPivot>::default())
}

/**
Serialize a field individually.

This method isn't intended to be used publicly, but is useful in the docs.
*/
#[inline]
pub fn standalone_field_ser<TMapping, TPivot>(_: TMapping) -> Result<String, serde_json::Error>
where
    TMapping: FieldMapping<TPivot>,
    TPivot: Default,
    DocumentField<TMapping, TPivot>: Serialize,
{
    serde_json::to_string(&DocumentField::<TMapping, TPivot>::default())
}

/** Parse a date string using an owned slice of items. */
pub fn parse_from_tokens<'a>(date: &str, fmt: Vec<Item<'a>>) -> Result<DateValue, ParseError> {
    let mut parsed = Parsed::new();
    match format::parse(&mut parsed, date, fmt.into_iter()) {
        Ok(_) => {
            // If the parsed result doesn't contain any time, set it to the default
            if parsed.hour_mod_12.is_none() {
                let _ = parsed.set_hour(0);
                let _ = parsed.set_minute(0);
            }

            // Set the DateTime result
            let naive_date = parsed.to_naive_datetime_with_offset(0)?;

            let date = DateTime::from_utc(naive_date, Utc);

            Ok(date.into())
        }
        Err(e) => Err(e.into()),
    }
}

/** Format a date string using an owned slice of items. */
pub fn format_with_tokens<'a>(date: &'a DateValue, fmt: Vec<Item<'a>>) -> FormattedDate<'a> {
    date.format_with_items(fmt.into_iter()).into()
}