[Add] Client settings

iota-storage/src/util/outgoing_relay.rs

@@ -0,0 +1,61 @@
+/* Accepts an outgoing relay only when its retry record and retention state commit together. */
+use crate::storage_error::StorageError;
+use crate::util::{chat_files, db, message_storage_policy, relay_queue, relay_replay};
+use iota_util::route_target::RouteTarget;
+use rusqlite::OptionalExtension;
+
+pub struct OutgoingRelay<'a> {
+    pub target: RouteTarget,
+    pub identity: &'a relay_queue::RelayIdentity,
+    pub frame: &'a [u8],
+    pub created_at: i64,
+    pub frame_id: u32,
+    pub type_map_version: &'a str,
+}
+
+pub fn commit_outgoing_relay(relay: OutgoingRelay<'_>) -> Result<(), StorageError> {
+    db::with_immediate_transaction(|tx| {
+        relay_queue::enqueue_in_tx(
+            tx,
+            relay.target,
+            relay.identity,
+            relay.frame,
+            relay.created_at,
+            relay.frame_id,
+            relay.type_map_version,
+        )?;
+        if message_storage_policy::get_in_tx(tx, relay.identity.signer_id)?.history_mode
+            == message_storage_policy::MessageHistoryMode::DeleteAfterClientDelivery
+        {
+            let message_id = tx.query_row("SELECT id FROM messages WHERE storage_owner = ?1 AND relay_signer_id = ?1 AND relay_message_id = ?2", rusqlite::params![relay.identity.signer_id, relay.identity.message_id], |row| row.get::<_, i64>(0)).optional()?;
+            if let Some(message_id) = message_id {
+                chat_files::remove_message_history_in_tx(tx, relay.identity.signer_id, message_id)?;
+            }
+        }
+        relay_replay::mark_queued_in_tx(tx, relay.identity.signer_id, &relay.identity.message_id)?;
+        Ok(())
+    })
+}
+
+pub fn apply_outgoing_history_policy(
+    storage_owner: i64,
+    relay_signer_id: i64,
+    relay_message_id: &str,
+) -> Result<(), StorageError> {
+    db::with_immediate_transaction(|tx| {
+        if message_storage_policy::get_in_tx(tx, storage_owner)?.history_mode
+            != message_storage_policy::MessageHistoryMode::DeleteAfterClientDelivery
+        {
+            return Ok(());
+        }
+        let message_id = tx.query_row(
+            "SELECT id FROM messages WHERE storage_owner = ?1 AND relay_signer_id = ?2 AND relay_message_id = ?3",
+            rusqlite::params![storage_owner, relay_signer_id, relay_message_id],
+            |row| row.get::<_, i64>(0),
+        ).optional()?;
+        if let Some(message_id) = message_id {
+            chat_files::remove_message_history_in_tx(tx, storage_owner, message_id)?;
+        }
+        Ok(())
+    })
+}