Import upstream v0.16.22, stripped

Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
2026-09-18 10:21:56 -07:00
commit 7dae9b29fd
1650 changed files with 485521 additions and 0 deletions
@@ -0,0 +1,94 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use crate::core::{Command, ResponseCode, Session, StatusResponse};
+use common::network::SessionStream;
+use email::sieve::SieveScript;
+use imap_proto::receiver::Request;
+use registry::schema::enums::Permission;
+use std::time::Instant;
+use store::{
+    ValueKey,
+    write::{AlignedBytes, Archive},
+};
+use trc::AddContext;
+use types::{blob::BlobSection, blob_hash::BlobHash, collection::Collection};
+
+impl<T: SessionStream> Session<T> {
+    pub async fn handle_getscript(&mut self, request: Request<Command>) -> trc::Result<Vec<u8>> {
+        // Validate access
+        self.assert_has_permission(Permission::SieveGetScript)?;
+
+        let op_start = Instant::now();
+        let name = request
+            .tokens
+            .into_iter()
+            .next()
+            .and_then(|s| s.unwrap_string().ok())
+            .ok_or_else(|| {
+                trc::ManageSieveEvent::Error
+                    .into_err()
+                    .details("Expected script name as a parameter.")
+            })?;
+        let account_id = self.state.access_token().account_id();
+        let document_id = self.get_script_id(account_id, &name).await?;
+        let sieve_ = self
+            .server
+            .store()
+            .get_value::<Archive<AlignedBytes>>(ValueKey::archive(
+                account_id,
+                Collection::SieveScript,
+                document_id,
+            ))
+            .await
+            .caused_by(trc::location!())?
+            .ok_or_else(|| {
+                trc::ManageSieveEvent::Error
+                    .into_err()
+                    .details("Script not found")
+                    .code(ResponseCode::NonExistent)
+            })?;
+        let sieve = sieve_
+            .unarchive::<SieveScript>()
+            .caused_by(trc::location!())?;
+        let blob_size = u32::from(sieve.size) as usize;
+        let script = self
+            .server
+            .get_blob_section(
+                &BlobHash::from(&sieve.blob_hash),
+                &BlobSection {
+                    size: blob_size,
+                    ..Default::default()
+                },
+            )
+            .await
+            .caused_by(trc::location!())?
+            .ok_or_else(|| {
+                trc::ManageSieveEvent::Error
+                    .into_err()
+                    .details("Script blob not found")
+                    .code(ResponseCode::NonExistent)
+            })?;
+        debug_assert_eq!(script.len(), blob_size);
+
+        let mut response = Vec::with_capacity(script.len() + 32);
+        response.push(b'{');
+        response.extend_from_slice(blob_size.to_string().as_bytes());
+        response.extend_from_slice(b"}\r\n");
+        response.extend(script);
+        response.extend_from_slice(b"\r\n");
+
+        trc::event!(
+            ManageSieve(trc::ManageSieveEvent::GetScript),
+            SpanId = self.session_id,
+            Id = name,
+            DocumentId = document_id,
+            Elapsed = op_start.elapsed()
+        );
+
+        Ok(StatusResponse::ok("").serialize(response))
+    }
+}