aboutsummaryrefslogtreecommitdiff
path: root/backend/src/db/bots.rs
blob: d359e28ac406939f562ff44442bba4ed61c56add (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
use diesel::prelude::*;
use serde::{Deserialize, Serialize};

use crate::schema::{bots, code_bundles};
use crate::DbConn;
use chrono;

#[derive(Insertable)]
#[table_name = "bots"]
pub struct NewBot<'a> {
    pub owner_id: i32,
    pub name: &'a str,
}

#[derive(Queryable, Debug, PartialEq, Serialize, Deserialize)]
pub struct Bot {
    pub id: i32,
    pub owner_id: i32,
    pub name: String,
}

pub fn create_bot(new_bot: &NewBot, conn: &PgConnection) -> QueryResult<Bot> {
    diesel::insert_into(bots::table)
        .values(new_bot)
        .get_result(conn)
}

pub fn find_bot(id: i32, conn: &PgConnection) -> QueryResult<Bot> {
    bots::table.find(id).first(conn)
}

#[derive(Insertable)]
#[table_name = "code_bundles"]
pub struct NewCodeBundle<'a> {
    pub bot_id: i32,
    pub path: &'a str,
}

#[derive(Queryable, Serialize, Deserialize, Debug)]
pub struct CodeBundle {
    pub id: i32,
    pub bot_id: i32,
    pub path: String,
    pub created_at: chrono::NaiveDateTime,
}

pub fn create_code_bundle(
    new_code_bundle: &NewCodeBundle,
    conn: &PgConnection,
) -> QueryResult<CodeBundle> {
    diesel::insert_into(code_bundles::table)
        .values(new_code_bundle)
        .get_result(conn)
}