Compare commits

..
Author SHA1 Message Date
Paul Makles cb813e91f0 chore: cut new 0.6.9 release 2023-10-28 14:39:10 +01:00
Paul Makles a6bc9fcbb3 refactor: implement Server::create 2023-10-28 14:25:29 +01:00
Paul Makles cd0737911b refactor: use fixtures to populate db
refactor: method for creating server channels
2023-10-27 16:31:37 +01:00
Paul Makles 7705354d4d ci: use reference db for generating spec 2023-10-26 15:07:03 +01:00
Paul Makles 39ac8b2278 refactor: don't create unnecessary reference 2023-10-26 14:12:28 +01:00
arslee07 c5494aa1ca fix: respect Permission::SendEmbeds when editing a message 2023-10-26 14:12:28 +01:00
Paul Makles 1a96bf5a4b chore: remove redundant file [skip ci] 2023-10-26 14:05:01 +01:00
Paul Makles 37a1244751 test: ignore hello world test because it breaks cargo test? 2023-10-26 14:05:01 +01:00
Paul Makles bf39f18f8d fix: always generate new test database 2023-10-26 14:05:01 +01:00
Paul Makles 95be7dc87a ci: use nextest because cargo test sig sevs 2023-10-26 14:05:01 +01:00
Paul Makles 9be171c7b6 fix: don't copy the old test file 2023-10-26 14:05:01 +01:00
Paul Makles 786fbb8ae4 chore: disable broken tests 2023-10-26 14:05:01 +01:00
Paul Makles 6c0210e8a1 feat: always use test configuration during tests 2023-10-26 14:05:01 +01:00
Paul Makles 456bf7b42a fix: check server ownership for text channel
refactor: minor cleanup of code
2023-10-26 14:05:01 +01:00
Paul Makles 21ffea0f9f fix(core/database): store member during permission query 2023-10-26 14:05:01 +01:00
Paul Makles 96d90215d2 fix(core/models): validator feature flag not properly gatekept 2023-10-26 14:05:01 +01:00
Paul Makles 29dcf714f2 chore: start work on test for permission issue 2023-10-26 14:05:01 +01:00
Paul Makles cb97004d3f feat(core): provide user profile where appropriate 2023-10-16 09:33:36 +01:00
Paul Makles 0b6e5dd9db chore: add nix file to vscode config 2023-10-16 09:23:11 +01:00
Paul Makles 7132877201 feat(core): validation for files in reference db 2023-10-16 09:22:53 +01:00
Paul Makles 4d42fb74e7 fix(core): publish user settings event 2023-10-16 09:22:17 +01:00
Paul Makles 866ace3381 refactor: use config for max bot count 2023-10-16 09:21:48 +01:00
Paul Makles 0798e81862 feat(core): implement ratelimit events for reference db 2023-10-16 09:21:34 +01:00
Paul Makles 2fa5ac41ac refactor: tests for more group routes 2023-09-22 19:04:17 +01:00
Paul Makles 25ae1555a2 chore: add nextest and fmt to nix 2023-09-22 19:03:49 +01:00
Paul Makles 881888bd1c chore: add timeout for nextest runs 2023-09-22 19:03:29 +01:00
Paul Makles 41f20c2239 ci: use custom Revolt.toml for tests 2023-09-10 15:46:00 +01:00
Paul Makles 38f1381fea chore: switch to resolver=2 2023-09-10 15:19:23 +01:00
Paul Makles 6838b0b1f0 refactor: ensure in-memory database consistency
test: update authifier to allow for in-memory tests
ci: explicitly run in-memory and mongo tests
2023-09-10 14:49:27 +01:00
Paul Makles 055deab105 fix: allow bots to authenticate with new routes 2023-09-10 14:48:13 +01:00
Paul Makles 6d2073120d fix: carry over send embed permission when sending [skip ci] 2023-09-10 12:41:23 +01:00
Paul Makles 9789909061 feat(delta): rewrite emoji routes w/o quark
#283
2023-09-05 12:36:18 +01:00
Paul Makles 895de86f1e fix(core/config): use uppercase filename 2023-09-03 19:00:17 +01:00
Paul Makles 98b8c5403b feat(delta): rewrite channel / message fetch 2023-09-03 18:27:21 +01:00
Paul Makles aa7c75c60e chore: ignore Revolt.toml 2023-09-03 18:12:15 +01:00
Paul Makles 4270f0c5d4 feat(core): implement create group 2023-09-03 18:12:08 +01:00
Paul Makles eb1f45d208 fix(core/database): use attachments collection name 2023-09-03 17:43:13 +01:00
Paul Makles 9929f7499a refactor(core): use core crates for sending messages 2023-09-03 16:31:39 +01:00
Paul Makles 279d9ef1b5 refactor(core): remove quark references from webhook routes 2023-09-03 15:52:18 +01:00
Paul Makles 5a9bb9e68d fix: convert index map/set to hash map/set
fix: idempotency token should use correct crate cfg
2023-08-27 15:51:20 +01:00
Paul Makles a8fc9e928a fix: import validator crate directly in derive 2023-08-27 15:45:23 +01:00
Paul Makles da231ae390 fix: remove indexmap from models 2023-08-27 15:43:57 +01:00
Paul Makles 7a31da4189 fix: add indexmap/serde to model depends 2023-08-27 15:40:05 +01:00
Paul Makles 7d0b5392b5 fix: should include crate in build script too 2023-08-27 15:33:28 +01:00
Paul Makles 1e1d44c0e4 fix: add config crate to Dockerfile
chore: bump release version
2023-08-27 15:30:53 +01:00
93 changed files with 2416 additions and 943 deletions
+2
View File
@@ -0,0 +1,2 @@
[profile.default]
slow-timeout = { period = "3s", terminate-after = 2 }
+20 -13
View File
@@ -20,6 +20,11 @@ jobs:
toolchain: stable toolchain: stable
override: true override: true
components: rustfmt, clippy components: rustfmt, clippy
- name: Install cargo-nextest
uses: baptiste0928/cargo-install@v1
with:
crate: cargo-nextest
locked: true
- name: Run cargo build - name: Run cargo build
uses: actions-rs/cargo@v1 uses: actions-rs/cargo@v1
@@ -30,25 +35,27 @@ jobs:
run: | run: |
docker-compose -f docker-compose.db.yml up -d docker-compose -f docker-compose.db.yml up -d
- name: Run cargo test
uses: actions-rs/cargo@v1
with:
command: test
- name: Run cargo test (with MongoDB)
uses: actions-rs/cargo@v1
env:
MONGODB: mongodb://localhost
with:
command: test
- name: Copy .env.example - name: Copy .env.example
if: github.event_name != 'pull_request' && github.ref_name == 'master'
run: | run: |
cp .env.example .env cp .env.example .env
- name: Run cargo test
env:
TEST_DB: REFERENCE
run: |
cargo nextest run
- name: Run cargo test (with MongoDB)
env:
TEST_DB: MONGODB
MONGODB: mongodb://localhost
run: |
cargo nextest run
- name: Start API in background - name: Start API in background
if: github.event_name != 'pull_request' && github.ref_name == 'master' if: github.event_name != 'pull_request' && github.ref_name == 'master'
env:
TEST_DB: REFERENCE
run: | run: |
cargo run --bin revolt-delta & cargo run --bin revolt-delta &
+2
View File
@@ -1,4 +1,6 @@
Rocket.toml Rocket.toml
Revolt.toml
target target
.data .data
.env .env
+2 -1
View File
@@ -1,5 +1,6 @@
{ {
"editor.formatOnSave": true, "editor.formatOnSave": true,
"rust-analyzer.checkOnSave.command": "clippy", "rust-analyzer.checkOnSave.command": "clippy",
"nixEnvSelector.suggestion": false "nixEnvSelector.suggestion": false,
"nixEnvSelector.nixFile": "${workspaceRoot}/default.nix"
} }
Generated
+15 -13
View File
@@ -347,9 +347,9 @@ dependencies = [
[[package]] [[package]]
name = "authifier" name = "authifier"
version = "1.0.7" version = "1.0.8"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a84411393a4326726ead660991ce8047f1865fa89900fb6c18e1e6c513cf1f9b" checksum = "30269caf0aaf1e1b542b150030e9688bf41d50026e09a51efd9408f332636c9d"
dependencies = [ dependencies = [
"async-std", "async-std",
"async-trait", "async-trait",
@@ -3494,7 +3494,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-bonfire" name = "revolt-bonfire"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-std", "async-std",
"async-tungstenite", "async-tungstenite",
@@ -3512,7 +3512,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-config" name = "revolt-config"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-std", "async-std",
"cached", "cached",
@@ -3524,7 +3524,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-database" name = "revolt-database"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-lock", "async-lock",
"async-recursion", "async-recursion",
@@ -3569,10 +3569,11 @@ dependencies = [
[[package]] [[package]]
name = "revolt-delta" name = "revolt-delta"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-channel", "async-channel",
"async-std", "async-std",
"authifier",
"bitfield", "bitfield",
"chrono", "chrono",
"dashmap", "dashmap",
@@ -3591,6 +3592,7 @@ dependencies = [
"redis-kiss", "redis-kiss",
"regex", "regex",
"reqwest", "reqwest",
"revolt-config",
"revolt-database", "revolt-database",
"revolt-models", "revolt-models",
"revolt-permissions", "revolt-permissions",
@@ -3613,7 +3615,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-models" name = "revolt-models"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"indexmap", "indexmap",
"iso8601-timestamp 0.2.11", "iso8601-timestamp 0.2.11",
@@ -3629,7 +3631,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-permissions" name = "revolt-permissions"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-std", "async-std",
"async-trait", "async-trait",
@@ -3644,7 +3646,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-presence" name = "revolt-presence"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-std", "async-std",
"log", "log",
@@ -3655,7 +3657,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-quark" name = "revolt-quark"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"async-lock", "async-lock",
"async-recursion", "async-recursion",
@@ -3708,7 +3710,7 @@ dependencies = [
[[package]] [[package]]
name = "revolt-result" name = "revolt-result"
version = "0.6.7" version = "0.6.9"
dependencies = [ dependencies = [
"revolt_okapi", "revolt_okapi",
"revolt_rocket_okapi", "revolt_rocket_okapi",
@@ -3858,9 +3860,9 @@ dependencies = [
[[package]] [[package]]
name = "rocket_authifier" name = "rocket_authifier"
version = "1.0.7" version = "1.0.8"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d4554a81937341d8c3cd3e43431ac2d23624444dc3a88e7ffd3ff66593779293" checksum = "5f89a12311f60e9288833fc3ce6029bce5d5c61870ceef74d4a50668a8b520ad"
dependencies = [ dependencies = [
"authifier", "authifier",
"iso8601-timestamp 0.1.10", "iso8601-timestamp 0.1.10",
+3
View File
@@ -1,7 +1,10 @@
[workspace] [workspace]
resolver = "2"
members = ["crates/delta", "crates/bonfire", "crates/quark", "crates/core/*"] members = ["crates/delta", "crates/bonfire", "crates/quark", "crates/core/*"]
[patch.crates-io] [patch.crates-io]
# mobc-redis = { git = "https://github.com/insertish/mobc", rev = "8b880bb59f2ba80b4c7bc40c649c113d8857a186" } # mobc-redis = { git = "https://github.com/insertish/mobc", rev = "8b880bb59f2ba80b4c7bc40c649c113d8857a186" }
redis22 = { package = "redis", version = "0.22.3", git = "https://github.com/revoltchat/redis-rs", rev = "1a41faf356fd21aebba71cea7eb7eb2653e5f0ef" } redis22 = { package = "redis", version = "0.22.3", git = "https://github.com/revoltchat/redis-rs", rev = "1a41faf356fd21aebba71cea7eb7eb2653e5f0ef" }
redis23 = { package = "redis", version = "0.23.1", git = "https://github.com/revoltchat/redis-rs", rev = "f8ca28ab85da59d2ccde526b4d2fb390eff5a5f9" } redis23 = { package = "redis", version = "0.23.1", git = "https://github.com/revoltchat/redis-rs", rev = "f8ca28ab85da59d2ccde526b4d2fb390eff5a5f9" }
# authifier = { package = "authifier", version = "1.0.8", path = "../authifier/crates/authifier" }
# rocket_authifier = { package = "rocket_authifier", version = "1.0.8", path = "../authifier/crates/rocket_authifier" }
+1
View File
@@ -20,6 +20,7 @@ COPY Cargo.toml Cargo.lock ./
COPY crates/bonfire/Cargo.toml ./crates/bonfire/ COPY crates/bonfire/Cargo.toml ./crates/bonfire/
COPY crates/delta/Cargo.toml ./crates/delta/ COPY crates/delta/Cargo.toml ./crates/delta/
COPY crates/quark/Cargo.toml ./crates/quark/ COPY crates/quark/Cargo.toml ./crates/quark/
COPY crates/core/config/Cargo.toml ./crates/core/config/
COPY crates/core/database/Cargo.toml ./crates/core/database/ COPY crates/core/database/Cargo.toml ./crates/core/database/
COPY crates/core/models/Cargo.toml ./crates/core/models/ COPY crates/core/models/Cargo.toml ./crates/core/models/
COPY crates/core/permissions/Cargo.toml ./crates/core/permissions/ COPY crates/core/permissions/Cargo.toml ./crates/core/permissions/
+1
View File
@@ -16,6 +16,7 @@ COPY Cargo.toml Cargo.lock ./
COPY crates/bonfire/Cargo.toml ./crates/bonfire/ COPY crates/bonfire/Cargo.toml ./crates/bonfire/
COPY crates/delta/Cargo.toml ./crates/delta/ COPY crates/delta/Cargo.toml ./crates/delta/
COPY crates/quark/Cargo.toml ./crates/quark/ COPY crates/quark/Cargo.toml ./crates/quark/
COPY crates/core/config/Cargo.toml ./crates/core/config/
COPY crates/core/database/Cargo.toml ./crates/core/database/ COPY crates/core/database/Cargo.toml ./crates/core/database/
COPY crates/core/models/Cargo.toml ./crates/core/models/ COPY crates/core/models/Cargo.toml ./crates/core/models/
COPY crates/core/permissions/Cargo.toml ./crates/core/permissions/ COPY crates/core/permissions/Cargo.toml ./crates/core/permissions/
+3
View File
@@ -1,6 +1,9 @@
disallowed-methods = [ disallowed-methods = [
# Shouldn't need to access these directly # Shouldn't need to access these directly
"revolt_database::models::bots::model::Bot::remove_field", "revolt_database::models::bots::model::Bot::remove_field",
"revolt_database::models::messages::model::Message::attach_sendable_embed",
"revolt_database::models::users::model::User::set_relationship",
"revolt_database::models::users::model::User::apply_relationship",
# Prefer to use Object::create() # Prefer to use Object::create()
"revolt_database::models::bots::ops::AbstractBots::insert_bot", "revolt_database::models::bots::ops::AbstractBots::insert_bot",
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-bonfire" name = "revolt-bonfire"
version = "0.6.7" version = "0.6.9"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
edition = "2021" edition = "2021"
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-config" name = "revolt-config"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
+3
View File
@@ -0,0 +1,3 @@
[database]
mongodb = "mongodb://localhost"
redis = "redis://localhost/"
+3 -1
View File
@@ -48,9 +48,11 @@ max_concurrent_connections = 50
[features.limits.default] [features.limits.default]
group_size = 100 group_size = 100
bots = 5 bots = 5
message_length = 2000
message_embeds = 5
message_replies = 5 message_replies = 5
message_attachments = 5 message_attachments = 5
message_embeds = 5 message_reactions = 20
servers = 100 servers = 100
server_emoji = 100 server_emoji = 100
server_roles = 200 server_roles = 200
+9 -2
View File
@@ -13,8 +13,13 @@ static CONFIG_BUILDER: Lazy<RwLock<Config>> = Lazy::new(|| {
FileFormat::Toml, FileFormat::Toml,
)); ));
if std::path::Path::new("revolt.toml").exists() { if std::env::var("TEST_DB").is_ok() {
builder = builder.add_source(File::new("revolt.toml", FileFormat::Toml)); builder = builder.add_source(File::from_str(
include_str!("../Revolt.test.toml"),
FileFormat::Toml,
));
} else if std::path::Path::new("Revolt.toml").exists() {
builder = builder.add_source(File::new("Revolt.toml", FileFormat::Toml));
} }
builder.build().unwrap() builder.build().unwrap()
@@ -97,9 +102,11 @@ pub struct Api {
pub struct FeaturesLimits { pub struct FeaturesLimits {
pub group_size: usize, pub group_size: usize,
pub bots: usize, pub bots: usize,
pub message_length: usize,
pub message_replies: usize, pub message_replies: usize,
pub message_attachments: usize, pub message_attachments: usize,
pub message_embeds: usize, pub message_embeds: usize,
pub message_reactions: usize,
pub servers: usize, pub servers: usize,
pub server_emoji: usize, pub server_emoji: usize,
pub server_roles: usize, pub server_roles: usize,
+7 -7
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-database" name = "revolt-database"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
@@ -23,11 +23,11 @@ default = ["mongodb", "async-std-runtime", "tasks"]
[dependencies] [dependencies]
# Core # Core
revolt-config = { version = "0.6.7", path = "../config" } revolt-config = { version = "0.6.9", path = "../config" }
revolt-result = { version = "0.6.7", path = "../result" } revolt-result = { version = "0.6.9", path = "../result" }
revolt-models = { version = "0.6.7", path = "../models" } revolt-models = { version = "0.6.9", path = "../models" }
revolt-presence = { version = "0.6.7", path = "../presence" } revolt-presence = { version = "0.6.9", path = "../presence" }
revolt-permissions = { version = "0.6.7", path = "../permissions", features = [ revolt-permissions = { version = "0.6.9", path = "../permissions", features = [
"serde", "serde",
"bson", "bson",
] } ] }
@@ -86,4 +86,4 @@ fcm = "0.9.2"
web-push = "0.10.0" web-push = "0.10.0"
# Authifier # Authifier
authifier = { version = "1.0" } authifier = { version = "1.0.8" }
@@ -0,0 +1,89 @@
[
{
"_object_type": "User",
"_id": "__ID:0__",
"username": "Owner",
"discriminator": "0001"
},
{
"_object_type": "User",
"_id": "__ID:1__",
"username": "Moderator",
"discriminator": "0001"
},
{
"_object_type": "User",
"_id": "__ID:2__",
"username": "User",
"discriminator": "0001"
},
{
"_object_type": "Channel",
"_id": "__ID:3__",
"channel_type": "TextChannel",
"name": "General",
"server": "__ID:4__",
"default_permissions": {
"a": 0,
"d": 1048576
},
"role_permissions": {
"__ID:5__": {
"a": 1048576,
"d": 0
}
}
},
{
"_object_type": "Server",
"_id": "__ID:4__",
"owner": "__ID:0__",
"name": "Server",
"channels": ["__ID:3__"],
"roles": {
"__ID:5__": {
"name": "Moderator",
"permissions": {
"a": 545270208,
"d": 0
},
"rank": 3
},
"__ID:6__": {
"name": "Owner",
"permissions": {
"a": 0,
"d": 0
},
"rank": 0
}
},
"default_permissions": 4000322560
},
{
"_object_type": "ServerMember",
"_id": {
"user": "__ID:0__",
"server": "__ID:4__"
},
"roles": ["__ID:6__"],
"joined_at": 1698318340195
},
{
"_object_type": "ServerMember",
"_id": {
"user": "__ID:1__",
"server": "__ID:4__"
},
"roles": ["__ID:5__"],
"joined_at": 1698318340195
},
{
"_object_type": "ServerMember",
"_id": {
"user": "__ID:2__",
"server": "__ID:4__"
},
"joined_at": 1698318340195
}
]
+32 -21
View File
@@ -1,6 +1,9 @@
mod mongodb; mod mongodb;
mod reference; mod reference;
use rand::Rng;
use revolt_config::config;
pub use self::mongodb::*; pub use self::mongodb::*;
pub use self::reference::*; pub use self::reference::*;
@@ -31,25 +34,44 @@ impl DatabaseInfo {
/// Create a database client from the given database information /// Create a database client from the given database information
#[async_recursion] #[async_recursion]
pub async fn connect(self) -> Result<Database, String> { pub async fn connect(self) -> Result<Database, String> {
let config = config().await;
Ok(match self { Ok(match self {
DatabaseInfo::Auto => { DatabaseInfo::Auto => {
if let Ok(uri) = std::env::var("MONGODB") { if std::env::var("TEST_DB").is_ok() {
return DatabaseInfo::MongoDb { DatabaseInfo::Test(format!(
uri, "revolt_test_{}",
rand::thread_rng().gen_range(1_000_000..10_000_000)
))
.connect()
.await?
} else if !config.database.mongodb.is_empty() {
DatabaseInfo::MongoDb {
uri: config.database.mongodb,
database_name: "revolt".to_string(), database_name: "revolt".to_string(),
} }
.connect() .connect()
.await; .await?
} } else {
DatabaseInfo::Reference.connect().await? DatabaseInfo::Reference.connect().await?
} }
}
DatabaseInfo::Test(database_name) => { DatabaseInfo::Test(database_name) => {
if let Ok(uri) = std::env::var("MONGODB") { match std::env::var("TEST_DB")
return DatabaseInfo::MongoDb { uri, database_name }.connect().await; .expect("`TEST_DB` environment variable should be set to REFERENCE or MONGODB")
.as_str()
{
"REFERENCE" => DatabaseInfo::Reference.connect().await?,
"MONGODB" => {
DatabaseInfo::MongoDb {
uri: config.database.mongodb,
database_name,
}
.connect()
.await?
}
_ => unreachable!("must specify REFERENCE or MONGODB"),
} }
DatabaseInfo::Reference.connect().await?
} }
DatabaseInfo::Reference => Database::Reference(Default::default()), DatabaseInfo::Reference => Database::Reference(Default::default()),
DatabaseInfo::MongoDb { uri, database_name } => { DatabaseInfo::MongoDb { uri, database_name } => {
@@ -65,14 +87,3 @@ impl DatabaseInfo {
}) })
} }
} }
impl From<Database> for authifier::Database {
fn from(value: Database) -> Self {
match value {
Database::Reference(_) => Default::default(),
Database::MongoDb(MongoDb(client, _)) => authifier::Database::MongoDb(
authifier::database::MongoDb(client.database("revolt")),
),
}
}
}
@@ -4,7 +4,7 @@ use futures::lock::Mutex;
use crate::{ use crate::{
Bot, Channel, ChannelCompositeKey, ChannelUnread, Emoji, File, Invite, Member, Bot, Channel, ChannelCompositeKey, ChannelUnread, Emoji, File, Invite, Member,
MemberCompositeKey, Message, Server, ServerBan, User, UserSettings, Webhook, MemberCompositeKey, Message, RatelimitEvent, Server, ServerBan, User, UserSettings, Webhook,
}; };
database_derived!( database_derived!(
@@ -19,6 +19,7 @@ database_derived!(
pub emojis: Arc<Mutex<HashMap<String, Emoji>>>, pub emojis: Arc<Mutex<HashMap<String, Emoji>>>,
pub files: Arc<Mutex<HashMap<String, File>>>, pub files: Arc<Mutex<HashMap<String, File>>>,
pub messages: Arc<Mutex<HashMap<String, Message>>>, pub messages: Arc<Mutex<HashMap<String, Message>>>,
pub ratelimit_events: Arc<Mutex<HashMap<String, RatelimitEvent>>>,
pub user_settings: Arc<Mutex<HashMap<String, UserSettings>>>, pub user_settings: Arc<Mutex<HashMap<String, UserSettings>>>,
pub users: Arc<Mutex<HashMap<String, User>>>, pub users: Arc<Mutex<HashMap<String, User>>>,
pub server_bans: Arc<Mutex<HashMap<MemberCompositeKey, ServerBan>>>, pub server_bans: Arc<Mutex<HashMap<MemberCompositeKey, ServerBan>>>,
+6 -11
View File
@@ -4,8 +4,8 @@ use serde::{Deserialize, Serialize};
use revolt_models::v0::{ use revolt_models::v0::{
AppendMessage, Channel, Emoji, FieldsChannel, FieldsMember, FieldsRole, FieldsServer, AppendMessage, Channel, Emoji, FieldsChannel, FieldsMember, FieldsRole, FieldsServer,
FieldsUser, FieldsWebhook, MemberCompositeKey, Message, PartialChannel, PartialMember, FieldsUser, FieldsWebhook, MemberCompositeKey, Message, PartialChannel, PartialMember,
PartialMessage, PartialRole, PartialServer, PartialUser, PartialWebhook, Server, UserSettings, PartialMessage, PartialRole, PartialServer, PartialUser, PartialWebhook, Server, User,
Webhook, UserSettings, Webhook,
}; };
use revolt_result::Error; use revolt_result::Error;
@@ -154,17 +154,12 @@ pub enum EventV1 {
event_id: Option<String>, event_id: Option<String>,
}, },
/*/// Relationship with another user changed /// Relationship with another user changed
UserRelationship { UserRelationship { id: String, user: User },
id: String,
user: User,
// ! this field can be deprecated
status: RelationshipStatus,
},*/
/// Settings updated remotely /// Settings updated remotely
UserSettingsUpdate { id: String, update: UserSettings }, UserSettingsUpdate { id: String, update: UserSettings },
/*/// User has been platform banned or deleted their account /// User has been platform banned or deleted their account
/// ///
/// Clients should remove the following associated data: /// Clients should remove the following associated data:
/// - Messages /// - Messages
@@ -173,7 +168,7 @@ pub enum EventV1 {
/// - Server Memberships /// - Server Memberships
/// ///
/// User flags are specified to explain why a wipe is occurring though not all reasons will necessarily ever appear. /// User flags are specified to explain why a wipe is occurring though not all reasons will necessarily ever appear.
UserPlatformWipe { user_id: String, flags: i32 }, */ UserPlatformWipe { user_id: String, flags: i32 },
/// New emoji /// New emoji
EmojiCreate(Emoji), EmojiCreate(Emoji),
@@ -18,7 +18,7 @@ mod tests {
db.migrate_database().await.unwrap(); db.migrate_database().await.unwrap();
// Migrate the existing database // Migrate the existing database
db.migrate_database().await.unwrap() db.migrate_database().await.unwrap();
}); });
} }
} }
@@ -1,3 +1,4 @@
use revolt_config::config;
use revolt_result::Result; use revolt_result::Result;
use ulid::Ulid; use ulid::Ulid;
@@ -79,9 +80,8 @@ impl Bot {
return Err(create_error!(IsBot)); return Err(create_error!(IsBot));
} }
// TODO: config let config = config().await;
let max_bot_count = 5; if db.get_number_of_bots_by_user(&owner.id).await? >= config.features.limits.default.bots {
if db.get_number_of_bots_by_user(&owner.id).await? >= max_bot_count {
return Err(create_error!(ReachedMaximumBots)); return Err(create_error!(ReachedMaximumBots));
} }
@@ -1,11 +1,15 @@
use std::collections::HashMap; use std::collections::HashMap;
use revolt_models::v0::MessageAuthor; use revolt_config::config;
use revolt_models::v0::{self, MessageAuthor};
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
use revolt_result::Result; use revolt_result::Result;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use crate::{events::client::EventV1, Database, File, IntoDocumentPath, SystemMessage, User}; use crate::{
events::client::EventV1, Database, File, IntoDocumentPath, PartialServer, Server,
SystemMessage, User,
};
auto_derived!( auto_derived!(
#[serde(tag = "channel_type")] #[serde(tag = "channel_type")]
@@ -168,7 +172,7 @@ auto_derived!(
#[allow(clippy::disallowed_methods)] #[allow(clippy::disallowed_methods)]
impl Channel { impl Channel {
/// Create a channel /* /// Create a channel
pub async fn create(&self, db: &Database) -> Result<()> { pub async fn create(&self, db: &Database) -> Result<()> {
db.insert_channel(self).await?; db.insert_channel(self).await?;
@@ -186,6 +190,109 @@ impl Channel {
} }
Ok(()) Ok(())
}*/
/// Create a new server channel
pub async fn create_server_channel(
db: &Database,
server: &mut Server,
data: v0::DataCreateServerChannel,
update_server: bool,
) -> Result<Channel> {
let config = config().await;
if server.channels.len() > config.features.limits.default.server_channels {
return Err(create_error!(TooManyChannels {
max: config.features.limits.default.server_channels,
}));
};
let id = ulid::Ulid::new().to_string();
let channel = match data.channel_type {
v0::LegacyServerChannelType::Text => Channel::TextChannel {
id: id.clone(),
server: server.id.to_owned(),
name: data.name,
description: data.description,
icon: None,
last_message_id: None,
default_permissions: None,
role_permissions: HashMap::new(),
nsfw: data.nsfw.unwrap_or(false),
},
v0::LegacyServerChannelType::Voice => Channel::VoiceChannel {
id: id.clone(),
server: server.id.to_owned(),
name: data.name,
description: data.description,
icon: None,
default_permissions: None,
role_permissions: HashMap::new(),
nsfw: data.nsfw.unwrap_or(false),
},
};
db.insert_channel(&channel).await?;
if update_server {
server
.update(
db,
PartialServer {
channels: Some([server.channels.clone(), [id].into()].concat()),
..Default::default()
},
vec![],
)
.await?;
EventV1::ChannelCreate(channel.clone().into())
.p(server.id.clone())
.await;
}
Ok(channel)
}
/// Create a group
pub async fn create_group(
db: &Database,
mut data: v0::DataCreateGroup,
owner_id: String,
) -> Result<Channel> {
data.users.insert(owner_id.to_string());
let config = config().await;
if data.users.len() > config.features.limits.default.group_size {
return Err(create_error!(GroupTooLarge {
max: config.features.limits.default.group_size,
}));
}
let recipients = data.users.into_iter().collect::<Vec<String>>();
let channel = Channel::Group {
id: ulid::Ulid::new().to_string(),
name: data.name,
owner: owner_id,
description: data.description,
recipients: recipients.clone(),
icon: None,
last_message_id: None,
permissions: None,
nsfw: data.nsfw.unwrap_or(false),
};
db.insert_channel(&channel).await?;
let event = EventV1::ChannelCreate(channel.clone().into());
for recipient in recipients {
event.clone().private(recipient).await;
}
Ok(channel)
} }
/// Add user to a group /// Add user to a group
@@ -214,10 +321,6 @@ impl Channel {
.p(id.to_string()) .p(id.to_string())
.await; .await;
EventV1::ChannelCreate(self.clone().into())
.private(user.id.to_string())
.await;
SystemMessage::UserAdded { SystemMessage::UserAdded {
id: user.id.to_string(), id: user.id.to_string(),
by: by_id.to_string(), by: by_id.to_string(),
@@ -235,6 +338,10 @@ impl Channel {
.await .await
.ok(); .ok();
EventV1::ChannelCreate(self.clone().into())
.private(user.id.to_string())
.await;
Ok(()) Ok(())
} }
_ => Err(create_error!(InvalidOperation)), _ => Err(create_error!(InvalidOperation)),
@@ -262,7 +369,7 @@ impl Channel {
} }
} }
/// Get a reference to this channel's id /// Clone this channel's id
pub fn id(&self) -> String { pub fn id(&self) -> String {
match self { match self {
Channel::DirectMessage { id, .. } Channel::DirectMessage { id, .. }
@@ -581,3 +688,36 @@ impl IntoDocumentPath for FieldsChannel {
}) })
} }
} }
#[cfg(test)]
mod tests {
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use crate::{fixture, util::permissions::DatabasePermissionQuery};
#[async_std::test]
async fn permissions_text_channel() {
database_test!(|db| async move {
fixture!(db, "server_with_roles",
owner user 0
moderator user 1
user user 2
channel channel 3);
let mut query = DatabasePermissionQuery::new(&db, &owner).channel(&channel);
assert!(calculate_channel_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::SendMessage));
let mut query = DatabasePermissionQuery::new(&db, &moderator).channel(&channel);
assert!(calculate_channel_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::SendMessage));
let mut query = DatabasePermissionQuery::new(&db, &user).channel(&channel);
assert!(!calculate_channel_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::SendMessage));
});
}
}
@@ -6,7 +6,7 @@ use crate::MongoDb;
use super::AbstractAttachments; use super::AbstractAttachments;
static COL: &str = "bots"; static COL: &str = "attachments";
#[async_trait] #[async_trait]
impl AbstractAttachments for MongoDb { impl AbstractAttachments for MongoDb {
@@ -22,18 +22,28 @@ impl AbstractAttachments for ReferenceDb {
async fn find_and_use_attachment( async fn find_and_use_attachment(
&self, &self,
id: &str, id: &str,
_tag: &str, tag: &str,
_parent_type: &str, parent_type: &str,
_parent_id: &str, parent_id: &str,
) -> Result<File> { ) -> Result<File> {
let mut files = self.files.lock().await; let mut files = self.files.lock().await;
if let Some(file) = files.get_mut(id) { if let Some(file) = files.get_mut(id) {
// TODO: check tag if file.tag == tag {
// TODO: set parent ID match parent_type {
"message" => file.message_id = Some(parent_id.to_owned()),
"user" => file.user_id = Some(parent_id.to_owned()),
"object" => file.object_id = Some(parent_id.to_owned()),
"server" => file.server_id = Some(parent_id.to_owned()),
_ => unreachable!(),
}
Ok(file.clone()) Ok(file.clone())
} else { } else {
Err(create_error!(NotFound)) Err(create_error!(NotFound))
} }
} else {
Err(create_error!(NotFound))
}
} }
/// Mark an attachment as having been reported. /// Mark an attachment as having been reported.
@@ -1,13 +1,21 @@
use std::collections::HashSet;
use indexmap::{IndexMap, IndexSet}; use indexmap::{IndexMap, IndexSet};
use iso8601_timestamp::Timestamp; use iso8601_timestamp::Timestamp;
use revolt_models::v0::{Embed, MessageAuthor, MessageSort, MessageWebhook, PushNotification}; use revolt_config::config;
use revolt_models::v0::{
self, DataMessageSend, Embed, MessageAuthor, MessageSort, MessageWebhook, PushNotification,
ReplyIntent, SendableEmbed, RE_MENTION,
};
use revolt_permissions::{ChannelPermission, PermissionValue};
use revolt_result::Result; use revolt_result::Result;
use ulid::Ulid; use ulid::Ulid;
use crate::{ use crate::{
events::client::EventV1, events::client::EventV1,
tasks::{self, ack::AckEvent}, tasks::{self, ack::AckEvent},
Channel, Database, File, util::idempotency::IdempotencyKey,
Channel, Database, Emoji, File,
}; };
auto_derived_partial!( auto_derived_partial!(
@@ -196,6 +204,162 @@ impl Default for Message {
#[allow(clippy::disallowed_methods)] #[allow(clippy::disallowed_methods)]
impl Message { impl Message {
/// Create message from API data
pub async fn create_from_api(
db: &Database,
channel: Channel,
data: DataMessageSend,
author: MessageAuthor<'_>,
mut idempotency: IdempotencyKey,
generate_embeds: bool,
) -> Result<Message> {
let config = config().await;
Message::validate_sum(
&data.content,
data.embeds.as_deref().unwrap_or_default(),
config.features.limits.default.message_length,
)?;
idempotency
.consume_nonce(data.nonce)
.await
.map_err(|_| create_error!(InvalidOperation))?;
// Check the message is not empty
if (data.content.as_ref().map_or(true, |v| v.is_empty()))
&& (data.attachments.as_ref().map_or(true, |v| v.is_empty()))
&& (data.embeds.as_ref().map_or(true, |v| v.is_empty()))
{
return Err(create_error!(EmptyMessage));
}
// Ensure restrict_reactions is not specified without reactions list
if let Some(interactions) = &data.interactions {
if interactions.restrict_reactions {
let disallowed = if let Some(list) = &interactions.reactions {
list.is_empty()
} else {
true
};
if disallowed {
return Err(create_error!(InvalidProperty));
}
}
}
let (author_id, webhook) = match &author {
MessageAuthor::User(user) => (user.id.clone(), None),
MessageAuthor::Webhook(webhook) => (webhook.id.clone(), Some((*webhook).clone())),
MessageAuthor::System { .. } => ("00000000000000000000000000".to_string(), None),
};
// Start constructing the message
let message_id = Ulid::new().to_string();
let mut message = Message {
id: message_id.clone(),
channel: channel.id(),
masquerade: data.masquerade.map(|masquerade| masquerade.into()),
interactions: data
.interactions
.map(|interactions| interactions.into())
.unwrap_or_default(),
author: author_id,
webhook: webhook.map(|w| w.into()),
..Default::default()
};
// Parse mentions in message.
let mut mentions = HashSet::new();
if let Some(content) = &data.content {
for capture in RE_MENTION.captures_iter(content) {
if let Some(mention) = capture.get(1) {
mentions.insert(mention.as_str().to_string());
}
}
}
// Verify replies are valid.
let mut replies = HashSet::new();
if let Some(entries) = data.replies {
if entries.len() > config.features.limits.default.message_replies {
return Err(create_error!(TooManyReplies {
max: config.features.limits.default.message_replies,
}));
}
for ReplyIntent { id, mention } in entries {
let message = db.fetch_message(&id).await?;
if mention {
mentions.insert(message.author.to_owned());
}
replies.insert(message.id);
}
}
if !mentions.is_empty() {
message.mentions.replace(mentions.into_iter().collect());
}
if !replies.is_empty() {
message
.replies
.replace(replies.into_iter().collect::<Vec<String>>());
}
// Add attachments to message.
let mut attachments = vec![];
if data
.attachments
.as_ref()
.is_some_and(|v| v.len() > config.features.limits.default.message_attachments)
{
return Err(create_error!(TooManyAttachments {
max: config.features.limits.default.message_attachments,
}));
}
if data
.embeds
.as_ref()
.is_some_and(|v| v.len() > config.features.limits.default.message_embeds)
{
return Err(create_error!(TooManyEmbeds {
max: config.features.limits.default.message_embeds,
}));
}
for attachment_id in data.attachments.as_deref().unwrap_or_default() {
attachments.push(
db.find_and_use_attachment(attachment_id, "attachments", "message", &message_id)
.await?,
);
}
if !attachments.is_empty() {
message.attachments.replace(attachments);
}
// Process included embeds.
for sendable_embed in data.embeds.unwrap_or_default() {
message.attach_sendable_embed(db, sendable_embed).await?;
}
// Set content
message.content = data.content;
// Pass-through nonce value for clients
message.nonce = Some(idempotency.into_key());
// Send the message
message.send(db, author, &channel, generate_embeds).await?;
Ok(message)
}
/// Send a message without any notifications /// Send a message without any notifications
pub async fn send_without_notifications( pub async fn send_without_notifications(
&mut self, &mut self,
@@ -293,6 +457,64 @@ impl Message {
Ok(()) Ok(())
} }
/// Convert sendable embed to text embed and attach to message
pub async fn attach_sendable_embed(
&mut self,
db: &Database,
embed: v0::SendableEmbed,
) -> Result<()> {
let media: Option<v0::File> = if let Some(id) = embed.media {
Some(
db.find_and_use_attachment(&id, "attachments", "message", &self.id)
.await?
.into(),
)
} else {
None
};
let embed = v0::Embed::Text(v0::Text {
icon_url: embed.icon_url,
url: embed.url,
title: embed.title,
description: embed.description,
media,
colour: embed.colour,
});
if let Some(embeds) = &mut self.embeds {
embeds.push(embed);
} else {
self.embeds = Some(vec![embed]);
}
Ok(())
}
/// Validate the sum of content of a message is under threshold
pub fn validate_sum(
content: &Option<String>,
embeds: &[SendableEmbed],
max_length: usize,
) -> Result<()> {
let mut running_total = 0;
if let Some(content) = content {
running_total += content.len();
}
for embed in embeds {
if let Some(desc) = &embed.description {
running_total += desc.len();
}
}
if running_total <= max_length {
Ok(())
} else {
Err(create_error!(PayloadTooLarge))
}
}
} }
impl SystemMessage { impl SystemMessage {
@@ -309,6 +531,27 @@ impl SystemMessage {
} }
impl Interactions { impl Interactions {
/// Validate interactions info is correct
pub async fn validate(&self, db: &Database, permissions: &PermissionValue) -> Result<()> {
let config = config().await;
if let Some(reactions) = &self.reactions {
permissions.throw_if_lacking_channel_permission(ChannelPermission::React)?;
if reactions.len() > config.features.limits.default.message_reactions {
return Err(create_error!(InvalidOperation));
}
for reaction in reactions {
if !Emoji::can_use(db, reaction).await? {
return Err(create_error!(InvalidOperation));
}
}
}
Ok(())
}
/// Check if we can use a given emoji to react /// Check if we can use a given emoji to react
pub fn can_use(&self, emoji: &str) -> bool { pub fn can_use(&self, emoji: &str) -> bool {
if self.restrict_reactions { if self.restrict_reactions {
@@ -60,7 +60,7 @@ impl AbstractMessages for ReferenceDb {
.cloned() .cloned()
.collect(); .collect();
// TODO: sorting, etc // TODO: sorting, etc (will be required for tests)
Ok(matched_messages) Ok(matched_messages)
@@ -1,28 +1,46 @@
use std::cmp::Ordering;
use std::time::Duration; use std::time::Duration;
use std::time::SystemTime;
use super::AbstractRatelimitEvents; use super::AbstractRatelimitEvents;
use crate::RatelimitEvent; use crate::RatelimitEvent;
use crate::RatelimitEventType; use crate::RatelimitEventType;
use crate::ReferenceDb; use crate::ReferenceDb;
use revolt_result::Result; use revolt_result::Result;
use ulid::Ulid;
#[async_trait] #[async_trait]
impl AbstractRatelimitEvents for ReferenceDb { impl AbstractRatelimitEvents for ReferenceDb {
/// Insert a new ratelimit event /// Insert a new ratelimit event
async fn insert_ratelimit_event(&self, _event: &RatelimitEvent) -> Result<()> { async fn insert_ratelimit_event(&self, event: &RatelimitEvent) -> Result<()> {
// TODO: implement let mut ratelimit_events = self.ratelimit_events.lock().await;
unimplemented!() if ratelimit_events.contains_key(&event.id) {
Err(create_database_error!("insert", "message"))
} else {
ratelimit_events.insert(event.id.to_string(), event.clone());
Ok(())
}
} }
/// Count number of events in given duration and check if we've hit the limit /// Count number of events in given duration and check if we've hit the limit
async fn has_ratelimited( async fn has_ratelimited(
&self, &self,
_target_id: &str, target_id: &str,
_event_type: RatelimitEventType, event_type: RatelimitEventType,
_period: Duration, period: Duration,
_count: usize, count: usize,
) -> Result<bool> { ) -> Result<bool> {
// TODO: implement let ratelimit_events = self.ratelimit_events.lock().await;
unimplemented!() let gte_cmp_id = Ulid::from_datetime(SystemTime::now() - period).to_string();
Ok(ratelimit_events
.iter()
.filter(|(id, event)| {
id.cmp(&&gte_cmp_id) == Ordering::Greater
&& event.target_id == target_id
&& event.event_type == event_type
})
.count()
>= count)
} }
} }
@@ -3,8 +3,8 @@ use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result}; use revolt_result::{create_error, Result};
use crate::{ use crate::{
events::client::EventV1, util::permissions::DatabasePermissionQuery, Database, File, Server, events::client::EventV1, util::permissions::DatabasePermissionQuery, Channel, Database, File,
SystemMessage, User, Server, SystemMessage, User,
}; };
auto_derived_partial!( auto_derived_partial!(
@@ -80,9 +80,8 @@ impl Member {
db: &Database, db: &Database,
server: &Server, server: &Server,
user: &User, user: &User,
// channels: Option<Vec<Channel>>, channels: Option<Vec<Channel>>,
//) -> Result<Vec<Channel>> { ) -> Result<Vec<Channel>> {
) -> Result<()> {
if db.fetch_ban(&server.id, &user.id).await.is_ok() { if db.fetch_ban(&server.id, &user.id).await.is_ok() {
return Err(create_error!(Banned)); return Err(create_error!(Banned));
} }
@@ -101,9 +100,10 @@ impl Member {
db.insert_member(&member).await?; db.insert_member(&member).await?;
let mut channels = vec![]; let should_fetch = channels.is_none();
let mut channels = channels.unwrap_or_default();
if true { if should_fetch {
let query = DatabasePermissionQuery::new(db, user).server(server); let query = DatabasePermissionQuery::new(db, user).server(server);
let existing_channels = db.fetch_channels(&server.channels).await?; let existing_channels = db.fetch_channels(&server.channels).await?;
@@ -152,8 +152,7 @@ impl Member {
.ok(); .ok();
} }
// Ok(channels) Ok(channels)
Ok(())
} }
/// Update member data /// Update member data
@@ -1,10 +1,11 @@
use std::collections::{HashMap, HashSet}; use std::collections::{HashMap, HashSet};
use revolt_permissions::OverrideField; use revolt_models::v0::{self, DataCreateServerChannel};
use revolt_permissions::{OverrideField, DEFAULT_PERMISSION_SERVER};
use revolt_result::Result; use revolt_result::Result;
use ulid::Ulid; use ulid::Ulid;
use crate::{events::client::EventV1, Database, File}; use crate::{events::client::EventV1, Channel, Database, File, User};
auto_derived_partial!( auto_derived_partial!(
/// Server /// Server
@@ -131,8 +132,51 @@ auto_derived!(
#[allow(clippy::disallowed_methods)] #[allow(clippy::disallowed_methods)]
impl Server { impl Server {
/// Create a server /// Create a server
pub async fn create(&self, db: &Database) -> Result<()> { pub async fn create(
db.insert_server(self).await db: &Database,
data: v0::DataCreateServer,
owner: &User,
create_default_channels: bool,
) -> Result<(Server, Vec<Channel>)> {
let mut server = Server {
id: ulid::Ulid::new().to_string(),
owner: owner.id.to_string(),
name: data.name,
description: data.description,
channels: vec![],
nsfw: data.nsfw.unwrap_or(false),
default_permissions: *DEFAULT_PERMISSION_SERVER as i64,
analytics: false,
banner: None,
categories: None,
discoverable: false,
flags: None,
icon: None,
roles: HashMap::new(),
system_messages: None,
};
let channels: Vec<Channel> = if create_default_channels {
vec![
Channel::create_server_channel(
db,
&mut server,
DataCreateServerChannel {
channel_type: v0::LegacyServerChannelType::Text,
name: "General".to_string(),
..Default::default()
},
false,
)
.await?,
]
} else {
vec![]
};
db.insert_server(&server).await?;
Ok((server, channels))
} }
/// Update server data /// Update server data
@@ -209,80 +253,7 @@ impl Server {
} }
} }
/* /// Create a new member in a server /*
pub async fn create_member(
&self,
db: &Database,
user: User,
channels: Option<Vec<Channel>>,
) -> Result<Vec<Channel>> {
if db.fetch_ban(&self.id, &user.id).await.is_ok() {
return Err(Error::Banned);
}
let member = Member {
id: MemberCompositeKey {
server: self.id.clone(),
user: user.id.clone(),
},
joined_at: Timestamp::now_utc(),
nickname: None,
avatar: None,
roles: vec![],
timeout: None,
};
db.insert_member(&member).await?;
let should_fetch = channels.is_none();
let mut channels = channels.unwrap_or_default();
if should_fetch {
let perm = perms(&user).server(self).member(&member);
let existing_channels = db.fetch_channels(&self.channels).await?;
for channel in existing_channels {
if perm
.clone()
.channel(&channel)
.has_permission(db, Permission::ViewChannel)
.await?
{
channels.push(channel);
}
}
}
/* // TODO: EventV1::ServerMemberJoin {
id: self.id.clone(),
user: user.id.clone(),
}
.p(self.id.clone())
.await;
EventV1::ServerCreate {
id: self.id.clone(),
server: self.clone(),
channels: channels.clone(),
}
.private(user.id.clone())
.await; */
if let Some(id) = self
.system_messages
.as_ref()
.and_then(|x| x.user_joined.as_ref())
{
SystemMessage::UserJoined {
id: user.id.clone(),
}
.into_message(id.to_string())
.create_no_web_push(db, id, false)
.await
.ok();
}
Ok(channels)
}
/// Remove a member from a server /// Remove a member from a server
pub async fn remove_member( pub async fn remove_member(
@@ -450,3 +421,36 @@ impl SystemMessageChannels {
ids ids
} }
} }
#[cfg(test)]
mod tests {
use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use crate::{fixture, util::permissions::DatabasePermissionQuery};
#[async_std::test]
async fn permissions() {
database_test!(|db| async move {
fixture!(db, "server_with_roles",
owner user 0
moderator user 1
user user 2
server server 4);
let mut query = DatabasePermissionQuery::new(&db, &owner).server(&server);
assert!(calculate_server_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::GrantAllSafe));
let mut query = DatabasePermissionQuery::new(&db, &moderator).server(&server);
assert!(calculate_server_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::BanMembers));
let mut query = DatabasePermissionQuery::new(&db, &user).server(&server);
assert!(!calculate_server_permissions(&mut query)
.await
.has_channel_permission(ChannelPermission::BanMembers));
});
}
}
@@ -1,6 +1,6 @@
use std::collections::HashMap; use std::collections::HashMap;
use crate::Database; use crate::{events::client::EventV1, Database};
use revolt_result::Result; use revolt_result::Result;
@@ -16,12 +16,12 @@ impl UserSettingsImpl for UserSettings {
async fn set(self, db: &Database, user: &str) -> Result<()> { async fn set(self, db: &Database, user: &str) -> Result<()> {
db.set_user_settings(user, &self).await?; db.set_user_settings(user, &self).await?;
/* // TODO: EventV1::UserSettingsUpdate { EventV1::UserSettingsUpdate {
id: user.to_string(), id: user.to_string(),
update: self, update: self,
} }
.private(user.to_string()) .private(user.to_string())
.await; */ .await;
Ok(()) Ok(())
} }
@@ -4,6 +4,7 @@ use crate::{events::client::EventV1, Database, File, RatelimitEvent};
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use rand::seq::SliceRandom; use rand::seq::SliceRandom;
use revolt_config::config;
use revolt_result::{create_error, Error, ErrorType, Result}; use revolt_result::{create_error, Error, ErrorType, Result};
use ulid::Ulid; use ulid::Ulid;
@@ -182,6 +183,28 @@ impl User {
Ok(user) Ok(user)
} }
/// Get the relationship with another user
pub fn relationship_with(&self, user_b: &str) -> RelationshipStatus {
if self.id == user_b {
return RelationshipStatus::User;
}
if let Some(relations) = &self.relations {
if let Some(relationship) = relations.iter().find(|x| x.id == user_b) {
return relationship.status.clone();
}
}
RelationshipStatus::None
}
pub fn is_friends_with(&self, user_b: &str) -> bool {
matches!(
self.relationship_with(user_b),
RelationshipStatus::Friend | RelationshipStatus::User
)
}
/// Check whether two users have a mutual connection /// Check whether two users have a mutual connection
/// ///
/// This will check if user and user_b share a server or a group. /// This will check if user and user_b share a server or a group.
@@ -196,6 +219,18 @@ impl User {
.is_empty()) .is_empty())
} }
/// Check if this user can acquire another server
pub async fn can_acquire_server(&self, db: &Database) -> Result<()> {
let config = config().await;
if db.fetch_server_count(&self.id).await? <= config.features.limits.default.servers {
Ok(())
} else {
Err(create_error!(TooManyServers {
max: config.features.limits.default.servers
}))
}
}
/// Sanitise and validate a username can be used /// Sanitise and validate a username can be used
pub fn validate_username(username: String) -> Result<String> { pub fn validate_username(username: String) -> Result<String> {
// Copy the username for validation // Copy the username for validation
@@ -326,6 +361,167 @@ impl User {
} }
} }
/// Set a relationship to another user
pub async fn set_relationship(
&mut self,
db: &Database,
user_b: &User,
status: RelationshipStatus,
) -> Result<()> {
db.set_relationship(&self.id, &user_b.id, &status).await?;
if let RelationshipStatus::None | RelationshipStatus::User = status {
if let Some(relations) = &mut self.relations {
relations.retain(|relation| relation.id != user_b.id);
}
} else {
let relation = Relationship {
id: user_b.id.to_string(),
status,
};
if let Some(relations) = &mut self.relations {
relations.retain(|relation| relation.id != user_b.id);
relations.push(relation);
} else {
self.relations = Some(vec![relation]);
}
}
Ok(())
}
/// Apply a certain relationship between two users
pub async fn apply_relationship(
&mut self,
db: &Database,
target: &mut User,
local: RelationshipStatus,
remote: RelationshipStatus,
) -> Result<()> {
target.set_relationship(db, self, remote).await?;
self.set_relationship(db, target, local).await?;
EventV1::UserRelationship {
id: target.id.clone(),
user: self.clone().into(db, Some(&*target)).await,
}
.private(target.id.clone())
.await;
EventV1::UserRelationship {
id: self.id.clone(),
user: target.clone().into(db, Some(&*self)).await,
}
.private(self.id.clone())
.await;
Ok(())
}
/// Add another user as a friend
pub async fn add_friend(&mut self, db: &Database, target: &mut User) -> Result<()> {
match self.relationship_with(&target.id) {
RelationshipStatus::User => Err(create_error!(NoEffect)),
RelationshipStatus::Friend => Err(create_error!(AlreadyFriends)),
RelationshipStatus::Outgoing => Err(create_error!(AlreadySentRequest)),
RelationshipStatus::Blocked => Err(create_error!(Blocked)),
RelationshipStatus::BlockedOther => Err(create_error!(BlockedByOther)),
RelationshipStatus::Incoming => {
self.apply_relationship(
db,
target,
RelationshipStatus::Friend,
RelationshipStatus::Friend,
)
.await
}
RelationshipStatus::None => {
self.apply_relationship(
db,
target,
RelationshipStatus::Outgoing,
RelationshipStatus::Incoming,
)
.await
}
}
}
/// Remove another user as a friend
pub async fn remove_friend(&mut self, db: &Database, target: &mut User) -> Result<()> {
match self.relationship_with(&target.id) {
RelationshipStatus::Friend
| RelationshipStatus::Outgoing
| RelationshipStatus::Incoming => {
self.apply_relationship(
db,
target,
RelationshipStatus::None,
RelationshipStatus::None,
)
.await
}
_ => Err(create_error!(NoEffect)),
}
}
/// Block another user
pub async fn block_user(&mut self, db: &Database, target: &mut User) -> Result<()> {
match self.relationship_with(&target.id) {
RelationshipStatus::User | RelationshipStatus::Blocked => Err(create_error!(NoEffect)),
RelationshipStatus::BlockedOther => {
self.apply_relationship(
db,
target,
RelationshipStatus::Blocked,
RelationshipStatus::Blocked,
)
.await
}
RelationshipStatus::None
| RelationshipStatus::Friend
| RelationshipStatus::Incoming
| RelationshipStatus::Outgoing => {
self.apply_relationship(
db,
target,
RelationshipStatus::Blocked,
RelationshipStatus::BlockedOther,
)
.await
}
}
}
/// Unblock another user
pub async fn unblock_user(&mut self, db: &Database, target: &mut User) -> Result<()> {
match self.relationship_with(&target.id) {
RelationshipStatus::Blocked => match target.relationship_with(&self.id) {
RelationshipStatus::Blocked => {
self.apply_relationship(
db,
target,
RelationshipStatus::BlockedOther,
RelationshipStatus::Blocked,
)
.await
}
RelationshipStatus::BlockedOther => {
self.apply_relationship(
db,
target,
RelationshipStatus::None,
RelationshipStatus::None,
)
.await
}
_ => Err(create_error!(InternalError)),
},
_ => Err(create_error!(NoEffect)),
}
}
/// Update user data /// Update user data
pub async fn update<'a>( pub async fn update<'a>(
&mut self, &mut self,
@@ -1,7 +1,7 @@
use revolt_result::Result; use revolt_result::Result;
use crate::ReferenceDb;
use crate::{FieldsUser, PartialUser, RelationshipStatus, User}; use crate::{FieldsUser, PartialUser, RelationshipStatus, User};
use crate::{ReferenceDb, Relationship};
use super::AbstractUsers; use super::AbstractUsers;
@@ -106,19 +106,49 @@ impl AbstractUsers for ReferenceDb {
/// Set relationship with another user /// Set relationship with another user
/// ///
/// This should use pull_relationship if relationship is None. /// This should use pull_relationship if relationship is None or User.
async fn set_relationship( async fn set_relationship(
&self, &self,
_user_id: &str, user_id: &str,
_target_id: &str, target_id: &str,
_relationship: &RelationshipStatus, relationship: &RelationshipStatus,
) -> Result<()> { ) -> Result<()> {
todo!() if let RelationshipStatus::User | RelationshipStatus::None = &relationship {
self.pull_relationship(user_id, target_id).await
} else {
let mut users = self.users.lock().await;
let user = users
.get_mut(user_id)
.ok_or_else(|| create_error!(NotFound))?;
let relation = Relationship {
id: target_id.to_string(),
status: relationship.clone(),
};
if let Some(relations) = &mut user.relations {
relations.retain(|relation| relation.id != target_id);
relations.push(relation);
} else {
user.relations = Some(vec![relation]);
}
Ok(())
}
} }
/// Remove relationship with another user /// Remove relationship with another user
async fn pull_relationship(&self, _user_id: &str, _target_id: &str) -> Result<()> { async fn pull_relationship(&self, user_id: &str, target_id: &str) -> Result<()> {
todo!() let mut users = self.users.lock().await;
let user = users
.get_mut(user_id)
.ok_or_else(|| create_error!(NotFound))?;
if let Some(relations) = &mut user.relations {
relations.retain(|relation| relation.id != target_id);
}
Ok(())
} }
/// Delete a user by their id /// Delete a user by their id
@@ -13,19 +13,19 @@ impl<'r> FromRequest<'r> for User {
.local_cache_async(async { .local_cache_async(async {
let db = request.rocket().state::<Database>().expect("`Database`"); let db = request.rocket().state::<Database>().expect("`Database`");
let _header_bot_token = request let header_bot_token = request
.headers() .headers()
.get("x-bot-token") .get("x-bot-token")
.next() .next()
.map(|x| x.to_string()); .map(|x| x.to_string());
/* if let Some(bot_token) = header_bot_token { if let Some(bot_token) = header_bot_token {
if let Ok(user) = User::from_token(db, &bot_token, UserHint::Bot).await { if let Ok(bot) = db.fetch_bot_by_token(&bot_token).await {
if let Ok(user) = db.fetch_user(&bot.id).await {
return Some(user); return Some(user);
} }
} else */ }
if let Outcome::Success(session) = request.guard::<Session>().await { } else if let Outcome::Success(session) = request.guard::<Session>().await {
// This uses a guard so can't really easily be refactored into from_token at this stage.
if let Ok(user) = db.fetch_user(&session.user_id).await { if let Ok(user) = db.fetch_user(&session.user_id).await {
return Some(user); return Some(user);
} }
+2 -2
View File
@@ -13,12 +13,12 @@ pub mod process_embeds;
pub mod web_push; pub mod web_push;
/// Spawn background workers /// Spawn background workers
pub async fn start_workers(db: Database) { pub async fn start_workers(db: Database, authifier_db: authifier::Database) {
for _ in 0..WORKER_COUNT { for _ in 0..WORKER_COUNT {
task::spawn(ack::worker(db.clone())); task::spawn(ack::worker(db.clone()));
task::spawn(last_message_id::worker(db.clone())); task::spawn(last_message_id::worker(db.clone()));
task::spawn(process_embeds::worker(db.clone())); task::spawn(process_embeds::worker(db.clone()));
task::spawn(web_push::worker(db.clone().into())); task::spawn(web_push::worker(authifier_db.clone()));
} }
} }
+95 -17
View File
@@ -1,4 +1,7 @@
use revolt_models::v0::*; use revolt_models::v0::*;
use revolt_permissions::{calculate_user_permissions, UserPermission};
use crate::{util::permissions::DatabasePermissionQuery, Database};
impl crate::Bot { impl crate::Bot {
pub fn into_public_bot(self, user: crate::User) -> PublicBot { pub fn into_public_bot(self, user: crate::User) -> PublicBot {
@@ -275,6 +278,15 @@ impl From<crate::EmojiParent> for EmojiParent {
} }
} }
impl From<EmojiParent> for crate::EmojiParent {
fn from(value: EmojiParent) -> Self {
match value {
EmojiParent::Detached => crate::EmojiParent::Detached,
EmojiParent::Server { id } => crate::EmojiParent::Server { id },
}
}
}
impl From<crate::File> for File { impl From<crate::File> for File {
fn from(value: crate::File) -> Self { fn from(value: crate::File) -> Self {
File { File {
@@ -329,7 +341,11 @@ impl From<crate::Message> for Message {
embeds: value.embeds, embeds: value.embeds,
mentions: value.mentions, mentions: value.mentions,
replies: value.replies, replies: value.replies,
reactions: value.reactions, reactions: value
.reactions
.into_iter()
.map(|(k, v)| (k, v.into_iter().collect()))
.collect(),
interactions: value.interactions.into(), interactions: value.interactions.into(),
masquerade: value.masquerade.map(|masq| masq.into()), masquerade: value.masquerade.map(|masq| masq.into()),
} }
@@ -353,7 +369,12 @@ impl From<crate::PartialMessage> for PartialMessage {
embeds: value.embeds, embeds: value.embeds,
mentions: value.mentions, mentions: value.mentions,
replies: value.replies, replies: value.replies,
reactions: value.reactions, reactions: value.reactions.map(|reactions| {
reactions
.into_iter()
.map(|(k, v)| (k, v.into_iter().collect()))
.collect()
}),
interactions: value.interactions.map(|interactions| interactions.into()), interactions: value.interactions.map(|interactions| interactions.into()),
masquerade: value.masquerade.map(|masq| masq.into()), masquerade: value.masquerade.map(|masq| masq.into()),
} }
@@ -385,7 +406,20 @@ impl From<crate::SystemMessage> for SystemMessage {
impl From<crate::Interactions> for Interactions { impl From<crate::Interactions> for Interactions {
fn from(value: crate::Interactions) -> Self { fn from(value: crate::Interactions) -> Self {
Interactions { Interactions {
reactions: value.reactions, reactions: value
.reactions
.map(|reactions| reactions.into_iter().collect()),
restrict_reactions: value.restrict_reactions,
}
}
}
impl From<Interactions> for crate::Interactions {
fn from(value: Interactions) -> Self {
crate::Interactions {
reactions: value
.reactions
.map(|reactions| reactions.into_iter().collect()),
restrict_reactions: value.restrict_reactions, restrict_reactions: value.restrict_reactions,
} }
} }
@@ -409,6 +443,16 @@ impl From<crate::Masquerade> for Masquerade {
} }
} }
impl From<Masquerade> for crate::Masquerade {
fn from(value: Masquerade) -> Self {
crate::Masquerade {
name: value.name,
avatar: value.avatar,
colour: value.colour,
}
}
}
impl From<crate::ServerBan> for ServerBan { impl From<crate::ServerBan> for ServerBan {
fn from(value: crate::ServerBan) -> Self { fn from(value: crate::ServerBan) -> Self {
ServerBan { ServerBan {
@@ -594,35 +638,69 @@ impl From<crate::FieldsRole> for FieldsRole {
} }
impl crate::User { impl crate::User {
pub async fn into<P>(self, perspective: P) -> User pub async fn into<'a, P>(self, db: &Database, perspective: P) -> User
where where
P: Into<Option<crate::User>>, P: Into<Option<&'a crate::User>>,
{ {
let relationship = if let Some(perspective) = perspective.into() { let perspective = perspective.into();
let (relationship, can_see_profile) = if self.bot.is_some() {
(RelationshipStatus::None, true)
} else if let Some(perspective) = perspective {
let mut query = DatabasePermissionQuery::new(db, perspective).user(&self);
if perspective.id == self.id {
(RelationshipStatus::User, true)
} else {
(
perspective perspective
.relations .relations
.unwrap_or_default() .as_ref()
.into_iter() .map(|relations| {
relations
.iter()
.find(|relationship| relationship.id == self.id) .find(|relationship| relationship.id == self.id)
.map(|relationship| relationship.status.into()) .map(|relationship| relationship.status.clone().into())
.unwrap_or_default() .unwrap_or_default()
})
.unwrap_or_default(),
calculate_user_permissions(&mut query)
.await
.has_user_permission(UserPermission::ViewProfile),
)
}
} else { } else {
RelationshipStatus::None (RelationshipStatus::None, false)
}; };
// do permission stuff here
// TODO: implement permissions =)
let can_see_profile = false;
User { User {
username: self.username, username: self.username,
discriminator: self.discriminator, discriminator: self.discriminator,
display_name: self.display_name, display_name: self.display_name,
avatar: self.avatar.map(|file| file.into()), avatar: self.avatar.map(|file| file.into()),
relations: vec![], relations: if let Some(crate::User { id, .. }) = perspective {
if id == &self.id {
self.relations
.unwrap_or_default()
.into_iter()
.map(|relation| relation.into())
.collect()
} else {
vec![]
}
} else {
vec![]
},
badges: self.badges.unwrap_or_default() as u32, badges: self.badges.unwrap_or_default() as u32,
status: None, status: if can_see_profile {
profile: None, self.status.map(|status| status.into())
} else {
None
},
profile: if can_see_profile {
self.profile.map(|profile| profile.into())
} else {
None
},
flags: self.flags.unwrap_or_default() as u32, flags: self.flags.unwrap_or_default() as u32,
privileged: self.privileged, privileged: self.privileged,
bot: self.bot.map(|bot| bot.into()), bot: self.bot.map(|bot| bot.into()),
+22 -7
View File
@@ -1,15 +1,12 @@
use std::num::NonZeroUsize; use std::num::NonZeroUsize;
use revolt_result::{create_error, Error, Result}; use revolt_result::{create_error, Result};
#[cfg(feature = "rocket-impl")]
use revolt_result::Error;
use async_std::sync::Mutex; use async_std::sync::Mutex;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use revolt_rocket_okapi::gen::OpenApiGenerator;
use revolt_rocket_okapi::request::{OpenApiFromRequest, RequestHeaderInput};
use revolt_rocket_okapi::revolt_okapi::openapi3::{Parameter, ParameterValue};
use rocket::http::Status;
use rocket::request::{FromRequest, Outcome};
use schemars::schema::{InstanceType, SchemaObject, SingleOrVec};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
@@ -42,6 +39,17 @@ impl IdempotencyKey {
} }
} }
#[cfg(feature = "rocket-impl")]
use revolt_rocket_okapi::{
gen::OpenApiGenerator,
request::{OpenApiFromRequest, RequestHeaderInput},
revolt_okapi::openapi3::{Parameter, ParameterValue},
};
#[cfg(feature = "rocket-impl")]
use schemars::schema::{InstanceType, SchemaObject, SingleOrVec};
#[cfg(feature = "rocket-impl")]
impl<'r> OpenApiFromRequest<'r> for IdempotencyKey { impl<'r> OpenApiFromRequest<'r> for IdempotencyKey {
fn from_request_input( fn from_request_input(
_gen: &mut OpenApiGenerator, _gen: &mut OpenApiGenerator,
@@ -71,6 +79,13 @@ impl<'r> OpenApiFromRequest<'r> for IdempotencyKey {
} }
} }
#[cfg(feature = "rocket-impl")]
use rocket::{
http::Status,
request::{FromRequest, Outcome},
};
#[cfg(feature = "rocket-impl")]
#[async_trait] #[async_trait]
impl<'r> FromRequest<'r> for IdempotencyKey { impl<'r> FromRequest<'r> for IdempotencyKey {
type Error = Error; type Error = Error;
+3
View File
@@ -2,3 +2,6 @@ pub mod bridge;
pub mod idempotency; pub mod idempotency;
pub mod permissions; pub mod permissions;
pub mod reference; pub mod reference;
#[cfg(test)]
pub mod test_fixtures;
+27 -12
View File
@@ -1,7 +1,8 @@
use std::borrow::Cow; use std::borrow::Cow;
use revolt_permissions::{ use revolt_permissions::{
calculate_user_permissions, ChannelType, Override, PermissionQuery, RelationshipStatus, calculate_user_permissions, ChannelType, Override, PermissionQuery, PermissionValue,
RelationshipStatus,
}; };
use crate::{Channel, Database, Member, Server, User}; use crate::{Channel, Database, Member, Server, User};
@@ -19,7 +20,8 @@ pub struct DatabasePermissionQuery<'a> {
member: Option<Cow<'a, Member>>, member: Option<Cow<'a, Member>>,
// flag_known_relationship: Option<&'a RelationshipStatus>, // flag_known_relationship: Option<&'a RelationshipStatus>,
cached_user_permission: Option<u32>, cached_user_permission: Option<PermissionValue>,
cached_mutual_connection: Option<bool>,
cached_permission: Option<u64>, cached_permission: Option<u64>,
} }
@@ -49,6 +51,10 @@ impl PermissionQuery for DatabasePermissionQuery<'_> {
/// Get the relationship with have with the currently selected user /// Get the relationship with have with the currently selected user
async fn user_relationship(&mut self) -> RelationshipStatus { async fn user_relationship(&mut self) -> RelationshipStatus {
if let Some(other_user) = &self.user { if let Some(other_user) = &self.user {
if self.perspective.id == other_user.id {
return RelationshipStatus::User;
}
if let Some(relations) = &self.perspective.relations { if let Some(relations) = &self.perspective.relations {
for entry in relations { for entry in relations {
if entry.id == other_user.id { if entry.id == other_user.id {
@@ -82,14 +88,17 @@ impl PermissionQuery for DatabasePermissionQuery<'_> {
/// Do we have a mutual connection with the currently selected user? /// Do we have a mutual connection with the currently selected user?
async fn have_mutual_connection(&mut self) -> bool { async fn have_mutual_connection(&mut self) -> bool {
if let Some(user) = &self.user { if let Some(value) = self.cached_mutual_connection {
// TODO: cache result? value
matches!( } else if let Some(user) = &self.user {
self.perspective let value = self
.perspective
.has_mutual_connection(self.database, &user.id) .has_mutual_connection(self.database, &user.id)
.await, .await
Ok(true) .unwrap_or_default();
)
self.cached_mutual_connection = Some(value);
matches!(value, true)
} else { } else {
false false
} }
@@ -111,11 +120,15 @@ impl PermissionQuery for DatabasePermissionQuery<'_> {
if let Some(server) = &self.server { if let Some(server) = &self.server {
if self.member.is_some() { if self.member.is_some() {
true true
} else { } else if let Ok(member) = self
self.database .database
.fetch_member(&server.id, &self.perspective.id) .fetch_member(&server.id, &self.perspective.id)
.await .await
.is_ok() {
self.member = Some(Cow::Owned(member));
true
} else {
false
} }
} else { } else {
false false
@@ -331,6 +344,7 @@ impl PermissionQuery for DatabasePermissionQuery<'_> {
| Cow::Owned(Channel::TextChannel { server, .. }) | Cow::Owned(Channel::TextChannel { server, .. })
| Cow::Borrowed(Channel::VoiceChannel { server, .. }) | Cow::Borrowed(Channel::VoiceChannel { server, .. })
| Cow::Owned(Channel::VoiceChannel { server, .. }) => { | Cow::Owned(Channel::VoiceChannel { server, .. }) => {
// FIXME: may double fetch
if let Ok(server) = self.database.fetch_server(server).await { if let Ok(server) = self.database.fetch_server(server).await {
self.server.replace(Cow::Owned(server)); self.server.replace(Cow::Owned(server));
} }
@@ -352,6 +366,7 @@ impl<'a> DatabasePermissionQuery<'a> {
server: None, server: None,
member: None, member: None,
cached_mutual_connection: None,
cached_user_permission: None, cached_user_permission: None,
cached_permission: None, cached_permission: None,
} }
+26 -1
View File
@@ -7,7 +7,7 @@ use schemars::{
JsonSchema, JsonSchema,
}; };
use crate::{Bot, Database, Webhook}; use crate::{Bot, Channel, Database, Emoji, Message, Server, User, Webhook};
/// Reference to some object in the database /// Reference to some object in the database
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
@@ -27,6 +27,31 @@ impl Reference {
db.fetch_bot(&self.id).await db.fetch_bot(&self.id).await
} }
/// Fetch emoji from Ref
pub async fn as_emoji(&self, db: &Database) -> Result<Emoji> {
db.fetch_emoji(&self.id).await
}
/// Fetch channel from Ref
pub async fn as_channel(&self, db: &Database) -> Result<Channel> {
db.fetch_channel(&self.id).await
}
/// Fetch message from Ref
pub async fn as_message(&self, db: &Database) -> Result<Message> {
db.fetch_message(&self.id).await
}
/// Fetch server from Ref
pub async fn as_server(&self, db: &Database) -> Result<Server> {
db.fetch_server(&self.id).await
}
/// Fetch user from Ref
pub async fn as_user(&self, db: &Database) -> Result<User> {
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
@@ -0,0 +1,103 @@
use std::collections::HashMap;
use once_cell::sync::Lazy;
use regex::{Captures, Regex};
use serde_json::from_str;
use crate::{Channel, Database, Member, Server, User};
static RE_ID: Lazy<Regex> = Lazy::new(|| Regex::new("__ID:(\\d+)__").unwrap());
#[derive(Debug, Deserialize)]
#[serde(tag = "_object_type")]
enum LoadedFixture {
User(User),
Channel(Channel),
Server(Server),
ServerMember(Member),
}
pub async fn load_fixture(db: &Database, input: &str) -> HashMap<String, String> {
let mut ids = HashMap::<String, String>::new();
let input = RE_ID.replace_all(input, |cap: &Captures| {
let d = cap.get(1).unwrap().as_str();
if !ids.contains_key(d) {
ids.insert(d.to_string(), ulid::Ulid::new().to_string());
}
ids.get(d).unwrap().clone()
});
// Deserialise the fixtures
let items: Vec<LoadedFixture> = from_str(&input).expect("Failed to deserialise fixture");
// Load all of the items within
for item in items {
#[allow(clippy::disallowed_methods)]
match item {
LoadedFixture::User(user) => db.insert_user(&user).await.unwrap(),
LoadedFixture::Channel(channel) => db.insert_channel(&channel).await.unwrap(),
LoadedFixture::Server(server) => db.insert_server(&server).await.unwrap(),
LoadedFixture::ServerMember(member) => db.insert_member(&member).await.unwrap(),
}
}
// Return IDs for ease of use
ids
}
#[async_trait]
pub trait FetchFixture {
async fn user(&self, db: &Database, d: usize) -> User;
async fn channel(&self, db: &Database, d: usize) -> Channel;
async fn server(&self, db: &Database, d: usize) -> Server;
async fn member(&self, db: &Database, d_server: usize, d_user: usize) -> Member;
}
#[async_trait]
impl FetchFixture for HashMap<String, String> {
async fn user(&self, db: &Database, d: usize) -> User {
db.fetch_user(self.get(&d.to_string()).unwrap())
.await
.unwrap()
}
async fn channel(&self, db: &Database, d: usize) -> Channel {
db.fetch_channel(self.get(&d.to_string()).unwrap())
.await
.unwrap()
}
async fn server(&self, db: &Database, d: usize) -> Server {
db.fetch_server(self.get(&d.to_string()).unwrap())
.await
.unwrap()
}
async fn member(&self, db: &Database, d_server: usize, d_user: usize) -> Member {
db.fetch_member(
self.get(&d_server.to_string()).unwrap(),
self.get(&d_user.to_string()).unwrap(),
)
.await
.unwrap()
}
}
#[macro_export]
macro_rules! fixture {
( $database:expr, $name:expr, $( $variable:ident $type:ident $id: expr )+ ) => {
use $crate::util::test_fixtures::FetchFixture;
let fixtures = $crate::util::test_fixtures::load_fixture(
&$database,
include_str!(concat!("../../../fixtures/", $name, ".json")),
)
.await;
$(
let $variable = fixtures.$type(&$database, $id).await;
)+
};
}
+4 -4
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-models" name = "revolt-models"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
@@ -9,7 +9,7 @@ description = "Revolt Backend: API Models"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features] [features]
serde = ["dep:serde", "revolt-permissions/serde"] serde = ["dep:serde", "revolt-permissions/serde", "indexmap/serde"]
schemas = ["dep:schemars", "revolt-permissions/schemas"] schemas = ["dep:schemars", "revolt-permissions/schemas"]
validator = ["dep:validator"] validator = ["dep:validator"]
partials = ["dep:revolt_optional_struct", "serde", "schemas"] partials = ["dep:revolt_optional_struct", "serde", "schemas"]
@@ -18,8 +18,8 @@ default = ["serde", "partials"]
[dependencies] [dependencies]
# Core # Core
revolt-config = { version = "0.6.7", path = "../config" } revolt-config = { version = "0.6.9", path = "../config" }
revolt-permissions = { version = "0.6.7", path = "../permissions" } revolt-permissions = { version = "0.6.9", path = "../permissions" }
# Utility # Utility
regex = "1" regex = "1"
+2 -4
View File
@@ -1,7 +1,5 @@
use super::User; use super::User;
use validator::Validate;
auto_derived!( auto_derived!(
/// Bot /// Bot
#[derive(Default)] #[derive(Default)]
@@ -103,7 +101,7 @@ auto_derived!(
/// Bot Details /// Bot Details
#[derive(Default)] #[derive(Default)]
#[cfg_attr(feature = "validator", derive(Validate))] #[cfg_attr(feature = "validator", derive(validator::Validate))]
pub struct DataCreateBot { pub struct DataCreateBot {
/// Bot username /// Bot username
#[cfg_attr( #[cfg_attr(
@@ -115,7 +113,7 @@ auto_derived!(
/// New Bot Details /// New Bot Details
#[derive(Default)] #[derive(Default)]
#[cfg_attr(feature = "validator", derive(Validate))] #[cfg_attr(feature = "validator", derive(validator::Validate))]
pub struct DataEditBot { pub struct DataEditBot {
/// Bot username /// Bot username
#[cfg_attr( #[cfg_attr(
+15 -2
View File
@@ -1,3 +1,6 @@
#[cfg(feature = "validator")]
use validator::Validate;
use super::File; use super::File;
auto_derived_partial!( auto_derived_partial!(
@@ -69,13 +72,23 @@ auto_derived!(
pub channel_id: String, pub channel_id: String,
/// The permissions for the webhook /// The permissions for the webhook
pub permissions: u64 pub permissions: u64,
} }
/// Optional fields on webhook object /// Optional fields on webhook object
pub enum FieldsWebhook { pub enum FieldsWebhook {
Avatar, Avatar,
} }
/// Information for the webhook
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct CreateWebhookBody {
#[validate(length(min = 1, max = 32))]
pub name: String,
#[validate(length(min = 1, max = 128))]
pub avatar: Option<String>,
}
); );
impl From<Webhook> for MessageWebhook { impl From<Webhook> for MessageWebhook {
@@ -94,7 +107,7 @@ impl From<Webhook> for ResponseWebhook {
name: value.name, name: value.name,
avatar: value.avatar.map(|file| file.id), avatar: value.avatar.map(|file| file.id),
channel_id: value.channel_id, channel_id: value.channel_id,
permissions: value.permissions permissions: value.permissions,
} }
} }
} }
+67 -1
View File
@@ -1,10 +1,11 @@
use super::File; use super::File;
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
use std::collections::HashMap; use std::collections::{HashMap, HashSet};
auto_derived!( auto_derived!(
/// Channel /// Channel
#[serde(tag = "channel_type")]
pub enum Channel { pub enum Channel {
/// Personal "Saved Notes" channel which allows users to save messages /// Personal "Saved Notes" channel which allows users to save messages
SavedMessages { SavedMessages {
@@ -205,4 +206,69 @@ auto_derived!(
#[cfg_attr(feature = "serde", serde(default))] #[cfg_attr(feature = "serde", serde(default))]
pub remove: Option<Vec<FieldsChannel>>, pub remove: Option<Vec<FieldsChannel>>,
} }
/// Create new group
#[derive(Default)]
#[cfg_attr(feature = "validator", derive(validator::Validate))]
pub struct DataCreateGroup {
/// Group name
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 32)))]
pub name: String,
/// Group description
#[cfg_attr(feature = "validator", validate(length(min = 0, max = 1024)))]
pub description: Option<String>,
/// Group icon
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 128)))]
pub icon: Option<String>,
/// Array of user IDs to add to the group
///
/// Must be friends with these users.
#[cfg_attr(feature = "validator", validate(length(min = 0, max = 49)))]
#[serde(default)]
pub users: HashSet<String>,
/// Whether this group is age-restricted
#[serde(skip_serializing_if = "Option::is_none")]
pub nsfw: Option<bool>,
}
/// Server Channel Type
#[derive(Default)]
pub enum LegacyServerChannelType {
/// Text Channel
#[default]
Text,
/// Voice Channel
Voice,
}
/// Create new server channel
#[derive(Default)]
#[cfg_attr(feature = "validator", derive(validator::Validate))]
pub struct DataCreateServerChannel {
/// Channel type
#[serde(rename = "type", default = "LegacyServerChannelType::default")]
pub channel_type: LegacyServerChannelType,
/// Channel name
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 32)))]
pub name: String,
/// Channel description
#[cfg_attr(feature = "validator", validate(length(min = 0, max = 1024)))]
pub description: Option<String>,
/// Whether this channel is age restricted
#[serde(skip_serializing_if = "Option::is_none")]
pub nsfw: Option<bool>,
}
); );
impl Channel {
/// Get a reference to this channel's id
pub fn id(&self) -> &str {
match self {
Channel::DirectMessage { id, .. }
| Channel::Group { id, .. }
| Channel::SavedMessages { id, .. }
| Channel::TextChannel { id, .. }
| Channel::VoiceChannel { id, .. } => id,
}
}
}
+24
View File
@@ -1,3 +1,14 @@
use once_cell::sync::Lazy;
use regex::Regex;
#[cfg(feature = "validator")]
use validator::Validate;
/// Regex for valid emoji names
///
/// Alphanumeric and underscores
pub static RE_EMOJI: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[a-z0-9_]+$").unwrap());
auto_derived!( auto_derived!(
/// Emoji /// Emoji
pub struct Emoji { pub struct Emoji {
@@ -30,4 +41,17 @@ auto_derived!(
Server { id: String }, Server { id: String },
Detached, Detached,
} }
/// Create a new emoji
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct DataCreateEmoji {
/// Server name
#[validate(length(min = 1, max = 32), regex = "RE_EMOJI")]
pub name: String,
/// Parent information
pub parent: EmojiParent,
/// Whether the emoji is mature
#[serde(default)]
pub nsfw: bool,
}
); );
+71 -6
View File
@@ -1,11 +1,21 @@
use std::time::SystemTime; use std::{
collections::{HashMap, HashSet},
time::SystemTime,
};
use once_cell::sync::Lazy;
use regex::Regex;
use revolt_config::config; use revolt_config::config;
use indexmap::{IndexMap, IndexSet}; #[cfg(feature = "validator")]
use validator::Validate;
use iso8601_timestamp::Timestamp; use iso8601_timestamp::Timestamp;
use super::{Embed, File, MessageWebhook, User, Webhook}; use super::{Embed, File, MessageWebhook, User, Webhook, RE_COLOUR};
pub static RE_MENTION: Lazy<Regex> =
Lazy::new(|| Regex::new(r"<@([0-9A-HJKMNP-TV-Z]{26})>").unwrap());
auto_derived_partial!( auto_derived_partial!(
/// Message /// Message
@@ -45,8 +55,8 @@ auto_derived_partial!(
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
pub replies: Option<Vec<String>>, pub replies: Option<Vec<String>>,
/// Hashmap of emoji IDs to array of user IDs /// Hashmap of emoji IDs to array of user IDs
#[serde(skip_serializing_if = "IndexMap::is_empty", default)] #[serde(skip_serializing_if = "HashMap::is_empty", default)]
pub reactions: IndexMap<String, IndexSet<String>>, pub reactions: HashMap<String, HashSet<String>>,
/// Information about how this message should be interacted with /// Information about how this message should be interacted with
#[serde(skip_serializing_if = "Interactions::is_default", default)] #[serde(skip_serializing_if = "Interactions::is_default", default)]
pub interactions: Interactions, pub interactions: Interactions,
@@ -86,7 +96,9 @@ auto_derived!(
} }
/// Name and / or avatar override information /// Name and / or avatar override information
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct Masquerade { pub struct Masquerade {
// FIXME: missing validation
/// Replace the display name shown on this message /// Replace the display name shown on this message
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>, pub name: Option<String>,
@@ -105,7 +117,7 @@ auto_derived!(
pub struct Interactions { pub struct Interactions {
/// Reactions which should always appear and be distinct /// Reactions which should always appear and be distinct
#[serde(skip_serializing_if = "Option::is_none", default)] #[serde(skip_serializing_if = "Option::is_none", default)]
pub reactions: Option<IndexSet<String>>, pub reactions: Option<HashSet<String>>,
/// Whether reactions should be restricted to the given list /// Whether reactions should be restricted to the given list
/// ///
/// Can only be set to true if reactions list is of at least length 1 /// Can only be set to true if reactions list is of at least length 1
@@ -152,6 +164,59 @@ auto_derived!(
/// URL to open when clicking notification /// URL to open when clicking notification
pub url: String, pub url: String,
} }
/// Representation of a text embed before it is sent.
#[derive(Default)]
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct SendableEmbed {
#[validate(length(min = 1, max = 128))]
pub icon_url: Option<String>,
#[validate(length(min = 1, max = 256))]
pub url: Option<String>,
#[validate(length(min = 1, max = 100))]
pub title: Option<String>,
#[validate(length(min = 1, max = 2000))]
pub description: Option<String>,
pub media: Option<String>,
#[validate(length(min = 1, max = 128), regex = "RE_COLOUR")]
pub colour: Option<String>,
}
/// What this message should reply to and how
pub struct ReplyIntent {
/// Message Id
pub id: String,
/// Whether this reply should mention the message's author
pub mention: bool,
}
/// Message to send
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct DataMessageSend {
/// Unique token to prevent duplicate message sending
///
/// **This is deprecated and replaced by `Idempotency-Key`!**
#[validate(length(min = 1, max = 64))]
pub nonce: Option<String>,
/// Message content to send
#[validate(length(min = 0, max = 2000))]
pub content: Option<String>,
/// Attachments to include in message
pub attachments: Option<Vec<String>>,
/// Messages to reply to
pub replies: Option<Vec<ReplyIntent>>,
/// Embeds to include in message
///
/// Text embed content contributes to the content length cap
#[validate]
pub embeds: Option<Vec<SendableEmbed>>,
/// Masquerade to apply to this message
#[validate]
pub masquerade: Option<Masquerade>,
/// Information about how this message should be interacted with
pub interactions: Option<Interactions>,
}
); );
/// Message Author Abstraction /// Message Author Abstraction
@@ -1,6 +1,27 @@
use super::File; use super::File;
use iso8601_timestamp::Timestamp; use iso8601_timestamp::Timestamp;
use once_cell::sync::Lazy;
use regex::Regex;
/// Regex for valid role colours
///
/// Allows the use of named colours, rgb(a), variables and all gradients.
///
/// Flags:
/// - Case-insensitive (`i`)
///
/// Source:
/// ```regex
/// VALUE = [a-z ]+|var\(--[a-z\d-]+\)|rgba?\([\d, ]+\)|#[a-f0-9]+
/// ADDITIONAL_VALUE = \d+deg
/// STOP = ([ ]+(\d{1,3}%|0))?
///
/// ^(?:VALUE|(repeating-)?(linear|conic|radial)-gradient\((VALUE|ADDITIONAL_VALUE)STOP(,[ ]*(VALUE)STOP)+\))$
/// ```
pub static RE_COLOUR: Lazy<Regex> = Lazy::new(|| {
Regex::new(r"(?i)^(?:[a-z ]+|var\(--[a-z\d-]+\)|rgba?\([\d, ]+\)|#[a-f0-9]+|(repeating-)?(linear|conic|radial)-gradient\(([a-z ]+|var\(--[a-z\d-]+\)|rgba?\([\d, ]+\)|#[a-f0-9]+|\d+deg)([ ]+(\d{1,3}%|0))?(,[ ]*([a-z ]+|var\(--[a-z\d-]+\)|rgba?\([\d, ]+\)|#[a-f0-9]+)([ ]+(\d{1,3}%|0))?)+\))$").unwrap()
});
auto_derived_partial!( auto_derived_partial!(
/// Server Member /// Server Member
+27 -1
View File
@@ -1,8 +1,11 @@
use super::File; use super::{Channel, File};
use revolt_permissions::OverrideField; use revolt_permissions::OverrideField;
use std::collections::HashMap; use std::collections::HashMap;
#[cfg(feature = "validator")]
use validator::Validate;
auto_derived_partial!( auto_derived_partial!(
/// Server /// Server
pub struct Server { pub struct Server {
@@ -141,4 +144,27 @@ auto_derived!(
#[cfg_attr(feature = "serde", serde(skip_serializing_if = "Option::is_none"))] #[cfg_attr(feature = "serde", serde(skip_serializing_if = "Option::is_none"))]
pub user_banned: Option<String>, pub user_banned: Option<String>,
} }
/// Information about new server to create
#[derive(Default)]
#[cfg_attr(feature = "validator", derive(Validate))]
pub struct DataCreateServer {
/// Server name
#[cfg_attr(feature = "validator", validate(length(min = 1, max = 32)))]
pub name: String,
/// Server description
#[cfg_attr(feature = "validator", validate(length(min = 0, max = 1024)))]
pub description: Option<String>,
/// Whether this server is age-restricted
#[cfg_attr(feature = "serde", serde(skip_serializing_if = "Option::is_none"))]
pub nsfw: Option<bool>,
}
/// Information returned when creating server
pub struct CreateServerLegacyResponse {
/// Server object
pub server: Server,
/// Default channels
pub channels: Vec<Channel>,
}
); );
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-permissions" name = "revolt-permissions"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
+15 -15
View File
@@ -5,37 +5,37 @@ use crate::{
}; };
/// Calculate permissions against a user /// Calculate permissions against a user
pub async fn calculate_user_permissions<P: PermissionQuery>(query: &mut P) -> u32 { pub async fn calculate_user_permissions<P: PermissionQuery>(query: &mut P) -> PermissionValue {
if query.are_we_privileged().await { if query.are_we_privileged().await {
return u32::MAX; return u64::MAX.into();
} }
if query.are_the_users_same().await { if query.are_the_users_same().await {
return u32::MAX; return u64::MAX.into();
} }
let mut permissions = 0_u32; let mut permissions = 0_u64;
match query.user_relationship().await { match query.user_relationship().await {
RelationshipStatus::Friend => return u32::MAX, RelationshipStatus::Friend => return u64::MAX.into(),
RelationshipStatus::Blocked | RelationshipStatus::BlockedOther => { RelationshipStatus::Blocked | RelationshipStatus::BlockedOther => {
return UserPermission::Access as u32 return (UserPermission::Access as u64).into()
} }
RelationshipStatus::Incoming | RelationshipStatus::Outgoing => { RelationshipStatus::Incoming | RelationshipStatus::Outgoing => {
permissions = UserPermission::Access as u32; permissions = UserPermission::Access as u64;
} }
_ => {} _ => {}
} }
if query.have_mutual_connection().await { if query.have_mutual_connection().await {
permissions = UserPermission::Access + UserPermission::ViewProfile; permissions = UserPermission::Access as u64 + UserPermission::ViewProfile as u64;
if query.user_is_bot().await || query.are_we_a_bot().await { if query.user_is_bot().await || query.are_we_a_bot().await {
permissions += UserPermission::SendMessage as u32; permissions += UserPermission::SendMessage as u64;
} }
permissions permissions.into()
} else { } else {
permissions permissions.into()
} }
} }
@@ -81,9 +81,7 @@ pub async fn calculate_channel_permissions<P: PermissionQuery>(query: &mut P) ->
query.set_recipient_as_user().await; query.set_recipient_as_user().await;
let permissions = calculate_user_permissions(query).await; let permissions = calculate_user_permissions(query).await;
if (permissions & UserPermission::SendMessage as u32) if permissions.has_user_permission(UserPermission::SendMessage) {
== UserPermission::SendMessage as u32
{
(*DEFAULT_PERMISSION_DIRECT_MESSAGE).into() (*DEFAULT_PERMISSION_DIRECT_MESSAGE).into()
} else { } else {
(*DEFAULT_PERMISSION_VIEW_ONLY).into() (*DEFAULT_PERMISSION_VIEW_ONLY).into()
@@ -106,7 +104,9 @@ pub async fn calculate_channel_permissions<P: PermissionQuery>(query: &mut P) ->
ChannelType::ServerChannel => { ChannelType::ServerChannel => {
query.set_server_from_channel().await; query.set_server_from_channel().await;
if query.are_we_a_member().await { if query.are_we_server_owner().await {
return ChannelPermission::GrantAllSafe.into();
} else if query.are_we_a_member().await {
let mut permissions = calculate_server_permissions(query).await; let mut permissions = calculate_server_permissions(query).await;
permissions.apply(query.get_default_channel_permissions().await); permissions.apply(query.get_default_channel_permissions().await);
+9 -7
View File
@@ -8,7 +8,7 @@ pub use server::*;
pub use user::*; pub use user::*;
/// Holds a permission value to manipulate. /// Holds a permission value to manipulate.
#[derive(Debug)] #[derive(Clone, Debug)]
pub struct PermissionValue(u64); pub struct PermissionValue(u64);
impl PermissionValue { impl PermissionValue {
@@ -39,20 +39,22 @@ impl PermissionValue {
} }
/// Check whether certain a permission has been granted /// Check whether certain a permission has been granted
pub fn has(&mut self, v: u64) -> bool { pub fn has(&self, v: u64) -> bool {
(self.0 & v) == v (self.0 & v) == v
} }
/// Check whether certain a user permission has been granted
pub fn has_user_permission(&self, permission: UserPermission) -> bool {
self.has(permission as u64)
}
/// Check whether certain a channel permission has been granted /// Check whether certain a channel permission has been granted
pub fn has_channel_permission(&mut self, permission: ChannelPermission) -> bool { pub fn has_channel_permission(&self, permission: ChannelPermission) -> bool {
self.has(permission as u64) self.has(permission as u64)
} }
/// Throw if missing channel permission /// Throw if missing channel permission
pub fn throw_if_lacking_channel_permission( pub fn throw_if_lacking_channel_permission(&self, permission: ChannelPermission) -> Result<()> {
&mut self,
permission: ChannelPermission,
) -> Result<()> {
if self.has_channel_permission(permission) { if self.has_channel_permission(permission) {
Ok(()) Ok(())
} else { } else {
+2 -2
View File
@@ -51,9 +51,9 @@ pub enum DataPermissionPoly {
#[cfg_attr(feature = "schemas", derive(JsonSchema))] #[cfg_attr(feature = "schemas", derive(JsonSchema))]
pub struct OverrideField { pub struct OverrideField {
/// Allow bit flags /// Allow bit flags
a: i64, pub a: i64,
/// Disallow bit flags /// Disallow bit flags
d: i64, pub d: i64,
} }
impl Override { impl Override {
+1 -1
View File
@@ -12,7 +12,7 @@ async fn validate_user_permissions() {
let mut query = Scenario {}; let mut query = Scenario {};
let perms = calculate_user_permissions(&mut query).await; let perms = calculate_user_permissions(&mut query).await;
assert_eq!(perms, u32::MAX); assert!(perms.has(u64::MAX));
let perms = calculate_channel_permissions(&mut query).await; let perms = calculate_channel_permissions(&mut query).await;
let value: u64 = perms.into(); let value: u64 = perms.into();
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-presence" name = "revolt-presence"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
-1
View File
@@ -221,7 +221,6 @@ mod tests {
// Create a few more sessions // Create a few more sessions
let (first_session, second_session_id) = create_session(&user_id, 0).await; let (first_session, second_session_id) = create_session(&user_id, 0).await;
assert!(!first_session); assert!(!first_session);
dbg!(second_session_id);
assert_eq!(second_session_id as u8 & 1, 0); assert_eq!(second_session_id as u8 & 1, 0);
let (first_session, other_session_id) = create_session(&other_id, 0).await; let (first_session, other_session_id) = create_session(&other_id, 0).await;
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-result" name = "revolt-result"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <me@insrt.uk>"] authors = ["Paul Makles <me@insrt.uk>"]
+3
View File
@@ -60,6 +60,9 @@ pub enum ErrorType {
TooManyAttachments { TooManyAttachments {
max: usize, max: usize,
}, },
TooManyEmbeds {
max: usize,
},
TooManyReplies { TooManyReplies {
max: usize, max: usize,
}, },
+1
View File
@@ -46,6 +46,7 @@ impl<'r> Responder<'r, 'static> for Error {
ErrorType::AlreadyInServer => Status::Conflict, ErrorType::AlreadyInServer => Status::Conflict,
ErrorType::TooManyServers { .. } => Status::BadRequest, ErrorType::TooManyServers { .. } => Status::BadRequest,
ErrorType::TooManyEmbeds { .. } => Status::BadRequest,
ErrorType::TooManyEmoji { .. } => Status::BadRequest, ErrorType::TooManyEmoji { .. } => Status::BadRequest,
ErrorType::TooManyChannels { .. } => Status::BadRequest, ErrorType::TooManyChannels { .. } => Status::BadRequest,
ErrorType::TooManyRoles { .. } => Status::BadRequest, ErrorType::TooManyRoles { .. } => Status::BadRequest,
+4 -2
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-delta" name = "revolt-delta"
version = "0.6.7" version = "0.6.9"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
authors = ["Paul Makles <paulmakles@gmail.com>"] authors = ["Paul Makles <paulmakles@gmail.com>"]
edition = "2018" edition = "2018"
@@ -57,7 +57,7 @@ rocket = { version = "0.5.0-rc.2", default-features = false, features = [
] } ] }
rocket_cors = { git = "https://github.com/lawliet89/rocket_cors", rev = "c17e8145baa4790319fdb6a473e465b960f55e7c" } rocket_cors = { git = "https://github.com/lawliet89/rocket_cors", rev = "c17e8145baa4790319fdb6a473e465b960f55e7c" }
rocket_empty = { version = "0.1.1", features = ["schema"] } rocket_empty = { version = "0.1.1", features = ["schema"] }
rocket_authifier = { version = "1.0.7" } rocket_authifier = { version = "1.0.8" }
rocket_prometheus = "0.10.0-rc.3" rocket_prometheus = "0.10.0-rc.3"
# spec generation # spec generation
@@ -68,6 +68,8 @@ revolt_rocket_okapi = { version = "0.9.1", features = ["swagger"] }
revolt-quark = { path = "../quark" } revolt-quark = { path = "../quark" }
# core # core
authifier = "1.0.8"
revolt-config = { path = "../core/config" }
revolt-database = { path = "../core/database", features = [ revolt-database = { path = "../core/database", features = [
"rocket-impl", "rocket-impl",
"redis-is-patched", "redis-is-patched",
+15 -3
View File
@@ -8,6 +8,7 @@ extern crate serde_json;
pub mod routes; pub mod routes;
pub mod util; pub mod util;
use revolt_database::{Database, MongoDb};
use rocket::{Build, Rocket}; use rocket::{Build, Rocket};
use rocket_cors::{AllowedOrigins, CorsOptions}; use rocket_cors::{AllowedOrigins, CorsOptions};
use rocket_prometheus::PrometheusMetrics; use rocket_prometheus::PrometheusMetrics;
@@ -33,7 +34,12 @@ pub async fn web() -> Rocket<Build> {
// Setup Authifier // Setup Authifier
let authifier = Authifier { let authifier = Authifier {
database: db.clone().into(), database: match db.clone() {
Database::Reference(_) => Default::default(),
Database::MongoDb(MongoDb(client, _)) => authifier::Database::MongoDb(
authifier::database::MongoDb(client.database("revolt")),
),
},
config: revolt_quark::util::authifier::config(), config: revolt_quark::util::authifier::config(),
event_channel: Some(sender), event_channel: Some(sender),
}; };
@@ -55,8 +61,14 @@ pub async fn web() -> Rocket<Build> {
}); });
// Launch background task workers // Launch background task workers
async_std::task::spawn(revolt_database::tasks::start_workers(db.clone())); async_std::task::spawn(revolt_database::tasks::start_workers(
async_std::task::spawn(revolt_quark::tasks::start_workers(legacy_db.clone())); db.clone(),
authifier.database.clone(),
));
async_std::task::spawn(revolt_quark::tasks::start_workers(
legacy_db.clone(),
authifier.database.clone(),
));
// Configure CORS // Configure CORS
let cors = CorsOptions { let cors = CorsOptions {
+1 -1
View File
@@ -48,7 +48,7 @@ mod test {
drop(response); drop(response);
let event = harness let event = harness
.wait_for_event(|event| match event { .wait_for_event(&bot.id, |event| match event {
EventV1::UserUpdate { id, .. } => id == &bot.id, EventV1::UserUpdate { id, .. } => id == &bot.id,
_ => false, _ => false,
}) })
+1 -1
View File
@@ -23,7 +23,7 @@ pub async fn fetch_bot(
} }
Ok(Json(FetchBotResponse { Ok(Json(FetchBotResponse {
user: db.fetch_user(&bot.id).await?.into(None).await, user: db.fetch_user(&bot.id).await?.into(db, None).await,
bot: bot.into(), bot: bot.into(),
})) }))
} }
+23 -36
View File
@@ -42,7 +42,7 @@ pub async fn invite_bot(
.await .await
.throw_if_lacking_channel_permission(ChannelPermission::ManageServer)?; .throw_if_lacking_channel_permission(ChannelPermission::ManageServer)?;
Member::create(db, &server, &bot_user) Member::create(db, &server, &bot_user, None)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }
@@ -66,7 +66,7 @@ pub async fn invite_bot(
mod test { mod test {
use crate::{rocket, util::test::TestHarness}; use crate::{rocket, util::test::TestHarness};
use revolt_database::{events::client::EventV1, Bot, Channel, Server}; use revolt_database::{events::client::EventV1, Bot, Channel, Server};
use revolt_models::v0; use revolt_models::v0::{self, DataCreateServer};
use rocket::http::{ContentType, Header, Status}; use rocket::http::{ContentType, Header, Status};
#[rocket::async_test] #[rocket::async_test]
@@ -78,20 +78,16 @@ mod test {
.await .await
.expect("`Bot`"); .expect("`Bot`");
// FIXME: Channel::create_group let group = Channel::create_group(
let group = Channel::Group { &harness.db,
id: ulid::Ulid::new().to_string(), v0::DataCreateGroup {
name: TestHarness::rand_string(), name: TestHarness::rand_string(),
owner: user.id.to_string(), ..Default::default()
description: None, },
last_message_id: None, user.id.to_string(),
icon: None, )
nsfw: false, .await
permissions: None, .unwrap();
recipients: vec![user.id.to_string()],
};
group.create(&harness.db).await.unwrap();
let response = harness let response = harness
.client .client
@@ -106,7 +102,7 @@ mod test {
drop(response); drop(response);
let event = harness let event = harness
.wait_for_event(|event| match event { .wait_for_event(&group.id(), |event| match event {
EventV1::ChannelGroupJoin { id, .. } => id == &group.id(), EventV1::ChannelGroupJoin { id, .. } => id == &group.id(),
_ => false, _ => false,
}) })
@@ -129,26 +125,17 @@ mod test {
.await .await
.expect("`Bot`"); .expect("`Bot`");
// FIXME: Server::create_server let (server, _) = Server::create(
let server = Server { &harness.db,
id: ulid::Ulid::new().to_string(), DataCreateServer {
name: TestHarness::rand_string(), name: TestHarness::rand_string(),
owner: user.id.to_string(), ..Default::default()
analytics: false, },
discoverable: false, &user,
nsfw: false, false,
banner: None, )
icon: None, .await
categories: None, .unwrap();
channels: vec![],
default_permissions: 0,
description: None,
flags: None,
roles: Default::default(),
system_messages: None,
};
server.create(&harness.db).await.unwrap();
let response = harness let response = harness
.client .client
@@ -168,7 +155,7 @@ mod test {
drop(response); drop(response);
let event = harness let event = harness
.wait_for_event(|event| match event { .wait_for_event(&server.id, |event| match event {
EventV1::ServerMemberJoin { id, .. } => id == &server.id, EventV1::ServerMemberJoin { id, .. } => id == &server.id,
_ => false, _ => false,
}) })
@@ -1,8 +1,11 @@
use revolt_quark::{ use revolt_database::{
models::{Channel, User}, util::{permissions::DatabasePermissionQuery, reference::Reference},
perms, Database, Permission, Ref, Result, Database, User,
}; };
use revolt_models::v0;
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::Result;
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
/// # Fetch Channel /// # Fetch Channel
@@ -10,12 +13,54 @@ use rocket::{serde::json::Json, State};
/// Fetch channel by its id. /// Fetch channel by its id.
#[openapi(tag = "Channel Information")] #[openapi(tag = "Channel Information")]
#[get("/<target>")] #[get("/<target>")]
pub async fn req(db: &State<Database>, user: User, target: Ref) -> Result<Json<Channel>> { pub async fn fetch_channel(
db: &State<Database>,
user: User,
target: Reference,
) -> Result<Json<v0::Channel>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
perms(&user)
.channel(&channel)
.throw_permission(db, Permission::ViewChannel)
.await?;
Ok(Json(channel)) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
calculate_channel_permissions(&mut query)
.await
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
Ok(Json(channel.into()))
}
#[cfg(test)]
mod test {
use crate::{rocket, util::test::TestHarness};
use revolt_database::Channel;
use revolt_models::v0;
use rocket::http::{Header, Status};
#[rocket::async_test]
async fn fetch_channel() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.get(format!("/channels/{}", group.id()))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::Ok);
let channel: v0::Channel = response.into_json().await.expect("`Channel`");
assert_eq!(channel, group.into());
}
} }
@@ -1,40 +1,184 @@
use revolt_quark::{ use revolt_database::{
get_relationship, util::{permissions::DatabasePermissionQuery, reference::Reference},
models::{user::RelationshipStatus, Channel, User}, Channel, Database, User,
perms, Db, EmptyResponse, Error, Permission, Ref, Result,
}; };
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::State;
use rocket_empty::EmptyResponse;
/// # Add Member to Group /// # Add Member to Group
/// ///
/// Adds another user to the group. /// Adds another user to the group.
#[openapi(tag = "Groups")] #[openapi(tag = "Groups")]
#[put("/<target>/recipients/<member>")] #[put("/<group_id>/recipients/<member_id>")]
pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> { pub async fn req(
db: &State<Database>,
user: User,
group_id: Reference,
member_id: Reference,
) -> Result<EmptyResponse> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
let mut channel = target.as_channel(db).await?; let mut channel = group_id.as_channel(db).await?;
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.channel(&channel) calculate_channel_permissions(&mut query)
.throw_permission_and_view_channel(db, Permission::InviteOthers) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::InviteOthers)?;
match &channel { match &channel {
Channel::Group { .. } => { Channel::Group { .. } => {
let member = member.as_user(db).await?; // FIXME: use permissions here?
if !matches!( // interesting if users could block new group invites
get_relationship(&user, &member.id), let member = member_id.as_user(db).await?;
RelationshipStatus::Friend if !user.is_friends_with(&member.id) {
) { return Err(create_error!(NotFriends));
return Err(Error::NotFriends);
} }
channel channel
.add_user_to_group(db, &member.id, &user.id) .add_user_to_group(db, &member, &user.id)
.await .await
.map(|_| EmptyResponse) .map(|_| EmptyResponse)
} }
_ => Err(Error::InvalidOperation), _ => Err(create_error!(InvalidOperation)),
}
}
#[cfg(test)]
mod test {
use crate::{rocket, util::test::TestHarness};
use revolt_database::{events::client::EventV1, Channel, RelationshipStatus};
use revolt_models::v0;
use rocket::http::{Header, Status};
#[rocket::async_test]
async fn success_add_member() {
let mut harness = TestHarness::new().await;
let (_, session, mut user) = harness.new_user().await;
let (_, _, mut other_user) = harness.new_user().await;
#[allow(clippy::disallowed_methods)]
user.apply_relationship(
&harness.db,
&mut other_user,
RelationshipStatus::Friend,
RelationshipStatus::Friend,
)
.await
.unwrap();
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.put(format!(
"/channels/{}/recipients/{}",
group.id(),
other_user.id
))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::NoContent);
drop(response);
harness
.wait_for_event(&format!("{}!", other_user.id), |event| match event {
EventV1::ChannelCreate(channel) => channel.id() == group.id(),
_ => false,
})
.await;
let event = harness
.wait_for_event(&group.id(), |event| match event {
EventV1::ChannelGroupJoin { id, .. } => id == &group.id(),
_ => false,
})
.await;
match event {
EventV1::ChannelGroupJoin { user, .. } => assert_eq!(user, other_user.id),
_ => unreachable!(),
};
let message = harness.wait_for_message(&group.id()).await;
assert_eq!(
message.system,
Some(v0::SystemMessage::UserAdded {
id: other_user.id.to_string(),
by: user.id.to_string()
})
);
}
#[rocket::async_test]
async fn fail_add_non_friend() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let (_, _, other_user) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.put(format!(
"/channels/{}/recipients/{}",
group.id(),
other_user.id
))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::Forbidden);
}
#[rocket::async_test]
async fn fail_add_already_in_group() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.put(format!("/channels/{}/recipients/{}", group.id(), user.id))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::Conflict);
} }
} }
@@ -1,84 +1,102 @@
use std::{collections::HashSet, iter::FromIterator}; use revolt_database::{Channel, Database, RelationshipStatus, User};
use revolt_models::v0;
use revolt_quark::{ use revolt_result::{create_error, Result};
get_relationship,
models::{user::RelationshipStatus, Channel, User},
variables::delta::MAX_GROUP_SIZE,
Db, Error, Result,
};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use serde::{Deserialize, Serialize}; use rocket::State;
use ulid::Ulid;
use validator::Validate; use validator::Validate;
/// # Group Data
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct DataCreateGroup {
/// Group name
#[validate(length(min = 1, max = 32))]
name: String,
/// Group description
#[validate(length(min = 0, max = 1024))]
description: Option<String>,
/// Array of user IDs to add to the group
///
/// Must be friends with these users.
#[validate(length(min = 0, max = 49))]
users: Vec<String>,
/// Whether this group is age-restricted
#[serde(skip_serializing_if = "Option::is_none")]
nsfw: Option<bool>,
}
/// # Create Group /// # Create Group
/// ///
/// Create a new group channel. /// Create a new group channel.
#[openapi(tag = "Groups")] #[openapi(tag = "Groups")]
#[post("/create", data = "<info>")] #[post("/create", data = "<data>")]
pub async fn req(db: &Db, user: User, info: Json<DataCreateGroup>) -> Result<Json<Channel>> { pub async fn create_group(
db: &State<Database>,
user: User,
data: Json<v0::DataCreateGroup>,
) -> Result<Json<v0::Channel>> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
let info = info.into_inner(); let data = data.into_inner();
info.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let mut set: HashSet<String> = HashSet::from_iter(info.users.into_iter()); for target in &data.users {
set.insert(user.id.clone()); match user.relationship_with(target) {
if set.len() > *MAX_GROUP_SIZE {
return Err(Error::GroupTooLarge {
max: *MAX_GROUP_SIZE,
});
}
for target in &set {
match get_relationship(&user, target) {
RelationshipStatus::Friend | RelationshipStatus::User => {} RelationshipStatus::Friend | RelationshipStatus::User => {}
_ => { _ => {
return Err(Error::NotFriends); return Err(create_error!(NotFriends));
} }
} }
} }
let group = Channel::Group { Ok(Json(Channel::create_group(db, data, user.id).await?.into()))
id: Ulid::new().to_string(), }
name: info.name, #[cfg(test)]
owner: user.id, mod test {
description: info.description, use crate::{rocket, util::test::TestHarness};
recipients: set.into_iter().collect::<Vec<String>>(), use revolt_database::events::client::EventV1;
use revolt_models::v0;
use rocket::http::{ContentType, Header, Status};
icon: None, #[rocket::async_test]
last_message_id: None, async fn create_group() {
let mut harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
permissions: None, let response = harness
.client
.post("/channels/create")
.header(Header::new("x-session-token", session.token.to_string()))
.header(ContentType::JSON)
.body(
json!(v0::DataCreateBot {
name: TestHarness::rand_string(),
})
.to_string(),
)
.dispatch()
.await;
nsfw: info.nsfw.unwrap_or(false), assert_eq!(response.status(), Status::Ok);
};
group.create(db).await?; let channel: v0::Channel = response.into_json().await.expect("`Channel`");
Ok(Json(group)) match channel {
v0::Channel::Group {
id,
owner,
recipients,
..
} => {
assert_eq!(owner, user.id);
assert_eq!(recipients.len(), 1);
assert!(harness.db.fetch_channel(&id).await.is_ok());
let event = harness
.wait_for_event(&format!("{}!", user.id), |event| match event {
EventV1::ChannelCreate(channel) => channel.id() == id,
_ => false,
})
.await;
match event {
EventV1::ChannelCreate(v0::Channel::Group {
owner: channel_owner,
..
}) => {
assert_eq!(owner, channel_owner);
}
_ => unreachable!(),
}
}
_ => unreachable!(),
}
}
} }
@@ -40,3 +40,151 @@ pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyR
_ => Err(Error::InvalidOperation), _ => Err(Error::InvalidOperation),
} }
} }
#[cfg(test)]
mod test {
use crate::{rocket, util::test::TestHarness};
use revolt_database::{events::client::EventV1, Channel, RelationshipStatus};
use revolt_models::v0;
use rocket::http::{Header, Status};
#[rocket::async_test]
async fn success_remove_member() {
let mut harness = TestHarness::new().await;
let (_, session, mut user) = harness.new_user().await;
let (_, _, mut other_user) = harness.new_user().await;
#[allow(clippy::disallowed_methods)]
user.apply_relationship(
&harness.db,
&mut other_user,
RelationshipStatus::Friend,
RelationshipStatus::Friend,
)
.await
.unwrap();
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.put(format!(
"/channels/{}/recipients/{}",
group.id(),
other_user.id
))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
assert_eq!(response.status(), Status::NoContent);
drop(response);
harness
.wait_for_event(&format!("{}!", other_user.id), |event| match event {
EventV1::ChannelCreate(channel) => channel.id() == group.id(),
_ => false,
})
.await;
let event = harness
.wait_for_event(&group.id(), |event| match event {
EventV1::ChannelGroupJoin { id, .. } => id == &group.id(),
_ => false,
})
.await;
match event {
EventV1::ChannelGroupJoin { user, .. } => assert_eq!(user, other_user.id),
_ => unreachable!(),
};
let message = harness.wait_for_message(&group.id()).await;
assert_eq!(
message.system,
Some(v0::SystemMessage::UserAdded {
id: other_user.id.to_string(),
by: user.id.to_string()
})
);
}
#[rocket::async_test]
async fn fail_not_in_group() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let (_, _, other_user) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let response = harness
.client
.delete(format!(
"/channels/{}/recipients/{}",
group.id(),
other_user.id
))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
dbg!(response.into_string().await);
// TODO: finish impl
// assert_eq!(response.status(), Status::NotFound);
}
#[rocket::async_test]
async fn fail_not_group_owner() {
let harness = TestHarness::new().await;
let (_, _, user) = harness.new_user().await;
let (_, session, other_user) = harness.new_user().await;
let (_, _, user_to_be_kicked) = harness.new_user().await;
let group = Channel::create_group(
&harness.db,
v0::DataCreateGroup {
name: TestHarness::rand_string(),
users: vec![&other_user.id, &user_to_be_kicked.id]
.into_iter()
.cloned()
.collect(),
..Default::default()
},
user.id.to_string(),
)
.await
.unwrap();
let _response = harness
.client
.delete(format!(
"/channels/{}/recipients/{}",
group.id(),
user_to_be_kicked.id
))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
// TODO: impl assert_eq!(response.status(), Status::Forbidden);
}
}
@@ -78,6 +78,11 @@ pub async fn req(
// 3. Replace if we are given new embeds // 3. Replace if we are given new embeds
if let Some(embeds) = edit.embeds { if let Some(embeds) = edit.embeds {
// Ensure we have permissions to send embeds
permissions
.throw_permission_and_view_channel(db, Permission::SendEmbeds)
.await?;
new_embeds.clear(); new_embeds.clear();
for embed in embeds { for embed in embeds {
@@ -89,7 +94,11 @@ pub async fn req(
message.update(db, partial).await?; message.update(db, partial).await?;
// Queue up a task for processing embeds // Queue up a task for processing embeds if the we have sufficient permissions
if permissions
.has_permission(db, Permission::SendEmbeds)
.await?
{
if let Some(content) = edit.content { if let Some(content) = edit.content {
revolt_quark::tasks::process_embeds::queue( revolt_quark::tasks::process_embeds::queue(
message.channel.to_string(), message.channel.to_string(),
@@ -98,6 +107,7 @@ pub async fn req(
) )
.await; .await;
} }
}
Ok(Json(message)) Ok(Json(message))
} }
@@ -1,26 +1,34 @@
use revolt_quark::{ use revolt_database::{
models::{Message, User}, util::{permissions::DatabasePermissionQuery, reference::Reference},
perms, Db, Error, Permission, Ref, Result, Database, User,
}; };
use revolt_models::v0;
use rocket::serde::json::Json; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State};
/// # Fetch Message /// # Fetch Message
/// ///
/// Retrieves a message by its id. /// Retrieves a message by its id.
#[openapi(tag = "Messaging")] #[openapi(tag = "Messaging")]
#[get("/<target>/messages/<msg>")] #[get("/<target>/messages/<msg>")]
pub async fn req(db: &Db, user: User, target: Ref, msg: Ref) -> Result<Json<Message>> { pub async fn req(
db: &State<Database>,
user: User,
target: Reference,
msg: Reference,
) -> Result<Json<v0::Message>> {
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
perms(&user)
.channel(&channel) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.throw_permission(db, Permission::ViewChannel) calculate_channel_permissions(&mut query)
.await?; .await
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
let message = msg.as_message(db).await?; let message = msg.as_message(db).await?;
if message.channel != channel.as_id() { if message.channel != channel.id() {
return Err(Error::NotFound); return Err(create_error!(NotFound));
} }
Ok(Json(message)) Ok(Json(message.into()))
} }
@@ -1,26 +0,0 @@
use revolt_quark::{models::User, Ref, Result};
use rocket::serde::json::Json;
use serde::{Deserialize, Serialize};
use validator::Validate;
/// # Query Parameters
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct OptionsQueryStale {
/// Array of message IDs
#[validate(length(min = 0, max = 150))]
ids: Vec<String>,
}
/// # Poll Message Changes
///
/// This route returns any changed message objects and tells you if any have been deleted.
///
/// Don't actually poll this route, instead use this to update your local database.
///
/// **DEPRECATED**
#[openapi(tag = "Messaging")]
#[post("/<_target>/messages/stale", data = "<_data>")]
pub async fn req(_user: User, _target: Ref, _data: Json<OptionsQueryStale>) -> Result<()> {
Ok(())
}
@@ -1,12 +1,13 @@
use revolt_database::util::idempotency::IdempotencyKey; use revolt_database::util::permissions::DatabasePermissionQuery;
use revolt_quark::{ use revolt_database::{
models::{message::DataMessageSend, Message, User}, util::idempotency::IdempotencyKey, util::reference::Reference, Database, User,
perms,
types::push::MessageAuthor,
Db, Error, Permission, Ref, Result,
}; };
use revolt_database::{Interactions, Message};
use revolt_models::v0;
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use rocket::State;
use validator::Validate; use validator::Validate;
/// # Send Message /// # Send Message
@@ -15,68 +16,63 @@ use validator::Validate;
#[openapi(tag = "Messaging")] #[openapi(tag = "Messaging")]
#[post("/<target>/messages", data = "<data>")] #[post("/<target>/messages", data = "<data>")]
pub async fn message_send( pub async fn message_send(
db: &Db, db: &State<Database>,
user: User, user: User,
target: Ref, target: Reference,
data: Json<DataMessageSend>, data: Json<v0::DataMessageSend>,
idempotency: IdempotencyKey, idempotency: IdempotencyKey,
) -> Result<Json<Message>> { ) -> Result<Json<v0::Message>> {
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
// Ensure we have permissions to send a message // Ensure we have permissions to send a message
let channel = target.as_channel(db).await?; let channel = target.as_channel(db).await?;
let mut permissions = perms(&user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
permissions let permissions = calculate_channel_permissions(&mut query).await;
.throw_permission_and_view_channel(db, Permission::SendMessage) permissions.throw_if_lacking_channel_permission(ChannelPermission::SendMessage)?;
.await?;
// Verify permissions for masquerade // Verify permissions for masquerade
if let Some(masq) = &data.masquerade { if let Some(masq) = &data.masquerade {
permissions permissions.throw_if_lacking_channel_permission(ChannelPermission::Masquerade)?;
.throw_permission(db, Permission::Masquerade)
.await?;
if masq.colour.is_some() { if masq.colour.is_some() {
permissions permissions.throw_if_lacking_channel_permission(ChannelPermission::ManageRole)?;
.throw_permission(db, Permission::ManageRole)
.await?;
} }
} }
// Check permissions for embeds // Check permissions for embeds
if data.embeds.as_ref().is_some_and(|v| !v.is_empty()) { if data.embeds.as_ref().is_some_and(|v| !v.is_empty()) {
permissions permissions.throw_if_lacking_channel_permission(ChannelPermission::SendEmbeds)?;
.throw_permission(db, Permission::SendEmbeds)
.await?;
} }
// Check permissions for files // Check permissions for files
if data.attachments.as_ref().is_some_and(|v| !v.is_empty()) { if data.attachments.as_ref().is_some_and(|v| !v.is_empty()) {
permissions permissions.throw_if_lacking_channel_permission(ChannelPermission::UploadFiles)?;
.throw_permission(db, Permission::UploadFiles)
.await?;
} }
// Ensure interactions information is correct // Ensure interactions information is correct
if let Some(interactions) = &data.interactions { if let Some(interactions) = &data.interactions {
interactions.validate(db, &mut permissions).await?; let interactions: Interactions = interactions.clone().into();
interactions.validate(db, &permissions).await?;
} }
// Create the message // Create the message
let message = channel let author: v0::User = user.clone().into(db, Some(&user)).await;
.send_message( Ok(Json(
Message::create_from_api(
db, db,
channel,
data, data,
MessageAuthor::User(&user), v0::MessageAuthor::User(&author),
idempotency, idempotency,
permissions permissions.has_channel_permission(ChannelPermission::SendEmbeds),
.has_permission(db, Permission::SendEmbeds)
.await?,
) )
.await?; .await?
.into(),
Ok(Json(message)) ))
} }
+2 -4
View File
@@ -16,7 +16,6 @@ mod message_delete;
mod message_edit; mod message_edit;
mod message_fetch; mod message_fetch;
mod message_query; mod message_query;
mod message_query_stale;
mod message_react; mod message_react;
mod message_search; mod message_search;
mod message_send; mod message_send;
@@ -30,7 +29,7 @@ mod webhook_fetch_all;
pub fn routes() -> (Vec<Route>, OpenApi) { pub fn routes() -> (Vec<Route>, OpenApi) {
openapi_get_routes_spec![ openapi_get_routes_spec![
channel_ack::req, channel_ack::req,
channel_fetch::req, channel_fetch::fetch_channel,
members_fetch::req, members_fetch::req,
channel_delete::req, channel_delete::req,
channel_edit::req, channel_edit::req,
@@ -38,12 +37,11 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
message_send::message_send, message_send::message_send,
message_query::req, message_query::req,
message_search::req, message_search::req,
message_query_stale::req,
message_fetch::req, message_fetch::req,
message_edit::req, message_edit::req,
message_bulk_delete::req, message_bulk_delete::req,
message_delete::req, message_delete::req,
group_create::req, group_create::create_group,
group_add_member::req, group_add_member::req,
group_remove_member::req, group_remove_member::req,
voice_join::req, voice_join::req,
@@ -1,23 +1,16 @@
use revolt_database::{Database, Webhook}; use revolt_database::{
use revolt_quark::{ util::{permissions::DatabasePermissionQuery, reference::Reference},
models::{Channel, User}, Channel, Database, User, Webhook,
perms, Db, Error, Permission, Ref, Result,
DEFAULT_WEBHOOK_PERMISSIONS,
}; };
use revolt_models::v0;
use revolt_permissions::{
calculate_channel_permissions, ChannelPermission, DEFAULT_WEBHOOK_PERMISSIONS,
};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
use serde::{Deserialize, Serialize};
use ulid::Ulid; use ulid::Ulid;
use validator::Validate; use validator::Validate;
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct CreateWebhookBody {
#[validate(length(min = 1, max = 32))]
name: String,
#[validate(length(min = 1, max = 128))]
avatar: Option<String>,
}
/// # Creates a webhook /// # Creates a webhook
/// ///
/// Creates a webhook which 3rd party platforms can use to send messages /// Creates a webhook which 3rd party platforms can use to send messages
@@ -25,33 +18,34 @@ pub struct CreateWebhookBody {
#[post("/<target>/webhooks", data = "<data>")] #[post("/<target>/webhooks", data = "<data>")]
pub async fn req( pub async fn req(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
user: User, user: User,
target: Ref, target: Reference,
data: Json<CreateWebhookBody>, data: Json<v0::CreateWebhookBody>,
) -> Result<Json<revolt_models::v0::Webhook>> { ) -> Result<Json<v0::Webhook>> {
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let channel = target.as_channel(legacy_db).await?; let channel = target.as_channel(db).await?;
if !matches!(channel, Channel::TextChannel { .. } | Channel::Group { .. }) { if !matches!(channel, Channel::TextChannel { .. } | Channel::Group { .. }) {
return Err(Error::InvalidOperation); return Err(create_error!(InvalidOperation));
} }
let mut permissions = perms(&user).channel(&channel); let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
permissions calculate_channel_permissions(&mut query)
.has_permission(legacy_db, Permission::ManageWebhooks) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageWebhooks)?;
let webhook_id = Ulid::new().to_string(); let webhook_id = Ulid::new().to_string();
let avatar = match &data.avatar { let avatar = match &data.avatar {
Some(id) => Some( Some(id) => Some(
db.find_and_use_attachment(id, "avatars", "user", &webhook_id) db.find_and_use_attachment(id, "avatars", "user", &webhook_id)
.await .await?,
.map_err(Error::from_core)?,
), ),
None => None, None => None,
}; };
@@ -60,12 +54,12 @@ pub async fn req(
id: webhook_id, id: webhook_id,
name: data.name, name: data.name,
avatar, avatar,
channel_id: channel.id().to_string(), channel_id: channel.id(),
permissions: *DEFAULT_WEBHOOK_PERMISSIONS, permissions: *DEFAULT_WEBHOOK_PERMISSIONS,
token: Some(nanoid::nanoid!(64)), token: Some(nanoid::nanoid!(64)),
}; };
webhook.create(db).await.map_err(Error::from_core)?; webhook.create(db).await?;
Ok(Json(webhook.into())) Ok(Json(webhook.into()))
} }
@@ -1,6 +1,10 @@
use revolt_database::Database; use revolt_database::{
util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, User,
};
use revolt_models::v0::Webhook; use revolt_models::v0::Webhook;
use revolt_quark::{models::User, perms, Db, Error, Permission, Ref, Result}; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::Result;
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
/// # Gets all webhooks /// # Gets all webhooks
@@ -10,20 +14,19 @@ use rocket::{serde::json::Json, State};
#[get("/<channel_id>/webhooks")] #[get("/<channel_id>/webhooks")]
pub async fn req( pub async fn req(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
user: User, user: User,
channel_id: Ref, channel_id: Reference,
) -> Result<Json<Vec<Webhook>>> { ) -> Result<Json<Vec<Webhook>>> {
let channel = channel_id.as_channel(legacy_db).await?; let channel = channel_id.as_channel(db).await?;
let mut permissions = perms(&user).channel(&channel);
permissions let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.has_permission(legacy_db, Permission::ManageWebhooks) calculate_channel_permissions(&mut query)
.await?; .await
.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
Ok(Json( Ok(Json(
db.fetch_webhooks_for_channel(channel.id()) db.fetch_webhooks_for_channel(&channel.id())
.await .await?
.map_err(Error::from_core)?
.into_iter() .into_iter()
.map(|v| v.into()) .map(|v| v.into())
.collect::<Vec<Webhook>>(), .collect::<Vec<Webhook>>(),
@@ -1,31 +1,11 @@
use once_cell::sync::Lazy; use revolt_config::config;
use regex::Regex; use revolt_database::{util::permissions::DatabasePermissionQuery, Database, Emoji, File, User};
use revolt_quark::models::emoji::EmojiParent; use revolt_models::v0;
use revolt_quark::models::{Emoji, File, User}; use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use revolt_quark::variables::delta::MAX_EMOJI_COUNT; use revolt_result::{create_error, Result};
use revolt_quark::{perms, Db, Error, Permission, Result};
use serde::Deserialize;
use validator::Validate; use validator::Validate;
use rocket::serde::json::Json; use rocket::{serde::json::Json, State};
/// Regex for valid emoji names
///
/// Alphanumeric and underscores
pub static RE_EMOJI: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[a-z0-9_]+$").unwrap());
/// # Emoji Data
#[derive(Validate, Deserialize, JsonSchema)]
pub struct DataCreateEmoji {
/// Server name
#[validate(length(min = 1, max = 32), regex = "RE_EMOJI")]
name: String,
/// Parent information
parent: EmojiParent,
/// Whether the emoji is mature
#[serde(default)]
nsfw: bool,
}
/// # Create New Emoji /// # Create New Emoji
/// ///
@@ -33,41 +13,45 @@ pub struct DataCreateEmoji {
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[put("/emoji/<id>", data = "<data>")] #[put("/emoji/<id>", data = "<data>")]
pub async fn create_emoji( pub async fn create_emoji(
db: &Db, db: &State<Database>,
user: User, user: User,
id: String, id: String,
data: Json<DataCreateEmoji>, data: Json<v0::DataCreateEmoji>,
) -> Result<Json<Emoji>> { ) -> Result<Json<v0::Emoji>> {
let config = config().await;
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
// Bots cannot manage emojis // Bots cannot manage emojis
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
// Validate we have permission to write into parent // Validate we have permission to write into parent
match &data.parent { match &data.parent {
EmojiParent::Server { id } => { v0::EmojiParent::Server { id } => {
let server = db.fetch_server(id).await?; let server = db.fetch_server(id).await?;
// Check for permission // Check for permission
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
.server(&server) calculate_server_permissions(&mut query)
.throw_permission(db, Permission::ManageCustomisation) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageCustomisation)?;
// Check that there are no more than 100 emoji // Check that we haven't hit the emoji limit
// ! FIXME: hardcoded upper limit
let emojis = db.fetch_emoji_by_parent_id(&server.id).await?; let emojis = db.fetch_emoji_by_parent_id(&server.id).await?;
if emojis.len() > *MAX_EMOJI_COUNT { if emojis.len() > config.features.limits.default.server_emoji {
return Err(Error::TooManyEmoji { return Err(create_error!(TooManyEmoji {
max: *MAX_EMOJI_COUNT, max: config.features.limits.default.server_emoji,
}); }));
} }
} }
EmojiParent::Detached => return Err(Error::InvalidOperation), v0::EmojiParent::Detached => return Err(create_error!(InvalidOperation)),
}; };
// Find the relevant attachment // Find the relevant attachment
@@ -76,7 +60,7 @@ pub async fn create_emoji(
// Create the emoji object // Create the emoji object
let emoji = Emoji { let emoji = Emoji {
id, id,
parent: data.parent, parent: data.parent.into(),
creator_id: user.id, creator_id: user.id,
name: data.name, name: data.name,
animated: "image/gif" == &attachment.content_type, animated: "image/gif" == &attachment.content_type,
@@ -85,5 +69,5 @@ pub async fn create_emoji(
// Save emoji // Save emoji
emoji.create(db).await?; emoji.create(db).await?;
Ok(Json(emoji)) Ok(Json(emoji.into()))
} }
@@ -1,19 +1,30 @@
use revolt_quark::models::{emoji::EmojiParent, User}; use revolt_database::{
use revolt_quark::{perms, Db, EmptyResponse, Error, Permission, Ref, Result}; util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, EmojiParent, User,
};
use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::State;
use rocket_empty::EmptyResponse;
/// # Delete Emoji /// # Delete Emoji
/// ///
/// Delete an emoji by its id. /// Delete an emoji by its id.
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[delete("/emoji/<id>")] #[delete("/emoji/<emoji_id>")]
pub async fn delete_emoji(db: &Db, user: User, id: Ref) -> Result<EmptyResponse> { pub async fn delete_emoji(
db: &State<Database>,
user: User,
emoji_id: Reference,
) -> Result<EmptyResponse> {
// Bots cannot manage emoji // Bots cannot manage emoji
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
// Fetch the emoji // Fetch the emoji
let emoji = id.as_emoji(db).await?; let emoji = emoji_id.as_emoji(db).await?;
// If we uploaded the emoji, then we have permission to delete it // If we uploaded the emoji, then we have permission to delete it
if emoji.creator_id != user.id { if emoji.creator_id != user.id {
@@ -23,10 +34,10 @@ pub async fn delete_emoji(db: &Db, user: User, id: Ref) -> Result<EmptyResponse>
let server = db.fetch_server(id).await?; let server = db.fetch_server(id).await?;
// Check for permission // Check for permission
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
.server(&server) calculate_server_permissions(&mut query)
.throw_permission(db, Permission::ManageCustomisation) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageCustomisation)?;
} }
EmojiParent::Detached => return Ok(EmptyResponse), EmojiParent::Detached => return Ok(EmptyResponse),
}; };
@@ -1,13 +1,18 @@
use revolt_quark::models::{Emoji, User}; use revolt_database::{util::reference::Reference, Database};
use revolt_quark::{Db, Ref, Result}; use revolt_models::v0;
use revolt_result::Result;
use rocket::serde::json::Json; use rocket::{serde::json::Json, State};
/// # Fetch Emoji /// # Fetch Emoji
/// ///
/// Fetch an emoji by its id. /// Fetch an emoji by its id.
#[openapi(tag = "Emojis")] #[openapi(tag = "Emojis")]
#[get("/emoji/<id>")] #[get("/emoji/<emoji_id>")]
pub async fn fetch_emoji(db: &Db, _user: User, id: Ref) -> Result<Json<Emoji>> { pub async fn fetch_emoji(db: &State<Database>, emoji_id: Reference) -> Result<Json<v0::Emoji>> {
id.as_emoji(db).await.map(Json) emoji_id
.as_emoji(db)
.await
.map(|emoji| emoji.into())
.map(Json)
} }
+1
View File
@@ -138,6 +138,7 @@ pub async fn root() -> Result<Json<RevoltConfig>> {
} }
#[cfg(test)] #[cfg(test)]
#[cfg(feature = "FIXME: THIS TEST CAUSES cargo test TO SEG FAULT, I HAVE NO CLUE HOW")]
mod test { mod test {
use crate::rocket; use crate::rocket;
use rocket::http::Status; use rocket::http::Status;
+27 -107
View File
@@ -1,119 +1,39 @@
use std::collections::HashMap; use revolt_database::util::permissions::DatabasePermissionQuery;
use revolt_database::{util::reference::Reference, Channel, Database, User};
use revolt_quark::{ use revolt_models::v0;
models::{server::PartialServer, Channel, User}, use revolt_permissions::{calculate_server_permissions, ChannelPermission};
perms, Db, Error, Permission, Ref, Result, variables::delta::MAX_CHANNEL_COUNT, use revolt_result::{create_error, Result};
};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use serde::{Deserialize, Serialize}; use rocket::State;
use ulid::Ulid;
use validator::Validate; use validator::Validate;
/// # Channel Type
#[derive(Serialize, Deserialize, JsonSchema, Default)]
enum ChannelType {
/// Text Channel
#[default]
Text,
/// Voice Channel
Voice,
}
/// # Channel Data
#[derive(Validate, Serialize, Deserialize, JsonSchema)]
pub struct DataCreateChannel {
/// Channel type
#[serde(rename = "type", default = "ChannelType::default")]
channel_type: ChannelType,
/// Channel name
#[validate(length(min = 1, max = 32))]
name: String,
/// Channel description
#[validate(length(min = 0, max = 1024))]
description: Option<String>,
/// Whether this channel is age restricted
#[serde(skip_serializing_if = "Option::is_none")]
nsfw: Option<bool>,
}
/// # Create Channel /// # Create Channel
/// ///
/// Create a new Text or Voice channel. /// Create a new Text or Voice channel.
#[openapi(tag = "Server Information")] #[openapi(tag = "Server Information")]
#[post("/<target>/channels", data = "<info>")] #[post("/<server>/channels", data = "<data>")]
pub async fn req( pub async fn create_server_channel(
db: &Db, db: &State<Database>,
user: User, user: User,
target: Ref, server: Reference,
info: Json<DataCreateChannel>, data: Json<v0::DataCreateServerChannel>,
) -> Result<Json<Channel>> { ) -> Result<Json<v0::Channel>> {
let info = info.into_inner(); let data = data.into_inner();
info.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let mut server = target.as_server(db).await?; let mut server = server.as_server(db).await?;
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
.server(&server) calculate_server_permissions(&mut query)
.throw_permission(db, Permission::ManageChannel) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageChannel)?;
if server.channels.len() > *MAX_CHANNEL_COUNT { Channel::create_server_channel(db, &mut server, data, true)
return Err(Error::TooManyChannels { max: *MAX_CHANNEL_COUNT }) .await
}; .map(|channel| channel.into())
.map(Json)
let id = Ulid::new().to_string();
let mut channels = server.channels.clone();
channels.push(id.clone());
let DataCreateChannel {
name,
description,
nsfw,
channel_type,
} = info;
let channel = match channel_type {
ChannelType::Text => Channel::TextChannel {
id,
server: server.id.clone(),
name,
description,
icon: None,
last_message_id: None,
default_permissions: None,
role_permissions: HashMap::new(),
nsfw: nsfw.unwrap_or(false),
},
ChannelType::Voice => Channel::VoiceChannel {
id,
server: server.id.clone(),
name,
description,
icon: None,
default_permissions: None,
role_permissions: HashMap::new(),
nsfw: nsfw.unwrap_or(false),
},
};
channel.create(db).await?;
server
.update(
db,
PartialServer {
channels: Some(channels),
..Default::default()
},
vec![],
)
.await?;
Ok(Json(channel))
} }
+2 -2
View File
@@ -25,12 +25,12 @@ mod server_fetch;
pub fn routes() -> (Vec<Route>, OpenApi) { pub fn routes() -> (Vec<Route>, OpenApi) {
openapi_get_routes_spec![ openapi_get_routes_spec![
server_create::req, server_create::create_server,
server_delete::req, server_delete::req,
server_fetch::req, server_fetch::req,
server_edit::req, server_edit::req,
server_ack::req, server_ack::req,
channel_create::req, channel_create::create_server_channel,
member_fetch_all::req, member_fetch_all::req,
member_remove::req, member_remove::req,
member_fetch::req, member_fetch::req,
@@ -1,102 +1,39 @@
use std::collections::HashMap; use revolt_database::{Database, Member, Server, User};
use revolt_models::v0;
use revolt_quark::{ use revolt_result::{create_error, Result};
models::{Channel, Server, User},
variables::delta::MAX_SERVER_COUNT,
Db, Error, Result, DEFAULT_PERMISSION_SERVER,
};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use serde::{Deserialize, Serialize}; use rocket::State;
use ulid::Ulid;
use validator::Validate; use validator::Validate;
/// # Server Data
#[derive(Validate, Deserialize, JsonSchema)]
pub struct DataCreateServer {
/// Server name
#[validate(length(min = 1, max = 32))]
name: String,
/// Server description
#[validate(length(min = 0, max = 1024))]
description: Option<String>,
/// Whether this server is age-restricted
#[serde(skip_serializing_if = "Option::is_none")]
nsfw: Option<bool>,
}
/// # Create Server Response
#[derive(Validate, Serialize, JsonSchema)]
pub struct CreateServerResponse {
/// Server object
server: Server,
/// Default channels
channels: Vec<Channel>,
}
/// # Create Server /// # Create Server
/// ///
/// Create a new server. /// Create a new server.
#[openapi(tag = "Server Information")] #[openapi(tag = "Server Information")]
#[post("/create", data = "<info>")] #[post("/create", data = "<data>")]
pub async fn req( pub async fn create_server(
db: &Db, db: &State<Database>,
user: User, user: User,
info: Json<DataCreateServer>, data: Json<v0::DataCreateServer>,
) -> Result<Json<CreateServerResponse>> { ) -> Result<Json<v0::CreateServerLegacyResponse>> {
if user.bot.is_some() { if user.bot.is_some() {
return Err(Error::IsBot); return Err(create_error!(IsBot));
} }
let info = info.into_inner(); let data = data.into_inner();
info.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
if !user.can_acquire_server(db).await? { user.can_acquire_server(db).await?;
return Err(Error::TooManyServers {
max: *MAX_SERVER_COUNT,
});
}
let DataCreateServer { let (server, channels) = Server::create(db, data, &user, true).await?;
name, let channels = Member::create(db, &server, &user, Some(channels)).await?;
description,
nsfw,
} = info;
let channel_id = Ulid::new().to_string(); Ok(Json(v0::CreateServerLegacyResponse {
let server_id = Ulid::new().to_string(); server: server.into(),
channels: channels.into_iter().map(|channel| channel.into()).collect(),
let channel = Channel::TextChannel { }))
id: channel_id.clone(),
server: server_id.clone(),
name: "General".into(),
description: None,
icon: None,
last_message_id: None,
default_permissions: None,
role_permissions: HashMap::new(),
nsfw: nsfw.unwrap_or(false),
};
db.insert_channel(&channel).await?;
let server = Server {
id: server_id.clone(),
owner: user.id.clone(),
name,
description,
channels: vec![channel_id],
nsfw: nsfw.unwrap_or(false),
default_permissions: *DEFAULT_PERMISSION_SERVER as i64,
..Default::default()
};
server.create(db).await?;
let channels = server.create_member(db, user, Some(vec![channel])).await?;
Ok(Json(CreateServerResponse { server, channels }))
} }
@@ -1,5 +1,9 @@
use revolt_database::{util::reference::Reference, Database}; use revolt_database::{
use revolt_quark::{models::User, perms, Db, Error, Permission, Result}; util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, User,
};
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::Result;
use rocket::State; use rocket::State;
use rocket_empty::EmptyResponse; use rocket_empty::EmptyResponse;
@@ -10,21 +14,16 @@ use rocket_empty::EmptyResponse;
#[delete("/<webhook_id>")] #[delete("/<webhook_id>")]
pub async fn webhook_delete( pub async fn webhook_delete(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
user: User, user: User,
webhook_id: Reference, webhook_id: Reference,
) -> Result<EmptyResponse> { ) -> Result<EmptyResponse> {
let webhook = webhook_id.as_webhook(db).await.map_err(Error::from_core)?; let webhook = webhook_id.as_webhook(db).await?;
let channel = legacy_db.fetch_channel(&webhook.channel_id).await?; let channel = db.fetch_channel(&webhook.channel_id).await?;
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.channel(&channel) calculate_channel_permissions(&mut query)
.throw_permission(legacy_db, Permission::ManageWebhooks)
.await?;
webhook
.delete(db)
.await .await
.map(|_| EmptyResponse) .throw_if_lacking_channel_permission(ChannelPermission::ManageWebhooks)?;
.map_err(Error::from_core)
webhook.delete(db).await.map(|_| EmptyResponse)
} }
@@ -1,6 +1,10 @@
use revolt_database::{util::reference::Reference, Database, PartialWebhook}; use revolt_database::{
util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, PartialWebhook, User,
};
use revolt_models::v0::{DataEditWebhook, Webhook}; use revolt_models::v0::{DataEditWebhook, Webhook};
use revolt_quark::{models::User, perms, Db, Error, Permission, Result}; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
use validator::Validate; use validator::Validate;
@@ -11,22 +15,24 @@ use validator::Validate;
#[patch("/<webhook_id>", data = "<data>")] #[patch("/<webhook_id>", data = "<data>")]
pub async fn webhook_edit( pub async fn webhook_edit(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
webhook_id: Reference, webhook_id: Reference,
user: User, user: User,
data: Json<DataEditWebhook>, data: Json<DataEditWebhook>,
) -> Result<Json<Webhook>> { ) -> Result<Json<Webhook>> {
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let mut webhook = webhook_id.as_webhook(db).await.map_err(Error::from_core)?; let mut webhook = webhook_id.as_webhook(db).await?;
let channel = legacy_db.fetch_channel(&webhook.channel_id).await?; let channel = db.fetch_channel(&webhook.channel_id).await?;
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.channel(&channel) calculate_channel_permissions(&mut query)
.throw_permission(legacy_db, Permission::ManageWebhooks) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ManageWebhooks)?;
if data.name.is_none() && data.avatar.is_none() && data.remove.is_empty() { if data.name.is_none() && data.avatar.is_none() && data.remove.is_empty() {
return Ok(Json(webhook.into())); return Ok(Json(webhook.into()));
@@ -48,16 +54,14 @@ pub async fn webhook_edit(
if let Some(avatar) = avatar { if let Some(avatar) = avatar {
let file = db let file = db
.find_and_use_attachment(&avatar, "avatars", "user", &webhook.id) .find_and_use_attachment(&avatar, "avatars", "user", &webhook.id)
.await .await?;
.map_err(Error::from_core)?;
partial.avatar = Some(file) partial.avatar = Some(file)
} }
webhook webhook
.update(db, partial, remove.into_iter().map(|v| v.into()).collect()) .update(db, partial, remove.into_iter().map(|v| v.into()).collect())
.await .await?;
.map_err(Error::from_core)?;
Ok(Json(webhook.into())) Ok(Json(webhook.into()))
} }
@@ -1,12 +1,10 @@
use revolt_database::{ use revolt_database::{
util::{idempotency::IdempotencyKey, reference::Reference}, util::{idempotency::IdempotencyKey, reference::Reference},
Database, Database, Message,
};
use revolt_quark::{
models::message::{DataMessageSend, Message},
types::push::MessageAuthor,
Db, Error, Result,
}; };
use revolt_models::v0;
use revolt_permissions::{ChannelPermission, PermissionValue};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
use validator::Validate; use validator::Validate;
@@ -18,31 +16,52 @@ use validator::Validate;
#[post("/<webhook_id>/<token>", data = "<data>")] #[post("/<webhook_id>/<token>", data = "<data>")]
pub async fn webhook_execute( pub async fn webhook_execute(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
webhook_id: Reference, webhook_id: Reference,
token: String, token: String,
data: Json<DataMessageSend>, data: Json<v0::DataMessageSend>,
idempotency: IdempotencyKey, idempotency: IdempotencyKey,
) -> Result<Json<Message>> { ) -> Result<Json<v0::Message>> {
let data = data.into_inner(); let data = data.into_inner();
data.validate() data.validate().map_err(|error| {
.map_err(|error| Error::FailedValidation { error })?; create_error!(FailedValidation {
error: error.to_string()
})
})?;
let webhook = webhook_id.as_webhook(db).await.map_err(Error::from_core)?; let webhook = webhook_id.as_webhook(db).await?;
webhook.assert_token(&token).map_err(Error::from_core)?; webhook.assert_token(&token)?;
data.validate_webhook_permissions(webhook.permissions)?; let permissions: PermissionValue = webhook.permissions.into();
permissions.throw_if_lacking_channel_permission(ChannelPermission::SendMessage)?;
let channel = legacy_db.fetch_channel(&webhook.channel_id).await?; if data.attachments.as_ref().map_or(false, |v| !v.is_empty()) {
let message = channel permissions.throw_if_lacking_channel_permission(ChannelPermission::UploadFiles)?;
.send_message( }
legacy_db,
if data.embeds.as_ref().map_or(false, |v| !v.is_empty()) {
permissions.throw_if_lacking_channel_permission(ChannelPermission::SendEmbeds)?;
}
if data.masquerade.is_some() {
permissions.throw_if_lacking_channel_permission(ChannelPermission::Masquerade)?;
}
if data.interactions.is_some() {
permissions.throw_if_lacking_channel_permission(ChannelPermission::React)?;
}
let channel = db.fetch_channel(&webhook.channel_id).await?;
Ok(Json(
Message::create_from_api(
db,
channel,
data, data,
MessageAuthor::Webhook(&webhook.into()), v0::MessageAuthor::Webhook(&webhook.into()),
idempotency, idempotency,
true, true,
) )
.await?; .await?
.into(),
Ok(Json(message)) ))
} }
@@ -1,10 +1,6 @@
use revolt_database::{util::reference::Reference, Database}; use revolt_database::{util::reference::Reference, Database, Message};
use revolt_models::v0::Webhook; use revolt_models::v0::{MessageAuthor, SendableEmbed, Webhook};
use revolt_quark::{ use revolt_result::{create_error, Error, Result};
models::{message::SendableEmbed, Message},
types::push::MessageAuthor,
Db, Error, Result,
};
use revolt_rocket_okapi::{ use revolt_rocket_okapi::{
gen::OpenApiGenerator, gen::OpenApiGenerator,
request::{OpenApiFromRequest, RequestHeaderInput}, request::{OpenApiFromRequest, RequestHeaderInput},
@@ -639,7 +635,7 @@ impl<'r> FromRequest<'r> for EventHeader<'r> {
async fn from_request(request: &'r Request<'_>) -> rocket::request::Outcome<Self, Self::Error> { async fn from_request(request: &'r Request<'_>) -> rocket::request::Outcome<Self, Self::Error> {
let headers = request.headers(); let headers = request.headers();
let Some(event) = headers.get_one("X-GitHub-Event") else { let Some(event) = headers.get_one("X-GitHub-Event") else {
return rocket::request::Outcome::Failure((Status::BadRequest, Error::InvalidOperation)) return rocket::request::Outcome::Failure((Status::BadRequest, create_error!(InvalidOperation)))
}; };
rocket::request::Outcome::Success(Self(event)) rocket::request::Outcome::Success(Self(event))
@@ -702,7 +698,7 @@ fn safe_from_str<T: for<'de> Deserialize<'de>>(data: &str) -> Result<T> {
Ok(output) => Ok(output), Ok(output) => Ok(output),
Err(err) => { Err(err) => {
log::error!("{err:?}"); log::error!("{err:?}");
Err(Error::InvalidOperation) Err(create_error!(InvalidOperation))
} }
} }
} }
@@ -727,7 +723,7 @@ fn convert_event(data: &str, event_name: &str) -> Result<Event> {
"issue_comment" => BaseEvent::IssueComment(safe_from_str(data)?), "issue_comment" => BaseEvent::IssueComment(safe_from_str(data)?),
"issues" => BaseEvent::Issues(safe_from_str(data)?), "issues" => BaseEvent::Issues(safe_from_str(data)?),
"pull_request" => BaseEvent::PullRequest(safe_from_str(data)?), "pull_request" => BaseEvent::PullRequest(safe_from_str(data)?),
_ => return Err(Error::InvalidOperation), _ => return Err(create_error!(InvalidOperation)),
}; };
let _Event { let _Event {
@@ -751,16 +747,15 @@ fn convert_event(data: &str, event_name: &str) -> Result<Event> {
#[post("/<webhook_id>/<token>/github", data = "<data>")] #[post("/<webhook_id>/<token>/github", data = "<data>")]
pub async fn webhook_execute_github( pub async fn webhook_execute_github(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
webhook_id: Reference, webhook_id: Reference,
token: String, token: String,
event: EventHeader<'_>, event: EventHeader<'_>,
data: String, data: String,
) -> Result<()> { ) -> Result<()> {
let webhook = webhook_id.as_webhook(db).await.map_err(Error::from_core)?; let webhook = webhook_id.as_webhook(db).await?;
webhook.assert_token(&token).map_err(Error::from_core)?; webhook.assert_token(&token)?;
let channel = legacy_db.fetch_channel(&webhook.channel_id).await?; let channel = db.fetch_channel(&webhook.channel_id).await?;
let event = convert_event(&data, &event)?; let event = convert_event(&data, &event)?;
let sendable_embed = match event.event { let sendable_embed = match event.event {
@@ -1058,30 +1053,25 @@ pub async fn webhook_execute_github(
}, },
}; };
sendable_embed sendable_embed.validate().map_err(|error| {
.validate() create_error!(FailedValidation {
.map_err(|error| Error::FailedValidation { error })?; error: error.to_string()
})
})?;
let message_id = Ulid::new().to_string(); let message_id = Ulid::new().to_string();
let embed = sendable_embed
.into_embed(legacy_db, &message_id)
.await?;
let mut message = Message { let mut message = Message {
id: message_id, id: message_id,
author: webhook.id.clone(), author: webhook.id.clone(),
channel: webhook.channel_id.clone(), channel: webhook.channel_id.clone(),
embeds: Some(vec![embed]),
webhook: Some(std::convert::Into::<Webhook>::into(webhook.clone()).into()), webhook: Some(std::convert::Into::<Webhook>::into(webhook.clone()).into()),
..Default::default() ..Default::default()
}; };
#[allow(clippy::disallowed_methods)]
message.attach_sendable_embed(db, sendable_embed).await?;
message message
.create( .send(db, MessageAuthor::Webhook(&webhook.into()), &channel, false)
legacy_db,
&channel,
Some(MessageAuthor::Webhook(&webhook.into())),
)
.await .await
} }
@@ -1,6 +1,10 @@
use revolt_database::{util::reference::Reference, Database}; use revolt_database::{
util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, User,
};
use revolt_models::v0::{ResponseWebhook, Webhook}; use revolt_models::v0::{ResponseWebhook, Webhook};
use revolt_quark::{models::User, perms, Db, Error, Permission, Result}; use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
use revolt_result::Result;
use rocket::{serde::json::Json, State}; use rocket::{serde::json::Json, State};
/// # Gets a webhook /// # Gets a webhook
@@ -10,17 +14,16 @@ use rocket::{serde::json::Json, State};
#[get("/<webhook_id>")] #[get("/<webhook_id>")]
pub async fn webhook_fetch( pub async fn webhook_fetch(
db: &State<Database>, db: &State<Database>,
legacy_db: &Db,
webhook_id: Reference, webhook_id: Reference,
user: User, user: User,
) -> Result<Json<ResponseWebhook>> { ) -> Result<Json<ResponseWebhook>> {
let webhook = webhook_id.as_webhook(db).await.map_err(Error::from_core)?; let webhook = webhook_id.as_webhook(db).await?;
let channel = legacy_db.fetch_channel(&webhook.channel_id).await?; let channel = db.fetch_channel(&webhook.channel_id).await?;
perms(&user) let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
.channel(&channel) calculate_channel_permissions(&mut query)
.throw_permission(legacy_db, Permission::ViewChannel) .await
.await?; .throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
Ok(Json(std::convert::Into::<Webhook>::into(webhook).into())) Ok(Json(std::convert::Into::<Webhook>::into(webhook).into()))
} }
+39 -14
View File
@@ -1,7 +1,8 @@
use futures::StreamExt; use futures::StreamExt;
use rand::Rng; use rand::Rng;
use redis_kiss::redis::aio::PubSub; use redis_kiss::redis::aio::PubSub;
use revolt_database::{events::client::EventV1, Database, DatabaseInfo, User}; use revolt_database::{events::client::EventV1, Database, User};
use revolt_models::v0;
use revolt_quark::authifier::{ use revolt_quark::authifier::{
models::{Account, Session}, models::{Account, Session},
Authifier, Authifier,
@@ -13,7 +14,7 @@ pub struct TestHarness {
authifier: Authifier, authifier: Authifier,
pub db: Database, pub db: Database,
sub: PubSub, sub: PubSub,
event_buffer: Vec<EventV1>, event_buffer: Vec<(String, EventV1)>,
} }
impl TestHarness { impl TestHarness {
@@ -24,20 +25,27 @@ impl TestHarness {
.await .await
.expect("valid rocket instance"); .expect("valid rocket instance");
let db = DatabaseInfo::Auto.connect().await.expect("`Database`");
let mut sub = redis_kiss::open_pubsub_connection() let mut sub = redis_kiss::open_pubsub_connection()
.await .await
.expect("`PubSub`"); .expect("`PubSub`");
sub.psubscribe("*").await.unwrap(); sub.psubscribe("*").await.unwrap();
let db = client
.rocket()
.state::<Database>()
.expect("`Database`")
.clone();
let authifier = client
.rocket()
.state::<Authifier>()
.expect("`Authifier`")
.clone();
TestHarness { TestHarness {
client, client,
authifier: Authifier { authifier,
database: db.clone().into(),
config: revolt_quark::util::authifier::config(),
event_channel: None,
},
db, db,
sub, sub,
event_buffer: vec![], event_buffer: vec![],
@@ -80,12 +88,12 @@ impl TestHarness {
(account, session, user) (account, session, user)
} }
pub async fn wait_for_event<F>(&mut self, predicate: F) -> EventV1 pub async fn wait_for_event<F>(&mut self, topic: &str, predicate: F) -> EventV1
where where
F: Fn(&EventV1) -> bool, F: Fn(&EventV1) -> bool,
{ {
for event in &self.event_buffer { for (msg_topic, event) in &self.event_buffer {
if predicate(event) { if topic == msg_topic && predicate(event) {
// does not remove from buffer // does not remove from buffer
return event.clone(); return event.clone();
} }
@@ -93,13 +101,15 @@ impl TestHarness {
let mut stream = self.sub.on_message(); let mut stream = self.sub.on_message();
while let Some(item) = stream.next().await { while let Some(item) = stream.next().await {
let payload: EventV1 = redis_kiss::decode_payload(&item.unwrap()).unwrap(); let item = item.unwrap();
let msg_topic = item.get_channel_name();
let payload: EventV1 = redis_kiss::decode_payload(&item).unwrap();
if predicate(&payload) { if topic == msg_topic && predicate(&payload) {
return payload; return payload;
} }
self.event_buffer.push(payload); self.event_buffer.push((msg_topic.to_string(), payload));
} }
// WARNING: if predicate is never satisfied, this will never return // WARNING: if predicate is never satisfied, this will never return
@@ -107,4 +117,19 @@ impl TestHarness {
unreachable!() unreachable!()
} }
pub async fn wait_for_message(&mut self, channel_id: &str) -> v0::Message {
dbg!(&self.event_buffer);
match self
.wait_for_event(channel_id, |event| match event {
EventV1::Message(v0::Message { channel, .. }) => channel == channel_id,
_ => false,
})
.await
{
EventV1::Message(message) => message,
_ => unreachable!(),
}
}
} }
+2 -2
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "revolt-quark" name = "revolt-quark"
version = "0.6.7" version = "0.6.9"
edition = "2021" edition = "2021"
license = "AGPL-3.0-or-later" license = "AGPL-3.0-or-later"
@@ -88,7 +88,7 @@ rocket_empty = { version = "0.1.1", optional = true, features = ["schema"] }
rocket_cors = { optional = true, git = "https://github.com/lawliet89/rocket_cors", rev = "c17e8145baa4790319fdb6a473e465b960f55e7c" } rocket_cors = { optional = true, git = "https://github.com/lawliet89/rocket_cors", rev = "c17e8145baa4790319fdb6a473e465b960f55e7c" }
# Authifier # Authifier
authifier = { version = "1.0.7", features = ["async-std-runtime"] } authifier = { version = "1.0.8", features = ["async-std-runtime"] }
# Sentry # Sentry
sentry = "0.31.5" sentry = "0.31.5"
+12 -11
View File
@@ -31,6 +31,18 @@ impl DatabaseInfo {
pub async fn connect(self) -> Result<Database, String> { pub async fn connect(self) -> Result<Database, String> {
Ok(match self { Ok(match self {
DatabaseInfo::Auto => { DatabaseInfo::Auto => {
if let Ok(test_db) = env::var("TEST_DB") {
return match test_db.as_str() {
"REFERENCE" => DatabaseInfo::Dummy.connect().await,
"MONGODB" => {
DatabaseInfo::MongoDb(env::var("MONGODB").expect("`MONGODB` env"))
.connect()
.await
}
_ => unreachable!("must specify REFERENCE or MONGODB"),
};
}
if let Ok(uri) = env::var("MONGODB") { if let Ok(uri) = env::var("MONGODB") {
return DatabaseInfo::MongoDb(uri).connect().await; return DatabaseInfo::MongoDb(uri).connect().await;
} }
@@ -61,17 +73,6 @@ impl Deref for Database {
} }
} }
impl From<Database> for authifier::Database {
fn from(val: Database) -> Self {
match val {
Database::Dummy(_) => authifier::Database::default(),
Database::MongoDb(MongoDb(client)) => authifier::Database::MongoDb(
authifier::database::MongoDb(client.database("revolt")),
),
}
}
}
impl From<Database> for revolt_database::Database { impl From<Database> for revolt_database::Database {
fn from(val: Database) -> Self { fn from(val: Database) -> Self {
match val { match val {
+2 -2
View File
@@ -13,12 +13,12 @@ pub mod process_embeds;
pub mod web_push; pub mod web_push;
/// Spawn background workers /// Spawn background workers
pub async fn start_workers(db: Database) { pub async fn start_workers(db: Database, authifier_db: authifier::Database) {
for _ in 0..WORKER_COUNT { for _ in 0..WORKER_COUNT {
task::spawn(ack::worker(db.clone())); task::spawn(ack::worker(db.clone()));
task::spawn(last_message_id::worker(db.clone())); task::spawn(last_message_id::worker(db.clone()));
task::spawn(process_embeds::worker(db.clone())); task::spawn(process_embeds::worker(db.clone()));
task::spawn(web_push::worker(db.clone().into())); task::spawn(web_push::worker(authifier_db.clone()));
} }
} }
+5 -1
View File
@@ -21,10 +21,14 @@ in pkgs.mkShell {
# Database # Database
# pkgs.mongodb # pkgs.mongodb
# Rust # Cargo
pkgs.cargo pkgs.cargo
pkgs.cargo-nextest
# Rust
pkgs.rustc pkgs.rustc
pkgs.clippy pkgs.clippy
pkgs.rustfmt
pkgs.pkgconfig pkgs.pkgconfig
pkgs.openssl.dev pkgs.openssl.dev
]; ];
+1
View File
@@ -1,4 +1,5 @@
publish: publish:
cargo publish --package revolt-config
cargo publish --package revolt-result cargo publish --package revolt-result
cargo publish --package revolt-permissions cargo publish --package revolt-permissions
cargo publish --package revolt-models cargo publish --package revolt-models
+3
View File
@@ -25,6 +25,7 @@ deps() {
crates/bonfire/src \ crates/bonfire/src \
crates/delta/src \ crates/delta/src \
crates/quark/src \ crates/quark/src \
crates/core/config/src \
crates/core/database/src \ crates/core/database/src \
crates/core/models/src \ crates/core/models/src \
crates/core/permissions/src \ crates/core/permissions/src \
@@ -35,6 +36,7 @@ deps() {
tee crates/delta/src/main.rs tee crates/delta/src/main.rs
echo '' | echo '' |
tee crates/quark/src/lib.rs | tee crates/quark/src/lib.rs |
tee crates/core/config/src/lib.rs |
tee crates/core/database/src/lib.rs | tee crates/core/database/src/lib.rs |
tee crates/core/models/src/lib.rs | tee crates/core/models/src/lib.rs |
tee crates/core/permissions/src/lib.rs | tee crates/core/permissions/src/lib.rs |
@@ -53,6 +55,7 @@ apps() {
crates/bonfire/src/main.rs \ crates/bonfire/src/main.rs \
crates/delta/src/main.rs \ crates/delta/src/main.rs \
crates/quark/src/lib.rs \ crates/quark/src/lib.rs \
crates/core/config/src/lib.rs \
crates/core/database/src/lib.rs \ crates/core/database/src/lib.rs \
crates/core/models/src/lib.rs \ crates/core/models/src/lib.rs \
crates/core/permissions/src/lib.rs \ crates/core/permissions/src/lib.rs \