49 lines
1.6 KiB
Rust
49 lines
1.6 KiB
Rust
use rusqlite::{params, Connection, Result};
|
|
pub mod types;
|
|
pub mod users;
|
|
pub mod messages;
|
|
|
|
use types::DB_PATH;
|
|
|
|
fn create_db(conn: &Connection) -> Result<()> {
|
|
match conn.execute(
|
|
"CREATE TABLE users (
|
|
id INTEGER PRIMARY KEY,
|
|
name TEXT UNIQUE NOT NULL,
|
|
sign_key BLOB NOT NULL,
|
|
insert_key BLOB NOT NULL,
|
|
messages_count INTEGER
|
|
)",
|
|
params![],
|
|
) {
|
|
Ok(_) => log::info!("USER table created successfully!"),
|
|
Err(e) => log::error!("failed to create USER table {:?}", e),
|
|
}
|
|
match conn.execute(
|
|
"CREATE TABLE messages (
|
|
id INTEGER PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
date datetime NOT NULL,
|
|
message TEXT NOT NULL
|
|
)",
|
|
params![],
|
|
) {
|
|
Ok(_) => log::info!("MESSAGES table created successfully!"),
|
|
Err(e) => log::error!("failed to create USER table {:?}", e),
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn start_db() -> Result<Connection> {
|
|
if !std::path::Path::new(DB_PATH).exists() {
|
|
let conn = Connection::open(DB_PATH)?;
|
|
println!("{}", conn.is_autocommit());
|
|
match create_db(&conn) {
|
|
Ok(_) => {log::info!("Successfully created DB!"); Ok(conn)},
|
|
Err(e) => {log::error!("Failed to create DB: {:?}",e ); Err(e)},
|
|
}
|
|
} else {
|
|
Connection::open(DB_PATH)
|
|
}
|
|
}
|