summaryrefslogtreecommitdiff
path: root/src/channel/repo/channels.rs
blob: 6fb0c23c676c66f37a07fa6fd2ce51d8a6b7d2d4 (plain)
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
use std::fmt;

use sqlx::{sqlite::Sqlite, SqliteConnection, Transaction};

use crate::error::BoxedError;
use crate::id::Id as BaseId;

pub trait Provider {
    fn channels(&mut self) -> Channels;
}

impl<'c> Provider for Transaction<'c, Sqlite> {
    fn channels(&mut self) -> Channels {
        Channels(self)
    }
}

pub struct Channels<'t>(&'t mut SqliteConnection);

#[derive(Debug)]
pub struct Channel {
    pub id: Id,
    pub name: String,
}

impl<'c> Channels<'c> {
    /// Create a new channel.
    pub async fn create(&mut self, name: &str) -> Result<Id, BoxedError> {
        let id = Id::generate();

        let channel = sqlx::query_scalar!(
            r#"
                insert
                into channel (id, name)
                values ($1, $2)
                returning id as "id: Id"
            "#,
            id,
            name,
        )
        .fetch_one(&mut *self.0)
        .await?;

        Ok(channel)
    }

    pub async fn all(&mut self) -> Result<Vec<Channel>, BoxedError> {
        let channels = sqlx::query_as!(
            Channel,
            r#"
                select
                    channel.id as "id: Id",
                    channel.name
                from channel
                order by channel.name
            "#,
        )
        .fetch_all(&mut *self.0)
        .await?;

        Ok(channels)
    }
}

/// Stable identifier for a [Channel]. Prefixed with `C`.
#[derive(Clone, Debug, Eq, Hash, PartialEq, sqlx::Type, serde::Deserialize, serde::Serialize)]
#[sqlx(transparent)]
#[serde(transparent)]
pub struct Id(BaseId);

impl From<BaseId> for Id {
    fn from(id: BaseId) -> Self {
        Self(id)
    }
}

impl Id {
    pub fn generate() -> Self {
        BaseId::generate("C")
    }
}

impl fmt::Display for Id {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.0.fmt(f)
    }
}