1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use crate::api::_generic::{handle_empty_response, handle_response};
use crate::api::permission::{RabbitMqPermission, RabbitMqTopicPermission};
use crate::errors::RabbitMqClientError;
use crate::RabbitMqClient;
use async_trait::async_trait;
use serde::{Deserialize, Serialize};

#[async_trait]
pub trait UserApi {
    async fn who_am_i(&self) -> Result<RabbitMqWhoAmI, RabbitMqClientError>;

    async fn list_users(&self) -> Result<Vec<RabbitMqUser>, RabbitMqClientError>;

    async fn list_users_without_permissions(
        &self,
    ) -> Result<Vec<RabbitMqUser>, RabbitMqClientError>;

    async fn bulk_delete_users(
        &self,
        users: RabbitMqUsersBulkDeleteRequest,
    ) -> Result<(), RabbitMqClientError>;

    async fn list_user_permissions(
        &self,
        user: String,
    ) -> Result<Vec<RabbitMqPermission>, RabbitMqClientError>;

    async fn list_user_topic_permissions(
        &self,
        user: String,
    ) -> Result<Vec<RabbitMqTopicPermission>, RabbitMqClientError>;
}

#[async_trait]
impl UserApi for RabbitMqClient {
    #[tracing::instrument(skip(self))]
    async fn who_am_i(&self) -> Result<RabbitMqWhoAmI, RabbitMqClientError> {
        let response = self
            .client
            .request(reqwest::Method::GET, format!("{}/api/whoami", self.api_url))
            .send()
            .await?;

        handle_response(response).await
    }

    #[tracing::instrument(skip(self))]
    async fn list_users(&self) -> Result<Vec<RabbitMqUser>, RabbitMqClientError> {
        let response = self
            .client
            .request(reqwest::Method::GET, format!("{}/api/users", self.api_url))
            .send()
            .await?;

        handle_response(response).await
    }

    #[tracing::instrument(skip(self))]
    async fn list_users_without_permissions(
        &self,
    ) -> Result<Vec<RabbitMqUser>, RabbitMqClientError> {
        let response = self
            .client
            .request(
                reqwest::Method::GET,
                format!("{}/api/users/without-permissions", self.api_url),
            )
            .send()
            .await?;

        handle_response(response).await
    }

    #[tracing::instrument(skip(self))]
    async fn bulk_delete_users(
        &self,
        users: RabbitMqUsersBulkDeleteRequest,
    ) -> Result<(), RabbitMqClientError> {
        let response = self
            .client
            .request(
                reqwest::Method::DELETE,
                format!("{}/api/users/bulk-delete", self.api_url),
            )
            .json(&users)
            .send()
            .await?;

        handle_empty_response(response).await
    }

    #[tracing::instrument(skip(self))]
    async fn list_user_permissions(
        &self,
        user: String,
    ) -> Result<Vec<RabbitMqPermission>, RabbitMqClientError> {
        let response = self
            .client
            .request(
                reqwest::Method::DELETE,
                format!("{}/api/users/{}/permissions", self.api_url, user),
            )
            .send()
            .await?;

        handle_response(response).await
    }

    #[tracing::instrument(skip(self))]
    async fn list_user_topic_permissions(
        &self,
        user: String,
    ) -> Result<Vec<RabbitMqTopicPermission>, RabbitMqClientError> {
        let response = self
            .client
            .request(
                reqwest::Method::DELETE,
                format!("{}/api/users/{}/topic-permissions", self.api_url, user),
            )
            .send()
            .await?;

        handle_response(response).await
    }
}

#[derive(Debug, Deserialize)]
pub struct RabbitMqWhoAmI {
    pub name: String,
    pub tags: Vec<String>,
}

#[derive(Debug, Deserialize)]
pub struct RabbitMqUser {
    pub name: String,
    pub password_hash: String,
    pub hashing_algorithm: String,
    pub tags: Vec<String>,
}

#[derive(Debug, Serialize)]
pub struct RabbitMqUsersBulkDeleteRequest {
    pub users: Vec<String>,
}