139 lines
4.2 KiB
Rust
139 lines
4.2 KiB
Rust
use rusqlite::{params, Connection, Result};
|
|
pub mod messages;
|
|
pub mod types;
|
|
pub mod users;
|
|
|
|
extern crate r2d2;
|
|
extern crate r2d2_sqlite;
|
|
extern crate rusqlite;
|
|
|
|
use core::marker::Send;
|
|
use r2d2_sqlite::SqliteConnectionManager;
|
|
|
|
pub use r2d2::{ManageConnection, Pool};
|
|
|
|
use types::DB_PATH;
|
|
|
|
fn create_db(conn: &Connection) -> Result<()> {
|
|
match conn.execute(
|
|
"CREATE TABLE users (
|
|
id BLOB PRIMARY KEY,
|
|
name TEXT UNIQUE NOT NULL,
|
|
sign_key BLOB UNIQUE NOT NULL,
|
|
insert_key BLOB UNIQUE NOT NULL,
|
|
messages_count INTEGER,
|
|
my_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 BLOB NOT NULL,
|
|
date datetime NOT NULL,
|
|
message TEXT NOT NULL,
|
|
from_me BOOL
|
|
)",
|
|
params![],
|
|
) {
|
|
Ok(_) => log::info!("MESSAGES table created successfully!"),
|
|
Err(e) => log::error!("failed to create USER table {:?}", e),
|
|
}
|
|
match conn.execute(
|
|
"CREATE TABLE my_messages (
|
|
id INTEGER PRIMARY KEY,
|
|
user_id BLOB NOT NULL,
|
|
date datetime NOT NULL,
|
|
message TEXT NOT NULL,
|
|
from_me BOOL
|
|
)",
|
|
params![],
|
|
) {
|
|
Ok(_) => log::info!("MY_MESSAGES table created successfully!"),
|
|
Err(e) => log::error!("failed to create USER table {:?}", e),
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn start_db() -> std::result::Result<Pool<SqliteConnectionManager>, r2d2::Error> where
|
|
{
|
|
let manager = r2d2_sqlite::SqliteConnectionManager::file(DB_PATH);
|
|
|
|
if !std::path::Path::new(DB_PATH).exists() {
|
|
let conn = Connection::open(DB_PATH).expect("failed to connect to db");
|
|
println!("{}", conn.is_autocommit());
|
|
create_db(&conn).expect("failed to create db");
|
|
Ok(r2d2::Pool::new(manager).unwrap())
|
|
} else {
|
|
Ok(r2d2::Pool::new(manager).unwrap())
|
|
}
|
|
}
|
|
|
|
pub fn old_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)
|
|
}
|
|
}
|
|
|
|
// ### A little stupid test ### //
|
|
//let conn = db::start_db().unwrap();
|
|
/*
|
|
users::add_user(db::types::User{
|
|
id: 9349,
|
|
name: "Nick".to_string(),
|
|
sign_key: "string".to_string(),
|
|
insert_key: fcpv2::types::SSK::parse("SSK@Rgt0qM8D24DltliV2-JE9tYLcrgGAKeDwkz41I3JBPs,p~c8c7FXcJjhcf2vA-Xm0Mjyw1o~xn7L2-T8zlBA1IU").unwrap(),
|
|
messages_count: 1,
|
|
}, &conn);
|
|
let time: chrono::DateTime<chrono::offset::FixedOffset> =
|
|
chrono::DateTime::parse_from_rfc3339("2021-03-18T04:22:42.501Z").unwrap();
|
|
db::messages::add_message(
|
|
db::types::Message {
|
|
user_id: 9349,
|
|
id: 4,
|
|
date: time.naive_utc(),
|
|
message: "HI?".to_string(),
|
|
},
|
|
&conn,
|
|
)
|
|
.unwrap();
|
|
db::messages::add_message(
|
|
db::types::Message {
|
|
user_id: 9349,
|
|
id: 5,
|
|
date: time.naive_utc(),
|
|
message: "I AM NICK!".to_string(),
|
|
},
|
|
&conn,
|
|
)
|
|
.unwrap();
|
|
db::messages::add_message(
|
|
db::types::Message {
|
|
user_id: 9349,
|
|
id: 6,
|
|
date: time.naive_utc(),
|
|
message: "I'LL FIND that".to_string(),
|
|
},
|
|
&conn,
|
|
)
|
|
.unwrap();
|
|
|
|
let messages = db::messages::select_message_by_id(9349, 3, &conn).unwrap();
|
|
*/
|