Compare commits

...
20 changed files with 343 additions and 192 deletions

No files matched your search

+2
View File
@@ -1,4 +1,6 @@
Rocket.toml Rocket.toml
Revolt.toml
target target
.data .data
.env .env
Generated
+1
View File
@@ -3591,6 +3591,7 @@ dependencies = [
"redis-kiss", "redis-kiss",
"regex", "regex",
"reqwest", "reqwest",
"revolt-config",
"revolt-database", "revolt-database",
"revolt-models", "revolt-models",
"revolt-permissions", "revolt-permissions",
+2 -2
View File
@@ -13,8 +13,8 @@ static CONFIG_BUILDER: Lazy<RwLock<Config>> = Lazy::new(|| {
FileFormat::Toml, FileFormat::Toml,
)); ));
if std::path::Path::new("revolt.toml").exists() { if std::path::Path::new("Revolt.toml").exists() {
builder = builder.add_source(File::new("revolt.toml", FileFormat::Toml)); builder = builder.add_source(File::new("Revolt.toml", FileFormat::Toml));
} }
builder.build().unwrap() builder.build().unwrap()
@@ -1,6 +1,6 @@
use std::collections::HashMap; use std::collections::HashMap;
use revolt_models::v0::MessageAuthor; use revolt_models::v0::{self, MessageAuthor};
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
use revolt_result::Result; use revolt_result::Result;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@@ -168,7 +168,7 @@ auto_derived!(
#[allow(clippy::disallowed_methods)] #[allow(clippy::disallowed_methods)]
impl Channel { impl Channel {
/// Create a channel /* /// Create a channel
pub async fn create(&self, db: &Database) -> Result<()> { pub async fn create(&self, db: &Database) -> Result<()> {
db.insert_channel(self).await?; db.insert_channel(self).await?;
@@ -186,6 +186,39 @@ impl Channel {
} }
Ok(()) Ok(())
}*/
/// Create a group
pub async fn create_group(
db: &Database,
data: v0::DataCreateGroup,
owner_id: String,
) -> Result<Channel> {
let recipients = data.users.into_iter().collect::<Vec<String>>();
let channel = Channel::Group {
id: ulid::Ulid::new().to_string(),
name: data.name,
owner: owner_id,
description: data.description,
recipients: recipients.clone(),
icon: None,
last_message_id: None,
permissions: None,
nsfw: data.nsfw.unwrap_or(false),
};
db.insert_channel(&channel).await?;
let event = EventV1::ChannelCreate(channel.clone().into());
for recipient in recipients {
event.clone().private(recipient).await;
}
Ok(channel)
} }
/// Add user to a group /// Add user to a group
@@ -6,7 +6,7 @@ use crate::MongoDb;
use super::AbstractAttachments; use super::AbstractAttachments;
static COL: &str = "bots"; static COL: &str = "attachments";
#[async_trait] #[async_trait]
impl AbstractAttachments for MongoDb { impl AbstractAttachments for MongoDb {
@@ -182,6 +182,21 @@ impl User {
Ok(user) Ok(user)
} }
/// Get the relationship with another user
pub fn relationship_with(&self, user_b: &str) -> RelationshipStatus {
if self.id == user_b {
return RelationshipStatus::User;
}
if let Some(relations) = &self.relations {
if let Some(relationship) = relations.iter().find(|x| x.id == user_b) {
return relationship.status.clone();
}
}
RelationshipStatus::None
}
/// Check whether two users have a mutual connection /// Check whether two users have a mutual connection
/// ///
/// This will check if user and user_b share a server or a group. /// This will check if user and user_b share a server or a group.
@@ -275,6 +275,15 @@ impl From<crate::EmojiParent> for EmojiParent {
} }
} }
impl From<EmojiParent> for crate::EmojiParent {
fn from(value: EmojiParent) -> Self {
match value {
EmojiParent::Detached => crate::EmojiParent::Detached,
EmojiParent::Server { id } => crate::EmojiParent::Server { id },
}
}
}
impl From<crate::File> for File { impl From<crate::File> for File {
fn from(value: crate::File) -> Self { fn from(value: crate::File) -> Self {
File { File {
+11 -1
View File
@@ -7,7 +7,7 @@ use schemars::{
JsonSchema, JsonSchema,
}; };
use crate::{Bot, Channel, Database, Webhook}; use crate::{Bot, Channel, Database, Emoji, Message, Webhook};
/// Reference to some object in the database /// Reference to some object in the database
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
@@ -27,11 +27,21 @@ impl Reference {
db.fetch_bot(&self.id).await db.fetch_bot(&self.id).await
} }
/// Fetch emoji from Ref
pub async fn as_emoji(&self, db: &Database) -> Result<Emoji> {
db.fetch_emoji(&self.id).await
}
/// Fetch channel from Ref /// Fetch channel from Ref
pub async fn as_channel(&self, db: &Database) -> Result<Channel> { pub async fn as_channel(&self, db: &Database) -> Result<Channel> {
db.fetch_channel(&self.id).await db.fetch_channel(&self.id).await
} }
/// Fetch message from Ref
pub async fn as_message(&self, db: &Database) -> Result<Message> {
db.fetch_message(&self.id).await
}
/// Fetch webhook from Ref /// Fetch webhook from Ref
pub async fn as_webhook(&self, db: &Database) -> Result<Webhook> { pub async fn as_webhook(&self, db: &Database) -> Result<Webhook> {
db.fetch_webhook(&self.id).await db.fetch_webhook(&self.id).await
+26 -1
View File
@@ -1,10 +1,11 @@
use super::File; use super::File;
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
use std::collections::HashMap; use std::collections::{HashMap, HashSet};
auto_derived!( auto_derived!(
/// Channel /// Channel
#[serde(tag = "channel_type")]
pub enum Channel { pub enum Channel {
/// Personal "Saved Notes" channel which allows users to save messages /// Personal "Saved Notes" channel which allows users to save messages
SavedMessages { SavedMessages {
@@ -205,4 +206,28 @@ auto_derived!(
#[cfg_attr(feature = "serde", serde(default))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsChannel>>, pub remove: Option<Vec<FieldsChannel>>,
} }
/// Create new group
#[derive(Default)]
#[cfg_attr(feature = "validator", derive(validator::Validate))]
pub struct DataCreateGroup {
/// Group name
#[validate(length(min = 1, max = 32))]
pub name: String,
/// Group description
#[validate(length(min = 0, max = 1024))]
pub description: Option<String>,
/// Group icon
#[validate(length(min = 1, max = 128))]
pub icon: Option<String>,
/// Array of user IDs to add to the group
///
/// Must be friends with these users.
#[validate(length(min = 0, max = 49))]
#[serde(default)]
pub users: HashSet<String>,
/// Whether this group is age-restricted
#[serde(skip_serializing_if = "Option::is_none")]
pub nsfw: Option<bool>,
}
); );
+22
View File
@@ -1,3 +1,12 @@
use once_cell::sync::Lazy;
use regex::Regex;
use validator::Validate;
/// Regex for valid emoji names
///
/// Alphanumeric and underscores
pub static RE_EMOJI: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[a-z0-9_]+$").unwrap());
auto_derived!( auto_derived!(
/// Emoji /// Emoji
pub struct Emoji { pub struct Emoji {
@@ -30,4 +39,17 @@ auto_derived!(
Server { id: String }, Server { id: String },
Detached, Detached,
} }
/// Create a new emoji
#[derive(Validate)]
pub struct DataCreateEmoji {
/// Server name
#[validate(length(min = 1, max = 32), regex = "RE_EMOJI")]
pub name: String,
/// Parent information
pub parent: EmojiParent,
/// Whether the emoji is mature
#[serde(default)]
pub nsfw: bool,
}
); );
+1
View File
@@ -68,6 +68,7 @@ revolt_rocket_okapi = { version = "0.9.1", features = ["swagger"] }
revolt-quark = { path = "../quark" } revolt-quark = { path = "../quark" }
# core # core
revolt-config = { path = "../core/config" }
revolt-database = { path = "../core/database", features = [ revolt-database = { path = "../core/database", features = [
"rocket-impl", "rocket-impl",
"redis-is-patched", "redis-is-patched",
+10 -14
View File
@@ -78,20 +78,16 @@ mod test {
.await .await
.expect("`Bot`"); .expect("`Bot`");
// FIXME: Channel::create_group let group = Channel::create_group(
let group = Channel::Group { &harness.db,
id: ulid::Ulid::new().to_string(), v0::DataCreateGroup {
name: TestHarness::rand_string(), name: TestHarness::rand_string(),
owner: user.id.to_string(), ..Default::default()
description: None, },
last_message_id: None, user.id.to_string(),
icon: None, )
nsfw: false, .await
permissions: None, .unwrap();
recipients: vec![user.id.to_string()],
};
group.create(&harness.db).await.unwrap();
let response = harness let response = harness
.client .client
@@ -1,8 +1,11 @@
use revolt_quark::{ use revolt_database::{
models::{Channel, User}, util::{permissions::DatabasePermissionQuery, reference::Reference},
perms, Database, Permission, Ref, Result, Database, User,
}; };
use revolt_models::v0;
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::Result;
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
/// # Fetch Channel /// # Fetch Channel
@@ -10,12 +13,54 @@ use rocket::{serde::json::Json, State};
/// Fetch channel by its id. /// Fetch channel by its id.
#[openapi(tag = "Channel Information")] #[openapi(tag = "Channel Information")]
#[get("/<target>")] #[get("/<target>")]
pub async fn req(db: &State<Database>, user: User, target: Ref) -> Result<Json<Channel>> { pub async fn fetch_channel(
db: &State<Database>,
user: User,
target: Reference,
) -> Result<Json<v0::Channel>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
perms(&user)
.channel(&channel)
.throw_permission(db, Permission::ViewChannel)
.await?;
Ok(Json(channel)) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
calculate_channel_permissions(&mut query)
.await
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
Ok(Json(channel.into()))
}
#[cfg(test)]
mod test {
use crate::{rocket, util::test::TestHarness};
use revolt_database::Channel;
use revolt_models::v0;
use rocket::http::{Header, Status};
#[rocket::async_test]
async fn fetch_channel() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.get(format!("/channels/{}", group.id()))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::Ok);
let channel: v0::Channel = response.into_json().await.expect("`Channel`");
assert_eq!(channel, group.into());
}
} }
@@ -1,84 +1,96 @@
use std::{collections::HashSet, iter::FromIterator}; use revolt_config::config;
use revolt_database::{Channel, Database, RelationshipStatus, User};
use revolt_quark::{ use revolt_models::v0;
get_relationship, use revolt_result::{create_error, Result};
models::{user::RelationshipStatus, Channel, User},
variables::delta::MAX_GROUP_SIZE,
Db, Error, Result,
};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use serde::{Deserialize, Serialize}; use rocket::State;
use ulid::Ulid;
use validator::Validate; use validator::Validate;
/// # Group Data
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct DataCreateGroup {
/// Group name
#[validate(length(min = 1, max = 32))]
name: String,
/// Group description
#[validate(length(min = 0, max = 1024))]
description: Option<String>,
/// Array of user IDs to add to the group
///
/// Must be friends with these users.
#[validate(length(min = 0, max = 49))]
users: Vec<String>,
/// Whether this group is age-restricted
#[serde(skip_serializing_if = "Option::is_none")]
nsfw: Option<bool>,
}
/// # Create Group /// # Create Group
/// ///
/// Create a new group channel. /// Create a new group channel.
#[openapi(tag = "Groups")] #[openapi(tag = "Groups")]
#[post("/create", data = "<info>")] #[post("/create", data = "<data>")]
pub async fn req(db: &Db, user: User, info: Json<DataCreateGroup>) -> Result<Json<Channel>> { pub async fn create_group(
db: &State<Database>,
user: User,
data: Json<v0::DataCreateGroup>,
) -> Result<Json<v0::Channel>> {
let config = config().await;
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
let info = info.into_inner(); let mut data = data.into_inner();
info.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let mut set: HashSet<String> = HashSet::from_iter(info.users.into_iter()); data.users.insert(user.id.to_string());
set.insert(user.id.clone());
if set.len() > *MAX_GROUP_SIZE { if data.users.len() > config.features.limits.default.group_size {
return Err(Error::GroupTooLarge { return Err(create_error!(GroupTooLarge {
max: *MAX_GROUP_SIZE, max: config.features.limits.default.group_size,
}); }));
} }
for target in &set { for target in &data.users {
match get_relationship(&user, target) { match user.relationship_with(target) {
RelationshipStatus::Friend | RelationshipStatus::User => {} RelationshipStatus::Friend | RelationshipStatus::User => {}
_ => { _ => {
return Err(Error::NotFriends); return Err(create_error!(NotFriends));
} }
} }
} }
let group = Channel::Group { Ok(Json(Channel::create_group(db, data, user.id).await?.into()))
id: Ulid::new().to_string(), }
name: info.name, #[cfg(test)]
owner: user.id, mod test {
description: info.description, use crate::{rocket, util::test::TestHarness};
recipients: set.into_iter().collect::<Vec<String>>(), use revolt_models::v0;
use rocket::http::{ContentType, Header, Status};
icon: None, #[rocket::async_test]
last_message_id: None, async fn create_group() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
permissions: None, let response = harness
.client
.post("/channels/create")
.header(Header::new("x-session-token", session.token.to_string()))
.header(ContentType::JSON)
.body(
json!(v0::DataCreateBot {
name: TestHarness::rand_string(),
})
.to_string(),
)
.dispatch()
.await;
nsfw: info.nsfw.unwrap_or(false), assert_eq!(response.status(), Status::Ok);
};
group.create(db).await?; let channel: v0::Channel = response.into_json().await.expect("`Channel`");
Ok(Json(group)) match channel {
v0::Channel::Group {
id,
owner,
recipients,
..
} => {
assert_eq!(owner, user.id);
assert_eq!(recipients.len(), 1);
assert!(harness.db.fetch_channel(&id).await.is_ok());
// TODO: does not check for events
}
_ => unreachable!(),
}
}
} }
@@ -1,26 +1,34 @@
use revolt_quark::{ use revolt_database::{
models::{Message, User}, util::{permissions::DatabasePermissionQuery, reference::Reference},
perms, Db, Error, Permission, Ref, Result, Database, User,
}; };
use revolt_models::v0;
use rocket::serde::json::Json; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State};
/// # Fetch Message /// # Fetch Message
/// ///
/// Retrieves a message by its id. /// Retrieves a message by its id.
#[openapi(tag = "Messaging")] #[openapi(tag = "Messaging")]
#[get("/<target>/messages/<msg>")] #[get("/<target>/messages/<msg>")]
pub async fn req(db: &Db, user: User, target: Ref, msg: Ref) -> Result<Json<Message>> { pub async fn req(
db: &State<Database>,
user: User,
target: Reference,
msg: Reference,
) -> Result<Json<v0::Message>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
perms(&user)
.channel(&channel) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.throw_permission(db, Permission::ViewChannel) calculate_channel_permissions(&mut query)
.await?; .await
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
let message = msg.as_message(db).await?; let message = msg.as_message(db).await?;
if message.channel != channel.as_id() { if message.channel != channel.id() {
return Err(Error::NotFound); return Err(create_error!(NotFound));
} }
Ok(Json(message)) Ok(Json(message.into()))
} }
@@ -1,26 +0,0 @@
use revolt_quark::{models::User, Ref, Result};
use rocket::serde::json::Json;
use serde::{Deserialize, Serialize};
use validator::Validate;
/// # Query Parameters
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct OptionsQueryStale {
/// Array of message IDs
#[validate(length(min = 0, max = 150))]
ids: Vec<String>,
}
/// # Poll Message Changes
///
/// This route returns any changed message objects and tells you if any have been deleted.
///
/// Don't actually poll this route, instead use this to update your local database.
///
/// **DEPRECATED**
#[openapi(tag = "Messaging")]
#[post("/<_target>/messages/stale", data = "<_data>")]
pub async fn req(_user: User, _target: Ref, _data: Json<OptionsQueryStale>) -> Result<()> {
Ok(())
}
+2 -4
View File
@@ -16,7 +16,6 @@ mod message_delete;
mod message_edit; mod message_edit;
mod message_fetch; mod message_fetch;
mod message_query; mod message_query;
mod message_query_stale;
mod message_react; mod message_react;
mod message_search; mod message_search;
mod message_send; mod message_send;
@@ -30,7 +29,7 @@ mod webhook_fetch_all;
pub fn routes() -> (Vec<Route>, OpenApi) { pub fn routes() -> (Vec<Route>, OpenApi) {
openapi_get_routes_spec![ openapi_get_routes_spec![
channel_ack::req, channel_ack::req,
channel_fetch::req, channel_fetch::fetch_channel,
members_fetch::req, members_fetch::req,
channel_delete::req, channel_delete::req,
channel_edit::req, channel_edit::req,
@@ -38,12 +37,11 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
message_send::message_send, message_send::message_send,
message_query::req, message_query::req,
message_search::req, message_search::req,
message_query_stale::req,
message_fetch::req, message_fetch::req,
message_edit::req, message_edit::req,
message_bulk_delete::req, message_bulk_delete::req,
message_delete::req, message_delete::req,
group_create::req, group_create::create_group,
group_add_member::req, group_add_member::req,
group_remove_member::req, group_remove_member::req,
voice_join::req, voice_join::req,
@@ -1,31 +1,11 @@
use once_cell::sync::Lazy; use revolt_config::config;
use regex::Regex; use revolt_database::{util::permissions::DatabasePermissionQuery, Database, Emoji, File, User};
use revolt_quark::models::emoji::EmojiParent; use revolt_models::v0;
use revolt_quark::models::{Emoji, File, User}; use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use revolt_quark::variables::delta::MAX_EMOJI_COUNT; use revolt_result::{create_error, Result};
use revolt_quark::{perms, Db, Error, Permission, Result};
use serde::Deserialize;
use validator::Validate; use validator::Validate;
use rocket::serde::json::Json; use rocket::{serde::json::Json, State};
/// Regex for valid emoji names
///
/// Alphanumeric and underscores
pub static RE_EMOJI: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[a-z0-9_]+$").unwrap());
/// # Emoji Data
#[derive(Validate, Deserialize, JsonSchema)]
pub struct DataCreateEmoji {
/// Server name
#[validate(length(min = 1, max = 32), regex = "RE_EMOJI")]
name: String,
/// Parent information
parent: EmojiParent,
/// Whether the emoji is mature
#[serde(default)]
nsfw: bool,
}
/// # Create New Emoji /// # Create New Emoji
/// ///
@@ -33,41 +13,45 @@ pub struct DataCreateEmoji {
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[put("/emoji/<id>", data = "<data>")] #[put("/emoji/<id>", data = "<data>")]
pub async fn create_emoji( pub async fn create_emoji(
db: &Db, db: &State<Database>,
user: User, user: User,
id: String, id: String,
data: Json<DataCreateEmoji>, data: Json<v0::DataCreateEmoji>,
) -> Result<Json<Emoji>> { ) -> Result<Json<v0::Emoji>> {
let config = config().await;
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
// Bots cannot manage emojis // Bots cannot manage emojis
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
// Validate we have permission to write into parent // Validate we have permission to write into parent
match &data.parent { match &data.parent {
EmojiParent::Server { id } => { v0::EmojiParent::Server { id } => {
let server = db.fetch_server(id).await?; let server = db.fetch_server(id).await?;
// Check for permission // Check for permission
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
.server(&server) calculate_server_permissions(&mut query)
.throw_permission(db, Permission::ManageCustomisation) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageCustomisation)?;
// Check that there are no more than 100 emoji // Check that we haven't hit the emoji limit
// ! FIXME: hardcoded upper limit
let emojis = db.fetch_emoji_by_parent_id(&server.id).await?; let emojis = db.fetch_emoji_by_parent_id(&server.id).await?;
if emojis.len() > *MAX_EMOJI_COUNT { if emojis.len() > config.features.limits.default.server_emoji {
return Err(Error::TooManyEmoji { return Err(create_error!(TooManyEmoji {
max: *MAX_EMOJI_COUNT, max: config.features.limits.default.server_emoji,
}); }));
} }
} }
EmojiParent::Detached => return Err(Error::InvalidOperation), v0::EmojiParent::Detached => return Err(create_error!(InvalidOperation)),
}; };
// Find the relevant attachment // Find the relevant attachment
@@ -76,7 +60,7 @@ pub async fn create_emoji(
// Create the emoji object // Create the emoji object
let emoji = Emoji { let emoji = Emoji {
id, id,
parent: data.parent, parent: data.parent.into(),
creator_id: user.id, creator_id: user.id,
name: data.name, name: data.name,
animated: "image/gif" == &attachment.content_type, animated: "image/gif" == &attachment.content_type,
@@ -85,5 +69,5 @@ pub async fn create_emoji(
// Save emoji // Save emoji
emoji.create(db).await?; emoji.create(db).await?;
Ok(Json(emoji)) Ok(Json(emoji.into()))
} }
@@ -1,19 +1,30 @@
use revolt_quark::models::{emoji::EmojiParent, User}; use revolt_database::{
use revolt_quark::{perms, Db, EmptyResponse, Error, Permission, Ref, Result}; util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, EmojiParent, User,
};
use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::State;
use rocket_empty::EmptyResponse;
/// # Delete Emoji /// # Delete Emoji
/// ///
/// Delete an emoji by its id. /// Delete an emoji by its id.
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[delete("/emoji/<id>")] #[delete("/emoji/<emoji_id>")]
pub async fn delete_emoji(db: &Db, user: User, id: Ref) -> Result<EmptyResponse> { pub async fn delete_emoji(
db: &State<Database>,
user: User,
emoji_id: Reference,
) -> Result<EmptyResponse> {
// Bots cannot manage emoji // Bots cannot manage emoji
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
// Fetch the emoji // Fetch the emoji
let emoji = id.as_emoji(db).await?; let emoji = emoji_id.as_emoji(db).await?;
// If we uploaded the emoji, then we have permission to delete it // If we uploaded the emoji, then we have permission to delete it
if emoji.creator_id != user.id { if emoji.creator_id != user.id {
@@ -23,10 +34,10 @@ pub async fn delete_emoji(db: &Db, user: User, id: Ref) -> Result<EmptyResponse>
let server = db.fetch_server(id).await?; let server = db.fetch_server(id).await?;
// Check for permission // Check for permission
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
.server(&server) calculate_server_permissions(&mut query)
.throw_permission(db, Permission::ManageCustomisation) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageCustomisation)?;
} }
EmojiParent::Detached => return Ok(EmptyResponse), EmojiParent::Detached => return Ok(EmptyResponse),
}; };
@@ -1,13 +1,18 @@
use revolt_quark::models::{Emoji, User}; use revolt_database::{util::reference::Reference, Database};
use revolt_quark::{Db, Ref, Result}; use revolt_models::v0;
use revolt_result::Result;
use rocket::serde::json::Json; use rocket::{serde::json::Json, State};
/// # Fetch Emoji /// # Fetch Emoji
/// ///
/// Fetch an emoji by its id. /// Fetch an emoji by its id.
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[get("/emoji/<id>")] #[get("/emoji/<emoji_id>")]
pub async fn fetch_emoji(db: &Db, _user: User, id: Ref) -> Result<Json<Emoji>> { pub async fn fetch_emoji(db: &State<Database>, emoji_id: Reference) -> Result<Json<v0::Emoji>> {
id.as_emoji(db).await.map(Json) emoji_id
.as_emoji(db)
.await
.map(|emoji| emoji.into())
.map(Json)
} }