feat: [#615] authorization layer added to the get torrent info method…

… of the torrent service
torrust · Aug 3, 2024 · f8e7570 · f8e7570
1 parent 1342dce
commit f8e7570
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/services/authorization.rs b/src/services/authorization.rs
@@ -49,6 +49,7 @@ pub enum ACTION {
     AddTorrent,
     GetCanonicalInfoHash,
     GenerateTorrentInfoListing,
+    GetTorrentInfo,
 }
 
 pub struct Service {
@@ -192,6 +193,7 @@ impl CasbinConfiguration {
                 admin, AddTorrent
                 admin, GetCanonicalInfoHash
                 admin, GenerateTorrentInfoListing
+                admin, GetTorrentInfo
                 registered, GetCategories
                 registered, GetImageByUrl
                 registered, GetPublicSettings
@@ -200,6 +202,7 @@ impl CasbinConfiguration {
                 registered, AddTorrent
                 registered, GetCanonicalInfoHash
                 registered, GenerateTorrentInfoListing
+                registered, GetTorrentInfo
                 guest, GetCategories
                 guest, GetTags
                 guest, GetAboutPage
@@ -208,6 +211,7 @@ impl CasbinConfiguration {
                 guest, GetTorrent
                 guest, GetCanonicalInfoHash
                 guest, GenerateTorrentInfoListing
+                guest, GetTorrentInfo
                 ",
             ),
         }

diff --git a/src/services/torrent.rs b/src/services/torrent.rs
@@ -333,6 +333,10 @@ impl Index {
         info_hash: &InfoHash,
         opt_user_id: Option<UserId>,
     ) -> Result<TorrentResponse, ServiceError> {
+        self.authorization_service
+            .authorize(ACTION::GetTorrentInfo, opt_user_id)
+            .await?;
+
         let torrent_listing = self.torrent_listing_generator.one_torrent_by_info_hash(info_hash).await?;
 
         let torrent_response = self