54 lines
1.2 KiB
Rust
54 lines
1.2 KiB
Rust
|
use diesel::prelude::*;
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
|
||
|
|
use crate::schema::{bots, code_bundles};
|
||
|
|
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)
|
||
|
|
}
|