notify.py 25.8 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

10
11
from typing import List, Tuple

12
13
from sqlalchemy import and_, or_

Lukas Fleischer's avatar
Lukas Fleischer committed
14
15
import aurweb.config
import aurweb.db
Lukas Fleischer's avatar
Lukas Fleischer committed
16
import aurweb.l10n
17

18
19
20
21
22
23
24
25
from aurweb import db, l10n, logging
from aurweb.models import PackageBase, User
from aurweb.models.package_comaintainer import PackageComaintainer
from aurweb.models.package_comment import PackageComment
from aurweb.models.package_notification import PackageNotification
from aurweb.models.package_request import PackageRequest
from aurweb.models.request_type import RequestType
from aurweb.models.tu_vote import TUVote
26

27
28
logger = logging.get_logger(__name__)

Lukas Fleischer's avatar
Lukas Fleischer committed
29
aur_location = aurweb.config.get('options', 'aur_location')
30
31


32
33
34
def headers_msgid(thread_id):
    return {'Message-ID': thread_id}

35

36
37
38
def headers_reply(thread_id):
    return {'In-Reply-To': thread_id, 'References': thread_id}

39

40
41
42
43
44
45
46
class Notification:
    def get_refs(self):
        return ()

    def get_headers(self):
        return {}

47
48
49
    def get_cc(self):
        return []

Lukas Fleischer's avatar
Lukas Fleischer committed
50
    def get_body_fmt(self, lang):
51
        body = ''
Lukas Fleischer's avatar
Lukas Fleischer committed
52
        for line in self.get_body(lang).splitlines():
53
54
            if line == '--':
                body += '--\n'
55
                continue
56
57
58
            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
59
        return body.rstrip()
60

61
    def _send(self) -> None:
62
        sendmail = aurweb.config.get('notifications', 'sendmail')
63
64
        sender = aurweb.config.get('notifications', 'sender')
        reply_to = aurweb.config.get('notifications', 'reply-to')
65
66
67
        reason = self.__class__.__name__
        if reason.endswith('Notification'):
            reason = reason[:-len('Notification')]
68
69

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

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

85
86
87
88
89
90
91
92
93
94
            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')
95
96
97
98
99
100
                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')
101
102
103
                user = aurweb.config.get('notifications', 'smtp-user')
                passwd = aurweb.config.get('notifications', 'smtp-password')

104
105
106
107
108
                classes = {
                    False: smtplib.SMTP,
                    True: smtplib.SMTP_SSL,
                }
                server = classes[use_ssl](server_addr, server_port)
109
110
111
112
113
114
115
116

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

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

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

123
124
125
126
127
128
129
130
    def send(self) -> None:
        try:
            self._send()
        except OSError as exc:
            logger.error("Unable to emit notification due to an "
                         "OSError (precise exception following).")
            logger.error(str(exc))

131

132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
class ServerErrorNotification(Notification):
    """ A notification used to represent an internal server error. """

    def __init__(self, traceback_id: int, version: str, utc: int):
        """
        Construct a ServerErrorNotification.

        :param traceback_id: Traceback ID
        :param version: aurweb version
        :param utc: UTC timestamp
        """
        self._tb_id = traceback_id
        self._version = version
        self._utc = utc

        postmaster = aurweb.config.get("notifications", "postmaster")
        self._to = postmaster

        super().__init__()

    def get_recipients(self) -> List[Tuple[str, str]]:
        from aurweb.auth import AnonymousUser
        user = (db.query(User).filter(User.Email == self._to).first()
                or AnonymousUser())
        return [(self._to, user.LangPreference)]

    def get_subject(self, lang: str) -> str:
        return l10n.translator.translate("AUR Server Error", lang)

    def get_body(self, lang: str) -> str:
        """ A forcibly English email body. """
        dt = aurweb.util.timestamp_to_datetime(self._utc)
        dts = dt.strftime("%Y-%m-%d %H:%M")
        return (f"Traceback ID: {self._tb_id}\n"
                f"Location: {aur_location}\n"
                f"Version: {self._version}\n"
                f"Datetime: {dts} UTC\n")

    def get_refs(self):
        return (aur_location,)


174
class ResetKeyNotification(Notification):
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
    def __init__(self, uid):

        user = db.query(User).filter(
            and_(User.ID == uid, User.Suspended == 0)
        ).with_entities(
            User.Username,
            User.Email,
            User.BackupEmail,
            User.LangPreference,
            User.ResetKey
        ).order_by(User.Username.asc()).first()

        self._username = user.Username
        self._to = user.Email
        self._backup = user.BackupEmail
        self._lang = user.LangPreference
        self._resetkey = user.ResetKey

Lukas Fleischer's avatar
Lukas Fleischer committed
193
        super().__init__()
194
195

    def get_recipients(self):
196
197
198
199
        if self._backup:
            return [(self._to, self._lang), (self._backup, self._lang)]
        else:
            return [(self._to, self._lang)]
200

Lukas Fleischer's avatar
Lukas Fleischer committed
201
    def get_subject(self, lang):
202
        return aurweb.l10n.translator.translate('AUR Password Reset', lang)
203

Lukas Fleischer's avatar
Lukas Fleischer committed
204
    def get_body(self, lang):
205
206
207
208
209
210
        return aurweb.l10n.translator.translate(
            '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)
211
212
213
214
215
216

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


class WelcomeNotification(ResetKeyNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
217
    def get_subject(self, lang):
218
219
220
        return aurweb.l10n.translator.translate(
            'Welcome to the Arch User Repository',
            lang)
221

Lukas Fleischer's avatar
Lukas Fleischer committed
222
    def get_body(self, lang):
223
224
225
226
227
        return aurweb.l10n.translator.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)
228
229
230


class CommentNotification(Notification):
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
    def __init__(self, uid, pkgbase_id, comment_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.CommentNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgcomment = db.query(PackageComment.Comments).filter(
            PackageComment.ID == comment_id).first()
        self._text = pkgcomment.Comments

Lukas Fleischer's avatar
Lukas Fleischer committed
253
        super().__init__()
254
255

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
256
        return self._recipients
257

Lukas Fleischer's avatar
Lukas Fleischer committed
258
    def get_subject(self, lang):
259
260
261
        return aurweb.l10n.translator.translate(
            'AUR Comment for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
262

Lukas Fleischer's avatar
Lukas Fleischer committed
263
    def get_body(self, lang):
264
265
266
        body = aurweb.l10n.translator.translate(
            '{user} [1] added the following comment to {pkgbase} [2]:',
            lang).format(user=self._user, pkgbase=self._pkgbase)
267
        body += '\n\n' + self._text + '\n\n--\n'
268
269
270
271
272
273
        dnlabel = aurweb.l10n.translator.translate(
            'Disable notifications', lang)
        body += aurweb.l10n.translator.translate(
            'If you no longer wish to receive notifications about this '
            'package, please go to the package page [2] and select '
            '"{label}".', lang).format(label=dnlabel)
274
275
276
277
278
279
280
281
282
283
284
285
286
        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):
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.UpdateNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
305
        super().__init__()
306
307

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
308
        return self._recipients
309

Lukas Fleischer's avatar
Lukas Fleischer committed
310
    def get_subject(self, lang):
311
312
313
        return aurweb.l10n.translator.translate(
            'AUR Package Update: {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
314

Lukas Fleischer's avatar
Lukas Fleischer committed
315
    def get_body(self, lang):
316
317
318
        body = aurweb.l10n.translator.translate(
            '{user} [1] pushed a new commit to {pkgbase} [2].',
            lang).format(user=self._user, pkgbase=self._pkgbase)
319
        body += '\n\n--\n'
320
321
322
323
324
325
        dnlabel = aurweb.l10n.translator.translate(
            'Disable notifications', lang)
        body += aurweb.l10n.translator.translate(
            'If you no longer wish to receive notifications about this '
            'package, please go to the package page [2] and select '
            '"{label}".', lang).format(label=dnlabel)
326
327
328
329
330
331
332
333
334
335
336
337
338
        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):
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageComaintainer, isouter=True).join(
            PackageBase,
            or_(PackageBase.MaintainerUID == User.ID,
                PackageBase.ID == PackageComaintainer.PackageBaseID)
        ).filter(
            and_(PackageBase.ID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgbase = db.query(PackageBase.FlaggerComment).filter(
            PackageBase.ID == pkgbase_id).first()
        self._text = pkgbase.FlaggerComment

Lukas Fleischer's avatar
Lukas Fleischer committed
363
        super().__init__()
364
365

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
366
        return self._recipients
367

Lukas Fleischer's avatar
Lukas Fleischer committed
368
    def get_subject(self, lang):
369
370
371
        return aurweb.l10n.translator.translate(
            'AUR Out-of-date Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
372

Lukas Fleischer's avatar
Lukas Fleischer committed
373
    def get_body(self, lang):
374
375
376
377
        body = aurweb.l10n.translator.translate(
            'Your package {pkgbase} [1] has been flagged out-of-date by '
            '{user} [2]:', lang).format(pkgbase=self._pkgbase,
                                        user=self._user)
378
379
380
381
382
383
384
385
386
        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):
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.OwnershipNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgbase = db.query(PackageBase.FlaggerComment).filter(
            PackageBase.ID == pkgbase_id).first()
        self._text = pkgbase.FlaggerComment

Lukas Fleischer's avatar
Lukas Fleischer committed
409
        super().__init__()
410
411

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
412
        return self._recipients
413

Lukas Fleischer's avatar
Lukas Fleischer committed
414
    def get_subject(self, lang):
415
416
417
        return aurweb.l10n.translator.translate(
            'AUR Ownership Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
418
419
420
421
422
423
424

    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
425
    def get_body(self, lang):
426
427
428
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was adopted by {user} [2].',
            lang).format(pkgbase=self._pkgbase, user=self._user)
429
430
431


class DisownNotification(OwnershipEventNotification):
432
    def get_body(self, lang):
433
434
435
436
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was disowned by {user} '
            '[2].', lang).format(pkgbase=self._pkgbase,
                                 user=self._user)
437
438
439


class ComaintainershipEventNotification(Notification):
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
    def __init__(self, uid, pkgbase_id):

        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        user = db.query(User).filter(
            and_(User.ID == uid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).first()

        self._to = user.Email
        self._lang = user.LangPreference

Lukas Fleischer's avatar
Lukas Fleischer committed
456
        super().__init__()
457
458

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

Lukas Fleischer's avatar
Lukas Fleischer committed
461
    def get_subject(self, lang):
462
463
464
        return aurweb.l10n.translator.translate(
            'AUR Co-Maintainer Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
465
466
467
468
469
470

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


class ComaintainerAddNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
471
    def get_body(self, lang):
472
473
474
        return aurweb.l10n.translator.translate(
            'You were added to the co-maintainer list of {pkgbase} [1].',
            lang).format(pkgbase=self._pkgbase)
475
476
477


class ComaintainerRemoveNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
478
    def get_body(self, lang):
479
480
481
        return aurweb.l10n.translator.translate(
            'You were removed from the co-maintainer list of {pkgbase} '
            '[1].', lang).format(pkgbase=self._pkgbase)
482
483
484


class DeleteNotification(Notification):
485
486
487
488
489
490
491
492
    def __init__(self, uid, old_pkgbase_id, new_pkgbase_id=None):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._old_pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == old_pkgbase_id).first().Name

        self._new_pkgbase = None
493
        if new_pkgbase_id:
494
495
496
497
498
499
500
501
502
503
504
505
506
            self._new_pkgbase = db.query(PackageBase.Name).filter(
                PackageBase.ID == new_pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == old_pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
507
        super().__init__()
508
509

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
510
        return self._recipients
511

Lukas Fleischer's avatar
Lukas Fleischer committed
512
    def get_subject(self, lang):
513
514
515
        return aurweb.l10n.translator.translate(
            'AUR Package deleted: {pkgbase}',
            lang).format(pkgbase=self._old_pkgbase)
516

Lukas Fleischer's avatar
Lukas Fleischer committed
517
    def get_body(self, lang):
518
        if self._new_pkgbase:
519
520
521
522
            dnlabel = aurweb.l10n.translator.translate(
                'Disable notifications', lang)
            return aurweb.l10n.translator.translate(
                '{user} [1] merged {old} [2] into {new} [3].\n\n'
523
                '--\n'
524
525
526
527
                'If you no longer wish receive notifications about the '
                'new package, please go to [3] and click "{label}".',
                lang).format(user=self._user, old=self._old_pkgbase,
                             new=self._new_pkgbase, label=dnlabel)
528
        else:
529
530
531
532
533
            return aurweb.l10n.translator.translate(
                '{user} [1] deleted {pkgbase} [2].\n\n'
                'You will no longer receive notifications about this '
                'package.', lang).format(user=self._user,
                                         pkgbase=self._old_pkgbase)
534
535
536
537
538
539
540
541
542
543

    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):
544
545
546
547
548
549
550
    def __init__(self, uid, reqid, reqtype, pkgbase_id, merge_into=None):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

551
        self._to = aurweb.config.get('options', 'aur_request_ml')
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573

        query = db.query(PackageRequest).join(PackageBase).join(
            PackageComaintainer,
            PackageComaintainer.PackageBaseID == PackageRequest.PackageBaseID,
            isouter=True
        ).join(
            User,
            or_(User.ID == PackageRequest.UsersID,
                User.ID == PackageBase.MaintainerUID,
                User.ID == PackageComaintainer.UsersID)
        ).filter(
            and_(PackageRequest.ID == reqid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email
        ).distinct()
        self._cc = [u.Email for u in query]

        pkgreq = db.query(PackageRequest.Comments).filter(
            PackageRequest.ID == reqid).first()

        self._text = pkgreq.Comments
574
575
576
577
578
        self._reqid = int(reqid)
        self._reqtype = reqtype
        self._merge_into = merge_into

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

581
582
583
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
584
    def get_subject(self, lang):
585
586
587
        return '[PRQ#%d] %s Request for %s' % \
               (self._reqid, self._reqtype.title(), self._pkgbase)

Lukas Fleischer's avatar
Lukas Fleischer committed
588
    def get_body(self, lang):
589
590
591
592
593
        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:
594
595
596
            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)
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
            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):
616
617
618
619
    def __init__(self, uid, reqid, reason):
        user = db.query(User.Username).filter(User.ID == uid).first()
        self._user = user.Username if user else None

620
        self._to = aurweb.config.get('options', 'aur_request_ml')
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650

        query = db.query(PackageRequest).join(PackageBase).join(
            PackageComaintainer,
            PackageComaintainer.PackageBaseID == PackageRequest.PackageBaseID,
            isouter=True
        ).join(
            User,
            or_(User.ID == PackageRequest.UsersID,
                User.ID == PackageBase.MaintainerUID,
                User.ID == PackageComaintainer.UsersID)
        ).filter(
            and_(PackageRequest.ID == reqid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email
        ).distinct()
        self._cc = [u.Email for u in query]

        pkgreq = db.query(PackageRequest).join(RequestType).filter(
            PackageRequest.ID == reqid
        ).with_entities(
            PackageRequest.ClosureComment,
            RequestType.Name,
            PackageRequest.PackageBaseName
        ).first()

        self._text = pkgreq.ClosureComment
        self._reqtype = pkgreq.Name
        self._pkgbase = pkgreq.PackageBaseName

651
652
653
654
        self._reqid = int(reqid)
        self._reason = reason

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

657
658
659
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
660
    def get_subject(self, lang):
661
662
663
664
        return '[PRQ#%d] %s Request for %s %s' % (self._reqid,
                                                  self._reqtype.title(),
                                                  self._pkgbase,
                                                  self._reason.title())
665

Lukas Fleischer's avatar
Lukas Fleischer committed
666
    def get_body(self, lang):
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
        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):
693
    def __init__(self, vote_id):
694
        self._vote_id = int(vote_id)
695
696
697
698
699
700
701
702
703
704
705

        subquery = db.query(TUVote.UserID).filter(TUVote.VoteID == vote_id)
        query = db.query(User).filter(
            and_(User.AccountTypeID.in_((2, 4)),
                 ~User.ID.in_(subquery),
                 User.Suspended == 0)
        ).with_entities(
            User.Email, User.LangPreference
        )
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
706
        super().__init__()
707
708

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
709
        return self._recipients
710

Lukas Fleischer's avatar
Lukas Fleischer committed
711
    def get_subject(self, lang):
712
713
714
        return aurweb.l10n.translator.translate(
            'TU Vote Reminder: Proposal {id}',
            lang).format(id=self._vote_id)
715

Lukas Fleischer's avatar
Lukas Fleischer committed
716
    def get_body(self, lang):
717
718
719
720
        return aurweb.l10n.translator.translate(
            '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)
721
722
723

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


726
def main():
727
    db.get_engine()
728
729
    action = sys.argv[1]
    action_map = {
730
731
732
733
734
735
736
737
738
739
740
741
742
        '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,
743
744
    }

745
746
    with db.begin():
        notification = action_map[action](*sys.argv[2:])
747
    notification.send()
748

749
750
751

if __name__ == '__main__':
    main()