src/database/mod.rs


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

use tracing::instrument;

pub mod comments;
pub mod friends;
pub mod likes;
pub mod posts;
pub mod sessions;
pub mod users;

pub fn connect() -> Result<rusqlite::Connection, rusqlite::Error> {
    rusqlite::Connection::open("xssbook.db")
}

pub fn init() -> Result<(), rusqlite::Error> {
    users::init()?;
    posts::init()?;
    sessions::init()?;
    likes::init()?;
    comments::init()?;
    friends::init()?;
    Ok(())
}

#[instrument()]
pub fn query(query: String) -> Result<usize, rusqlite::Error> {
    tracing::trace!("Running custom query");
    let conn = connect()?;
    conn.execute(&query, [])
}