@@ -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 {
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use schemars::{
|
|||||||
JsonSchema,
|
JsonSchema,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{Bot, Channel, Database, Message, 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,6 +27,11 @@ 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
|
||||||
|
|||||||
@@ -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,
|
||||||
|
}
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -87,6 +87,8 @@ mod test {
|
|||||||
assert_eq!(owner, user.id);
|
assert_eq!(owner, user.id);
|
||||||
assert_eq!(recipients.len(), 1);
|
assert_eq!(recipients.len(), 1);
|
||||||
assert!(harness.db.fetch_channel(&id).await.is_ok());
|
assert!(harness.db.fetch_channel(&id).await.is_ok());
|
||||||
|
|
||||||
|
// TODO: does not check for events
|
||||||
}
|
}
|
||||||
_ => unreachable!(),
|
_ => unreachable!(),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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(())
|
|
||||||
}
|
|
||||||
@@ -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;
|
||||||
@@ -38,7 +37,6 @@ 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,
|
||||||
|
|||||||
@@ -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)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user