surrealpatch/lib/src/fnc/time.rs

use crate::err::Error;
use crate::sql::datetime::Datetime;
use crate::sql::value::Value;
use chrono::offset::TimeZone;
use chrono::Datelike;
use chrono::DurationRound;
use chrono::Local;
use chrono::Timelike;
use chrono::Utc;

pub fn day((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.day().into())
}

pub fn floor((datetime, duration): (Value, Value)) -> Result<Value, Error> {
	Ok(match (datetime, duration) {
		(Value::Datetime(v), Value::Duration(w)) => match chrono::Duration::from_std(*w) {
			Ok(d) => match v.duration_trunc(d) {
				Ok(v) => v.into(),
				_ => Value::None,
			},
			_ => Value::None,
		},
		_ => Value::None,
	})
}

pub fn format((datetime, format): (Value, String)) -> Result<Value, Error> {
	Ok(match datetime {
		Value::Datetime(v) => v.format(&format).to_string().into(),
		_ => Value::None,
	})
}

pub fn group((datetime, strand): (Value, Value)) -> Result<Value, Error> {
	match datetime {
		Value::Datetime(v) => match strand {
			Value::Strand(g) => match g.as_str() {
				"year" => Ok(Utc
					.with_ymd_and_hms(v.year(), 1, 1, 0,0,0)
					.earliest()
					.unwrap()
					.into()),
				"month" => Ok(Utc
					.with_ymd_and_hms(v.year(), v.month(), 1, 0,0,0)
					.earliest()
					.unwrap()
					.into()),
				"day" => Ok(Utc
					.with_ymd_and_hms(v.year(), v.month(), v.day(), 0,0,0)
					.earliest()
					.unwrap()
					.into()),
				"hour" => Ok(Utc
					.with_ymd_and_hms(v.year(), v.month(), v.day(), v.hour(),0,0)
					.earliest()
					.unwrap()
					.into()),
				"minute" => Ok(Utc
					.with_ymd_and_hms(v.year(), v.month(), v.day(), v.hour(),v.minute(),0)
					.earliest()
					.unwrap()
					.into()),
				"second" => Ok(Utc
					.with_ymd_and_hms(v.year(), v.month(), v.day(), v.hour(), v.minute(), v.second())
					.earliest()
					.unwrap()
					.into()),
				_ => Err(Error::InvalidArguments {
					name: String::from("time::group"),
					message: String::from("The second argument must be a string, and can be one of 'year', 'month', 'day', 'hour', 'minute', or 'second'."),
				}),
			},
			_ => Ok(Value::None),
		},
		_ => Ok(Value::None),
	}
}

pub fn hour((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.hour().into())
}

pub fn minute((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.minute().into())
}

pub fn month((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.month().into())
}

pub fn nano((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.timestamp_nanos().into())
}

pub fn now(_: ()) -> Result<Value, Error> {
	Ok(Datetime::default().into())
}

pub fn round((datetime, duration): (Value, Value)) -> Result<Value, Error> {
	Ok(match (datetime, duration) {
		(Value::Datetime(v), Value::Duration(w)) => match chrono::Duration::from_std(*w) {
			Ok(d) => match v.duration_round(d) {
				Ok(v) => v.into(),
				_ => Value::None,
			},
			_ => Value::None,
		},
		_ => Value::None,
	})
}

pub fn second((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.second().into())
}

pub fn timezone(_: ()) -> Result<Value, Error> {
	Ok(Local::now().offset().to_string().into())
}

pub fn unix((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.timestamp().into())
}

pub fn wday((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.weekday().number_from_monday().into())
}

pub fn week((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.iso_week().week().into())
}

pub fn yday((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.ordinal().into())
}

pub fn year((datetime,): (Option<Value>,)) -> Result<Value, Error> {
	let date = match datetime {
		Some(Value::Datetime(v)) => v,
		None => Datetime::default(),
		Some(_) => return Ok(Value::None),
	};
	Ok(date.year().into())
}