feat(media): blocking

src/service/media/data.rs

@@ -1,7 +1,9 @@
 use ruma::{OwnedServerName, ServerName, UserId};
 use sha2::{digest::Output, Sha256};
 
-use crate::Result;
+use crate::{Error, Result};
+
+use super::BlockedMediaInfo;
 
 use super::DbFileMeta;
 
@@ -16,6 +18,7 @@ pub trait Data: Send + Sync {
         filename: Option<&str>,
         content_type: Option<&str>,
         user_id: Option<&UserId>,
+        is_blocked_filehash: bool,
     ) -> Result<()>;
 
     fn search_file_metadata(&self, servername: &ServerName, media_id: &str) -> Result<DbFileMeta>;
@@ -62,4 +65,32 @@ pub trait Data: Send + Sync {
         force_filehash: bool,
         after: Option<u64>,
     ) -> Vec<Result<String>>;
+
+    fn is_blocked(&self, server_name: &ServerName, media_id: &str) -> Result<bool>;
+
+    fn block(
+        &self,
+        media: &[(OwnedServerName, String)],
+        unix_secs: u64,
+        reason: Option<String>,
+    ) -> Vec<Error>;
+
+    fn block_from_user(
+        &self,
+        user_id: &UserId,
+        now: u64,
+        reason: &str,
+        after: Option<u64>,
+    ) -> Vec<Error>;
+
+    fn unblock(&self, media: &[(OwnedServerName, String)]) -> Vec<Error>;
+
+    /// Returns a Vec of:
+    /// - The server the media is from
+    /// - The media id
+    /// - The time it was blocked, in unix seconds
+    /// - The optional reason why it was blocked
+    fn list_blocked(&self) -> Vec<Result<BlockedMediaInfo>>;
+
+    fn is_blocked_filehash(&self, sha256_digest: &[u8]) -> Result<bool>;
 }