path: root/src/models/screening.rs

/*
    Social program for Ramaskrik.
    Copyright (C) 2019  Harald Eilertsen <haraldei@anduin.net>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

use crate::models::{Film, Room};
use crate::schema::*;
use serde::{Deserialize, Serialize};

joinable!(screenings -> rooms (room_id));
joinable!(screenings -> films (film_id));

#[derive(Deserialize, Insertable)]
#[table_name = "screenings"]
pub struct NewScreening {
    pub film_id: i32,
    pub room_id: i32,
    pub start_time: chrono::DateTime<chrono::Utc>,
    pub end_time: chrono::DateTime<chrono::Utc>,
}

#[derive(Deserialize, Identifiable, PartialEq, Serialize, Queryable)]
#[table_name = "screenings"]
pub struct Screening {
    pub id: i32,
    pub film_id: i32,
    pub room_id: i32,
    pub start_time: chrono::DateTime<chrono::Utc>,
    pub end_time: chrono::DateTime<chrono::Utc>,
}

/// Aggregate screening, room and film info into one struct.
#[derive(Deserialize, Serialize)]
pub struct AggregatedScreening {
    pub id: i32,
    pub film: Film,
    pub room: Room,
    pub start_time: chrono::DateTime<chrono::Utc>,
    pub end_time: chrono::DateTime<chrono::Utc>,
}

impl AggregatedScreening {
    pub fn from_query(s: Vec<(Screening, Option<Film>, Option<Room>)>) -> Vec<AggregatedScreening> {
        s.iter().map(|(s, f, r)| {
            AggregatedScreening {
                id: s.id,
                film: f.as_ref().unwrap().clone().to_owned(),
                room: r.as_ref().unwrap().clone().to_owned(),
                start_time: s.start_time,
                end_time: s.end_time,
            }
        }).collect()
    }
}