ontv 0.1.2

A rich desktop application for tracking tv shows
Documentation 
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

use std::collections::{BTreeMap, BTreeSet};

use chrono::{DateTime, Utc};

use crate::database::iter::Iter;
use crate::model::{Pending, SeriesId};

#[derive(Default)]
pub(crate) struct Database {
    /// Pending by series.
    data: BTreeMap<SeriesId, Pending>,
    /// Index by timestamp.
    by_timestamp: BTreeSet<(DateTime<Utc>, SeriesId)>,
}

impl Database {
    /// Get an existing pending element.
    #[inline]
    pub(crate) fn get(&self, id: &SeriesId) -> Option<&Pending> {
        self.data.get(id)
    }

    /// Export data from the database.
    pub(crate) fn export(&self) -> impl IntoIterator<Item = Pending> {
        let mut export = Vec::with_capacity(self.by_timestamp.len());

        for (_, pending) in &self.data {
            export.push(pending.clone());
        }

        export
    }

    /// Remove by previously looked up index.
    #[inline]
    pub(crate) fn remove(&mut self, id: &SeriesId) -> Option<Pending> {
        let p = self.data.remove(id)?;
        self.by_timestamp.remove(&(p.timestamp, p.series));
        Some(p)
    }

    /// Iterate immutably over pending entries in timestamp order.
    #[inline]
    pub(crate) fn iter(&self) -> impl DoubleEndedIterator<Item = &Pending> + Clone {
        Iter::new(self.by_timestamp.iter().map(|(_, key)| key), &self.data)
    }

    /// Get pending by series.
    #[inline]
    pub(crate) fn by_series(&self, id: &SeriesId) -> Option<&Pending> {
        self.data.get(id)
    }
}

impl Extend<Pending> for Database {
    fn extend<T>(&mut self, iter: T)
    where
        T: IntoIterator<Item = Pending>,
    {
        for p in iter {
            if let Some(p) = self.data.remove(&p.series) {
                let _ = self.by_timestamp.remove(&(p.timestamp, p.series));
            }

            self.by_timestamp.insert((p.timestamp, p.series));
            self.data.insert(p.series, p);
        }
    }
}