Merge remote-tracking branch 'origin/main' into feat/livekit

This commit is contained in:
Zomatree
2025-08-18 07:44:52 +01:00
120 changed files with 372 additions and 351 deletions

3
Cargo.lock generated
View File

@@ -6535,7 +6535,6 @@ dependencies = [
"bson", "bson",
"deadqueue", "deadqueue",
"decancer", "decancer",
"fcm_v1",
"futures", "futures",
"indexmap 1.9.3", "indexmap 1.9.3",
"isahc", "isahc",
@@ -6558,7 +6557,6 @@ dependencies = [
"revolt-permissions", "revolt-permissions",
"revolt-presence", "revolt-presence",
"revolt-result", "revolt-result",
"revolt_a2",
"revolt_okapi", "revolt_okapi",
"revolt_optional_struct", "revolt_optional_struct",
"revolt_rocket_okapi", "revolt_rocket_okapi",
@@ -6570,7 +6568,6 @@ dependencies = [
"unicode-segmentation", "unicode-segmentation",
"url-escape", "url-escape",
"validator 0.16.1", "validator 0.16.1",
"web-push",
] ]
[[package]] [[package]]

View File

@@ -34,7 +34,7 @@ services:
- minio - minio
entrypoint: > entrypoint: >
/bin/sh -c "while ! /usr/bin/mc ready minio; do /bin/sh -c "while ! /usr/bin/mc ready minio; do
/usr/bin/mc config host add minio http://minio:9000 minioautumn minioautumn; /usr/bin/mc alias set minio http://minio:9000 minioautumn minioautumn;
echo 'Waiting minio...' && sleep 1; echo 'Waiting minio...' && sleep 1;
done; /usr/bin/mc mb minio/revolt-uploads; exit 0;" done; /usr/bin/mc mb minio/revolt-uploads; exit 0;"

View File

@@ -11,8 +11,9 @@ description = "Revolt Backend: Configuration"
[features] [features]
anyhow = ["dep:sentry-anyhow"] anyhow = ["dep:sentry-anyhow"]
report-macros = ["revolt-result"] report-macros = ["revolt-result"]
sentry = ["dep:sentry"]
test = ["async-std"] test = ["async-std"]
default = ["test", "anyhow"] default = ["test", "sentry"]
[dependencies] [dependencies]
# Utility # Utility
@@ -32,7 +33,7 @@ log = "0.4.14"
pretty_env_logger = "0.4.0" pretty_env_logger = "0.4.0"
# Sentry # Sentry
sentry = "0.31.5" sentry = { version = "0.31.5", optional = true }
sentry-anyhow = { version = "0.38.1", optional = true } sentry-anyhow = { version = "0.38.1", optional = true }
# Core # Core

View File

@@ -6,10 +6,12 @@ use futures_locks::RwLock;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use serde::Deserialize; use serde::Deserialize;
#[cfg(feature = "sentry")]
pub use sentry::{capture_error, capture_message, Level}; pub use sentry::{capture_error, capture_message, Level};
#[cfg(feature = "anyhow")]
pub use sentry_anyhow::capture_anyhow; pub use sentry_anyhow::capture_anyhow;
#[cfg(feature = "report-macros")] #[cfg(all(feature = "report-macros", feature = "sentry"))]
#[macro_export] #[macro_export]
macro_rules! report_error { macro_rules! report_error {
( $expr: expr, $error: ident $( $tt:tt )? ) => { ( $expr: expr, $error: ident $( $tt:tt )? ) => {
@@ -24,7 +26,7 @@ macro_rules! report_error {
}; };
} }
#[cfg(feature = "report-macros")] #[cfg(all(feature = "report-macros", feature = "sentry"))]
#[macro_export] #[macro_export]
macro_rules! capture_internal_error { macro_rules! capture_internal_error {
( $expr: expr ) => { ( $expr: expr ) => {
@@ -35,7 +37,7 @@ macro_rules! capture_internal_error {
}; };
} }
#[cfg(feature = "report-macros")] #[cfg(all(feature = "report-macros", feature = "sentry"))]
#[macro_export] #[macro_export]
macro_rules! report_internal_error { macro_rules! report_internal_error {
( $expr: expr ) => { ( $expr: expr ) => {
@@ -442,6 +444,7 @@ pub async fn config() -> Settings {
} }
/// Configure logging and common Rust variables /// Configure logging and common Rust variables
#[cfg(feature = "sentry")]
pub async fn setup_logging(release: &'static str, dsn: String) -> Option<sentry::ClientInitGuard> { pub async fn setup_logging(release: &'static str, dsn: String) -> Option<sentry::ClientInitGuard> {
if std::env::var("RUST_LOG").is_err() { if std::env::var("RUST_LOG").is_err() {
std::env::set_var("RUST_LOG", "info"); std::env::set_var("RUST_LOG", "info");
@@ -467,6 +470,7 @@ pub async fn setup_logging(release: &'static str, dsn: String) -> Option<sentry:
} }
} }
#[cfg(feature = "sentry")]
#[macro_export] #[macro_export]
macro_rules! configure { macro_rules! configure {
($application: ident) => { ($application: ident) => {

View File

@@ -10,12 +10,12 @@ description = "Revolt Backend: Database Implementation"
[features] [features]
# Databases # Databases
mongodb = ["dep:mongodb", "bson"] mongodb = ["dep:mongodb", "bson", "authifier/database-mongodb"]
# ... Other # ... Other
tasks = ["isahc", "linkify", "url-escape"] tasks = ["isahc", "linkify", "url-escape"]
async-std-runtime = ["async-std"] async-std-runtime = ["async-std", "authifier/async-std-runtime"]
rocket-impl = ["rocket", "schemars", "revolt_okapi", "revolt_rocket_okapi"] rocket-impl = ["rocket", "schemars", "revolt_okapi", "revolt_rocket_okapi", "authifier/rocket_impl"]
axum-impl = ["axum"] axum-impl = ["axum"]
redis-is-patched = ["revolt-presence/redis-is-patched"] redis-is-patched = ["revolt-presence/redis-is-patched"]
@@ -91,13 +91,8 @@ rocket = { version = "0.5.1", default-features = false, features = [
revolt_okapi = { version = "0.9.1", optional = true } revolt_okapi = { version = "0.9.1", optional = true }
revolt_rocket_okapi = { version = "0.10.0", optional = true } revolt_rocket_okapi = { version = "0.10.0", optional = true }
# Notifications
fcm_v1 = "0.3.0"
web-push = "0.10.0"
revolt_a2 = { version = "0.10", default-features = false, features = ["ring"] }
# Authifier # Authifier
authifier = { version = "1.0.15", features = ["rocket_impl"] } authifier = { version = "1.0.15" }
# RabbitMQ # RabbitMQ
amqprs = { version = "1.7.0" } amqprs = { version = "1.7.0" }

View File

@@ -1,3 +1,4 @@
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;
@@ -13,6 +14,7 @@ use authifier::Authifier;
use rand::Rng; use rand::Rng;
use revolt_config::config; use revolt_config::config;
#[cfg(feature = "mongodb")]
pub use self::mongodb::*; pub use self::mongodb::*;
pub use self::reference::*; pub use self::reference::*;
@@ -25,8 +27,10 @@ pub enum DatabaseInfo {
/// Use the mock database /// Use the mock database
Reference, Reference,
/// Connect to MongoDB /// Connect to MongoDB
#[cfg(feature = "mongodb")]
MongoDb { uri: String, database_name: String }, MongoDb { uri: String, database_name: String },
/// Use existing MongoDB connection /// Use existing MongoDB connection
#[cfg(feature = "mongodb")]
MongoDbFromClient(::mongodb::Client, String), MongoDbFromClient(::mongodb::Client, String),
} }
@@ -36,6 +40,7 @@ pub enum Database {
/// Mock database /// Mock database
Reference(ReferenceDb), Reference(ReferenceDb),
/// MongoDB database /// MongoDB database
#[cfg(feature = "mongodb")]
MongoDb(MongoDb), MongoDb(MongoDb),
} }
@@ -45,7 +50,7 @@ impl DatabaseInfo {
pub async fn connect(self) -> Result<Database, String> { pub async fn connect(self) -> Result<Database, String> {
let config = config().await; let config = config().await;
Ok(match self { match self {
DatabaseInfo::Auto => { DatabaseInfo::Auto => {
if std::env::var("TEST_DB").is_ok() { if std::env::var("TEST_DB").is_ok() {
DatabaseInfo::Test(format!( DatabaseInfo::Test(format!(
@@ -53,16 +58,20 @@ impl DatabaseInfo {
rand::thread_rng().gen_range(1_000_000..10_000_000) rand::thread_rng().gen_range(1_000_000..10_000_000)
)) ))
.connect() .connect()
.await? .await
} else if !config.database.mongodb.is_empty() { } else if !config.database.mongodb.is_empty() {
DatabaseInfo::MongoDb { #[cfg(feature = "mongodb")]
return DatabaseInfo::MongoDb {
uri: config.database.mongodb, uri: config.database.mongodb,
database_name: "revolt".to_string(), database_name: "revolt".to_string(),
} }
.connect() .connect()
.await? .await;
#[cfg(not(feature = "mongodb"))]
return Err("MongoDB not enabled.".to_string())
} else { } else {
DatabaseInfo::Reference.connect().await? DatabaseInfo::Reference.connect().await
} }
} }
DatabaseInfo::Test(database_name) => { DatabaseInfo::Test(database_name) => {
@@ -70,30 +79,36 @@ impl DatabaseInfo {
.expect("`TEST_DB` environment variable should be set to REFERENCE or MONGODB") .expect("`TEST_DB` environment variable should be set to REFERENCE or MONGODB")
.as_str() .as_str()
{ {
"REFERENCE" => DatabaseInfo::Reference.connect().await?, "REFERENCE" => DatabaseInfo::Reference.connect().await,
"MONGODB" => { "MONGODB" => {
DatabaseInfo::MongoDb { #[cfg(feature = "mongodb")]
return DatabaseInfo::MongoDb {
uri: config.database.mongodb, uri: config.database.mongodb,
database_name, database_name,
} }
.connect() .connect()
.await? .await;
#[cfg(not(feature = "mongodb"))]
return Err("MongoDB not enabled.".to_string())
} }
_ => unreachable!("must specify REFERENCE or MONGODB"), _ => unreachable!("must specify REFERENCE or MONGODB"),
} }
} }
DatabaseInfo::Reference => Database::Reference(Default::default()), DatabaseInfo::Reference => Ok(Database::Reference(Default::default())),
#[cfg(feature = "mongodb")]
DatabaseInfo::MongoDb { uri, database_name } => { DatabaseInfo::MongoDb { uri, database_name } => {
let client = ::mongodb::Client::with_uri_str(uri) let client = ::mongodb::Client::with_uri_str(uri)
.await .await
.map_err(|_| "Failed to init db connection.".to_string())?; .map_err(|_| "Failed to init db connection.".to_string())?;
Database::MongoDb(MongoDb(client, database_name)) Ok(Database::MongoDb(MongoDb(client, database_name)))
} }
#[cfg(feature = "mongodb")]
DatabaseInfo::MongoDbFromClient(client, database_name) => { DatabaseInfo::MongoDbFromClient(client, database_name) => {
Database::MongoDb(MongoDb(client, database_name)) Ok(Database::MongoDb(MongoDb(client, database_name)))
} }
}) }
} }
} }
@@ -219,12 +234,16 @@ impl Database {
Authifier { Authifier {
database: match self { database: match self {
Database::Reference(_) => Default::default(), Database::Reference(_) => Default::default(),
#[cfg(feature = "mongodb")]
Database::MongoDb(MongoDb(client, _)) => authifier::Database::MongoDb( Database::MongoDb(MongoDb(client, _)) => authifier::Database::MongoDb(
authifier::database::MongoDb(client.database("revolt")), authifier::database::MongoDb(client.database("revolt")),
), ),
}, },
config: auth_config, config: auth_config,
#[cfg(feature = "tasks")]
event_channel: Some(crate::tasks::authifier_relay::sender()), event_channel: Some(crate::tasks::authifier_relay::sender()),
#[cfg(not(feature = "tasks"))]
event_channel: None,
} }
} }
} }

View File

@@ -10,7 +10,6 @@ use serde::de::DeserializeOwned;
use serde::Serialize; use serde::Serialize;
database_derived!( database_derived!(
#[cfg(feature = "mongodb")]
/// MongoDB implementation /// MongoDB implementation
#[derive(Debug)] #[derive(Debug)]
pub struct MongoDb(pub ::mongodb::Client, pub String); pub struct MongoDb(pub ::mongodb::Client, pub String);

View File

@@ -25,6 +25,9 @@ pub use mongodb;
#[macro_use] #[macro_use]
extern crate bson; extern crate bson;
#[cfg(not(feature = "async-std-runtime"))]
compile_error!("async-std-runtime feature must be enabled.");
#[macro_export] #[macro_export]
#[cfg(debug_assertions)] #[cfg(debug_assertions)]
macro_rules! query { macro_rules! query {
@@ -103,6 +106,7 @@ pub mod util;
pub use models::*; pub use models::*;
pub mod events; pub mod events;
#[cfg(feature = "tasks")]
pub mod tasks; pub mod tasks;
mod amqp; mod amqp;

View File

@@ -1,3 +1,4 @@
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::{Bot, FieldsBot, PartialBot}; use crate::{Bot, FieldsBot, PartialBot};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::Invite; use crate::Invite;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::ChannelUnread; use crate::ChannelUnread;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::{FieldsWebhook, PartialWebhook, Webhook}; use crate::{FieldsWebhook, PartialWebhook, Webhook};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -9,10 +9,13 @@ use serde::{Deserialize, Serialize};
use ulid::Ulid; use ulid::Ulid;
use crate::{ use crate::{
events::client::EventV1, tasks::ack::AckEvent, Database, File, IntoDocumentPath, PartialServer, events::client::EventV1, Database, File, PartialServer,
Server, SystemMessage, User, AMQP, Server, SystemMessage, User, AMQP,
}; };
#[cfg(feature = "mongodb")]
use crate::IntoDocumentPath;
auto_derived!( auto_derived!(
#[serde(tag = "channel_type")] #[serde(tag = "channel_type")]
pub enum Channel { pub enum Channel {
@@ -670,10 +673,11 @@ impl Channel {
.private(user.to_string()) .private(user.to_string())
.await; .await;
#[cfg(feature = "tasks")]
crate::tasks::ack::queue_ack( crate::tasks::ack::queue_ack(
self.id().to_string(), self.id().to_string(),
user.to_string(), user.to_string(),
AckEvent::AckMessage { crate::tasks::ack::AckEvent::AckMessage {
id: message.to_string(), id: message.to_string(),
}, },
) )
@@ -790,6 +794,7 @@ impl Channel {
} }
} }
#[cfg(feature = "mongodb")]
impl IntoDocumentPath for FieldsChannel { impl IntoDocumentPath for FieldsChannel {
fn as_path(&self) -> Option<&'static str> { fn as_path(&self) -> Option<&'static str> {
Some(match self { Some(match self {

View File

@@ -1,5 +1,7 @@
use crate::{revolt_result::Result, Channel, FieldsChannel, PartialChannel}; use crate::{revolt_result::Result, Channel, FieldsChannel, PartialChannel};
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::Emoji; use crate::Emoji;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::FileHash; use crate::FileHash;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -4,6 +4,7 @@ use crate::File;
use super::FileUsedFor; use super::FileUsedFor;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -14,7 +14,6 @@ use validator::Validate;
use crate::{ use crate::{
events::client::EventV1, events::client::EventV1,
tasks::{self, ack::AckEvent},
util::{ util::{
bulk_permissions::BulkDatabasePermissionQuery, idempotency::IdempotencyKey, bulk_permissions::BulkDatabasePermissionQuery, idempotency::IdempotencyKey,
permissions::DatabasePermissionQuery, permissions::DatabasePermissionQuery,
@@ -22,6 +21,9 @@ use crate::{
Channel, Database, Emoji, File, User, AMQP, Channel, Database, Emoji, File, User, AMQP,
}; };
#[cfg(feature = "tasks")]
use crate::tasks::{self, ack::AckEvent};
auto_derived_partial!( auto_derived_partial!(
/// Message /// Message
pub struct Message { pub struct Message {
@@ -619,9 +621,11 @@ impl Message {
.await; .await;
// Update last_message_id // Update last_message_id
#[cfg(feature = "tasks")]
tasks::last_message_id::queue(self.channel.to_string(), self.id.to_string(), is_dm).await; tasks::last_message_id::queue(self.channel.to_string(), self.id.to_string(), is_dm).await;
// Add mentions for affected users // Add mentions for affected users
#[cfg(feature = "tasks")]
if !mentions_elsewhere { if !mentions_elsewhere {
if let Some(mentions) = &self.mentions { if let Some(mentions) = &self.mentions {
tasks::ack::queue_message( tasks::ack::queue_message(
@@ -640,6 +644,7 @@ impl Message {
} }
// Generate embeds // Generate embeds
#[cfg(feature = "tasks")]
if generate_embeds { if generate_embeds {
if let Some(content) = &self.content { if let Some(content) = &self.content {
tasks::process_embeds::queue( tasks::process_embeds::queue(
@@ -676,10 +681,12 @@ impl Message {
) )
.await?; .await?;
if !self.has_suppressed_notifications() if !self.has_suppressed_notifications()
&& (self.mentions.is_some() || self.contains_mass_push_mention()) && (self.mentions.is_some() || self.contains_mass_push_mention())
{ {
// send Push notifications // send Push notifications
#[cfg(feature = "tasks")]
tasks::ack::queue_message( tasks::ack::queue_message(
self.channel.to_string(), self.channel.to_string(),
AckEvent::ProcessMessage { AckEvent::ProcessMessage {

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::{AppendMessage, FieldsMessage, Message, MessageQuery, PartialMessage}; use crate::{AppendMessage, FieldsMessage, Message, MessageQuery, PartialMessage};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -38,7 +38,10 @@ pub use servers::*;
pub use user_settings::*; pub use user_settings::*;
pub use users::*; pub use users::*;
use crate::{Database, MongoDb, ReferenceDb}; use crate::{Database, ReferenceDb};
#[cfg(feature = "mongodb")]
use crate::MongoDb;
pub trait AbstractDatabase: pub trait AbstractDatabase:
Sync Sync
@@ -66,6 +69,8 @@ pub trait AbstractDatabase:
} }
impl AbstractDatabase for ReferenceDb {} impl AbstractDatabase for ReferenceDb {}
#[cfg(feature = "mongodb")]
impl AbstractDatabase for MongoDb {} impl AbstractDatabase for MongoDb {}
impl std::ops::Deref for Database { impl std::ops::Deref for Database {
@@ -74,6 +79,7 @@ impl std::ops::Deref for Database {
fn deref(&self) -> &Self::Target { fn deref(&self) -> &Self::Target {
match &self { match &self {
Database::Reference(dummy) => dummy, Database::Reference(dummy) => dummy,
#[cfg(feature = "mongodb")]
Database::MongoDb(mongo) => mongo, Database::MongoDb(mongo) => mongo,
} }
} }

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::PolicyChange; use crate::PolicyChange;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -1,6 +1,8 @@
use std::time::Duration; use std::time::Duration;
use crate::{revolt_result::Result, RatelimitEvent, RatelimitEventType}; use crate::{revolt_result::Result, RatelimitEvent, RatelimitEventType};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::Report; use crate::Report;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::Snapshot; use crate::Snapshot;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::{MemberCompositeKey, ServerBan}; use crate::{MemberCompositeKey, ServerBan};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -1,16 +1,20 @@
use ::mongodb::SessionCursor; #[cfg(feature = "mongodb")]
use ::mongodb::{ClientSession, SessionCursor};
use revolt_result::Result; use revolt_result::Result;
use crate::{FieldsMember, Member, MemberCompositeKey, PartialMember}; use crate::{FieldsMember, Member, MemberCompositeKey, PartialMember};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;
#[derive(Debug)] #[derive(Debug)]
#[allow(clippy::large_enum_variant)]
pub enum ChunkedServerMembersGenerator { pub enum ChunkedServerMembersGenerator {
#[cfg(feature = "mongodb")] #[cfg(feature = "mongodb")]
MongoDb { MongoDb {
session: ::mongodb::ClientSession, session: ClientSession,
cursor: Option<SessionCursor<Member>>, cursor: Option<SessionCursor<Member>>,
}, },
@@ -22,7 +26,7 @@ pub enum ChunkedServerMembersGenerator {
impl ChunkedServerMembersGenerator { impl ChunkedServerMembersGenerator {
#[cfg(feature = "mongodb")] #[cfg(feature = "mongodb")]
pub fn new_mongo(session: ::mongodb::ClientSession, cursor: SessionCursor<Member>) -> Self { pub fn new_mongo(session: ClientSession, cursor: SessionCursor<Member>) -> Self {
ChunkedServerMembersGenerator::MongoDb { ChunkedServerMembersGenerator::MongoDb {
session, session,
cursor: Some(cursor), cursor: Some(cursor),

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::{FieldsRole, FieldsServer, PartialRole, PartialServer, Role, Server}; use crate::{FieldsRole, FieldsServer, PartialRole, PartialServer, Role, Server};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -2,6 +2,7 @@ use revolt_result::Result;
use crate::UserSettings; use crate::UserSettings;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -4,6 +4,7 @@ use revolt_result::Result;
use crate::{FieldsUser, PartialUser, RelationshipStatus, User}; use crate::{FieldsUser, PartialUser, RelationshipStatus, User};
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;

View File

@@ -212,16 +212,34 @@ impl AbstractUsers for MongoDb {
partial: &PartialUser, partial: &PartialUser,
remove: Vec<FieldsUser>, remove: Vec<FieldsUser>,
) -> Result<()> { ) -> Result<()> {
query!( if remove.contains(&FieldsUser::StatusText) && partial.status.is_some() {
self, // stupid-ass workaround to fix mongo conflicting the same item
update_one_by_id, let _: Result<()> = query!(
COL, self,
id, update_one_by_id,
partial, COL,
remove.iter().map(|x| x as &dyn IntoDocumentPath).collect(), id,
None PartialUser {
) ..Default::default()
.map(|_| ()) },
remove.iter().map(|x| x as &dyn IntoDocumentPath).collect(),
None
)
.map(|_| ());
query!(self, update_one_by_id, COL, id, partial, vec![], None).map(|_| ())
} else {
query!(
self,
update_one_by_id,
COL,
id,
partial,
remove.iter().map(|x| x as &dyn IntoDocumentPath).collect(),
None
)
.map(|_| ())
}
} }
/// Set relationship with another user /// Set relationship with another user

View File

@@ -14,41 +14,40 @@ use crate::{
}; };
/// Reference to some object in the database /// Reference to some object in the database
#[derive(Serialize, Deserialize)] pub struct Reference<'a> {
pub struct Reference {
/// Id of object /// Id of object
pub id: String, pub id: &'a str,
} }
impl Reference { impl<'a> Reference<'a> {
/// Create a Ref from an unchecked string /// Create a Ref from an unchecked string
pub fn from_unchecked(id: String) -> Reference { pub fn from_unchecked(id: &'a str) -> Reference<'a> {
Reference { id } Reference { id }
} }
/// Fetch ban from Ref /// Fetch ban from Ref
pub async fn as_ban(&self, db: &Database, server: &str) -> Result<ServerBan> { pub async fn as_ban(&self, db: &Database, server: &str) -> Result<ServerBan> {
db.fetch_ban(server, &self.id).await db.fetch_ban(server, self.id).await
} }
/// Fetch bot from Ref /// Fetch bot from Ref
pub async fn as_bot(&self, db: &Database) -> Result<Bot> { pub async fn as_bot(&self, db: &Database) -> Result<Bot> {
db.fetch_bot(&self.id).await db.fetch_bot(self.id).await
} }
/// Fetch emoji from Ref /// Fetch emoji from Ref
pub async fn as_emoji(&self, db: &Database) -> Result<Emoji> { pub async fn as_emoji(&self, db: &Database) -> Result<Emoji> {
db.fetch_emoji(&self.id).await 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 invite from Ref or create invite to server if discoverable /// Fetch invite from Ref or create invite to server if discoverable
pub async fn as_invite(&self, db: &Database) -> Result<Invite> { pub async fn as_invite(&self, db: &Database) -> Result<Invite> {
if ulid::Ulid::from_str(&self.id).is_ok() { if ulid::Ulid::from_str(self.id).is_ok() {
let server = self.as_server(db).await?; let server = self.as_server(db).await?;
if !server.discoverable { if !server.discoverable {
return Err(create_error!(NotFound)); return Err(create_error!(NotFound));
@@ -65,18 +64,18 @@ impl Reference {
.ok_or(create_error!(NotFound))?, .ok_or(create_error!(NotFound))?,
}) })
} else { } else {
db.fetch_invite(&self.id).await db.fetch_invite(self.id).await
} }
} }
/// Fetch message from Ref /// Fetch message from Ref
pub async fn as_message(&self, db: &Database) -> Result<Message> { pub async fn as_message(&self, db: &Database) -> Result<Message> {
db.fetch_message(&self.id).await db.fetch_message(self.id).await
} }
/// Fetch message from Ref and validate channel /// Fetch message from Ref and validate channel
pub async fn as_message_in_channel(&self, db: &Database, channel: &str) -> Result<Message> { pub async fn as_message_in_channel(&self, db: &Database, channel: &str) -> Result<Message> {
let msg = db.fetch_message(&self.id).await?; let msg = db.fetch_message(self.id).await?;
if msg.channel != channel { if msg.channel != channel {
return Err(create_error!(NotFound)); return Err(create_error!(NotFound));
} }
@@ -86,36 +85,36 @@ impl Reference {
/// Fetch member from Ref /// Fetch member from Ref
pub async fn as_member(&self, db: &Database, server: &str) -> Result<Member> { pub async fn as_member(&self, db: &Database, server: &str) -> Result<Member> {
db.fetch_member(server, &self.id).await db.fetch_member(server, self.id).await
} }
/// Fetch server from Ref /// Fetch server from Ref
pub async fn as_server(&self, db: &Database) -> Result<Server> { pub async fn as_server(&self, db: &Database) -> Result<Server> {
db.fetch_server(&self.id).await db.fetch_server(self.id).await
} }
/// Fetch user from Ref /// Fetch user from Ref
pub async fn as_user(&self, db: &Database) -> Result<User> { pub async fn as_user(&self, db: &Database) -> Result<User> {
db.fetch_user(&self.id).await db.fetch_user(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
} }
} }
#[cfg(feature = "rocket-impl")] #[cfg(feature = "rocket-impl")]
impl<'r> FromParam<'r> for Reference { impl<'r> FromParam<'r> for Reference<'r> {
type Error = &'r str; type Error = &'r str;
fn from_param(param: &'r str) -> Result<Self, Self::Error> { fn from_param(param: &'r str) -> Result<Self, Self::Error> {
Ok(Reference::from_unchecked(param.into())) Ok(Reference::from_unchecked(param))
} }
} }
#[cfg(feature = "rocket-impl")] #[cfg(feature = "rocket-impl")]
impl JsonSchema for Reference { impl<'a> JsonSchema for Reference<'a> {
fn schema_name() -> String { fn schema_name() -> String {
"Id".to_string() "Id".to_string()
} }

View File

@@ -362,7 +362,7 @@ pub async fn sync_voice_permissions(
.iter() .iter()
.flatten() .flatten()
{ {
let user = Reference::from_unchecked(user_id.clone()) let user = Reference::from_unchecked(&user_id)
.as_user(db) .as_user(db)
.await?; .await?;
@@ -387,7 +387,7 @@ pub async fn sync_user_voice_permissions(
let member = match server_id { let member = match server_id {
Some(server_id) => Some( Some(server_id) => Some(
Reference::from_unchecked(user.id.clone()) Reference::from_unchecked(&user.id)
.as_member(db, server_id) .as_member(db, server_id)
.await?, .await?,
), ),

View File

@@ -132,8 +132,8 @@ auto_derived!(
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 2048)))] #[cfg_attr(feature = "validator", validate(length(min = 1, max = 2048)))]
pub interactions_url: Option<String>, pub interactions_url: Option<String>,
/// Fields to remove from bot object /// Fields to remove from bot object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsBot>>, pub remove: Vec<FieldsBot>,
} }
/// Where we are inviting a bot to /// Where we are inviting a bot to

View File

@@ -227,7 +227,7 @@ auto_derived!(
/// Fields to remove from channel /// Fields to remove from channel
#[cfg_attr(feature = "serde", serde(default))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsChannel>>, pub remove: Vec<FieldsChannel>,
} }
/// Create new group /// Create new group

View File

@@ -217,7 +217,7 @@ auto_derived!(
#[derive(Default)] #[derive(Default)]
#[cfg_attr(feature = "validator", derive(Validate))] #[cfg_attr(feature = "validator", derive(Validate))]
pub struct SendableEmbed { pub struct SendableEmbed {
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 128)))] #[cfg_attr(feature = "validator", validate(length(min = 1, max = 256)))]
pub icon_url: Option<String>, pub icon_url: Option<String>,
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 256)))] #[cfg_attr(feature = "validator", validate(length(min = 1, max = 256)))]
pub url: Option<String>, pub url: Option<String>,

View File

@@ -140,8 +140,7 @@ auto_derived!(
/// voice channel to move to if already in a voice channel /// voice channel to move to if already in a voice channel
pub voice_channel: Option<String>, pub voice_channel: Option<String>,
/// Fields to remove from channel object /// Fields to remove from channel object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsMember>>, pub remove: Vec<FieldsMember>,
} }
); );

View File

@@ -175,6 +175,8 @@ auto_derived!(
/// Ranking position /// Ranking position
/// ///
/// Smaller values take priority. /// Smaller values take priority.
///
/// **Removed** - no effect, use the edit server role positions route
pub rank: Option<i64>, pub rank: Option<i64>,
} }
@@ -247,8 +249,8 @@ auto_derived!(
pub analytics: Option<bool>, pub analytics: Option<bool>,
/// Fields to remove from server object /// Fields to remove from server object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsServer>>, pub remove: Vec<FieldsServer>,
} }
/// New role information /// New role information
@@ -270,8 +272,8 @@ auto_derived!(
/// **Removed** - no effect, use the edit server role positions route /// **Removed** - no effect, use the edit server role positions route
pub rank: Option<i64>, pub rank: Option<i64>,
/// Fields to remove from role object /// Fields to remove from role object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsRole>>, pub remove: Vec<FieldsRole>,
} }
/// New role permissions /// New role permissions

View File

@@ -245,8 +245,8 @@ auto_derived!(
pub flags: Option<i32>, pub flags: Option<i32>,
/// Fields to remove from user object /// Fields to remove from user object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsUser>>, pub remove: Vec<FieldsUser>,
} }
/// User flag reponse /// User flag reponse

View File

@@ -8,6 +8,7 @@ license = "AGPL-3.0-or-later"
revolt-result = { version = "0.8.8", path = "../../core/result" } revolt-result = { version = "0.8.8", path = "../../core/result" }
revolt-config = { version = "0.8.8", path = "../../core/config", features = [ revolt-config = { version = "0.8.8", path = "../../core/config", features = [
"report-macros", "report-macros",
"anyhow"
] } ] }
revolt-database = { version = "0.8.8", path = "../../core/database" } revolt-database = { version = "0.8.8", path = "../../core/database" }
revolt-models = { version = "0.8.8", path = "../../core/models", features = [ revolt-models = { version = "0.8.8", path = "../../core/models", features = [

View File

@@ -59,7 +59,7 @@ pub async fn ingress(
let channel_id = channel_id.to_internal_error()?; let channel_id = channel_id.to_internal_error()?;
let user_id = user_id.to_internal_error()?; let user_id = user_id.to_internal_error()?;
let channel = Reference::from_unchecked(channel_id.clone()) let channel = Reference::from_unchecked(&channel_id)
.as_channel(db) .as_channel(db)
.await?; .await?;
@@ -68,30 +68,30 @@ pub async fn ingress(
// Only publish one event when a user is moved from one channel to another. // Only publish one event when a user is moved from one channel to another.
if let Some(moved_from) = get_user_moved_to_voice(channel_id, user_id).await? { if let Some(moved_from) = get_user_moved_to_voice(channel_id, user_id).await? {
EventV1::VoiceChannelMove { EventV1::VoiceChannelMove {
user: user_id.clone(), user: user_id.to_string(),
from: moved_from, from: moved_from,
to: channel_id.clone(), to: channel_id.to_string(),
state: voice_state, state: voice_state,
} }
.p(channel_id.clone()) .p(channel_id.to_string())
.await; .await;
} else { } else {
EventV1::VoiceChannelJoin { EventV1::VoiceChannelJoin {
id: channel_id.clone(), id: channel_id.to_string(),
state: voice_state, state: voice_state,
} }
.p(channel_id.clone()) .p(channel_id.to_string())
.await; .await;
}; };
// First user who joined - send call started system message. // First user who joined - send call started system message.
if event.room.as_ref().unwrap().num_participants == 1 { if event.room.as_ref().unwrap().num_participants == 1 {
let user = Reference::from_unchecked(user_id.clone()) let user = Reference::from_unchecked(&user_id)
.as_user(db) .as_user(db)
.await?; .await?;
let mut call_started_message = SystemMessage::CallStarted { let mut call_started_message = SystemMessage::CallStarted {
by: user_id.clone(), by: user_id.to_string(),
finished_at: None, finished_at: None,
} }
.into_message(channel.id().to_string()); .into_message(channel.id().to_string());
@@ -120,7 +120,7 @@ pub async fn ingress(
let channel_id = channel_id.to_internal_error()?; let channel_id = channel_id.to_internal_error()?;
let user_id = user_id.to_internal_error()?; let user_id = user_id.to_internal_error()?;
let channel = Reference::from_unchecked(channel_id.clone()) let channel = Reference::from_unchecked(&channel_id)
.as_channel(db) .as_channel(db)
.await?; .await?;
@@ -147,7 +147,7 @@ pub async fn ingress(
take_channel_call_started_system_message(channel_id).await? take_channel_call_started_system_message(channel_id).await?
{ {
// Could have been deleted // Could have been deleted
if let Ok(mut message) = Reference::from_unchecked(system_message_id) if let Ok(mut message) = Reference::from_unchecked(&system_message_id)
.as_message(db) .as_message(db)
.await .await
{ {
@@ -179,11 +179,11 @@ pub async fn ingress(
let user_id = user_id.to_internal_error()?; let user_id = user_id.to_internal_error()?;
let track = event.track.as_ref().to_internal_error()?; let track = event.track.as_ref().to_internal_error()?;
let channel = Reference::from_unchecked(channel_id.clone()) let channel = Reference::from_unchecked(&channel_id)
.as_channel(db) .as_channel(db)
.await?; .await?;
let user = Reference::from_unchecked(user_id.clone()) let user = Reference::from_unchecked(&user_id)
.as_user(db) .as_user(db)
.await?; .await?;

View File

@@ -11,7 +11,7 @@ use rocket_empty::EmptyResponse;
pub async fn delete_bot( pub async fn delete_bot(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let bot = target.as_bot(db).await?; let bot = target.as_bot(db).await?;
if bot.owner != user.id { if bot.owner != user.id {

View File

@@ -14,7 +14,7 @@ use validator::Validate;
pub async fn edit_bot( pub async fn edit_bot(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<DataEditBot>, data: Json<DataEditBot>,
) -> Result<Json<v0::BotWithUserResponse>> { ) -> Result<Json<v0::BotWithUserResponse>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -37,7 +37,7 @@ pub async fn edit_bot(
if data.public.is_none() if data.public.is_none()
&& data.analytics.is_none() && data.analytics.is_none()
&& data.interactions_url.is_none() && data.interactions_url.is_none()
&& data.remove.is_none() && data.remove.is_empty()
{ {
return Ok(Json(v0::BotWithUserResponse { return Ok(Json(v0::BotWithUserResponse {
bot: bot.into(), bot: bot.into(),
@@ -64,7 +64,6 @@ pub async fn edit_bot(
db, db,
partial, partial,
remove remove
.unwrap_or_default()
.into_iter() .into_iter()
.map(|v| v.into()) .map(|v| v.into())
.collect(), .collect(),
@@ -100,7 +99,7 @@ mod test {
.body( .body(
json!(v0::DataEditBot { json!(v0::DataEditBot {
public: Some(true), public: Some(true),
remove: Some(vec![FieldsBot::Token]), remove: vec![FieldsBot::Token],
..Default::default() ..Default::default()
}) })
.to_string(), .to_string(),

View File

@@ -11,7 +11,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch_bot( pub async fn fetch_bot(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
bot: Reference, bot: Reference<'_>,
) -> Result<Json<FetchBotResponse>> { ) -> Result<Json<FetchBotResponse>> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));

View File

@@ -13,9 +13,9 @@ use rocket::State;
pub async fn fetch_public_bot( pub async fn fetch_public_bot(
db: &State<Database>, db: &State<Database>,
user: Option<User>, user: Option<User>,
target: Reference, target: Reference<'_>,
) -> Result<Json<PublicBot>> { ) -> Result<Json<PublicBot>> {
let bot = db.fetch_bot(&target.id).await?; let bot = db.fetch_bot(target.id).await?;
if !bot.public && user.is_none_or(|x| x.id != bot.owner) { if !bot.public && user.is_none_or(|x| x.id != bot.owner) {
return Err(create_error!(NotFound)); return Err(create_error!(NotFound));
} }

View File

@@ -20,7 +20,7 @@ pub async fn invite_bot(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
dest: Json<v0::InviteBotDestination>, dest: Json<v0::InviteBotDestination>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {

View File

@@ -15,8 +15,8 @@ use rocket_empty::EmptyResponse;
pub async fn ack( pub async fn ack(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
message: Reference, message: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));
@@ -29,7 +29,7 @@ pub async fn ack(
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?; .throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
channel channel
.ack(&user.id, &message.id) .ack(&user.id, message.id)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }

View File

@@ -17,7 +17,7 @@ pub async fn delete(
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: v0::OptionsChannelDelete, options: v0::OptionsChannelDelete,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let mut channel = target.as_channel(db).await?; let mut channel = target.as_channel(db).await?;

View File

@@ -17,7 +17,7 @@ pub async fn edit(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataEditChannel>, data: Json<v0::DataEditChannel>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -39,7 +39,7 @@ pub async fn edit(
&& data.nsfw.is_none() && data.nsfw.is_none()
&& data.owner.is_none() && data.owner.is_none()
&& data.voice.is_none() && data.voice.is_none()
&& data.remove.is_none() && data.remove.is_empty()
{ {
return Ok(Json(channel.into())); return Ok(Json(channel.into()));
} }
@@ -97,23 +97,21 @@ pub async fn edit(
nsfw, nsfw,
.. ..
} => { } => {
if let Some(fields) = &data.remove { if data.remove.contains(&v0::FieldsChannel::Icon) {
if fields.contains(&v0::FieldsChannel::Icon) { if let Some(icon) = &icon {
if let Some(icon) = &icon { db.mark_attachment_as_deleted(&icon.id).await?;
db.mark_attachment_as_deleted(&icon.id).await?;
}
} }
}
for field in fields { for field in &data.remove {
match field { match field {
v0::FieldsChannel::Description => { v0::FieldsChannel::Description => {
description.take(); description.take();
}
v0::FieldsChannel::Icon => {
icon.take();
}
_ => {}
} }
v0::FieldsChannel::Icon => {
icon.take();
}
_ => {}
} }
} }
@@ -202,23 +200,21 @@ pub async fn edit(
voice, voice,
.. ..
} => { } => {
if let Some(fields) = &data.remove { if data.remove.contains(&v0::FieldsChannel::Icon) {
if fields.contains(&v0::FieldsChannel::Icon) { if let Some(icon) = &icon {
if let Some(icon) = &icon { db.mark_attachment_as_deleted(&icon.id).await?;
db.mark_attachment_as_deleted(&icon.id).await?;
}
} }
}
for field in fields { for field in &data.remove {
match field { match field {
v0::FieldsChannel::Description => { v0::FieldsChannel::Description => {
description.take(); description.take();
}
v0::FieldsChannel::Icon => {
icon.take();
}
_ => {}
} }
v0::FieldsChannel::Icon => {
icon.take();
}
_ => {}
} }
} }
@@ -255,7 +251,6 @@ pub async fn edit(
db, db,
partial, partial,
data.remove data.remove
.unwrap_or_default()
.into_iter() .into_iter()
.map(|f| f.into()) .map(|f| f.into())
.collect(), .collect(),

View File

@@ -16,7 +16,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch( pub async fn fetch(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;

View File

@@ -17,8 +17,8 @@ pub async fn add_member(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
group_id: Reference, group_id: Reference<'_>,
member_id: Reference, member_id: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));

View File

@@ -15,8 +15,8 @@ pub async fn remove_member(
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
member: Reference, member: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));

View File

@@ -18,7 +18,7 @@ use rocket::{serde::json::Json, State};
pub async fn create_invite( pub async fn create_invite(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::Invite>> { ) -> Result<Json<v0::Invite>> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));

View File

@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch_members( pub async fn fetch_members(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<Vec<v0::User>>> { ) -> Result<Json<Vec<v0::User>>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);

View File

@@ -22,7 +22,7 @@ use validator::Validate;
pub async fn bulk_delete_messages( pub async fn bulk_delete_messages(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: Json<v0::OptionsBulkDelete>, options: Json<v0::OptionsBulkDelete>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let options = options.into_inner(); let options = options.into_inner();
@@ -51,7 +51,7 @@ pub async fn bulk_delete_messages(
.await .await
.throw_if_lacking_channel_permission(ChannelPermission::ManageMessages)?; .throw_if_lacking_channel_permission(ChannelPermission::ManageMessages)?;
Message::bulk_delete(db, &target.id, options.ids) Message::bulk_delete(db, target.id, options.ids)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }

View File

@@ -17,8 +17,8 @@ use rocket_empty::EmptyResponse;
pub async fn clear_reactions( pub async fn clear_reactions(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);

View File

@@ -15,10 +15,10 @@ use rocket_empty::EmptyResponse;
pub async fn delete( pub async fn delete(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let message = msg.as_message_in_channel(db, &target.id).await?; let message = msg.as_message_in_channel(db, target.id).await?;
if message.author != user.id { if message.author != user.id {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;

View File

@@ -18,8 +18,8 @@ use validator::Validate;
pub async fn edit( pub async fn edit(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
edit: Json<v0::DataEditMessage>, edit: Json<v0::DataEditMessage>,
) -> Result<Json<v0::Message>> { ) -> Result<Json<v0::Message>> {
let edit = edit.into_inner(); let edit = edit.into_inner();

View File

@@ -15,8 +15,8 @@ use rocket::{serde::json::Json, State};
pub async fn fetch( pub async fn fetch(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
) -> Result<Json<v0::Message>> { ) -> Result<Json<v0::Message>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);

View File

@@ -1,4 +1,7 @@
use revolt_database::{util::{permissions::DatabasePermissionQuery, reference::Reference}, Channel, Database, PartialMessage, SystemMessage, User, AMQP}; use revolt_database::{
util::{permissions::DatabasePermissionQuery, reference::Reference},
Channel, Database, PartialMessage, SystemMessage, User, AMQP,
};
use revolt_models::v0::MessageAuthor; use revolt_models::v0::MessageAuthor;
use revolt_permissions::{calculate_channel_permissions, ChannelPermission}; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result}; use revolt_result::{create_error, Result};
@@ -14,8 +17,8 @@ pub async fn message_pin(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
@@ -169,7 +172,7 @@ mod test {
}) })
.await; .await;
let updated_message = Reference::from_unchecked(message.id) let updated_message = Reference::from_unchecked(&message.id)
.as_message(&harness.db) .as_message(&harness.db)
.await .await
.expect("Failed to find updated message"); .expect("Failed to find updated message");

View File

@@ -16,7 +16,7 @@ use validator::Validate;
pub async fn query( pub async fn query(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: v0::OptionsQueryMessages, options: v0::OptionsQueryMessages,
) -> Result<Json<v0::BulkMessageResponse>> { ) -> Result<Json<v0::BulkMessageResponse>> {
options.validate().map_err(|error| { options.validate().map_err(|error| {

View File

@@ -15,9 +15,9 @@ use rocket_empty::EmptyResponse;
pub async fn react_message( pub async fn react_message(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
emoji: Reference, emoji: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
@@ -30,7 +30,7 @@ pub async fn react_message(
// Add the reaction // Add the reaction
message message
.add_reaction(db, &user, &emoji.id) .add_reaction(db, &user, emoji.id)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }

View File

@@ -16,7 +16,7 @@ use validator::Validate;
pub async fn search( pub async fn search(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: Json<v0::DataMessageSearch>, options: Json<v0::DataMessageSearch>,
) -> Result<Json<v0::BulkMessageResponse>> { ) -> Result<Json<v0::BulkMessageResponse>> {
if user.bot.is_some() { if user.bot.is_some() {

View File

@@ -21,7 +21,7 @@ pub async fn message_send(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataMessageSend>, data: Json<v0::DataMessageSend>,
idempotency: IdempotencyKey, idempotency: IdempotencyKey,
) -> Result<Json<v0::Message>> { ) -> Result<Json<v0::Message>> {
@@ -204,7 +204,7 @@ mod test {
Member::create(&harness.db, &server, &user, Some(channels.clone())) Member::create(&harness.db, &server, &user, Some(channels.clone()))
.await .await
.expect("Failed to create member"); .expect("Failed to create member");
let member = Reference::from_unchecked(user.id.clone()) let member = Reference::from_unchecked(&user.id)
.as_member(&harness.db, &server.id) .as_member(&harness.db, &server.id)
.await .await
.expect("Failed to get member"); .expect("Failed to get member");
@@ -244,7 +244,7 @@ mod test {
Member::create(&harness.db, &server, &second_user, Some(channels.clone())) Member::create(&harness.db, &server, &second_user, Some(channels.clone()))
.await .await
.expect("Failed to create second member"); .expect("Failed to create second member");
let mut second_member = Reference::from_unchecked(second_user.id.clone()) let mut second_member = Reference::from_unchecked(&second_user.id)
.as_member(&harness.db, &server.id) .as_member(&harness.db, &server.id)
.await .await
.expect("Failed to get second member"); .expect("Failed to get second member");

View File

@@ -14,8 +14,8 @@ pub async fn message_unpin(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
@@ -91,7 +91,7 @@ mod test {
Member::create(&harness.db, &server, &user, Some(channels.clone())) Member::create(&harness.db, &server, &user, Some(channels.clone()))
.await .await
.expect("Failed to create member"); .expect("Failed to create member");
let member = Reference::from_unchecked(user.id.clone()) let member = Reference::from_unchecked(&user.id)
.as_member(&harness.db, &server.id) .as_member(&harness.db, &server.id)
.await .await
.expect("Failed to get member"); .expect("Failed to get member");
@@ -174,7 +174,7 @@ mod test {
}) })
.await; .await;
let updated_message = Reference::from_unchecked(message.id) let updated_message = Reference::from_unchecked(&message.id)
.as_message(&harness.db) .as_message(&harness.db)
.await .await
.expect("Failed to find updated message"); .expect("Failed to find updated message");

View File

@@ -18,9 +18,9 @@ use rocket_empty::EmptyResponse;
pub async fn unreact_message( pub async fn unreact_message(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
msg: Reference, msg: Reference<'_>,
emoji: Reference, emoji: Reference<'_>,
options: v0::OptionsUnreact, options: v0::OptionsUnreact,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
@@ -41,14 +41,14 @@ pub async fn unreact_message(
// Check if we should wipe all of this reaction // Check if we should wipe all of this reaction
if remove_all { if remove_all {
return message return message
.clear_reaction(db, &emoji.id) .clear_reaction(db, emoji.id)
.await .await
.map(|_| EmptyResponse); .map(|_| EmptyResponse);
} }
// Remove the reaction // Remove the reaction
message message
.remove_reaction(db, options.user_id.as_ref().unwrap_or(&user.id), &emoji.id) .remove_reaction(db, options.user_id.as_ref().unwrap_or(&user.id), emoji.id)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }

View File

@@ -50,7 +50,7 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
group_remove_member::remove_member, group_remove_member::remove_member,
voice_join::call, voice_join::call,
permissions_set::set_role_permissions, permissions_set::set_role_permissions,
permissions_set_default::set_default_permissions, permissions_set_default::set_default_channel_permissions,
message_react::react_message, message_react::react_message,
message_unreact::unreact_message, message_unreact::unreact_message,
message_clear_reactions::clear_reactions, message_clear_reactions::clear_reactions,

View File

@@ -17,7 +17,7 @@ pub async fn set_role_permissions(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
role_id: String, role_id: String,
data: Json<v0::DataSetRolePermissions>, data: Json<v0::DataSetRolePermissions>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {

View File

@@ -13,11 +13,11 @@ use rocket::{serde::json::Json, State};
/// Channel must be a `Group`, `TextChannel` or `VoiceChannel`. /// Channel must be a `Group`, `TextChannel` or `VoiceChannel`.
#[openapi(tag = "Channel Permissions")] #[openapi(tag = "Channel Permissions")]
#[put("/<target>/permissions/default", data = "<data>", rank = 1)] #[put("/<target>/permissions/default", data = "<data>", rank = 1)]
pub async fn set_default_permissions( pub async fn set_default_channel_permissions(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataDefaultChannelPermissions>, data: Json<v0::DataDefaultChannelPermissions>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -72,7 +72,7 @@ pub async fn set_default_permissions(
} }
let server = match channel.server() { let server = match channel.server() {
Some(server_id) => Some(Reference::from_unchecked(server_id.to_string()).as_server(db).await?), Some(server_id) => Some(Reference::from_unchecked(server_id).as_server(db).await?),
None => None None => None
}; };

View File

@@ -22,7 +22,7 @@ pub async fn call(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataJoinCall>, data: Json<v0::DataJoinCall>,
) -> Result<Json<v0::CreateVoiceUserResponse>> { ) -> Result<Json<v0::CreateVoiceUserResponse>> {
if !voice_client.is_enabled() { if !voice_client.is_enabled() {
@@ -79,7 +79,7 @@ pub async fn call(
for channel_id in get_user_voice_channels(&user.id).await? { for channel_id in get_user_voice_channels(&user.id).await? {
let node = get_channel_node(&channel_id).await?.unwrap(); let node = get_channel_node(&channel_id).await?.unwrap();
let channel = Reference::from_unchecked(channel_id.clone()) let channel = Reference::from_unchecked(&channel_id)
.as_channel(db) .as_channel(db)
.await?; .await?;

View File

@@ -19,7 +19,7 @@ use validator::Validate;
pub async fn create_webhook( pub async fn create_webhook(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::CreateWebhookBody>, data: Json<v0::CreateWebhookBody>,
) -> Result<Json<v0::Webhook>> { ) -> Result<Json<v0::Webhook>> {
let data = data.into_inner(); let data = data.into_inner();

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch_webhooks( pub async fn fetch_webhooks(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
channel_id: Reference, channel_id: Reference<'_>,
) -> Result<Json<Vec<Webhook>>> { ) -> Result<Json<Vec<Webhook>>> {
let channel = channel_id.as_channel(db).await?; let channel = channel_id.as_channel(db).await?;

View File

@@ -16,7 +16,7 @@ use rocket_empty::EmptyResponse;
pub async fn delete_emoji( pub async fn delete_emoji(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
emoji_id: Reference, emoji_id: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
// Fetch the emoji // Fetch the emoji
let emoji = emoji_id.as_emoji(db).await?; let emoji = emoji_id.as_emoji(db).await?;
@@ -26,7 +26,7 @@ pub async fn delete_emoji(
// Otherwise, validate we have permission to delete from parent // Otherwise, validate we have permission to delete from parent
match &emoji.parent { match &emoji.parent {
EmojiParent::Server { id } => { EmojiParent::Server { id } => {
let server = db.fetch_server(id).await?; let server = db.fetch_server(id.as_str()).await?;
// Check for permission // Check for permission
let mut query = DatabasePermissionQuery::new(db, &user).server(&server); let mut query = DatabasePermissionQuery::new(db, &user).server(&server);

View File

@@ -9,7 +9,7 @@ use rocket::{serde::json::Json, State};
/// Fetch an emoji by its id. /// Fetch an emoji by its id.
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[get("/emoji/<emoji_id>")] #[get("/emoji/<emoji_id>")]
pub async fn fetch_emoji(db: &State<Database>, emoji_id: Reference) -> Result<Json<v0::Emoji>> { pub async fn fetch_emoji(db: &State<Database>, emoji_id: Reference<'_>) -> Result<Json<v0::Emoji>> {
emoji_id emoji_id
.as_emoji(db) .as_emoji(db)
.await .await

View File

@@ -12,7 +12,7 @@ use rocket_empty::EmptyResponse;
/// Delete an invite by its id. /// Delete an invite by its id.
#[openapi(tag = "Invites")] #[openapi(tag = "Invites")]
#[delete("/<target>")] #[delete("/<target>")]
pub async fn delete(db: &State<Database>, user: User, target: Reference) -> Result<EmptyResponse> { pub async fn delete(db: &State<Database>, user: User, target: Reference<'_>) -> Result<EmptyResponse> {
let invite = target.as_invite(db).await?; let invite = target.as_invite(db).await?;
if user.id == invite.creator() { if user.id == invite.creator() {

View File

@@ -8,7 +8,7 @@ use rocket::{serde::json::Json, State};
/// Fetch an invite by its id. /// Fetch an invite by its id.
#[openapi(tag = "Invites")] #[openapi(tag = "Invites")]
#[get("/<target>")] #[get("/<target>")]
pub async fn fetch(db: &State<Database>, target: Reference) -> Result<Json<v0::InviteResponse>> { pub async fn fetch(db: &State<Database>, target: Reference<'_>) -> Result<Json<v0::InviteResponse>> {
Ok(Json(match target.as_invite(db).await? { Ok(Json(match target.as_invite(db).await? {
Invite::Server { Invite::Server {
channel, creator, .. channel, creator, ..
@@ -27,7 +27,7 @@ pub async fn fetch(db: &State<Database>, target: Reference) -> Result<Json<v0::I
let server = db.fetch_server(&server).await?; let server = db.fetch_server(&server).await?;
v0::InviteResponse::Server { v0::InviteResponse::Server {
code: target.id, code: target.id.to_string(),
member_count: db.fetch_member_count(&server.id).await? as i64, member_count: db.fetch_member_count(&server.id).await? as i64,
server_id: server.id, server_id: server.id,
server_name: server.name, server_name: server.name,
@@ -57,7 +57,7 @@ pub async fn fetch(db: &State<Database>, target: Reference) -> Result<Json<v0::I
description, description,
.. ..
} => v0::InviteResponse::Group { } => v0::InviteResponse::Group {
code: target.id, code: target.id.to_string(),
channel_id: id, channel_id: id,
channel_name: name, channel_name: name,
channel_description: description, channel_description: description,

View File

@@ -12,7 +12,7 @@ pub async fn join(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::InviteJoinResponse>> { ) -> Result<Json<v0::InviteJoinResponse>> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));

View File

@@ -19,8 +19,8 @@ pub async fn ban(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
server: Reference, server: Reference<'_>,
target: Reference, target: Reference<'_>,
data: Json<v0::DataBanCreate>, data: Json<v0::DataBanCreate>,
) -> Result<Json<v0::ServerBan>> { ) -> Result<Json<v0::ServerBan>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -65,7 +65,7 @@ pub async fn ban(
} }
} }
ServerBan::create(db, &server, &target.id, data.reason) ServerBan::create(db, &server, target.id, data.reason)
.await .await
.map(Into::into) .map(Into::into)
.map(Json) .map(Json)

View File

@@ -17,7 +17,7 @@ use rocket::State;
pub async fn list( pub async fn list(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::BanListResult>> { ) -> Result<Json<v0::BanListResult>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).server(&server); let mut query = DatabasePermissionQuery::new(db, &user).server(&server);

View File

@@ -15,8 +15,8 @@ use rocket_empty::EmptyResponse;
pub async fn unban( pub async fn unban(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
server: Reference, server: Reference<'_>,
target: Reference, target: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let server = server.as_server(db).await?; let server = server.as_server(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).server(&server); let mut query = DatabasePermissionQuery::new(db, &user).server(&server);

View File

@@ -16,7 +16,7 @@ use validator::Validate;
pub async fn create_server_channel( pub async fn create_server_channel(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
server: Reference, server: Reference<'_>,
data: Json<v0::DataCreateServerChannel>, data: Json<v0::DataCreateServerChannel>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {
let data = data.into_inner(); let data = data.into_inner();

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn list_emoji( pub async fn list_emoji(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<Vec<v0::Emoji>>> { ) -> Result<Json<Vec<v0::Emoji>>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).server(&server); let mut query = DatabasePermissionQuery::new(db, &user).server(&server);

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn invites( pub async fn invites(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<Vec<v0::Invite>>> { ) -> Result<Json<Vec<v0::Invite>>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).server(&server); let mut query = DatabasePermissionQuery::new(db, &user).server(&server);

View File

@@ -31,8 +31,8 @@ pub async fn edit(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
server: Reference, server: Reference<'_>,
member: Reference, member: Reference<'_>,
data: Json<v0::DataMemberEdit>, data: Json<v0::DataMemberEdit>,
) -> Result<Json<v0::Member>> { ) -> Result<Json<v0::Member>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -52,13 +52,7 @@ pub async fn edit(
let permissions = calculate_server_permissions(&mut query).await; let permissions = calculate_server_permissions(&mut query).await;
// Check permissions in server // Check permissions in server
if data.nickname.is_some() if data.nickname.is_some() || data.remove.contains(&v0::FieldsMember::Nickname) {
|| data
.remove
.as_ref()
.map(|x| x.contains(&v0::FieldsMember::Nickname))
.unwrap_or_default()
{
if user.id == member.id.user { if user.id == member.id.user {
permissions.throw_if_lacking_channel_permission(ChannelPermission::ChangeNickname)?; permissions.throw_if_lacking_channel_permission(ChannelPermission::ChangeNickname)?;
} else { } else {
@@ -66,13 +60,7 @@ pub async fn edit(
} }
} }
if data.avatar.is_some() if data.avatar.is_some() || data.remove.contains(&v0::FieldsMember::Avatar) {
|| data
.remove
.as_ref()
.map(|x| x.contains(&v0::FieldsMember::Avatar))
.unwrap_or_default()
{
if user.id == member.id.user { if user.id == member.id.user {
permissions.throw_if_lacking_channel_permission(ChannelPermission::ChangeAvatar)?; permissions.throw_if_lacking_channel_permission(ChannelPermission::ChangeAvatar)?;
} else { } else {
@@ -80,23 +68,11 @@ pub async fn edit(
} }
} }
if data.roles.is_some() if data.roles.is_some() || data.remove.contains(&v0::FieldsMember::Roles) {
|| data
.remove
.as_ref()
.map(|x| x.contains(&v0::FieldsMember::Roles))
.unwrap_or_default()
{
permissions.throw_if_lacking_channel_permission(ChannelPermission::AssignRoles)?; permissions.throw_if_lacking_channel_permission(ChannelPermission::AssignRoles)?;
} }
if data.timeout.is_some() if data.timeout.is_some() || data.remove.contains(&v0::FieldsMember::Timeout) {
|| data
.remove
.as_ref()
.map(|x| x.contains(&v0::FieldsMember::Timeout))
.unwrap_or_default()
{
if data.timeout.is_some() && member.id.user == user.id { if data.timeout.is_some() && member.id.user == user.id {
return Err(create_error!(CannotTimeoutYourself)); return Err(create_error!(CannotTimeoutYourself));
} }
@@ -121,7 +97,7 @@ pub async fn edit(
// ensure the channel we are moving them to is in the server and is a voice channel // ensure the channel we are moving them to is in the server and is a voice channel
let channel = Reference::from_unchecked(new_channel.clone()) let channel = Reference::from_unchecked(new_channel)
.as_channel(db) .as_channel(db)
.await .await
.map_err(|_| create_error!(UnknownChannel))?; .map_err(|_| create_error!(UnknownChannel))?;
@@ -192,11 +168,9 @@ pub async fn edit(
}; };
// 1. Remove fields from object // 1. Remove fields from object
if let Some(fields) = &remove { if remove.contains(&v0::FieldsMember::Avatar) {
if fields.contains(&v0::FieldsMember::Avatar) { if let Some(avatar) = &member.avatar {
if let Some(avatar) = &member.avatar { db.mark_attachment_as_deleted(&avatar.id).await?;
db.mark_attachment_as_deleted(&avatar.id).await?;
}
} }
} }
@@ -205,19 +179,10 @@ pub async fn edit(
partial.avatar = Some(File::use_user_avatar(db, &avatar, &user.id, &user.id).await?); partial.avatar = Some(File::use_user_avatar(db, &avatar, &user.id, &user.id).await?);
} }
let remove_contains_voice = remove let remove_contains_voice = remove.contains(FieldsMember::CanPublish) || remove.contains(FieldsMember::CanReceive);
.as_ref()
.map(|r| r.contains(FieldsMember::CanPublish) || r.contains(FieldsMember::CanReceive))
.unwrap_or_default();
member member
.update( .update(db, partial, remove.into_iter().map(Into::into).collect())
db,
partial,
remove
.map(|v| v.into_iter().map(Into::into).collect())
.unwrap_or_default(),
)
.await?; .await?;
if let Some(new_voice_channel) = new_voice_channel { if let Some(new_voice_channel) = new_voice_channel {
@@ -261,7 +226,7 @@ pub async fn edit(
if let Some(channel) = get_user_voice_channel_in_server(&target_user.id, &server.id).await? if let Some(channel) = get_user_voice_channel_in_server(&target_user.id, &server.id).await?
{ {
let node = get_channel_node(&channel).await?.unwrap(); let node = get_channel_node(&channel).await?.unwrap();
let channel = Reference::from_unchecked(channel).as_channel(db).await?; let channel = Reference::from_unchecked(&channel).as_channel(db).await?;
sync_user_voice_permissions( sync_user_voice_permissions(
db, db,

View File

@@ -36,7 +36,7 @@ pub struct MemberQueryResponse {
pub async fn member_experimental_query( pub async fn member_experimental_query(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: OptionsQueryMembers, options: OptionsQueryMembers,
) -> Result<Json<MemberQueryResponse>> { ) -> Result<Json<MemberQueryResponse>> {
if !options.experimental_api { if !options.experimental_api {

View File

@@ -15,8 +15,8 @@ use rocket::{serde::json::Json, State};
pub async fn fetch( pub async fn fetch(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
member: Reference, member: Reference<'_>,
roles: Option<bool>, roles: Option<bool>,
) -> Result<Json<v0::MemberResponse>> { ) -> Result<Json<v0::MemberResponse>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch_all( pub async fn fetch_all(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: v0::OptionsFetchAllMembers, options: v0::OptionsFetchAllMembers,
) -> Result<Json<v0::AllMemberResponse>> { ) -> Result<Json<v0::AllMemberResponse>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

View File

@@ -17,8 +17,8 @@ pub async fn kick(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
member: Reference, member: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

View File

@@ -17,8 +17,8 @@ mod permissions_set_default;
mod roles_create; mod roles_create;
mod roles_delete; mod roles_delete;
mod roles_edit; mod roles_edit;
mod roles_fetch;
mod roles_edit_positions; mod roles_edit_positions;
mod roles_fetch;
mod server_ack; mod server_ack;
mod server_create; mod server_create;
mod server_delete; mod server_delete;
@@ -47,7 +47,7 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
roles_fetch::fetch, roles_fetch::fetch,
roles_delete::delete, roles_delete::delete,
permissions_set::set_role_permission, permissions_set::set_role_permission,
permissions_set_default::set_default_permissions, permissions_set_default::set_default_server_permissions,
emoji_list::list_emoji, emoji_list::list_emoji,
roles_edit_positions::edit_role_ranks roles_edit_positions::edit_role_ranks
] ]

View File

@@ -15,7 +15,7 @@ pub async fn set_role_permission(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
role_id: String, role_id: String,
data: Json<v0::DataSetServerRolePermission>, data: Json<v0::DataSetServerRolePermission>,
) -> Result<Json<v0::Server>> { ) -> Result<Json<v0::Server>> {
@@ -41,7 +41,7 @@ pub async fn set_role_permission(
.await?; .await?;
for channel_id in &server.channels { for channel_id in &server.channels {
let channel = Reference::from_unchecked(channel_id.clone()).as_channel(db).await?; let channel = Reference::from_unchecked(channel_id).as_channel(db).await?;
sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?; sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?;
}; };

View File

@@ -13,11 +13,11 @@ use rocket::{serde::json::Json, State};
/// Sets permissions for the default role in this server. /// Sets permissions for the default role in this server.
#[openapi(tag = "Server Permissions")] #[openapi(tag = "Server Permissions")]
#[put("/<target>/permissions/default", data = "<data>", rank = 1)] #[put("/<target>/permissions/default", data = "<data>", rank = 1)]
pub async fn set_default_permissions( pub async fn set_default_server_permissions(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<DataPermissionsValue>, data: Json<DataPermissionsValue>,
) -> Result<Json<v0::Server>> { ) -> Result<Json<v0::Server>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -40,7 +40,7 @@ pub async fn set_default_permissions(
.await?; .await?;
for channel_id in &server.channels { for channel_id in &server.channels {
let channel = Reference::from_unchecked(channel_id.clone()).as_channel(db).await?; let channel = Reference::from_unchecked(channel_id).as_channel(db).await?;
sync_voice_permissions(db, voice_client, &channel, Some(&server), None).await?; sync_voice_permissions(db, voice_client, &channel, Some(&server), None).await?;
}; };

View File

@@ -17,7 +17,7 @@ use validator::Validate;
pub async fn create( pub async fn create(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataCreateRole>, data: Json<v0::DataCreateRole>,
) -> Result<Json<v0::NewRoleResponse>> { ) -> Result<Json<v0::NewRoleResponse>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -40,20 +40,10 @@ pub async fn create(
})); }));
}; };
let member_rank = query.get_member_rank();
let rank = if let Some(given_rank) = data.rank {
if given_rank <= member_rank.unwrap_or(i64::MIN) {
return Err(create_error!(NotElevated));
}
given_rank
} else {
member_rank.unwrap_or(0).saturating_add(1)
};
let role = Role { let role = Role {
name: data.name, name: data.name,
rank, // Rank of the new role should be below the lowest role
rank: server.roles.len() as i64,
colour: None, colour: None,
hoist: false, hoist: false,
permissions: Default::default(), permissions: Default::default(),

View File

@@ -16,7 +16,7 @@ use rocket_empty::EmptyResponse;
pub async fn delete( pub async fn delete(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
role_id: String, role_id: String,
voice_client: &State<VoiceClient> voice_client: &State<VoiceClient>
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
@@ -34,7 +34,7 @@ pub async fn delete(
} }
for channel_id in &server.channels { for channel_id in &server.channels {
let channel = Reference::from_unchecked(channel_id.clone()).as_channel(db).await?; let channel = Reference::from_unchecked(channel_id).as_channel(db).await?;
sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?; sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?;
}; };

View File

@@ -18,7 +18,7 @@ pub async fn edit(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
role_id: String, role_id: String,
data: Json<v0::DataEditRole>, data: Json<v0::DataEditRole>,
) -> Result<Json<v0::Role>> { ) -> Result<Json<v0::Role>> {
@@ -63,14 +63,12 @@ pub async fn edit(
&server.id, &server.id,
&role_id, &role_id,
partial, partial,
remove remove.into_iter().map(Into::into).collect(),
.map(|v| v.into_iter().map(Into::into).collect())
.unwrap_or_default(),
) )
.await?; .await?;
for channel_id in &server.channels { for channel_id in &server.channels {
let channel = Reference::from_unchecked(channel_id.clone()).as_channel(db).await?; let channel = Reference::from_unchecked(channel_id).as_channel(db).await?;
sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?; sync_voice_permissions(db, voice_client, &channel, Some(&server), Some(&role_id)).await?;
}; };

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn edit_role_ranks( pub async fn edit_role_ranks(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataEditRoleRanks>, data: Json<v0::DataEditRoleRanks>,
) -> Result<Json<v0::Server>> { ) -> Result<Json<v0::Server>> {
let data = data.into_inner(); let data = data.into_inner();

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch( pub async fn fetch(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
role_id: String, role_id: String,
) -> Result<Json<v0::Role>> { ) -> Result<Json<v0::Role>> {
let mut server = target.as_server(db).await?; let mut server = target.as_server(db).await?;

View File

@@ -12,7 +12,7 @@ use rocket_empty::EmptyResponse;
/// Mark all channels in a server as read. /// Mark all channels in a server as read.
#[openapi(tag = "Server Information")] #[openapi(tag = "Server Information")]
#[put("/<target>/ack")] #[put("/<target>/ack")]
pub async fn ack(db: &State<Database>, user: User, target: Reference) -> Result<EmptyResponse> { pub async fn ack(db: &State<Database>, user: User, target: Reference<'_>) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));
} }

View File

@@ -18,11 +18,11 @@ pub async fn delete(
db: &State<Database>, db: &State<Database>,
voice_client: &State<VoiceClient>, voice_client: &State<VoiceClient>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: v0::OptionsServerDelete, options: v0::OptionsServerDelete,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;
let member = db.fetch_member(&target.id, &user.id).await?; let member = db.fetch_member(target.id, &user.id).await?;
if server.owner == user.id { if server.owner == user.id {
for channel_id in &server.channels { for channel_id in &server.channels {

View File

@@ -18,7 +18,7 @@ use validator::Validate;
pub async fn edit( pub async fn edit(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataEditServer>, data: Json<v0::DataEditServer>,
) -> Result<Json<v0::Server>> { ) -> Result<Json<v0::Server>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -43,7 +43,7 @@ pub async fn edit(
&& data.flags.is_none() && data.flags.is_none()
&& data.analytics.is_none() && data.analytics.is_none()
&& data.discoverable.is_none() && data.discoverable.is_none()
&& data.remove.is_none() && data.remove.is_empty()
{ {
return Ok(Json(server.into())); return Ok(Json(server.into()));
} else if data.name.is_some() } else if data.name.is_some()
@@ -52,7 +52,7 @@ pub async fn edit(
|| data.banner.is_some() || data.banner.is_some()
|| data.system_messages.is_some() || data.system_messages.is_some()
|| data.analytics.is_some() || data.analytics.is_some()
|| data.remove.is_some() || !data.remove.is_empty()
{ {
permissions.throw_if_lacking_channel_permission(ChannelPermission::ManageServer)?; permissions.throw_if_lacking_channel_permission(ChannelPermission::ManageServer)?;
} }
@@ -96,17 +96,15 @@ pub async fn edit(
}; };
// 1. Remove fields from object // 1. Remove fields from object
if let Some(fields) = &remove { if remove.contains(&v0::FieldsServer::Banner) {
if fields.contains(&v0::FieldsServer::Banner) { if let Some(banner) = &server.banner {
if let Some(banner) = &server.banner { db.mark_attachment_as_deleted(&banner.id).await?;
db.mark_attachment_as_deleted(&banner.id).await?;
}
} }
}
if fields.contains(&v0::FieldsServer::Icon) { if remove.contains(&v0::FieldsServer::Icon) {
if let Some(icon) = &server.icon { if let Some(icon) = &server.icon {
db.mark_attachment_as_deleted(&icon.id).await?; db.mark_attachment_as_deleted(&icon.id).await?;
}
} }
} }
@@ -149,13 +147,7 @@ pub async fn edit(
} }
server server
.update( .update(db, partial, remove.into_iter().map(Into::into).collect())
db,
partial,
remove
.map(|v| v.into_iter().map(Into::into).collect())
.unwrap_or_default(),
)
.await?; .await?;
Ok(Json(server.into())) Ok(Json(server.into()))

View File

@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn fetch( pub async fn fetch(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
options: v0::OptionsFetchServer, options: v0::OptionsFetchServer,
) -> Result<Json<v0::FetchServerResponse>> { ) -> Result<Json<v0::FetchServerResponse>> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

Some files were not shown because too many files have changed in this diff Show More