Add much stuff for rewrite

src/db/model_utils.rs

@@ -1,16 +1,14 @@
 use crate::db::models::Periods;
 use chrono::{NaiveTime, Timelike};
-use diesel::backend::Backend;
 use diesel::deserialize::FromSql;
 use diesel::serialize::{IsNull, Output, ToSql};
 use diesel::sql_types::Binary;
 use diesel::sqlite::Sqlite;
 use diesel::{deserialize, serialize};
 use serde::{Deserialize, Serialize};
-use std::io::Write;
 
 #[derive(Debug, Serialize, Deserialize, AsExpression, FromSqlRow, PartialEq, Clone)]
-#[sql_type = "Binary"]
+#[diesel(sql_type = Binary)]
 pub struct Period {
 	#[serde(with = "period_format")]
 	pub start: NaiveTime,
@@ -41,23 +39,51 @@ mod period_format {
 	}
 }
 
-impl ToSql<Binary, Sqlite> for Periods {
-	fn to_sql<W: Write>(&self, out: &mut Output<W, Sqlite>) -> serialize::Result {
-		for period in self.0.iter() {
-			out.write_all(&[
-				period.start.hour() as u8,
-				period.start.minute() as u8,
-				period.end.hour() as u8,
-				period.end.minute() as u8,
-			])?;
+impl Period {
+	pub fn new(start: NaiveTime, end: NaiveTime) -> Self {
+		Period { start, end }
+	}
+
+	pub fn new_on() -> Self {
+		Period {
+			start: NaiveTime::from_hms_opt(0, 0, 0).unwrap(),
+			end: NaiveTime::from_hms_opt(0, 0, 0).unwrap(),
 		}
+	}
+}
+
+impl ToSql<Binary, Sqlite> for Periods
+where
+	Vec<u8>: ToSql<Binary, Sqlite>,
+{
+	fn to_sql<'b>(&'b self, out: &mut Output<'b, '_, Sqlite>) -> serialize::Result {
+		let periods_u8: Vec<u8> = self
+			.0
+			.iter()
+			.flat_map(|period| {
+				let vec = vec![
+					period.start.hour() as u8,
+					period.start.minute() as u8,
+					period.end.hour() as u8,
+					period.end.minute() as u8,
+				];
+				vec
+			})
+			.collect();
+
+		out.set_value(periods_u8);
+
 		Ok(IsNull::No)
 	}
 }
 
-impl FromSql<Binary, Sqlite> for Periods {
-	fn from_sql(bytes: Option<&<Sqlite as Backend>::RawValue>) -> deserialize::Result<Self> {
-		let blob = bytes.unwrap().read_blob();
+impl<DB> FromSql<Binary, DB> for Periods
+where
+	DB: diesel::backend::Backend,
+	Vec<u8>: FromSql<Binary, DB>,
+{
+	fn from_sql(bytes: DB::RawValue<'_>) -> deserialize::Result<Self> {
+		let blob: Vec<u8> = Vec::from_sql(bytes).unwrap();
 
 		let mut vec = Vec::new();
 		for i in (3..blob.len()).step_by(4) {
@@ -66,8 +92,8 @@ impl FromSql<Binary, Sqlite> for Periods {
 			let end_val_h: u32 = blob[i - 1] as u32;
 			let end_val_m: u32 = blob[i] as u32;
 			vec.push(Period {
-				start: NaiveTime::from_hms(start_val_h, start_val_m, 0),
-				end: NaiveTime::from_hms(end_val_h, end_val_m, 0),
+				start: NaiveTime::from_hms_opt(start_val_h, start_val_m, 0).unwrap(),
+				end: NaiveTime::from_hms_opt(end_val_h, end_val_m, 0).unwrap(),
 			});
 		}
 		Ok(Periods(vec))