Send notifications when changing co-maintainership

Implements FS#45590.

Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>

2 files changed, 33 insertions, 0 deletions

diff --git a/scripts/notify.py b/scripts/notify.py
index ab652772..4832baf7 100755
--- a/scripts/notify.py
+++ b/scripts/notify.py
@@ -65,6 +65,10 @@ def pkgbase_from_pkgreq(cur, reqid):
                 [reqid])
     return cur.fetchone()[0]
 
+def get_user_email(cur, uid):
+    cur.execute('SELECT Email FROM Users WHERE ID = %s', [uid])
+    return cur.fetchone()[0]
+
 def get_maintainer_email(cur, pkgbase_id):
     cur.execute('SELECT Users.Email FROM Users ' +
                 'INNER JOIN PackageBases ' +
@@ -157,6 +161,31 @@ def flag(cur, uid, pkgbase_id):
 
     send_notification(to, subject, body, refs)
 
+def comaintainer_add(cur, pkgbase_id, uid):
+    pkgbase = pkgbase_from_id(cur, pkgbase_id)
+    to = [get_user_email(cur, uid)]
+
+    pkgbase_uri = aur_location + '/pkgbase/' + pkgbase + '/'
+
+    subject = 'AUR Co-Maintainer Notification for %s' % (pkgbase)
+    body = 'You were added to the co-maintainer list of %s [1].' % (pkgbase)
+    refs = '[1] ' + pkgbase_uri + '\n'
+
+    send_notification(to, subject, body, refs)
+
+def comaintainer_remove(cur, pkgbase_id, uid):
+    pkgbase = pkgbase_from_id(cur, pkgbase_id)
+    to = [get_user_email(cur, uid)]
+
+    pkgbase_uri = aur_location + '/pkgbase/' + pkgbase + '/'
+
+    subject = 'AUR Co-Maintainer Notification for %s' % (pkgbase)
+    body = 'You were removed from the co-maintainer list of %s [1].' % \
+            (pkgbase)
+    refs = '[1] ' + pkgbase_uri + '\n'
+
+    send_notification(to, subject, body, refs)
+
 def delete(cur, uid, old_pkgbase_id, new_pkgbase_id=None):
     user = username_from_id(cur, uid)
     old_pkgbase = pkgbase_from_id(cur, old_pkgbase_id)
@@ -246,6 +275,8 @@ if __name__ == '__main__':
         'welcome': welcome,
         'comment': comment,
         'flag': flag,
+        'comaintainer-add': comaintainer_add,
+        'comaintainer-remove': comaintainer_remove,
         'delete': delete,
         'request-open': request_open,
         'request-close': request_close,
diff --git a/web/lib/pkgbasefuncs.inc.php b/web/lib/pkgbasefuncs.inc.php
index bb9d2415..aad9d14d 100644
--- a/web/lib/pkgbasefuncs.inc.php
+++ b/web/lib/pkgbasefuncs.inc.php
@@ -1048,6 +1048,7 @@ function pkgbase_set_comaintainers($base_id, $users) {
 	foreach ($uids_new as $uid) {
 		if (in_array($uid, $uids_add)) {
 			$q = sprintf("INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (%d, %d, %d)", $base_id, $uid, $i);
+			notify(array('comaintainer-add', $base_id, $uid));
 		} else {
 			$q = sprintf("UPDATE PackageComaintainers SET Priority = %d WHERE PackageBaseID = %d AND UsersID = %d", $i, $base_id, $uid);
 		}
@@ -1059,6 +1060,7 @@ function pkgbase_set_comaintainers($base_id, $users) {
 	foreach ($uids_rem as $uid) {
 		$q = sprintf("DELETE FROM PackageComaintainers WHERE PackageBaseID = %d AND UsersID = %d", $base_id, $uid);
 		$dbh->exec($q);
+		notify(array('comaintainer-remove', $base_id, $uid));
 	}
 
 	return array(true, __("The package base co-maintainers have been updated."));