79 lines
2.6 KiB
Rust
79 lines
2.6 KiB
Rust
use chrono::Utc;
|
|
use hyper::{Response, Body, StatusCode};
|
|
use hyper::header::HeaderValue;
|
|
use mysql_async::{Conn, Pool, error::Error as SqlError};
|
|
use mysql_async::prelude::{params, Queryable};
|
|
use serde::Serialize;
|
|
|
|
use db::{UBigInt, BigInt, Integer, VarChar};
|
|
use db::member::STATUS_ONLINE;
|
|
use db::common::FromDB;
|
|
use crate::auth;
|
|
|
|
#[derive(Serialize)]
|
|
pub struct Member {
|
|
pub id: UBigInt,
|
|
pub secret: VarChar,
|
|
pub name: VarChar,
|
|
pub joindate: BigInt,
|
|
pub status: Integer,
|
|
pub permissions: UBigInt,
|
|
}
|
|
|
|
|
|
pub async fn insert_new_member(p: &Pool, name: VarChar, perms: u64) -> Result<Member, SqlError> {
|
|
use crate::auth::generate_secret;
|
|
|
|
let conn: Conn = p.get_conn().await?;
|
|
let secret_raw: String = generate_secret();
|
|
let secret = match bcrypt::hash(&secret_raw, auth::BCRYPT_COST) {
|
|
Ok(value) => value,
|
|
Err(e) => panic!("\tCould not insert member due to bcrypt failure:\n\t\t{}",e)
|
|
};
|
|
let now: BigInt = Utc::now().timestamp();
|
|
|
|
let conn = conn.drop_exec(
|
|
"INSERT INTO members(secret, name, joindate, status, permissions)
|
|
VALUES(:secret, :name, :joindate, :status, :permissions)",
|
|
mysql_async::params!{
|
|
"secret" => secret.clone(),
|
|
"name" => name.clone(),
|
|
"joindate" => now,
|
|
"status" => STATUS_ONLINE,
|
|
"permissions" => perms
|
|
}).await?;
|
|
|
|
// now pull back the user from our db and return that row
|
|
let db_row_result: (Conn, Option<UBigInt>) = conn.first_exec(
|
|
"SELECT id FROM members WHERE secret = :secret",
|
|
params!{
|
|
"secret" => secret.clone()
|
|
}).await?;
|
|
|
|
Ok(Member {
|
|
id: db_row_result.1.unwrap(), // if we made it this far this shouldn't fail (i hope)
|
|
secret: secret_raw,
|
|
name: name,
|
|
joindate: now,
|
|
status: 0,
|
|
permissions: perms
|
|
})
|
|
}
|
|
|
|
|
|
pub async fn get_online_members(p: &Pool, response: &mut Response<Body>) {
|
|
// TODO: at some point we should provide a way of not querying literally every user in
|
|
// existance
|
|
// TODO: loggin at some point or something idklol
|
|
return match db::channels::Channel::filter(p, STATUS_ONLINE).await {
|
|
db::Response::Set(users) => {
|
|
response.headers_mut().insert("Content-Type",
|
|
HeaderValue::from_static("application/json"));
|
|
|
|
*response.body_mut() = Body::from(serde_json::to_string(&users).unwrap_or("[]".into()));
|
|
},
|
|
db::Response::Other(_msg) => *response.status_mut() = StatusCode::INTERNAL_SERVER_ERROR,
|
|
_ => *response.status_mut() = StatusCode::INTERNAL_SERVER_ERROR
|
|
}
|
|
}
|