aboutsummaryrefslogtreecommitdiff
path: root/src/db/categories.rs
blob: 31cb3d05428d143439f34bb489fb4b4d5b9f689d (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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
use tokio_rusqlite::{named_params, Connection, Row};

use crate::db::utils;
use crate::model::category::{Category, Create, Update};

fn row_to_category(row: &Row) -> Result<Category, rusqlite::Error> {
    Ok(Category {
        id: row.get(0)?,
        name: row.get(1)?,
        color: row.get(2)?,
    })
}

pub async fn list(conn: &Connection) -> Vec<Category> {
    let query = r#"
        SELECT
            id,
            name,
            color
        FROM categories
        WHERE deleted_at IS NULL
        ORDER BY name
    "#;

    let res = conn
        .call(move |conn| {
            let mut stmt = conn.prepare(query)?;

            let users = stmt
                .query_map([], row_to_category)?
                .collect::<Result<Vec<Category>, _>>()?;

            Ok(users)
        })
        .await;

    match res {
        Ok(categories) => categories,
        Err(err) => {
            log::error!("Error listing categories: {:?}", err);
            vec![]
        }
    }
}

pub async fn get(conn: &Connection, id: i64) -> Option<Category> {
    let query = r#"
        SELECT
            id,
            name,
            color
        FROM categories
        WHERE
            id = :id
            AND deleted_at IS NULL
    "#;

    let res = conn
        .call(move |conn| {
            let mut stmt = conn.prepare(query)?;
            let mut iter =
                stmt.query_map(named_params![":id": id], row_to_category)?;
            utils::one(&mut iter)
        })
        .await;

    match res {
        Ok(category) => Some(category),
        Err(err) => {
            log::error!("Error looking for category {}: {:?}", id, err);
            None
        }
    }
}

pub async fn create(conn: &Connection, c: Create) -> Option<i64> {
    let query = r#"INSERT INTO categories(name, color) VALUES (:name, :color)"#;

    let res = conn
        .call(move |conn| {
            conn.execute(
                query,
                named_params![":name": c.name, ":color": c.color],
            )?;
            Ok(conn.last_insert_rowid())
        })
        .await;

    match res {
        Ok(category_id) => Some(category_id),
        Err(err) => {
            log::error!("Error creating category: {:?}", err);
            None
        }
    }
}

pub async fn update(conn: &Connection, id: i64, c: Update) -> bool {
    let query = r#"
        UPDATE categories
        SET
            name = :name,
            color = :color,
            updated_at = datetime()
        WHERE id = :id
    "#;

    let res = conn
        .call(move |conn| {
            Ok(conn.execute(
                query,
                named_params![":name": c.name, ":color": c.color, ":id": id],
            )?)
        })
        .await;

    match res {
        Ok(_) => true,
        Err(err) => {
            log::error!("Error updating category {}: {:?}", id, err);
            false
        }
    }
}

pub async fn delete(conn: &Connection, id: i64) -> bool {
    let res = conn
        .call(move |conn| {
            Ok(conn.execute(
                r#"
            UPDATE
                categories
            SET
                deleted_at = datetime()
            WHERE
                id = :id
            "#,
                named_params![":id": id],
            )?)
        })
        .await;

    match res {
        Ok(_) => true,
        Err(err) => {
            log::error!("Error deleting category {}: {:?}", id, err);
            false
        }
    }
}