From 59478e06c25303a790a0840ab2ac30662c4ef781 Mon Sep 17 00:00:00 2001
From: Hugues Hiegel <root@paranoid>
Date: Tue, 5 Aug 2014 16:46:22 +0200
Subject:  c'est la merde..

---
 plugins/markasjunk2/drivers/cmd_learn.php | 74 +++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 plugins/markasjunk2/drivers/cmd_learn.php

(limited to 'plugins/markasjunk2/drivers/cmd_learn.php')

diff --git a/plugins/markasjunk2/drivers/cmd_learn.php b/plugins/markasjunk2/drivers/cmd_learn.php
new file mode 100644
index 000000000..f27adc1d1
--- /dev/null
+++ b/plugins/markasjunk2/drivers/cmd_learn.php
@@ -0,0 +1,74 @@
+<?php
+
+/**
+ * Command line learn driver
+ * @version 2.0
+ * @author Philip Weir
+ * Patched by Julien Vehent to support DSPAM
+ * Enhanced support for DSPAM by Stevan Bajic <stevan@bajic.ch>
+ */
+
+class markasjunk2_cmd_learn
+{
+	public function spam($uids)
+	{
+		$this->_do_salearn($uids, true);
+	}
+
+	public function ham($uids)
+	{
+		$this->_do_salearn($uids, false);
+	}
+
+	private function _do_salearn($uids, $spam)
+	{
+		$rcmail = rcube::get_instance();
+		$temp_dir = realpath($rcmail->config->get('temp_dir'));
+
+		if ($spam)
+			$command = $rcmail->config->get('markasjunk2_spam_cmd');
+		else
+			$command = $rcmail->config->get('markasjunk2_ham_cmd');
+
+		if (!$command)
+			return;
+
+		$command = str_replace('%u', $_SESSION['username'], $command);
+		$command = str_replace('%l', $rcmail->user->get_username('local'), $command);
+		$command = str_replace('%d', $rcmail->user->get_username('domain'), $command);
+		if (preg_match('/%i/', $command)) {
+			$identity_arr = $rcmail->user->get_identity();
+			$command = str_replace('%i', $identity_arr['email'], $command);
+		}
+
+		foreach (explode(",", $uids) as $uid) {
+			// get DSPAM signature from header (if %xds macro is used)
+			if (preg_match('/%xds/', $command)) {
+				if (preg_match('/^X\-DSPAM\-Signature:\s+((\d+,)?([a-f\d]+))\s*$/im', $rcmail->storage->get_raw_headers($uid), $dspam_signature))
+					$tmp_command = str_replace('%xds', $dspam_signature[1], $command);
+				else
+					continue; // no DSPAM signature found in headers -> continue with next uid/message
+			}
+
+			if (preg_match('/%f/', $command)) {
+				$tmpfname = tempnam($temp_dir, 'rcmSALearn');
+				file_put_contents($tmpfname, $rcmail->storage->get_raw_body($uid));
+				$tmp_command = str_replace('%f', $tmpfname, $command);
+			}
+
+			exec($tmp_command, $output);
+
+			if ($rcmail->config->get('markasjunk2_debug')) {
+				rcube::write_log('markasjunk2', $tmp_command);
+				rcube::write_log('markasjunk2', $output);
+			}
+
+			if (preg_match('/%f/', $command))
+				unlink($tmpfname);
+
+			$output = '';
+		}
+	}
+}
+
+?>
\ No newline at end of file
-- 
cgit v1.2.3