notify.py 23.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, to, msg.as_bytes())
122
                server.quit()
123
124
125
126


class ResetKeyNotification(Notification):
    def __init__(self, conn, uid):
127
128
129
130
        cur = conn.execute('SELECT UserName, Email, BackupEmail, ' +
                           'LangPreference, ResetKey ' +
                           'FROM Users WHERE ID = ?', [uid])
        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
176
177
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.CommentNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
                           'PackageNotifications.PackageBaseID = ?',
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
178
        self._recipients = cur.fetchall()
179
180
181
        cur = conn.execute('SELECT Comments FROM PackageComments WHERE ID = ?',
                           [comment_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
182
        super().__init__()
183
184

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

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

Lukas Fleischer's avatar
Lukas Fleischer committed
191
192
    def get_body(self, lang):
        body = self._l10n.translate(
193
194
                '{user} [1] added the following comment to {pkgbase} [2]:',
                lang).format(user=self._user, pkgbase=self._pkgbase)
195
        body += '\n\n' + self._text + '\n\n-- \n'
Lukas Fleischer's avatar
Lukas Fleischer committed
196
197
198
199
        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 '
200
                '"{label}".', lang).format(label=dnlabel)
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
        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
217
218
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
219
220
221
222
223
224
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.UpdateNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
                           'PackageNotifications.PackageBaseID = ?',
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
225
226
        self._recipients = cur.fetchall()
        super().__init__()
227
228

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
229
        return self._recipients
230

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

Lukas Fleischer's avatar
Lukas Fleischer committed
235
    def get_body(self, lang):
236
237
238
239
        body = self._l10n.translate('{user} [1] pushed a new commit to '
                                    '{pkgbase} [2].', lang).format(
                                            user=self._user,
                                            pkgbase=self._pkgbase)
240
        body += '\n\n-- \n'
Lukas Fleischer's avatar
Lukas Fleischer committed
241
242
243
244
        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 '
245
                '"{label}".', lang).format(label=dnlabel)
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
        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
262
263
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
264
265
266
267
268
269
                           'LEFT JOIN PackageComaintainers ' +
                           'ON PackageComaintainers.UsersID = Users.ID ' +
                           'INNER JOIN PackageBases ' +
                           'ON PackageBases.MaintainerUID = Users.ID OR ' +
                           'PackageBases.ID = PackageComaintainers.PackageBaseID ' +
                           'WHERE PackageBases.ID = ?', [pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
270
        self._recipients = cur.fetchall()
271
272
273
        cur = conn.execute('SELECT FlaggerComment FROM PackageBases WHERE ' +
                           'ID = ?', [pkgbase_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
274
        super().__init__()
275
276

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
277
        return self._recipients
278

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

Lukas Fleischer's avatar
Lukas Fleischer committed
284
285
    def get_body(self, lang):
        body = self._l10n.translate(
286
287
288
                'Your package {pkgbase} [1] has been flagged out-of-date by '
                '{user} [2]:', lang).format(pkgbase=self._pkgbase,
                                            user=self._user)
289
290
291
292
293
294
295
296
297
298
299
300
        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
301
302
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
303
304
305
306
307
308
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'Users.OwnershipNotify = 1 AND ' +
                           'PackageNotifications.UserID != ? AND ' +
                           'PackageNotifications.PackageBaseID = ?',
                           [uid, pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
309
        self._recipients = cur.fetchall()
310
311
312
        cur = conn.execute('SELECT FlaggerComment FROM PackageBases WHERE ' +
                           'ID = ?', [pkgbase_id])
        self._text = cur.fetchone()[0]
Lukas Fleischer's avatar
Lukas Fleischer committed
313
        super().__init__()
314
315

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
316
        return self._recipients
317

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

    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
328
329
    def get_body(self, lang):
        return self._l10n.translate(
330
331
                'The package {pkgbase} [1] was adopted by {user} [2].',
                lang).format(pkgbase=self._pkgbase, user=self._user)
332
333
334


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


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
345
346
347
348
        cur = conn.execute('SELECT Email, LangPreference FROM Users ' +
                           'WHERE ID = ?', [uid])
        self._to, self._lang = cur.fetchone()
        super().__init__()
349
350

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

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

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


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


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


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
384
385
        cur = conn.execute('SELECT DISTINCT Users.Email, ' +
                           'Users.LangPreference FROM Users ' +
386
387
388
389
390
                           'INNER JOIN PackageNotifications ' +
                           'ON PackageNotifications.UserID = Users.ID WHERE ' +
                           'PackageNotifications.UserID != ? AND ' +
                           'PackageNotifications.PackageBaseID = ?',
                           [uid, old_pkgbase_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
391
392
        self._recipients = cur.fetchall()
        super().__init__()
393
394

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
395
        return self._recipients
396

Lukas Fleischer's avatar
Lukas Fleischer committed
397
    def get_subject(self, lang):
398
399
        return self._l10n.translate('AUR Package deleted: {pkgbase}',
                                    lang).format(pkgbase=self._old_pkgbase)
400

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

    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 ' +
                           'INNER JOIN Users ' +
                           'ON Users.ID = PackageRequests.UsersID ' +
                           'OR Users.ID = PackageBases.MaintainerUID ' +
                           'WHERE PackageRequests.ID = ?', [reqid])
        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
447
        return [(self._to, 'en')]
448

449
450
451
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
452
    def get_subject(self, lang):
453
454
455
        return '[PRQ#%d] %s Request for %s' % \
               (self._reqid, self._reqtype.title(), self._pkgbase)

Lukas Fleischer's avatar
Lukas Fleischer committed
456
    def get_body(self, lang):
457
458
459
460
461
        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:
462
463
464
            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)
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
            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 ' +
                           'INNER JOIN Users ' +
                           'ON Users.ID = PackageRequests.UsersID ' +
                           'OR Users.ID = PackageBases.MaintainerUID ' +
                           'WHERE PackageRequests.ID = ?', [reqid])
        self._to = aurweb.config.get('options', 'aur_request_ml')
        self._cc = [row[0] for row in cur.fetchall()]
495
496
497
498
499
500
501
502
        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()
503
504
505
506
        self._reqid = int(reqid)
        self._reason = reason

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

509
510
511
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
512
    def get_subject(self, lang):
513
514
515
516
        return '[PRQ#%d] %s Request for %s %s' % (self._reqid,
                                                  self._reqtype.title(),
                                                  self._pkgbase,
                                                  self._reason.title())
517

Lukas Fleischer's avatar
Lukas Fleischer committed
518
    def get_body(self, lang):
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
        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
547
        cur = conn.execute('SELECT Email, LangPreference FROM Users ' +
548
549
550
                           'WHERE AccountTypeID IN (2, 4) AND ID NOT IN ' +
                           '(SELECT UserID FROM TU_Votes ' +
                           'WHERE TU_Votes.VoteID = ?)', [vote_id])
Lukas Fleischer's avatar
Lukas Fleischer committed
551
552
        self._recipients = cur.fetchall()
        super().__init__()
553
554

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
555
        return self._recipients
556

Lukas Fleischer's avatar
Lukas Fleischer committed
557
    def get_subject(self, lang):
558
559
        return self._l10n.translate('TU Vote Reminder: Proposal {id}',
                                    lang).format(id=self._vote_id)
560

Lukas Fleischer's avatar
Lukas Fleischer committed
561
562
    def get_body(self, lang):
        return self._l10n.translate(
563
564
565
                '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)
566
567
568

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


571
def main():
572
573
    action = sys.argv[1]
    action_map = {
574
575
576
577
578
579
580
581
582
583
584
585
586
        '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,
587
588
    }

Lukas Fleischer's avatar
Lukas Fleischer committed
589
    conn = aurweb.db.Connection()
590

591
592
    notification = action_map[action](conn, *sys.argv[2:])
    notification.send()
593

Lukas Fleischer's avatar
Lukas Fleischer committed
594
595
    conn.commit()
    conn.close()
596
597
598
599


if __name__ == '__main__':
    main()