From f9cd95fdce180037fbb50101066d959ba3e083ca Mon Sep 17 00:00:00 2001
From: Jordan Schelew <jordan@websavers.ca>
Date: Fri, 28 Feb 2025 10:40:40 -0400
Subject: [PATCH] Add Sieve support to auto domain grab plugin

---
 plugins/auto-domain-grab/index.php | 35 +++++++++++++++++++++++++-----
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/plugins/auto-domain-grab/index.php b/plugins/auto-domain-grab/index.php
index ee2458948..4ac8dad07 100644
--- a/plugins/auto-domain-grab/index.php
+++ b/plugins/auto-domain-grab/index.php
@@ -1,10 +1,10 @@
 <?php
 
 /**
- * This extension automatically detects the IMAP and SMTP settings by
+ * This extension automatically detects the IMAP, SMTP, and Sieve settings by
  * extracting them from the email address itself. For example, if the user
- * attemps to login as 'info@example.com', then the IMAP and SMTP host would
- * be set to to 'example.com'.
+ * attemps to login as 'info@example.com', then the IMAP, SMTP, and Sieve host 
+ * would be set to to 'example.com'.
  *
  * Based on:
  * https://github.com/the-djmaze/snappymail/blob/master/plugins/override-smtp-credentials/index.php
@@ -19,15 +19,17 @@ class AutoDomainGrabPlugin extends \RainLoop\Plugins\AbstractPlugin
 		RELEASE = '2022-11-11',
 		REQUIRED = '2.21.0',
 		CATEGORY = 'General',
-		DESCRIPTION = 'Sets the IMAP/SMTP host based on the user\'s login';
+		DESCRIPTION = 'Sets the IMAP, SMTP, Sieve host based on the user\'s login';
 
 	private $imap_prefix = 'mail.';
 	private $smtp_prefix = 'mail.';
-
+	private $sieve_prefix = 'mail.';
+	
 	public function Init() : void
 	{
-		$this->addHook('smtp.before-connect', 'FilterSmtpCredentials');
 		$this->addHook('imap.before-connect', 'FilterImapCredentials');
+		$this->addHook('smtp.before-connect', 'FilterSmtpCredentials');
+		$this->addHook('sieve.before-connect', 'FilterSieveCredentials');
 	}
 
 	/**
@@ -71,4 +73,25 @@ class AutoDomainGrabPlugin extends \RainLoop\Plugins\AbstractPlugin
 			}
 		}
 	}
+
+	/**
+	 * This function detects the Sieve Host, and if it is set to 'auto', replaces it with the MX or email domain.
+	 */
+	public function FilterSieveCredentials(\RainLoop\Model\Account $oAccount, \MailSo\Sieve\SieveClient $oSieveClient, \MailSo\Sieve\Settings $oSettings)
+	{
+		// Check for mail.$DOMAIN as entered value in RL settings
+		if ('auto' === $oSettings->host)
+		{
+			$domain = \substr(\strrchr($oAccount->Email(), '@'), 1);
+			$mxhosts = array();
+			if (\getmxrr($domain, $mxhosts) && $mxhosts)
+			{
+				$oSettings->host = $mxhosts[0];
+			}
+			else
+			{
+				$oSettings->host = $this->sieve_prefix.$domain;
+			}
+		}
+	}
 }