Run cargo fmt.
This commit is contained in:
@@ -190,7 +190,7 @@ impl Channel {
|
|||||||
if let Channel::TextChannel { server, .. } = &self {
|
if let Channel::TextChannel { server, .. } = &self {
|
||||||
get_collection("servers")
|
get_collection("servers")
|
||||||
.update_one(
|
.update_one(
|
||||||
doc !{
|
doc! {
|
||||||
"_id": server
|
"_id": server
|
||||||
},
|
},
|
||||||
doc! {
|
doc! {
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
use crate::database::*;
|
use crate::database::*;
|
||||||
use crate::notifications::events::ClientboundNotification;
|
use crate::notifications::events::ClientboundNotification;
|
||||||
use crate::util::result::{Error, Result};
|
use crate::util::result::{Error, Result};
|
||||||
use mongodb::bson::Document;
|
|
||||||
use mongodb::bson::doc;
|
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
|
use mongodb::bson::doc;
|
||||||
use mongodb::bson::to_document;
|
use mongodb::bson::to_document;
|
||||||
|
use mongodb::bson::Document;
|
||||||
use mongodb::options::FindOptions;
|
use mongodb::options::FindOptions;
|
||||||
use rocket_contrib::json::JsonValue;
|
use rocket_contrib::json::JsonValue;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
@@ -149,9 +149,9 @@ impl Server {
|
|||||||
operation: "delete_many",
|
operation: "delete_many",
|
||||||
with: "messages",
|
with: "messages",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
// Delete all channels, members, bans and invites.
|
// Delete all channels, members, bans and invites.
|
||||||
for with in [ "channels", "invites" ] {
|
for with in ["channels", "invites"] {
|
||||||
get_collection(with)
|
get_collection(with)
|
||||||
.delete_many(
|
.delete_many(
|
||||||
doc! {
|
doc! {
|
||||||
@@ -166,7 +166,7 @@ impl Server {
|
|||||||
})?;
|
})?;
|
||||||
}
|
}
|
||||||
|
|
||||||
for with in [ "server_members", "server_bans" ] {
|
for with in ["server_members", "server_bans"] {
|
||||||
get_collection(with)
|
get_collection(with)
|
||||||
.delete_many(
|
.delete_many(
|
||||||
doc! {
|
doc! {
|
||||||
@@ -194,7 +194,10 @@ impl Server {
|
|||||||
with: "server",
|
with: "server",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
ClientboundNotification::ServerDelete { id: self.id.clone() }.publish(self.id.clone());
|
ClientboundNotification::ServerDelete {
|
||||||
|
id: self.id.clone(),
|
||||||
|
}
|
||||||
|
.publish(self.id.clone());
|
||||||
|
|
||||||
if let Some(attachment) = &self.icon {
|
if let Some(attachment) = &self.icon {
|
||||||
attachment.delete().await?;
|
attachment.delete().await?;
|
||||||
@@ -240,15 +243,16 @@ impl Server {
|
|||||||
"_id.server": &self.id,
|
"_id.server": &self.id,
|
||||||
"_id.user": &id
|
"_id.user": &id
|
||||||
},
|
},
|
||||||
None
|
None,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| Error::DatabaseError {
|
.map_err(|_| Error::DatabaseError {
|
||||||
operation: "find_one",
|
operation: "find_one",
|
||||||
with: "server_bans",
|
with: "server_bans",
|
||||||
})?
|
})?
|
||||||
.is_some() {
|
.is_some()
|
||||||
return Err(Error::Banned)
|
{
|
||||||
|
return Err(Error::Banned);
|
||||||
}
|
}
|
||||||
|
|
||||||
get_collection("server_members")
|
get_collection("server_members")
|
||||||
@@ -266,10 +270,10 @@ impl Server {
|
|||||||
operation: "insert_one",
|
operation: "insert_one",
|
||||||
with: "server_members",
|
with: "server_members",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
ClientboundNotification::ServerMemberJoin {
|
ClientboundNotification::ServerMemberJoin {
|
||||||
id: self.id.clone(),
|
id: self.id.clone(),
|
||||||
user: id.to_string()
|
user: id.to_string(),
|
||||||
}
|
}
|
||||||
.publish(self.id.clone());
|
.publish(self.id.clone());
|
||||||
|
|
||||||
@@ -292,11 +296,11 @@ impl Server {
|
|||||||
operation: "delete_one",
|
operation: "delete_one",
|
||||||
with: "server_members",
|
with: "server_members",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
if result.deleted_count > 0 {
|
if result.deleted_count > 0 {
|
||||||
ClientboundNotification::ServerMemberLeave {
|
ClientboundNotification::ServerMemberLeave {
|
||||||
id: self.id.clone(),
|
id: self.id.clone(),
|
||||||
user: id.to_string()
|
user: id.to_string(),
|
||||||
}
|
}
|
||||||
.publish(self.id.clone());
|
.publish(self.id.clone());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -78,10 +78,12 @@ impl Ref {
|
|||||||
})?
|
})?
|
||||||
.ok_or_else(|| Error::NotFound)?;
|
.ok_or_else(|| Error::NotFound)?;
|
||||||
|
|
||||||
Ok(from_document::<Member>(doc).map_err(|_| Error::DatabaseError {
|
Ok(
|
||||||
operation: "from_document",
|
from_document::<Member>(doc).map_err(|_| Error::DatabaseError {
|
||||||
with: "server_member",
|
operation: "from_document",
|
||||||
})?)
|
with: "server_member",
|
||||||
|
})?,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn fetch_ban(&self, server: &str) -> Result<Ban> {
|
pub async fn fetch_ban(&self, server: &str) -> Result<Ban> {
|
||||||
|
|||||||
@@ -9,12 +9,12 @@ use std::ops;
|
|||||||
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
||||||
#[repr(u32)]
|
#[repr(u32)]
|
||||||
pub enum ChannelPermission {
|
pub enum ChannelPermission {
|
||||||
View = 0b00000000000000000000000000000001, // 1
|
View = 0b00000000000000000000000000000001, // 1
|
||||||
SendMessage = 0b00000000000000000000000000000010, // 2
|
SendMessage = 0b00000000000000000000000000000010, // 2
|
||||||
ManageMessages = 0b00000000000000000000000000000100, // 4
|
ManageMessages = 0b00000000000000000000000000000100, // 4
|
||||||
ManageChannel = 0b00000000000000000000000000001000, // 8
|
ManageChannel = 0b00000000000000000000000000001000, // 8
|
||||||
VoiceCall = 0b00000000000000000000000000010000, // 16
|
VoiceCall = 0b00000000000000000000000000010000, // 16
|
||||||
InviteOthers = 0b00000000000000000000000000100000, // 32
|
InviteOthers = 0b00000000000000000000000000100000, // 32
|
||||||
}
|
}
|
||||||
|
|
||||||
impl_op_ex!(+ |a: &ChannelPermission, b: &ChannelPermission| -> u32 { *a as u32 | *b as u32 });
|
impl_op_ex!(+ |a: &ChannelPermission, b: &ChannelPermission| -> u32 { *a as u32 | *b as u32 });
|
||||||
|
|||||||
@@ -8,18 +8,18 @@ use std::ops;
|
|||||||
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
||||||
#[repr(u32)]
|
#[repr(u32)]
|
||||||
pub enum ServerPermission {
|
pub enum ServerPermission {
|
||||||
View = 0b00000000000000000000000000000001, // 1
|
View = 0b00000000000000000000000000000001, // 1
|
||||||
ManageMembers = 0b00000000000000000000000000000010, // 2
|
ManageMembers = 0b00000000000000000000000000000010, // 2
|
||||||
ManageChannels = 0b00000000000000000000000000000100, // 4
|
ManageChannels = 0b00000000000000000000000000000100, // 4
|
||||||
ManageServer = 0b00000000000000000000000000001000, // 8
|
ManageServer = 0b00000000000000000000000000001000, // 8
|
||||||
KickMembers = 0b00000000000000000000000000010000, // 16
|
KickMembers = 0b00000000000000000000000000010000, // 16
|
||||||
BanMembers = 0b00000000000000000000000000100000, // 32
|
BanMembers = 0b00000000000000000000000000100000, // 32
|
||||||
// 6 bits of space
|
// 6 bits of space
|
||||||
ChangeNickname = 0b00000000000000000001000000000000, // 4096
|
ChangeNickname = 0b00000000000000000001000000000000, // 4096
|
||||||
ManageNicknames = 0b00000000000000000010000000000000, // 8192
|
ManageNicknames = 0b00000000000000000010000000000000, // 8192
|
||||||
ChangeAvatar = 0b00000000000000000100000000000000, // 16392
|
ChangeAvatar = 0b00000000000000000100000000000000, // 16392
|
||||||
RemoveAvatars = 0b00000000000000001000000000000000, // 32784
|
RemoveAvatars = 0b00000000000000001000000000000000, // 32784
|
||||||
// 16 bits of space
|
// 16 bits of space
|
||||||
}
|
}
|
||||||
|
|
||||||
impl_op_ex!(+ |a: &ServerPermission, b: &ServerPermission| -> u32 { *a as u32 | *b as u32 });
|
impl_op_ex!(+ |a: &ServerPermission, b: &ServerPermission| -> u32 { *a as u32 | *b as u32 });
|
||||||
@@ -52,7 +52,9 @@ impl<'a> PermissionCalculator<'a> {
|
|||||||
if self.perspective.id == server.owner {
|
if self.perspective.id == server.owner {
|
||||||
Ok(u32::MAX)
|
Ok(u32::MAX)
|
||||||
} else {
|
} else {
|
||||||
Ok(ServerPermission::View + ServerPermission::ChangeNickname + ServerPermission::ChangeAvatar)
|
Ok(ServerPermission::View
|
||||||
|
+ ServerPermission::ChangeNickname
|
||||||
|
+ ServerPermission::ChangeAvatar)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -10,10 +10,10 @@ use std::ops;
|
|||||||
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
|
||||||
#[repr(u32)]
|
#[repr(u32)]
|
||||||
pub enum UserPermission {
|
pub enum UserPermission {
|
||||||
Access = 0b00000000000000000000000000000001, // 1
|
Access = 0b00000000000000000000000000000001, // 1
|
||||||
ViewProfile = 0b00000000000000000000000000000010, // 2
|
ViewProfile = 0b00000000000000000000000000000010, // 2
|
||||||
SendMessage = 0b00000000000000000000000000000100, // 4
|
SendMessage = 0b00000000000000000000000000000100, // 4
|
||||||
Invite = 0b00000000000000000000000000001000, // 8
|
Invite = 0b00000000000000000000000000001000, // 8
|
||||||
}
|
}
|
||||||
|
|
||||||
bitfield! {
|
bitfield! {
|
||||||
|
|||||||
@@ -53,7 +53,7 @@ pub enum RemoveServerField {
|
|||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
pub enum RemoveMemberField {
|
pub enum RemoveMemberField {
|
||||||
Nickname,
|
Nickname,
|
||||||
Avatar
|
Avatar,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
@@ -216,35 +216,23 @@ pub async fn posthandle_hook(notification: &ClientboundNotification) {
|
|||||||
get_hive().hive.drop_topic(&id).ok();
|
get_hive().hive.drop_topic(&id).ok();
|
||||||
}
|
}
|
||||||
ClientboundNotification::ChannelGroupLeave { id, user } => {
|
ClientboundNotification::ChannelGroupLeave { id, user } => {
|
||||||
get_hive()
|
get_hive().hive.unsubscribe(user, id).ok();
|
||||||
.hive
|
|
||||||
.unsubscribe(user, id)
|
|
||||||
.ok();
|
|
||||||
}
|
}
|
||||||
ClientboundNotification::ServerDelete { id } => {
|
ClientboundNotification::ServerDelete { id } => {
|
||||||
get_hive().hive.drop_topic(&id).ok();
|
get_hive().hive.drop_topic(&id).ok();
|
||||||
}
|
}
|
||||||
ClientboundNotification::ServerMemberLeave { id, user } => {
|
ClientboundNotification::ServerMemberLeave { id, user } => {
|
||||||
get_hive()
|
get_hive().hive.unsubscribe(user, id).ok();
|
||||||
.hive
|
|
||||||
.unsubscribe(user, id)
|
|
||||||
.ok();
|
|
||||||
|
|
||||||
if let Ok(server) = Ref::from_unchecked(id.clone()).fetch_server().await {
|
if let Ok(server) = Ref::from_unchecked(id.clone()).fetch_server().await {
|
||||||
for channel in server.channels {
|
for channel in server.channels {
|
||||||
get_hive()
|
get_hive().hive.unsubscribe(user, &channel).ok();
|
||||||
.hive
|
|
||||||
.unsubscribe(user, &channel)
|
|
||||||
.ok();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ClientboundNotification::UserRelationship { id, user, status } => {
|
ClientboundNotification::UserRelationship { id, user, status } => {
|
||||||
if status == &RelationshipStatus::None {
|
if status == &RelationshipStatus::None {
|
||||||
get_hive()
|
get_hive().hive.unsubscribe(id, &user.id).ok();
|
||||||
.hive
|
|
||||||
.unsubscribe(id, &user.id)
|
|
||||||
.ok();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ pub async fn req(user: User, target: Ref) -> Result<()> {
|
|||||||
.with_channel(&target)
|
.with_channel(&target)
|
||||||
.for_channel()
|
.for_channel()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_view() {
|
if !perm.get_view() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -45,8 +45,7 @@ pub async fn req(user: User, target: Ref, data: Json<Data>) -> Result<()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
match &target {
|
match &target {
|
||||||
Channel::Group { id, icon, .. } |
|
Channel::Group { id, icon, .. } | Channel::TextChannel { id, icon, .. } => {
|
||||||
Channel::TextChannel { id, icon, .. } => {
|
|
||||||
let mut set = doc! {};
|
let mut set = doc! {};
|
||||||
let mut unset = doc! {};
|
let mut unset = doc! {};
|
||||||
|
|
||||||
|
|||||||
@@ -21,8 +21,8 @@ pub async fn req(user: User, target: Ref) -> Result<()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
target.delete().await
|
target.delete().await
|
||||||
},
|
}
|
||||||
_ => unreachable!()
|
_ => unreachable!(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,8 +18,8 @@ pub enum InviteResponse {
|
|||||||
channel_description: Option<String>,
|
channel_description: Option<String>,
|
||||||
user_name: String,
|
user_name: String,
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
user_avatar: Option<File>
|
user_avatar: Option<File>,
|
||||||
}
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
#[get("/<target>")]
|
#[get("/<target>")]
|
||||||
@@ -28,14 +28,18 @@ pub async fn req(target: Ref) -> Result<JsonValue> {
|
|||||||
|
|
||||||
match target {
|
match target {
|
||||||
Invite::Server {
|
Invite::Server {
|
||||||
channel,
|
channel, creator, ..
|
||||||
creator,
|
|
||||||
..
|
|
||||||
} => {
|
} => {
|
||||||
let channel = Ref::from_unchecked(channel).fetch_channel().await?;
|
let channel = Ref::from_unchecked(channel).fetch_channel().await?;
|
||||||
let creator = Ref::from_unchecked(creator).fetch_user().await?;
|
let creator = Ref::from_unchecked(creator).fetch_user().await?;
|
||||||
|
|
||||||
if let Channel::TextChannel { server, name, description, .. } = channel {
|
if let Channel::TextChannel {
|
||||||
|
server,
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
..
|
||||||
|
} = channel
|
||||||
|
{
|
||||||
let server = Ref::from_unchecked(server).fetch_server().await?;
|
let server = Ref::from_unchecked(server).fetch_server().await?;
|
||||||
|
|
||||||
Ok(json!(InviteResponse::Server {
|
Ok(json!(InviteResponse::Server {
|
||||||
|
|||||||
@@ -8,10 +8,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
let target = target.fetch_invite().await?;
|
let target = target.fetch_invite().await?;
|
||||||
|
|
||||||
match target {
|
match target {
|
||||||
Invite::Server {
|
Invite::Server { channel, .. } => {
|
||||||
channel,
|
|
||||||
..
|
|
||||||
} => {
|
|
||||||
let channel = Ref::from_unchecked(channel).fetch_channel().await?;
|
let channel = Ref::from_unchecked(channel).fetch_channel().await?;
|
||||||
let server = if let Channel::TextChannel { server, .. } = &channel {
|
let server = if let Channel::TextChannel { server, .. } = &channel {
|
||||||
Ref::from_unchecked(server.clone()).fetch_server().await?
|
Ref::from_unchecked(server.clone()).fetch_server().await?
|
||||||
|
|||||||
@@ -2,9 +2,9 @@ use crate::database::*;
|
|||||||
use crate::util::result::{Error, Result};
|
use crate::util::result::{Error, Result};
|
||||||
|
|
||||||
use mongodb::bson::doc;
|
use mongodb::bson::doc;
|
||||||
use validator::Validate;
|
|
||||||
use rocket_contrib::json::Json;
|
use rocket_contrib::json::Json;
|
||||||
use serde::{Serialize, Deserialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use validator::Validate;
|
||||||
|
|
||||||
#[derive(Validate, Serialize, Deserialize)]
|
#[derive(Validate, Serialize, Deserialize)]
|
||||||
pub struct Data {
|
pub struct Data {
|
||||||
@@ -17,14 +17,14 @@ pub async fn req(user: User, server: Ref, target: Ref, data: Json<Data>) -> Resu
|
|||||||
let data = data.into_inner();
|
let data = data.into_inner();
|
||||||
data.validate()
|
data.validate()
|
||||||
.map_err(|error| Error::FailedValidation { error })?;
|
.map_err(|error| Error::FailedValidation { error })?;
|
||||||
|
|
||||||
let server = server.fetch_server().await?;
|
let server = server.fetch_server().await?;
|
||||||
|
|
||||||
let perm = permissions::PermissionCalculator::new(&user)
|
let perm = permissions::PermissionCalculator::new(&user)
|
||||||
.with_server(&server)
|
.with_server(&server)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_ban_members() {
|
if !perm.get_ban_members() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
@@ -42,15 +42,12 @@ pub async fn req(user: User, server: Ref, target: Ref, data: Json<Data>) -> Resu
|
|||||||
}
|
}
|
||||||
|
|
||||||
get_collection("server_bans")
|
get_collection("server_bans")
|
||||||
.insert_one(
|
.insert_one(document, None)
|
||||||
document,
|
|
||||||
None
|
|
||||||
)
|
|
||||||
.await
|
.await
|
||||||
.map_err(|_| Error::DatabaseError {
|
.map_err(|_| Error::DatabaseError {
|
||||||
operation: "insert_one",
|
operation: "insert_one",
|
||||||
with: "server_ban"
|
with: "server_ban",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
server.remove_member(&target.id).await
|
server.remove_member(&target.id).await
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ use crate::database::*;
|
|||||||
use crate::util::result::{Error, Result};
|
use crate::util::result::{Error, Result};
|
||||||
|
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
use rocket_contrib::json::JsonValue;
|
|
||||||
use mongodb::bson::{doc, from_document};
|
use mongodb::bson::{doc, from_document};
|
||||||
|
use rocket_contrib::json::JsonValue;
|
||||||
|
|
||||||
#[get("/<target>/bans")]
|
#[get("/<target>/bans")]
|
||||||
pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
||||||
@@ -13,7 +13,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_manage_members() {
|
if !perm.get_manage_members() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
@@ -30,7 +30,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
operation: "find",
|
operation: "find",
|
||||||
with: "server_bans",
|
with: "server_bans",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
let mut bans = vec![];
|
let mut bans = vec![];
|
||||||
while let Some(result) = cursor.next().await {
|
while let Some(result) = cursor.next().await {
|
||||||
if let Ok(doc) = result {
|
if let Ok(doc) = result {
|
||||||
@@ -39,6 +39,6 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(json!(bans))
|
Ok(json!(bans))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,24 +4,24 @@ use crate::util::result::{Error, Result};
|
|||||||
use mongodb::bson::doc;
|
use mongodb::bson::doc;
|
||||||
|
|
||||||
#[delete("/<server>/bans/<target>")]
|
#[delete("/<server>/bans/<target>")]
|
||||||
pub async fn req(user: User, server: Ref, target: Ref) -> Result<()> {
|
pub async fn req(user: User, server: Ref, target: Ref) -> Result<()> {
|
||||||
let server = server.fetch_server().await?;
|
let server = server.fetch_server().await?;
|
||||||
|
|
||||||
let perm = permissions::PermissionCalculator::new(&user)
|
let perm = permissions::PermissionCalculator::new(&user)
|
||||||
.with_server(&server)
|
.with_server(&server)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_ban_members() {
|
if !perm.get_ban_members() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
|
|
||||||
if target.id == user.id {
|
if target.id == user.id {
|
||||||
return Err(Error::InvalidOperation)
|
return Err(Error::InvalidOperation);
|
||||||
}
|
}
|
||||||
|
|
||||||
if target.id == server.owner {
|
if target.id == server.owner {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
|
|
||||||
let target = target.fetch_ban(&server.id).await?;
|
let target = target.fetch_ban(&server.id).await?;
|
||||||
@@ -31,13 +31,13 @@ pub async fn req(user: User, server: Ref, target: Ref) -> Result<()> {
|
|||||||
"_id.server": &server.id,
|
"_id.server": &server.id,
|
||||||
"_id.user": &target.id.user
|
"_id.user": &target.id.user
|
||||||
},
|
},
|
||||||
None
|
None,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| Error::DatabaseError {
|
.map_err(|_| Error::DatabaseError {
|
||||||
operation: "delete_one",
|
operation: "delete_one",
|
||||||
with: "server_ban"
|
with: "server_ban",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
use crate::database::*;
|
use crate::database::*;
|
||||||
use crate::util::result::{Error, Result};
|
use crate::util::result::{Error, Result};
|
||||||
|
|
||||||
use rocket_contrib::json::JsonValue;
|
|
||||||
use serde::{Serialize, Deserialize};
|
|
||||||
use mongodb::bson::{doc, from_document};
|
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
|
use mongodb::bson::{doc, from_document};
|
||||||
|
use rocket_contrib::json::JsonValue;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
pub struct ServerInvite {
|
pub struct ServerInvite {
|
||||||
@@ -22,7 +22,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_manage_server() {
|
if !perm.get_manage_server() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
@@ -39,7 +39,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
operation: "find",
|
operation: "find",
|
||||||
with: "invites",
|
with: "invites",
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
let mut invites = vec![];
|
let mut invites = vec![];
|
||||||
while let Some(result) = cursor.next().await {
|
while let Some(result) = cursor.next().await {
|
||||||
if let Ok(doc) = result {
|
if let Ok(doc) = result {
|
||||||
@@ -48,6 +48,6 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(json!(invites))
|
Ok(json!(invites))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,8 +21,7 @@ pub async fn req(user: User, server: Ref, target: String, data: Json<Data>) -> R
|
|||||||
data.validate()
|
data.validate()
|
||||||
.map_err(|error| Error::FailedValidation { error })?;
|
.map_err(|error| Error::FailedValidation { error })?;
|
||||||
|
|
||||||
if data.nickname.is_none() && data.avatar.is_none() && data.remove.is_none()
|
if data.nickname.is_none() && data.avatar.is_none() && data.remove.is_none() {
|
||||||
{
|
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -35,30 +34,31 @@ pub async fn req(user: User, server: Ref, target: String, data: Json<Data>) -> R
|
|||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if target.id.user == user.id {
|
if target.id.user == user.id {
|
||||||
if (data.nickname.is_some() && !perm.get_change_nickname()) ||
|
if (data.nickname.is_some() && !perm.get_change_nickname())
|
||||||
(data.avatar.is_some() && !perm.get_change_avatar()) {
|
|| (data.avatar.is_some() && !perm.get_change_avatar())
|
||||||
return Err(Error::MissingPermission)
|
{
|
||||||
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(remove) = &data.remove {
|
if let Some(remove) = &data.remove {
|
||||||
if match remove {
|
if match remove {
|
||||||
RemoveMemberField::Avatar => !perm.get_change_avatar(),
|
RemoveMemberField::Avatar => !perm.get_change_avatar(),
|
||||||
RemoveMemberField::Nickname => !perm.get_change_nickname()
|
RemoveMemberField::Nickname => !perm.get_change_nickname(),
|
||||||
} {
|
} {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if data.avatar.is_some() || (data.nickname.is_some() && !perm.get_manage_nicknames()) {
|
if data.avatar.is_some() || (data.nickname.is_some() && !perm.get_manage_nicknames()) {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(remove) = &data.remove {
|
if let Some(remove) = &data.remove {
|
||||||
if match remove {
|
if match remove {
|
||||||
RemoveMemberField::Avatar => !perm.get_remove_avatars(),
|
RemoveMemberField::Avatar => !perm.get_remove_avatars(),
|
||||||
RemoveMemberField::Nickname => !perm.get_manage_nicknames()
|
RemoveMemberField::Nickname => !perm.get_manage_nicknames(),
|
||||||
} {
|
} {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -84,7 +84,8 @@ pub async fn req(user: User, server: Ref, target: String, data: Json<Data>) -> R
|
|||||||
}
|
}
|
||||||
|
|
||||||
if let Some(attachment_id) = &data.avatar {
|
if let Some(attachment_id) = &data.avatar {
|
||||||
let attachment = File::find_and_use(&attachment_id, "avatars", "user", &target.id.user).await?;
|
let attachment =
|
||||||
|
File::find_and_use(&attachment_id, "avatars", "user", &target.id.user).await?;
|
||||||
set.insert(
|
set.insert(
|
||||||
"avatar",
|
"avatar",
|
||||||
to_document(&attachment).map_err(|_| Error::DatabaseError {
|
to_document(&attachment).map_err(|_| Error::DatabaseError {
|
||||||
@@ -107,7 +108,11 @@ pub async fn req(user: User, server: Ref, target: String, data: Json<Data>) -> R
|
|||||||
|
|
||||||
if operations.len() > 0 {
|
if operations.len() > 0 {
|
||||||
get_collection("server_members")
|
get_collection("server_members")
|
||||||
.update_one(doc! { "_id.server": &server.id, "_id.user": &target.id.user }, operations, None)
|
.update_one(
|
||||||
|
doc! { "_id.server": &server.id, "_id.user": &target.id.user },
|
||||||
|
operations,
|
||||||
|
None,
|
||||||
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| Error::DatabaseError {
|
.map_err(|_| Error::DatabaseError {
|
||||||
operation: "update_one",
|
operation: "update_one",
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ pub async fn req(user: User, target: Ref, member: String) -> Result<JsonValue> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_view() {
|
if !perm.get_view() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,8 +2,8 @@ use crate::database::*;
|
|||||||
use crate::util::result::{Error, Result};
|
use crate::util::result::{Error, Result};
|
||||||
|
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
|
use mongodb::bson::{doc, from_document, Document};
|
||||||
use rocket_contrib::json::JsonValue;
|
use rocket_contrib::json::JsonValue;
|
||||||
use mongodb::bson::{Document, doc, from_document};
|
|
||||||
|
|
||||||
// ! FIXME: this is a temporary route while permissions are being worked on.
|
// ! FIXME: this is a temporary route while permissions are being worked on.
|
||||||
|
|
||||||
@@ -15,7 +15,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_view() {
|
if !perm.get_view() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,18 +11,18 @@ pub async fn req(user: User, target: Ref, member: String) -> Result<()> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_manage_members() {
|
if !perm.get_manage_members() {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
|
|
||||||
let member = Ref::from(member)?.fetch_member(&target.id).await?;
|
let member = Ref::from(member)?.fetch_member(&target.id).await?;
|
||||||
if member.id.user == user.id {
|
if member.id.user == user.id {
|
||||||
return Err(Error::InvalidOperation)
|
return Err(Error::InvalidOperation);
|
||||||
}
|
}
|
||||||
|
|
||||||
if target.id == target.owner {
|
if target.id == target.owner {
|
||||||
return Err(Error::MissingPermission)
|
return Err(Error::MissingPermission);
|
||||||
}
|
}
|
||||||
|
|
||||||
target.remove_member(&member.id.user).await
|
target.remove_member(&member.id.user).await
|
||||||
|
|||||||
@@ -2,19 +2,19 @@ use rocket::Route;
|
|||||||
|
|
||||||
mod server_create;
|
mod server_create;
|
||||||
mod server_delete;
|
mod server_delete;
|
||||||
mod server_fetch;
|
|
||||||
mod server_edit;
|
mod server_edit;
|
||||||
|
mod server_fetch;
|
||||||
|
|
||||||
mod channel_create;
|
mod channel_create;
|
||||||
|
|
||||||
|
mod member_edit;
|
||||||
|
mod member_fetch;
|
||||||
mod member_fetch_all;
|
mod member_fetch_all;
|
||||||
mod member_remove;
|
mod member_remove;
|
||||||
mod member_fetch;
|
|
||||||
mod member_edit;
|
|
||||||
|
|
||||||
mod ban_create;
|
mod ban_create;
|
||||||
mod ban_remove;
|
|
||||||
mod ban_list;
|
mod ban_list;
|
||||||
|
mod ban_remove;
|
||||||
|
|
||||||
mod invites_fetch;
|
mod invites_fetch;
|
||||||
|
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
|
|||||||
.with_server(&target)
|
.with_server(&target)
|
||||||
.for_server()
|
.for_server()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !perm.get_view() {
|
if !perm.get_view() {
|
||||||
Err(Error::MissingPermission)?
|
Err(Error::MissingPermission)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ pub enum Error {
|
|||||||
// ? Server related errors.
|
// ? Server related errors.
|
||||||
UnknownServer,
|
UnknownServer,
|
||||||
Banned,
|
Banned,
|
||||||
|
|
||||||
// ? General errors.
|
// ? General errors.
|
||||||
TooManyIds,
|
TooManyIds,
|
||||||
FailedValidation {
|
FailedValidation {
|
||||||
|
|||||||
Reference in New Issue
Block a user