Compare commits

...
3 Commits
108 changed files with 236 additions and 168 deletions

No files matched your search

Generated
-3
View File
@@ -6291,7 +6291,6 @@ dependencies = [
"bson", "bson",
"deadqueue", "deadqueue",
"decancer", "decancer",
"fcm_v1",
"futures", "futures",
"indexmap 1.9.3", "indexmap 1.9.3",
"isahc", "isahc",
@@ -6311,7 +6310,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",
@@ -6323,7 +6321,6 @@ dependencies = [
"unicode-segmentation", "unicode-segmentation",
"url-escape", "url-escape",
"validator 0.16.1", "validator 0.16.1",
"web-push",
] ]
[[package]] [[package]]
+3 -2
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
+7 -3
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 ) => {
@@ -419,6 +421,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");
@@ -444,6 +447,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) => {
+4 -9
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" }
+31 -12
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,
} }
} }
} }
@@ -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
pub struct MongoDb(pub ::mongodb::Client, pub String); pub struct MongoDb(pub ::mongodb::Client, pub String);
); );
+4
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;
@@ -1,3 +1,4 @@
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;
@@ -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;
@@ -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;
@@ -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;
@@ -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;
@@ -8,10 +8,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 {
@@ -646,10 +649,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(),
}, },
) )
@@ -766,6 +770,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 {
@@ -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;
@@ -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;
@@ -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;
@@ -4,6 +4,7 @@ use crate::File;
use super::FileUsedFor; use super::FileUsedFor;
#[cfg(feature = "mongodb")]
mod mongodb; mod mongodb;
mod reference; mod reference;
@@ -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 {
@@ -616,9 +618,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(
@@ -637,6 +641,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(
@@ -673,10 +678,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 {
@@ -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;
+7 -1
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,
} }
} }
@@ -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;
@@ -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;
@@ -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;
@@ -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;
@@ -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;
@@ -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),
@@ -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;
@@ -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;
@@ -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;
+19 -20
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()
} }
+1
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 = [
+1 -1
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 {
+1 -1
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();
+1 -1
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));
+2 -2
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));
} }
+1 -1
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() {
@@ -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)
} }
@@ -17,7 +17,7 @@ pub async fn delete(
db: &State<Database>, db: &State<Database>,
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?;
@@ -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();
@@ -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?;
@@ -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));
@@ -14,8 +14,8 @@ pub async fn remove_member(
db: &State<Database>, db: &State<Database>,
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));
@@ -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));
@@ -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);
@@ -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)
} }
@@ -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);
@@ -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?;
@@ -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();
@@ -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);
@@ -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");
@@ -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| {
@@ -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)
} }
@@ -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() {
@@ -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>> {
@@ -202,7 +202,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");
@@ -242,7 +242,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");
@@ -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");
@@ -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)
} }
@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn set_role_permissions( pub async fn set_role_permissions(
db: &State<Database>, db: &State<Database>,
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>> {
@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn set_default_channel_permissions( pub async fn set_default_channel_permissions(
db: &State<Database>, db: &State<Database>,
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();
@@ -16,7 +16,7 @@ use rocket::{serde::json::Json, State};
pub async fn call( pub async fn call(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::LegacyCreateVoiceUserResponse>> { ) -> Result<Json<v0::LegacyCreateVoiceUserResponse>> {
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);
@@ -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();
@@ -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?;
@@ -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);
@@ -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
@@ -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() {
@@ -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, ..
@@ -34,7 +34,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,
@@ -64,7 +64,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,
@@ -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));
@@ -17,8 +17,8 @@ use validator::Validate;
pub async fn ban( pub async fn ban(
db: &State<Database>, db: &State<Database>,
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();
@@ -56,7 +56,7 @@ pub async fn ban(
.await?; .await?;
} }
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)
+1 -1
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);
@@ -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);
@@ -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();
@@ -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);
@@ -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);
@@ -19,8 +19,8 @@ use validator::Validate;
pub async fn edit( pub async fn edit(
db: &State<Database>, db: &State<Database>,
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();
@@ -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 {
@@ -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?;
@@ -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?;
@@ -15,8 +15,8 @@ use rocket_empty::EmptyResponse;
pub async fn kick( pub async fn kick(
db: &State<Database>, db: &State<Database>,
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?;
@@ -15,7 +15,7 @@ use rocket::{serde::json::Json, State};
pub async fn set_role_permission( pub async fn set_role_permission(
db: &State<Database>, db: &State<Database>,
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>> {
@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn set_default_server_permissions( pub async fn set_default_server_permissions(
db: &State<Database>, db: &State<Database>,
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();
@@ -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();
@@ -15,7 +15,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,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let mut server = target.as_server(db).await?; let mut server = target.as_server(db).await?;
@@ -16,7 +16,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<'_>,
role_id: String, role_id: String,
data: Json<v0::DataEditRole>, data: Json<v0::DataEditRole>,
) -> Result<Json<v0::Role>> { ) -> Result<Json<v0::Role>> {
@@ -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();
@@ -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?;
@@ -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));
} }
@@ -13,11 +13,11 @@ 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<'_>,
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 {
server.delete(db).await server.delete(db).await
@@ -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();
@@ -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?;
+1 -1
View File
@@ -14,7 +14,7 @@ pub async fn add(
db: &State<Database>, db: &State<Database>,
amqp: &State<AMQP>, amqp: &State<AMQP>,
mut user: User, mut user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::User>> { ) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?; let mut target = target.as_user(db).await?;
+1 -1
View File
@@ -13,7 +13,7 @@ use rocket::State;
pub async fn block( pub async fn block(
db: &State<Database>, db: &State<Database>,
mut user: User, mut user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::User>> { ) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?; let mut target = target.as_user(db).await?;
+1 -1
View File
@@ -14,7 +14,7 @@ use validator::Validate;
pub async fn edit( pub async fn edit(
db: &State<Database>, db: &State<Database>,
mut user: User, mut user: User,
target: Reference, target: Reference<'_>,
data: Json<v0::DataEditUser>, data: Json<v0::DataEditUser>,
) -> Result<Json<v0::User>> { ) -> Result<Json<v0::User>> {
let data = data.into_inner(); let data = data.into_inner();
@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn profile( pub async fn profile(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::UserProfile>> { ) -> Result<Json<v0::UserProfile>> {
if user.id == target.id { if user.id == target.id {
return Ok(Json(user.profile.map(Into::into).unwrap_or_default())); return Ok(Json(user.profile.map(Into::into).unwrap_or_default()));
+1 -1
View File
@@ -13,7 +13,7 @@ use rocket::{serde::json::Json, State};
/// Retrieve a user's information. /// Retrieve a user's information.
#[openapi(tag = "User Information")] #[openapi(tag = "User Information")]
#[get("/<target>")] #[get("/<target>")]
pub async fn fetch(db: &State<Database>, user: User, target: Reference) -> Result<Json<v0::User>> { pub async fn fetch(db: &State<Database>, user: User, target: Reference<'_>) -> Result<Json<v0::User>> {
if user.id == target.id { if user.id == target.id {
return Ok(Json(user.into_self(false).await)); return Ok(Json(user.into_self(false).await));
} }
@@ -10,7 +10,7 @@ use rocket::{serde::json::Json, State};
#[get("/<target>/flags")] #[get("/<target>/flags")]
pub async fn fetch_user_flags( pub async fn fetch_user_flags(
db: &State<Database>, db: &State<Database>,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::FlagResponse>> { ) -> Result<Json<v0::FlagResponse>> {
let flags = if let Ok(target) = target.as_user(db).await { let flags = if let Ok(target) = target.as_user(db).await {
target.flags.unwrap_or_default() target.flags.unwrap_or_default()
+1 -1
View File
@@ -16,7 +16,7 @@ use rocket::State;
pub async fn mutual( pub async fn mutual(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::MutualResponse>> { ) -> Result<Json<v0::MutualResponse>> {
if target.id == user.id { if target.id == user.id {
return Err(create_error!(InvalidOperation)); return Err(create_error!(InvalidOperation));
+1 -1
View File
@@ -17,7 +17,7 @@ use rocket::{serde::json::Json, State};
pub async fn open_dm( pub async fn open_dm(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::Channel>> { ) -> Result<Json<v0::Channel>> {
let target = target.as_user(db).await?; let target = target.as_user(db).await?;
@@ -13,7 +13,7 @@ use rocket::State;
pub async fn remove( pub async fn remove(
db: &State<Database>, db: &State<Database>,
mut user: User, mut user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::User>> { ) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?; let mut target = target.as_user(db).await?;
@@ -13,7 +13,7 @@ use rocket::State;
pub async fn unblock( pub async fn unblock(
db: &State<Database>, db: &State<Database>,
mut user: User, mut user: User,
target: Reference, target: Reference<'_>,
) -> Result<Json<v0::User>> { ) -> Result<Json<v0::User>> {
let mut target = target.as_user(db).await?; let mut target = target.as_user(db).await?;
@@ -15,7 +15,7 @@ use rocket_empty::EmptyResponse;
pub async fn webhook_delete( pub async fn webhook_delete(
db: &State<Database>, db: &State<Database>,
user: User, user: User,
webhook_id: Reference, webhook_id: Reference<'_>,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let webhook = webhook_id.as_webhook(db).await?; let webhook = webhook_id.as_webhook(db).await?;
let channel = db.fetch_channel(&webhook.channel_id).await?; let channel = db.fetch_channel(&webhook.channel_id).await?;
@@ -10,7 +10,7 @@ use rocket_empty::EmptyResponse;
#[delete("/<webhook_id>/<token>")] #[delete("/<webhook_id>/<token>")]
pub async fn webhook_delete_token( pub async fn webhook_delete_token(
db: &State<Database>, db: &State<Database>,
webhook_id: Reference, webhook_id: Reference<'_>,
token: String, token: String,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let webhook = webhook_id.as_webhook(db).await?; let webhook = webhook_id.as_webhook(db).await?;
Loaded 100 of 108 files, more files were not shown because too many files have changed in this diff. Show more