stoatchat/src/routes/users/unblock_user.rs

use crate::{
    database::entities::RelationshipStatus, database::entities::User, database::get_collection,
    database::guards::reference::Ref, database::permissions::get_relationship, util::result::Error,
};
use crate::{
    notifications::{events::ClientboundNotification, hive},
    util::result::Result,
};
use futures::try_join;
use hive_pubsub::PubSub;
use mongodb::bson::doc;
use rocket_contrib::json::JsonValue;

#[delete("/<target>/block")]
pub async fn req(user: User, target: Ref) -> Result<JsonValue> {
    let col = get_collection("users");

    match get_relationship(&user, &target) {
        RelationshipStatus::Blocked => {
            match get_relationship(&target.fetch_user().await?, &user.as_ref()) {
                RelationshipStatus::Blocked => {
                    col.update_one(
                        doc! {
                            "_id": &user.id,
                            "relations._id": &target.id
                        },
                        doc! {
                            "$set": {
                                "relations.$.status": "BlockedOther"
                            }
                        },
                        None,
                    )
                    .await
                    .map_err(|_| Error::DatabaseError {
                        operation: "update_one",
                        with: "user",
                    })?;

                    ClientboundNotification::UserRelationship {
                        id: user.id.clone(),
                        user: target.id.clone(),
                        status: RelationshipStatus::BlockedOther,
                    }
                    .publish(user.id.clone())
                    .await
                    .ok();

                    Ok(json!({ "status": "BlockedOther" }))
                }
                RelationshipStatus::BlockedOther => {
                    match try_join!(
                        col.update_one(
                            doc! {
                                "_id": &user.id
                            },
                            doc! {
                                "$pull": {
                                    "relations": {
                                        "_id": &target.id
                                    }
                                }
                            },
                            None
                        ),
                        col.update_one(
                            doc! {
                                "_id": &target.id
                            },
                            doc! {
                                "$pull": {
                                    "relations": {
                                        "_id": &user.id
                                    }
                                }
                            },
                            None
                        )
                    ) {
                        Ok(_) => {
                            try_join!(
                                ClientboundNotification::UserRelationship {
                                    id: user.id.clone(),
                                    user: target.id.clone(),
                                    status: RelationshipStatus::None
                                }
                                .publish(user.id.clone()),
                                ClientboundNotification::UserRelationship {
                                    id: target.id.clone(),
                                    user: user.id.clone(),
                                    status: RelationshipStatus::None
                                }
                                .publish(target.id.clone())
                            )
                            .ok();

                            let hive = hive::get_hive();
                            hive.unsubscribe(&user.id, &target.id).ok();
                            hive.unsubscribe(&target.id, &user.id).ok();

                            Ok(json!({ "status": "None" }))
                        }
                        Err(_) => Err(Error::DatabaseError {
                            operation: "update_one",
                            with: "user",
                        }),
                    }
                }
                _ => Err(Error::InternalError),
            }
        }
        _ => Err(Error::NoEffect),
    }
}