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 { 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) = 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) { // 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 } }