From c438411a3d5fa4b168be1ba8d4c7f2a7a8ab60e5 Mon Sep 17 00:00:00 2001
From: andryyy <andre.peters@debinux.de>
Date: Wed, 6 Feb 2019 09:21:54 +0100
Subject: [PATCH] [Dovecot] Extend quarantine template: add score and
 release/delete buttons - depending on acl

---
 data/Dockerfiles/dovecot/quarantine_notify.py | 17 +++++++++--------
 data/assets/templates/quarantine.tpl          | 14 ++++++++++++--
 2 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/data/Dockerfiles/dovecot/quarantine_notify.py b/data/Dockerfiles/dovecot/quarantine_notify.py
index 2d466235..b79bab2f 100755
--- a/data/Dockerfiles/dovecot/quarantine_notify.py
+++ b/data/Dockerfiles/dovecot/quarantine_notify.py
@@ -13,6 +13,7 @@ import json
 import redis
 import time
 import html2text
+import socket
 
 while True:
   try:
@@ -29,7 +30,7 @@ time_now = int(time.time())
 def query_mysql(query, headers = True, update = False):
   while True:
     try:
-      cnx = mysql.connector.connect(unix_socket = '/var/run/mysqld/mysqld.sock', user='__DBUSER__', passwd='__DBPASS__', database='__DBNAME__', charset="utf8")
+      cnx = mysql.connector.connect(unix_socket = '/var/run/mysqld/mysqld.sock', user='mailcow', passwd='9S6jd31ijEytZE9lQ8sWOxK34tg5', database='mailcow', charset="utf8")
     except Exception as ex:
       print '%s - trying again...'  % (ex)
       time.sleep(3)
@@ -53,8 +54,8 @@ def query_mysql(query, headers = True, update = False):
     cur.close()
     cnx.close()
 
-def notify_rcpt(rcpt, msg_count):
-  meta_query = query_mysql('SELECT id, subject, sender, created FROM quarantine WHERE notified = 0 AND rcpt = "%s"' % (rcpt))
+def notify_rcpt(rcpt, msg_count, quarantine_acl):
+  meta_query = query_mysql('SELECT SHA2(CONCAT(id, qid), 256) AS qhash, id, subject, score, sender, created FROM quarantine WHERE notified = 0 AND rcpt = "%s"' % (rcpt))
   if r.get('Q_HTML'):
     try:
       template = Template(r.get('Q_HTML'))
@@ -65,7 +66,7 @@ def notify_rcpt(rcpt, msg_count):
   else:
     with open('/templates/quarantine.tpl') as file_:
       template = Template(file_.read())
-  html = template.render(meta=meta_query, counter=msg_count)
+  html = template.render(meta=meta_query, counter=msg_count, hostname=socket.gethostname(), quarantine_acl=quarantine_acl)
   text = html2text.html2text(html)
   count = 0
   while count < 15:
@@ -92,7 +93,7 @@ def notify_rcpt(rcpt, msg_count):
       print '%s'  % (ex)
       time.sleep(3)
 
-records = query_mysql('SELECT count(id) AS counter, rcpt FROM quarantine WHERE notified = 0 GROUP BY rcpt')
+records = query_mysql('SELECT IFNULL(user_acl.quarantine, 0) AS quarantine_acl, count(id) AS counter, rcpt FROM quarantine LEFT OUTER JOIN user_acl ON user_acl.username = rcpt WHERE notified = 0 AND rcpt in (SELECT username FROM mailbox) GROUP BY rcpt')
 
 for record in records:
   attrs = ''
@@ -113,14 +114,14 @@ for record in records:
   if attrs['quarantine_notification'] == 'hourly':
     if last_notification == 0 or (last_notification + 3600) < time_now:
       print "Notifying %s about %d new items in quarantine" % (record['rcpt'], record['counter'])
-      notify_rcpt(record['rcpt'], record['counter'])
+      notify_rcpt(record['rcpt'], record['counter'], record['quarantine_acl'])
   elif attrs['quarantine_notification'] == 'daily':
     if last_notification == 0 or (last_notification + 86400) < time_now:
       print "Notifying %s about %d new items in quarantine" % (record['rcpt'], record['counter'])
-      notify_rcpt(record['rcpt'], record['counter'])
+      notify_rcpt(record['rcpt'], record['counter'], record['quarantine_acl'])
   elif attrs['quarantine_notification'] == 'weekly':
     if last_notification == 0 or (last_notification + 604800) < time_now:
       print "Notifying %s about %d new items in quarantine" % (record['rcpt'], record['counter'])
-      notify_rcpt(record['rcpt'], record['counter'])
+      notify_rcpt(record['rcpt'], record['counter'], record['quarantine_acl'])
   else:
     break
diff --git a/data/assets/templates/quarantine.tpl b/data/assets/templates/quarantine.tpl
index 2a0a310e..787a043f 100644
--- a/data/assets/templates/quarantine.tpl
+++ b/data/assets/templates/quarantine.tpl
@@ -2,7 +2,7 @@
   <head>
   <style>
   body {
-    font-family: sans-serif;
+    font-family: Helvetica, Arial, Sans-Serif;
   }
   table {
     border-collapse: collapse;
@@ -15,18 +15,28 @@
     border-bottom: 1px solid #ddd;
     vertical-align: top;
   }
+  th {
+    background-color: #56B04C;
+    color: white;
+  }
+  tr:nth-child(even){background-color: #f2f2f2}
+
   </style>
   </head>
   <body>
     <p>Hi!<br>
     There are {{counter}} new messages waiting in quarantine:<br>
     <table>
-    <tr><th>Subject</th><th>Sender</th><th>Arrived on</th></tr>
+    <tr><th>Subject</th><th>Sender</th><th>Score</th><th>Arrived on</th>{% if quarantine_acl == 1 %}<th>Actions</th>{% endif %}</tr>
     {% for line in meta %}
     <tr>
     <td>{{ line.subject|e }}</td>
     <td>{{ line.sender|e }}</td>
+    <td>{{ line.score }}</td>
     <td>{{ line.created }}</td>
+    {% if quarantine_acl == 1 %}
+    <td><a href="https://{{ hostname }}/qhandler/release/{{ line.qhash }}">release</a> | <a href="https://{{ hostname }}/qhandler/delete/{{ line.qhash }}">delete</a></td>
+    {% endif %}
     </tr>
     {% endfor %}
     </table>