notify.py 24.5 KB
Newer Older
1
#!/usr/bin/env python3
2
3

import email.mime.text
4
5
import email.utils
import smtplib
6
7
8
9
import subprocess
import sys
import textwrap

Lukas Fleischer's avatar
Lukas Fleischer committed
10
11
import aurweb.config
import aurweb.db
Lukas Fleischer's avatar
Lukas Fleischer committed
12
import aurweb.l10n
13

Lukas Fleischer's avatar
Lukas Fleischer committed
14
aur_location = aurweb.config.get('options', 'aur_location')
15
16


17
18
19
def headers_msgid(thread_id):
    return {'Message-ID': thread_id}

20

21
22
23
def headers_reply(thread_id):
    return {'In-Reply-To': thread_id, 'References': thread_id}

24

Lukas Fleischer's avatar
Lukas Fleischer committed
25
26
def username_from_id(conn, uid):
    cur = conn.execute('SELECT UserName FROM Users WHERE ID = ?', [uid])
27
28
    return cur.fetchone()[0]

29

Lukas Fleischer's avatar
Lukas Fleischer committed
30
31
32
def pkgbase_from_id(conn, pkgbase_id):
    cur = conn.execute('SELECT Name FROM PackageBases WHERE ID = ?',
                       [pkgbase_id])
33
34
    return cur.fetchone()[0]

35

Lukas Fleischer's avatar
Lukas Fleischer committed
36
37
38
def pkgbase_from_pkgreq(conn, reqid):
    cur = conn.execute('SELECT PackageBaseID FROM PackageRequests ' +
                       'WHERE ID = ?', [reqid])
39
40
    return cur.fetchone()[0]

41

42
class Notification:
Lukas Fleischer's avatar
Lukas Fleischer committed
43
44
45
    def __init__(self):
        self._l10n = aurweb.l10n.Translator()

46
47
48
49
50
51
    def get_refs(self):
        return ()

    def get_headers(self):
        return {}

52
53
54
    def get_cc(self):
        return []

Lukas Fleischer's avatar
Lukas Fleischer committed
55
    def get_body_fmt(self, lang):
56
        body = ''
Lukas Fleischer's avatar
Lukas Fleischer committed
57
        for line in self.get_body(lang).splitlines():
58
59
60
            if line == '-- ':
                body += '-- \n'
                continue
61
62
63
            body += textwrap.fill(line, break_long_words=False) + '\n'
        for i, ref in enumerate(self.get_refs()):
            body += '\n' + '[%d] %s' % (i + 1, ref)
Lukas Fleischer's avatar
Lukas Fleischer committed
64
        return body.rstrip()
65

Lukas Fleischer's avatar
Lukas Fleischer committed
66
    def send(self):
67
        sendmail = aurweb.config.get('notifications', 'sendmail')
68
69
        sender = aurweb.config.get('notifications', 'sender')
        reply_to = aurweb.config.get('notifications', 'reply-to')
70
71
72
        reason = self.__class__.__name__
        if reason.endswith('Notification'):
            reason = reason[:-len('Notification')]
73
74

        for recipient in self.get_recipients():
Lukas Fleischer's avatar
Lukas Fleischer committed
75
76
77
78
            to, lang = recipient
            msg = email.mime.text.MIMEText(self.get_body_fmt(lang),
                                           'plain', 'utf-8')
            msg['Subject'] = self.get_subject(lang)
79
80
            msg['From'] = sender
            msg['Reply-to'] = reply_to
Lukas Fleischer's avatar
Lukas Fleischer committed
81
            msg['To'] = to
82
83
            if self.get_cc():
                msg['Cc'] = str.join(', ', self.get_cc())
84
            msg['X-AUR-Reason'] = reason
85
            msg['Date'] = email.utils.formatdate(localtime=True)
86
87
88
89

            for key, value in self.get_headers().items():
                msg[key] = value

90
91
92
93
94
95
96
97
98
99
            sendmail = aurweb.config.get('notifications', 'sendmail')
            if sendmail:
                # send email using the sendmail binary specified in the
                # configuration file
                p = subprocess.Popen([sendmail, '-t', '-oi'],
                                     stdin=subprocess.PIPE)
                p.communicate(msg.as_bytes())
            else:
                # send email using smtplib; no local MTA required
                server_addr = aurweb.config.get('notifications', 'smtp-server')
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
                server_port = aurweb.config.getint('notifications', 'smtp-port')
                use_ssl = aurweb.config.getboolean('notifications', 'smtp-use-ssl')
                use_starttls = aurweb.config.getboolean('notifications', 'smtp-use-starttls')
                user = aurweb.config.get('notifications', 'smtp-user')
                passwd = aurweb.config.get('notifications', 'smtp-password')

                if use_ssl:
                    server = smtplib.SMTP_SSL(server_addr, server_port)
                else:
                    server = smtplib.SMTP(server_addr, server_port)

                if use_starttls:
                    server.ehlo()
                    server.starttls()
                    server.ehlo()

                if user and passwd:
                    server.login(user, passwd)
118
119

                server.set_debuglevel(0)
120
                deliver_to = [to] + self.get_cc()
121
                server.sendmail(sender, deliver_to, msg.as_bytes())
122
                server.quit()
123
124
125
126


class ResetKeyNotification(Notification):
    def __init__(self, conn, uid):
127
128
        cur = conn.execute('SELECT UserName, Email, BackupEmail, ' +
                           'LangPreference, ResetKey ' +
129
                           'FROM Users WHERE ID = ? AND Suspended = 0', [uid])
130
        self._username, self._to, self._backup, self._lang, self._resetkey = cur.fetchone()
Lukas Fleischer's avatar
Lukas Fleischer committed
131
        super().__init__()
132
133

    def get_recipients(self):
134
135
136
137
        if self._backup:
            return [(self._to, self._lang), (self._backup, self._lang)]
        else:
            return [(self._to, self._lang)]
138

Lukas Fleischer's avatar
Lukas Fleischer committed
139
140
    def get_subject(self, lang):
        return self._l10n.translate('AUR Password Reset', lang)
141

Lukas Fleischer's avatar
Lukas Fleischer committed
142
143
    def get_body(self, lang):
        return self._l10n.translate(
144
145
146
147
148
                'A password reset request was submitted for the account '
                '{user} associated with your email address. If you wish to '
                'reset your password follow the link [1] below, otherwise '
                'ignore this message and nothing will happen.',
                lang).format(user=self._username)
149
150
151
152
153
154

    def get_refs(self):
        return (aur_location + '/passreset/?resetkey=' + self._resetkey,)


class WelcomeNotification(ResetKeyNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
155
156
157
    def get_subject(self, lang):
        return self._l10n.translate('Welcome to the Arch User Repository',
                                    lang)
158

Lukas Fleischer's avatar
Lukas Fleischer committed
159
160
161
162
163
164
    def get_body(self, lang):
        return self._l10n.translate(
                'Welcome to the Arch User Repository! In order to set an '
                'initial password for your new account, please click the '
                'link [1] below. If the link does not work, try copying and '
                'pasting it into your browser.', lang)
165
166
167
168
169
170


class CommentNotification(Notification):
    def __init__(self, conn, uid, pkgbase_id, comment_id):
        self._user = username_from_id(conn, uid)
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
171
172
        cur = conn.execute('SELECT DISTINCT Users.Email, Users.LangPreference '
                           'FROM Users INNER JOIN PackageNotifications ' +
173
174
175
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.CommentNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
176
177
                           'PackageNotifications.PackageBaseID = ? AND ' +
                           'Users.Suspended = 0',
178
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
179
        self._recipients = cur.fetchall()
180
181
182
        cur = conn.execute('SELECT Comments FROM PackageComments WHERE ID = ?',
                           [comment_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
183
        super().__init__()
184
185

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
186
        return self._recipients
187

Lukas Fleischer's avatar
Lukas Fleischer committed
188
    def get_subject(self, lang):
189
190
        return self._l10n.translate('AUR Comment for {pkgbase}',
                                    lang).format(pkgbase=self._pkgbase)
191

Lukas Fleischer's avatar
Lukas Fleischer committed
192
193
    def get_body(self, lang):
        body = self._l10n.translate(
194
195
                '{user} [1] added the following comment to {pkgbase} [2]:',
                lang).format(user=self._user, pkgbase=self._pkgbase)
196
        body += '\n\n' + self._text + '\n\n-- \n'
Lukas Fleischer's avatar
Lukas Fleischer committed
197
198
199
200
        dnlabel = self._l10n.translate('Disable notifications', lang)
        body += self._l10n.translate(
                'If you no longer wish to receive notifications about this '
                'package, please go to the package page [2] and select '
201
                '"{label}".', lang).format(label=dnlabel)
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
        return body

    def get_refs(self):
        return (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')

    def get_headers(self):
        thread_id = '<pkg-notifications-' + self._pkgbase + \
                    '@aur.archlinux.org>'
        return headers_reply(thread_id)


class UpdateNotification(Notification):
    def __init__(self, conn, uid, pkgbase_id):
        self._user = username_from_id(conn, uid)
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
218
219
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
220
221
222
223
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.UpdateNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
224
225
                           'PackageNotifications.PackageBaseID = ? AND ' +
                           'Users.Suspended = 0',
226
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
227
228
        self._recipients = cur.fetchall()
        super().__init__()
229
230

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
231
        return self._recipients
232

Lukas Fleischer's avatar
Lukas Fleischer committed
233
    def get_subject(self, lang):
234
235
        return self._l10n.translate('AUR Package Update: {pkgbase}',
                                    lang).format(pkgbase=self._pkgbase)
236

Lukas Fleischer's avatar
Lukas Fleischer committed
237
    def get_body(self, lang):
238
239
240
241
        body = self._l10n.translate('{user} [1] pushed a new commit to '
                                    '{pkgbase} [2].', lang).format(
                                            user=self._user,
                                            pkgbase=self._pkgbase)
242
        body += '\n\n-- \n'
Lukas Fleischer's avatar
Lukas Fleischer committed
243
244
245
246
        dnlabel = self._l10n.translate('Disable notifications', lang)
        body += self._l10n.translate(
                'If you no longer wish to receive notifications about this '
                'package, please go to the package page [2] and select '
247
                '"{label}".', lang).format(label=dnlabel)
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
        return body

    def get_refs(self):
        return (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')

    def get_headers(self):
        thread_id = '<pkg-notifications-' + self._pkgbase + \
                    '@aur.archlinux.org>'
        return headers_reply(thread_id)


class FlagNotification(Notification):
    def __init__(self, conn, uid, pkgbase_id):
        self._user = username_from_id(conn, uid)
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
264
265
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
266
267
268
269
270
                           'LEFT JOIN PackageComaintainers ' +
                           'ON PackageComaintainers.UsersID = Users.ID ' +
                           'INNER JOIN PackageBases ' +
                           'ON PackageBases.MaintainerUID = Users.ID OR ' +
                           'PackageBases.ID = PackageComaintainers.PackageBaseID ' +
271
272
                           'WHERE PackageBases.ID = ? AND ' +
                           'Users.Suspended = 0', [pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
273
        self._recipients = cur.fetchall()
274
275
276
        cur = conn.execute('SELECT FlaggerComment FROM PackageBases WHERE ' +
                           'ID = ?', [pkgbase_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
277
        super().__init__()
278
279

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
280
        return self._recipients
281

Lukas Fleischer's avatar
Lukas Fleischer committed
282
    def get_subject(self, lang):
283
284
285
        return self._l10n.translate('AUR Out-of-date Notification for '
                                    '{pkgbase}',
                                    lang).format(pkgbase=self._pkgbase)
286

Lukas Fleischer's avatar
Lukas Fleischer committed
287
288
    def get_body(self, lang):
        body = self._l10n.translate(
289
290
291
                'Your package {pkgbase} [1] has been flagged out-of-date by '
                '{user} [2]:', lang).format(pkgbase=self._pkgbase,
                                            user=self._user)
292
293
294
295
296
297
298
299
300
301
302
303
        body += '\n\n' + self._text
        return body

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',
                aur_location + '/account/' + self._user + '/')


class OwnershipEventNotification(Notification):
    def __init__(self, conn, uid, pkgbase_id):
        self._user = username_from_id(conn, uid)
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
304
305
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
306
307
308
309
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.OwnershipNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
310
311
                           'PackageNotifications.PackageBaseID = ? AND ' +
                           'Users.Suspended = 0',
312
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
313
        self._recipients = cur.fetchall()
314
315
316
        cur = conn.execute('SELECT FlaggerComment FROM PackageBases WHERE ' +
                           'ID = ?', [pkgbase_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
317
        super().__init__()
318
319

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
320
        return self._recipients
321

Lukas Fleischer's avatar
Lukas Fleischer committed
322
    def get_subject(self, lang):
323
324
        return self._l10n.translate('AUR Ownership Notification for {pkgbase}',
                                    lang).format(pkgbase=self._pkgbase)
325
326
327
328
329
330
331

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',
                aur_location + '/account/' + self._user + '/')


class AdoptNotification(OwnershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
332
333
    def get_body(self, lang):
        return self._l10n.translate(
334
335
                'The package {pkgbase} [1] was adopted by {user} [2].',
                lang).format(pkgbase=self._pkgbase, user=self._user)
336
337
338


class DisownNotification(OwnershipEventNotification):
339
    def get_body(self, lang):
Lukas Fleischer's avatar
Lukas Fleischer committed
340
        return self._l10n.translate(
341
342
343
                'The package {pkgbase} [1] was disowned by {user} '
                '[2].', lang).format(pkgbase=self._pkgbase,
                                     user=self._user)
344
345
346
347
348


class ComaintainershipEventNotification(Notification):
    def __init__(self, conn, uid, pkgbase_id):
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
349
        cur = conn.execute('SELECT Email, LangPreference FROM Users ' +
350
                           'WHERE ID = ? AND Suspended = 0', [uid])
Lukas Fleischer's avatar
Lukas Fleischer committed
351
352
        self._to, self._lang = cur.fetchone()
        super().__init__()
353
354

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
355
        return [(self._to, self._lang)]
356

Lukas Fleischer's avatar
Lukas Fleischer committed
357
    def get_subject(self, lang):
358
359
360
        return self._l10n.translate('AUR Co-Maintainer Notification for '
                                    '{pkgbase}',
                                    lang).format(pkgbase=self._pkgbase)
361
362
363
364
365
366

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',)


class ComaintainerAddNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
367
368
    def get_body(self, lang):
        return self._l10n.translate(
369
370
                'You were added to the co-maintainer list of {pkgbase} [1].',
                lang).format(pkgbase=self._pkgbase)
371
372
373


class ComaintainerRemoveNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
374
375
    def get_body(self, lang):
        return self._l10n.translate(
376
377
                'You were removed from the co-maintainer list of {pkgbase} '
                '[1].', lang).format(pkgbase=self._pkgbase)
378
379
380
381
382
383
384
385
386
387


class DeleteNotification(Notification):
    def __init__(self, conn, uid, old_pkgbase_id, new_pkgbase_id=None):
        self._user = username_from_id(conn, uid)
        self._old_pkgbase = pkgbase_from_id(conn, old_pkgbase_id)
        if new_pkgbase_id:
            self._new_pkgbase = pkgbase_from_id(conn, new_pkgbase_id)
        else:
            self._new_pkgbase = None
Lukas Fleischer's avatar
Lukas Fleischer committed
388
389
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
390
391
392
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'PackageNotifications.UserID != ? AND ' +
393
394
                           'PackageNotifications.PackageBaseID = ? AND ' +
                           'Users.Suspended = 0',
395
                           [uid, old_pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
396
397
        self._recipients = cur.fetchall()
        super().__init__()
398
399

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
400
        return self._recipients
401

Lukas Fleischer's avatar
Lukas Fleischer committed
402
    def get_subject(self, lang):
403
404
        return self._l10n.translate('AUR Package deleted: {pkgbase}',
                                    lang).format(pkgbase=self._old_pkgbase)
405

Lukas Fleischer's avatar
Lukas Fleischer committed
406
    def get_body(self, lang):
407
        if self._new_pkgbase:
Lukas Fleischer's avatar
Lukas Fleischer committed
408
409
            dnlabel = self._l10n.translate('Disable notifications', lang)
            return self._l10n.translate(
410
                    '{user} [1] merged {old} [2] into {new} [3].\n\n'
411
                    '-- \n'
Lukas Fleischer's avatar
Lukas Fleischer committed
412
                    'If you no longer wish receive notifications about the '
413
414
415
                    'new package, please go to [3] and click "{label}".',
                    lang).format(user=self._user, old=self._old_pkgbase,
                                 new=self._new_pkgbase, label=dnlabel)
416
        else:
Lukas Fleischer's avatar
Lukas Fleischer committed
417
            return self._l10n.translate(
418
                    '{user} [1] deleted {pkgbase} [2].\n\n'
Lukas Fleischer's avatar
Lukas Fleischer committed
419
                    'You will no longer receive notifications about this '
420
421
                    'package.', lang).format(user=self._user,
                                             pkgbase=self._old_pkgbase)
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437

    def get_refs(self):
        refs = (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._old_pkgbase + '/')
        if self._new_pkgbase:
            refs += (aur_location + '/pkgbase/' + self._new_pkgbase + '/',)
        return refs


class RequestOpenNotification(Notification):
    def __init__(self, conn, uid, reqid, reqtype, pkgbase_id, merge_into=None):
        self._user = username_from_id(conn, uid)
        self._pkgbase = pkgbase_from_id(conn, pkgbase_id)
        cur = conn.execute('SELECT DISTINCT Users.Email FROM PackageRequests ' +
                           'INNER JOIN PackageBases ' +
                           'ON PackageBases.ID = PackageRequests.PackageBaseID ' +
438
439
                           'LEFT JOIN PackageComaintainers ' +
                           'ON PackageComaintainers.PackageBaseID = PackageRequests.PackageBaseID ' +
440
441
442
                           'INNER JOIN Users ' +
                           'ON Users.ID = PackageRequests.UsersID ' +
                           'OR Users.ID = PackageBases.MaintainerUID ' +
443
                           'OR Users.ID = PackageComaintainers.UsersID ' +
444
445
                           'WHERE PackageRequests.ID = ? AND ' +
                           'Users.Suspended = 0', [reqid])
446
447
448
449
450
451
452
453
454
455
        self._to = aurweb.config.get('options', 'aur_request_ml')
        self._cc = [row[0] for row in cur.fetchall()]
        cur = conn.execute('SELECT Comments FROM PackageRequests WHERE ID = ?',
                           [reqid])
        self._text = cur.fetchone()[0]
        self._reqid = int(reqid)
        self._reqtype = reqtype
        self._merge_into = merge_into

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
456
        return [(self._to, 'en')]
457

458
459
460
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
461
    def get_subject(self, lang):
462
463
464
        return '[PRQ#%d] %s Request for %s' % \
               (self._reqid, self._reqtype.title(), self._pkgbase)

Lukas Fleischer's avatar
Lukas Fleischer committed
465
    def get_body(self, lang):
466
467
468
469
470
        if self._merge_into:
            body = '%s [1] filed a request to merge %s [2] into %s [3]:' % \
                   (self._user, self._pkgbase, self._merge_into)
            body += '\n\n' + self._text
        else:
471
472
473
            an = 'an' if self._reqtype[0] in 'aeiou' else 'a'
            body = '%s [1] filed %s %s request for %s [2]:' % \
                   (self._user, an, self._reqtype, self._pkgbase)
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
            body += '\n\n' + self._text
        return body

    def get_refs(self):
        refs = (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')
        if self._merge_into:
            refs += (aur_location + '/pkgbase/' + self._merge_into + '/',)
        return refs

    def get_headers(self):
        thread_id = '<pkg-request-' + str(self._reqid) + '@aur.archlinux.org>'
        # Use a deterministic Message-ID for the first email referencing a
        # request.
        headers = headers_msgid(thread_id)
        return headers


class RequestCloseNotification(Notification):
    def __init__(self, conn, uid, reqid, reason):
        self._user = username_from_id(conn, uid) if int(uid) else None
        cur = conn.execute('SELECT DISTINCT Users.Email FROM PackageRequests ' +
                           'INNER JOIN PackageBases ' +
                           'ON PackageBases.ID = PackageRequests.PackageBaseID ' +
498
499
                           'LEFT JOIN PackageComaintainers ' +
                           'ON PackageComaintainers.PackageBaseID = PackageRequests.PackageBaseID ' +
500
501
502
                           'INNER JOIN Users ' +
                           'ON Users.ID = PackageRequests.UsersID ' +
                           'OR Users.ID = PackageBases.MaintainerUID ' +
503
                           'OR Users.ID = PackageComaintainers.UsersID ' +
504
505
                           'WHERE PackageRequests.ID = ? AND ' +
                           'Users.Suspended = 0', [reqid])
506
507
        self._to = aurweb.config.get('options', 'aur_request_ml')
        self._cc = [row[0] for row in cur.fetchall()]
508
509
510
511
512
513
514
515
        cur = conn.execute('SELECT PackageRequests.ClosureComment, ' +
                           'RequestTypes.Name, ' +
                           'PackageRequests.PackageBaseName ' +
                           'FROM PackageRequests ' +
                           'INNER JOIN RequestTypes ' +
                           'ON RequestTypes.ID = PackageRequests.ReqTypeID ' +
                           'WHERE PackageRequests.ID = ?', [reqid])
        self._text, self._reqtype, self._pkgbase = cur.fetchone()
516
517
518
519
        self._reqid = int(reqid)
        self._reason = reason

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
520
        return [(self._to, 'en')]
521

522
523
524
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
525
    def get_subject(self, lang):
526
527
528
529
        return '[PRQ#%d] %s Request for %s %s' % (self._reqid,
                                                  self._reqtype.title(),
                                                  self._pkgbase,
                                                  self._reason.title())
530

Lukas Fleischer's avatar
Lukas Fleischer committed
531
    def get_body(self, lang):
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
        if self._user:
            body = 'Request #%d has been %s by %s [1]' % \
                   (self._reqid, self._reason, self._user)
        else:
            body = 'Request #%d has been %s automatically by the Arch User ' \
                   'Repository package request system' % \
                   (self._reqid, self._reason)
        if self._text.strip() == '':
            body += '.'
        else:
            body += ':\n\n' + self._text
        return body

    def get_refs(self):
        if self._user:
            return (aur_location + '/account/' + self._user + '/',)
        else:
            return ()

    def get_headers(self):
        thread_id = '<pkg-request-' + str(self._reqid) + '@aur.archlinux.org>'
        headers = headers_reply(thread_id)
        return headers


class TUVoteReminderNotification(Notification):
    def __init__(self, conn, vote_id):
        self._vote_id = int(vote_id)
Lukas Fleischer's avatar
Lukas Fleischer committed
560
        cur = conn.execute('SELECT Email, LangPreference FROM Users ' +
561
562
                           'WHERE AccountTypeID IN (2, 4) AND ID NOT IN ' +
                           '(SELECT UserID FROM TU_Votes ' +
563
564
                           'WHERE TU_Votes.VoteID = ?) AND ' +
                           'Users.Suspended = 0', [vote_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
565
566
        self._recipients = cur.fetchall()
        super().__init__()
567
568

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
569
        return self._recipients
570

Lukas Fleischer's avatar
Lukas Fleischer committed
571
    def get_subject(self, lang):
572
573
        return self._l10n.translate('TU Vote Reminder: Proposal {id}',
                                    lang).format(id=self._vote_id)
574

Lukas Fleischer's avatar
Lukas Fleischer committed
575
576
    def get_body(self, lang):
        return self._l10n.translate(
577
578
579
                'Please remember to cast your vote on proposal {id} [1]. '
                'The voting period ends in less than 48 hours.',
                lang).format(id=self._vote_id)
580
581
582

    def get_refs(self):
        return (aur_location + '/tu/?id=' + str(self._vote_id),)
583
584


585
def main():
586
587
    action = sys.argv[1]
    action_map = {
588
589
590
591
592
593
594
595
596
597
598
599
600
        'send-resetkey': ResetKeyNotification,
        'welcome': WelcomeNotification,
        'comment': CommentNotification,
        'update': UpdateNotification,
        'flag': FlagNotification,
        'adopt': AdoptNotification,
        'disown': DisownNotification,
        'comaintainer-add': ComaintainerAddNotification,
        'comaintainer-remove': ComaintainerRemoveNotification,
        'delete': DeleteNotification,
        'request-open': RequestOpenNotification,
        'request-close': RequestCloseNotification,
        'tu-vote-reminder': TUVoteReminderNotification,
601
602
    }

Lukas Fleischer's avatar
Lukas Fleischer committed
603
    conn = aurweb.db.Connection()
604

605
606
    notification = action_map[action](conn, *sys.argv[2:])
    notification.send()
607

Lukas Fleischer's avatar
Lukas Fleischer committed
608
609
    conn.commit()
    conn.close()
610
611
612
613


if __name__ == '__main__':
    main()