fix: fixes for full mongo implementation

This commit is contained in:
Paul Makles
2022-02-11 21:44:59 +00:00
parent 7a9adb45f7
commit 4801b51cef
15 changed files with 45 additions and 36 deletions

25
Cargo.lock generated
View File

@@ -1318,7 +1318,7 @@ checksum = "527e8c9ac747e28542699a951517aa9a6945af506cd1f2e1b53a576c17b6cc11"
dependencies = [ dependencies = [
"bytes 1.1.0", "bytes 1.1.0",
"fnv", "fnv",
"itoa", "itoa 0.4.8",
] ]
[[package]] [[package]]
@@ -1384,7 +1384,7 @@ dependencies = [
"http-body 0.3.1", "http-body 0.3.1",
"httparse", "httparse",
"httpdate 0.3.2", "httpdate 0.3.2",
"itoa", "itoa 0.4.8",
"pin-project", "pin-project",
"socket2 0.3.19", "socket2 0.3.19",
"tokio 0.2.25", "tokio 0.2.25",
@@ -1408,7 +1408,7 @@ dependencies = [
"http-body 0.4.3", "http-body 0.4.3",
"httparse", "httparse",
"httpdate 1.0.1", "httpdate 1.0.1",
"itoa", "itoa 0.4.8",
"pin-project-lite 0.2.7", "pin-project-lite 0.2.7",
"socket2 0.4.2", "socket2 0.4.2",
"tokio 1.11.0", "tokio 1.11.0",
@@ -1540,6 +1540,12 @@ version = "0.4.8"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b71991ff56294aa922b450139ee08b3bfc70982c6b2c7562771375cf73542dd4" checksum = "b71991ff56294aa922b450139ee08b3bfc70982c6b2c7562771375cf73542dd4"
[[package]]
name = "itoa"
version = "1.0.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1aab8fc367588b89dcee83ab0fd66b72b50b72fa1904d7095045ace2b0c81c35"
[[package]] [[package]]
name = "js-sys" name = "js-sys"
version = "0.3.55" version = "0.3.55"
@@ -2761,7 +2767,7 @@ dependencies = [
"combine", "combine",
"dtoa", "dtoa",
"futures-util", "futures-util",
"itoa", "itoa 0.4.8",
"percent-encoding", "percent-encoding",
"pin-project-lite 0.2.7", "pin-project-lite 0.2.7",
"sha1", "sha1",
@@ -2782,7 +2788,7 @@ dependencies = [
"combine", "combine",
"dtoa", "dtoa",
"futures-util", "futures-util",
"itoa", "itoa 0.4.8",
"percent-encoding", "percent-encoding",
"pin-project-lite 0.2.7", "pin-project-lite 0.2.7",
"sha1", "sha1",
@@ -2956,6 +2962,7 @@ dependencies = [
"rauth", "rauth",
"rocket", "rocket",
"serde", "serde",
"serde_json",
"ulid 0.5.0", "ulid 0.5.0",
"validator", "validator",
] ]
@@ -3282,12 +3289,12 @@ dependencies = [
[[package]] [[package]]
name = "serde_json" name = "serde_json"
version = "1.0.68" version = "1.0.78"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0f690853975602e1bfe1ccbf50504d67174e3bcf340f23b5ea9992e0587a52d8" checksum = "d23c1ba4cf0efd44be32017709280b32d1cea5c3f1275c3b6d9e8bc54f758085"
dependencies = [ dependencies = [
"indexmap", "indexmap",
"itoa", "itoa 1.0.1",
"ryu", "ryu",
"serde", "serde",
] ]
@@ -3299,7 +3306,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "edfa57a7f8d9c1d260a549e7224100f6c43d43f9103e06dd8b4095a9b2b43ce9" checksum = "edfa57a7f8d9c1d260a549e7224100f6c43d43f9103e06dd8b4095a9b2b43ce9"
dependencies = [ dependencies = [
"form_urlencoded", "form_urlencoded",
"itoa", "itoa 0.4.8",
"ryu", "ryu",
"serde", "serde",
] ]

View File

@@ -121,7 +121,10 @@ async fn launch_web() {
}; };
} }
let db = DatabaseInfo::MongoDb("mongodb://localhost").connect().await.unwrap(); let db = DatabaseInfo::MongoDb("mongodb://localhost")
.connect()
.await
.unwrap();
db.migrate_database().await.unwrap(); db.migrate_database().await.unwrap();
let mongo_db = mongodb::Client::with_uri_str("mongodb://localhost") let mongo_db = mongodb::Client::with_uri_str("mongodb://localhost")

View File

@@ -83,7 +83,7 @@ pub async fn edit_bot(db: &Db, user: User, target: Ref, data: Json<Data>) -> Res
db.update_bot(&bot.id, &partial, remove.unwrap_or_else(Vec::new)) db.update_bot(&bot.id, &partial, remove.unwrap_or_else(Vec::new))
.await?; .await?;
bot.apply_options(partial); bot.apply_options(partial);
Ok(Json(bot)) Ok(Json(bot))
} }

View File

@@ -3,7 +3,6 @@ use revolt_quark::{
perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result,
}; };
#[delete("/<target>")] #[delete("/<target>")]
pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;

View File

@@ -3,7 +3,6 @@ use revolt_quark::{
perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result,
}; };
#[put("/<target>/recipients/<member>")] #[put("/<target>/recipients/<member>")]
pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;

View File

@@ -3,7 +3,6 @@ use revolt_quark::{
Db, EmptyResponse, Error, Ref, Result, Db, EmptyResponse, Error, Ref, Result,
}; };
#[delete("/<target>/recipients/<member>")] #[delete("/<target>/recipients/<member>")]
pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;

View File

@@ -1,6 +1,5 @@
use revolt_quark::{models::User, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result}; use revolt_quark::{models::User, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result};
#[delete("/<target>/messages/<msg>")] #[delete("/<target>/messages/<msg>")]
pub async fn req(db: &Db, user: User, target: Ref, msg: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref, msg: Ref) -> Result<EmptyResponse> {
let message = msg.as_message(db).await?; let message = msg.as_message(db).await?;

View File

@@ -74,6 +74,6 @@ pub async fn req(
message.embeds = Some(new_embeds); message.embeds = Some(new_embeds);
partial.embeds = message.embeds.clone(); partial.embeds = message.embeds.clone();
db.update_message(&message.id, &partial, vec![]).await?; db.update_message(&message.id, &partial).await?;
Ok(Json(message)) Ok(Json(message))
} }

View File

@@ -1,4 +1,7 @@
use revolt_quark::{Error, Result, Ref, models::{User, Invite}, Db}; use revolt_quark::{
models::{Invite, User},
Db, Error, Ref, Result,
};
use rocket::serde::json::Value; use rocket::serde::json::Value;
@@ -7,32 +10,34 @@ use crate::util::variables::MAX_SERVER_COUNT;
#[post("/<target>")] #[post("/<target>")]
pub async fn req(db: &Db, user: User, target: Ref) -> Result<Value> { pub async fn req(db: &Db, user: User, target: Ref) -> Result<Value> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot) return Err(Error::IsBot);
} }
if !user.can_acquire_server(db).await? { if !user.can_acquire_server(db).await? {
return Err(Error::TooManyServers { return Err(Error::TooManyServers {
max: *MAX_SERVER_COUNT max: *MAX_SERVER_COUNT,
}) });
} }
let invite = target.as_invite(db).await?; let invite = target.as_invite(db).await?;
match &invite { match &invite {
Invite::Server { channel, server, .. } => { Invite::Server {
channel, server, ..
} => {
let server = db.fetch_server(server).await?; let server = db.fetch_server(server).await?;
if db.fetch_ban(&server.id, &user.id).await.is_ok() { if db.fetch_ban(&server.id, &user.id).await.is_ok() {
return Err(Error::Banned) return Err(Error::Banned);
} }
let channel = db.fetch_channel(channel).await?; let channel = db.fetch_channel(channel).await?;
db.insert_member(&server.id, &user.id).await?; db.insert_member(&server.id, &user.id).await?;
Ok(json!({ Ok(json!({
"type": "Server", "type": "Server",
"channel": channel, "channel": channel,
"server": server "server": server
})) }))
} }
_ => unreachable!() _ => unreachable!(),
} }
} }

View File

@@ -1,4 +1,4 @@
use revolt_quark::models::{File, User, ServerBan}; use revolt_quark::models::{File, ServerBan, User};
use revolt_quark::{perms, Db, Error, Ref, Result, ServerPermission}; use revolt_quark::{perms, Db, Error, Ref, Result, ServerPermission};
use rocket::serde::json::Json; use rocket::serde::json::Json;

View File

@@ -1,6 +1,5 @@
use revolt_quark::{models::User, perms, Db, EmptyResponse, Error, Ref, Result, ServerPermission}; use revolt_quark::{models::User, perms, Db, EmptyResponse, Error, Ref, Result, ServerPermission};
#[delete("/<target>/members/<member>")] #[delete("/<target>/members/<member>")]
pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

View File

@@ -1,6 +1,5 @@
use revolt_quark::{models::User, Db, EmptyResponse, Ref, Result}; use revolt_quark::{models::User, Db, EmptyResponse, Ref, Result};
#[delete("/<target>")] #[delete("/<target>")]
pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> { pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> {
let server = target.as_server(db).await?; let server = target.as_server(db).await?;

View File

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use rocket::State; use rocket::State;
#[put("/<username>/block")] #[put("/<target>/block")]
pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> { pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user_by_username(&username).await?; let mut target = db.fetch_user(&target).await?;
user.block_user(db, &mut target).await?; user.block_user(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await)) Ok(Json(target.with_auto_perspective(db, &user).await))
} }

View File

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use rocket::State; use rocket::State;
#[delete("/<username>/friend")] #[delete("/<target>/friend")]
pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> { pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user_by_username(&username).await?; let mut target = db.fetch_user(&target).await?;
user.remove_friend(db, &mut target).await?; user.remove_friend(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await)) Ok(Json(target.with_auto_perspective(db, &user).await))
} }

View File

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use rocket::State; use rocket::State;
#[delete("/<username>/block")] #[delete("/<target>/block")]
pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> { pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
let mut target = db.fetch_user_by_username(&username).await?; let mut target = db.fetch_user(&target).await?;
user.unblock_user(db, &mut target).await?; user.unblock_user(db, &mut target).await?;
Ok(Json(target.with_auto_perspective(db, &user).await)) Ok(Json(target.with_auto_perspective(db, &user).await))
} }