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,137 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use common::{KV_LOCK_TASK, Server};
+use registry::schema::enums::TaskType;
+use registry::schema::structs::Task;
+use registry::types::EnumImpl;
+use std::future::Future;
+use std::time::Instant;
+use store::ahash::AHashMap;
+use store::write::Operation;
+use tokio::sync::mpsc;
+use trc::TaskManagerEvent;
+
+pub mod acme;
+pub mod alarm;
+pub mod destroy_account;
+pub mod dkim;
+pub mod dns;
+pub mod imip;
+pub mod index;
+pub mod lock;
+pub mod maintenance;
+pub mod manager;
+pub mod merge_threads;
+pub mod report;
+pub mod restore_item;
+pub mod scheduler;
+pub mod spam_classifier;
+
+const QUEUE_REFRESH_INTERVAL: u64 = 60 * 5; // 5 minutes
+const DEFAULT_LOCK_EXPIRY: u64 = 60 * 60; // 1 hour
+
+pub(crate) struct TaskManagerIpc {
+    txs: [mpsc::Sender<TaskJob>; TaskType::COUNT],
+    locked: AHashMap<u64, Locked>,
+    revision: u64,
+}
+
+#[derive(Debug)]
+pub(crate) struct Locked {
+    expires: Instant,
+    due: u64,
+    revision: u64,
+}
+
+#[derive(Debug)]
+pub(crate) struct TaskDetails {
+    task: Task,
+    info: TaskJob,
+}
+
+#[derive(Debug)]
+pub(crate) struct TaskJob {
+    id: u64,
+    due: u64,
+    typ: TaskType,
+}
+
+#[derive(Debug, PartialEq, Eq)]
+pub(crate) enum TaskResult {
+    Success(Vec<Task>),
+    Update([Operation; 2]),
+    Failure {
+        typ: TaskFailureType,
+        message: String,
+        max_attempts: Option<u64>,
+    },
+    Ignored,
+}
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+#[allow(dead_code)]
+pub(crate) enum TaskFailureType {
+    Retry(u64),
+    Temporary,
+    Perpetual,
+    Permanent,
+}
+
+pub(crate) trait TaskInfo {
+    fn name(&self) -> &'static str;
+}
+
+impl TaskInfo for Task {
+    fn name(&self) -> &'static str {
+        match self {
+            Task::IndexDocument(_) => "IndexDocument",
+            Task::UnindexDocument(_) => "UnindexDocument",
+            Task::IndexTrace(_) => "IndexTrace",
+            Task::CalendarAlarmEmail(_) => "CalendarAlarmEmail",
+            Task::CalendarAlarmNotification(_) => "CalendarAlarmNotification",
+            Task::CalendarItipMessage(_) => "CalendarItipMessage",
+            Task::MergeThreads(_) => "MergeThreads",
+            Task::DmarcReport(_) => "DmarcReport",
+            Task::TlsReport(_) => "TlsReport",
+            Task::RestoreArchivedItem(_) => "RestoreArchivedItem",
+            Task::DestroyAccount(_) => "DestroyAccount",
+            Task::AccountMaintenance(_) => "AccountMaintenance",
+            Task::StoreMaintenance(_) => "StoreMaintenance",
+            Task::SpamFilterMaintenance(_) => "SpamFilterMaintenance",
+            Task::AcmeRenewal(_) => "AcmeRenewal",
+            Task::DkimManagement(_) => "DkimManagement",
+            Task::DnsManagement(_) => "DnsManagement",
+            Task::TenantMaintenance(_) => "TenantMaintenance",
+        }
+    }
+}
+
+impl TaskResult {
+    pub fn permanent(message: impl Into<String>) -> Self {
+        TaskResult::Failure {
+            typ: TaskFailureType::Permanent,
+            message: message.into(),
+            max_attempts: None,
+        }
+    }
+
+    pub fn temporary(message: impl Into<String>) -> Self {
+        TaskResult::Failure {
+            typ: TaskFailureType::Temporary,
+            message: message.into(),
+            max_attempts: None,
+        }
+    }
+
+    pub fn perpetual(message: impl Into<String>) -> Self {
+        TaskResult::Failure {
+            typ: TaskFailureType::Perpetual,
+            message: message.into(),
+            max_attempts: None,
+        }
+    }
+}