aboutsummaryrefslogtreecommitdiffstats
path: root/src/models/screening.rs
blob: feb98cc0f35030637c7b542ce36a277fd48ca8bd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
    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_derive::{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 date: chrono::NaiveDate,
    pub start_time: chrono::NaiveTime,
    pub end_time: chrono::NaiveTime,
}

#[derive(Deserialize, Identifiable, PartialEq, Serialize, Queryable)]
#[table_name = "screenings"]
pub struct ScreeningRecord {
    pub id: i32,
    pub film_id: i32,
    pub room_id: i32,
    pub date: chrono::NaiveDate,
    pub start_time: chrono::NaiveTime,
    pub end_time: chrono::NaiveTime,
}

/// 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 date: chrono::NaiveDate,
    pub start_time: chrono::NaiveTime,
    pub end_time: chrono::NaiveTime,
}

impl AggregatedScreening {
    pub fn from_query(s: Vec<(ScreeningRecord, 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(),
                date: s.date,
                start_time: s.start_time,
                end_time: s.end_time,
            }
        }).collect()
    }
}