use mysql_async::{params, Pool}; use mysql_async::prelude::Queryable; use mysql_async::Error as SqlError; use crate::{Channel, UBigInt, Integer}; use crate::Response; use rand::RngCore; pub const VOICE_CHANNEL: Integer = 1; pub const TEXT_CHANNEL: Integer = 2; pub const MAX_NAME_LEN: usize = 256; pub const MAX_DESCRIPTION_LEN: usize = 2048; impl Channel { pub async fn filter(p: &Pool, kind: Integer) -> Result, SqlError> { //! @returns -> on success : Ok(Response::Set(Vec)) //! @throw -> on sql fail : Err(SqlError) if kind == VOICE_CHANNEL || kind ==TEXT_CHANNEL { let mut conn = p.get_conn().await?; let q = "SELECT id, name, description, kind FROM channels WHERE kind = :kind"; let params = params!{"kind" => kind}; let channels = conn.exec_map(q, params, |(id, name, description, kind)| { Channel { id, name, description, kind } }).await?; Ok(Response::Set(channels)) } else { return Ok(Response::RestrictedInput("Channel kind is invalid".into())); } } pub async fn add(p: &Pool, name: &str, description: &str, kind: Integer) -> Result, SqlError> { //! @returns on success -> Response::Row //! @returns on partial success -> Response::Empty //! @returns on failure -> Response::Other //! @returns on user failure -> Response::RestrictedInput(msg) // bounds are literally [1, 2] if kind == TEXT_CHANNEL || kind == VOICE_CHANNEL { let mut conn = p.get_conn().await?; let q = "INSERT INTO channels (id, name, description, kind) VALUES (:i, :n, :d, :k)"; let id = rand::rngs::OsRng.next_u64(); // generate id's randomly for channels conn.exec_drop(q, params!{ "i" => id, "n" => name, "d" => description, "k" => kind }).await?; Ok(Response::Row(Self { id, name: name.to_string(), description: Some(description.to_string()), kind })) } else { return Ok(Response::RestrictedInput(String::from("Invalid channel type"))); } } pub async fn delete(p: &Pool, id: UBigInt) -> Result, SqlError> { //! Deletes channel given UBigInt as the row key //! @param p -> SqlPool //! @param id -> UBigInt //! @return on success -> Response::Success //! @return on server failure -> Response::Other let mut conn = p.get_conn().await?; let q = "DELETE FROM channels WHERE id = :id"; conn.exec_drop(q, params!{"id" => id}).await?; Ok(Response::Success) } }