hgext/notify.py

patch queue: notify.patch

from mercurial.demandload import *
from mercurial.i18n import gettext as _
from mercurial.node import *
demandload(globals(), 'email.MIMEText mercurial:templater,util fnmatch socket')
demandload(globals(), 'time')

class notifier(object):
    def __init__(self, ui, repo):
        self.ui = ui
        self.ui.readconfig(self.ui.config('notify', 'config'))
        self.repo = repo
        self.stripcount = self.ui.config('notify', 'strip')
        self.root = self.strip(self.repo.root)

    def strip(self, path):
        path = util.pconvert(path)
        count = self.stripcount
        while path and count >= 0:
            c = path.find('/')
            if c == -1:
                break
            path = path[c+1:]
            count -= 1
        return path

    def subscribers(self):
        subs = []
        for user, pat in self.ui.configitems('usersubs'):
            if fnmatch.fnmatch(self.root, pat):
                subs.append(user)
        for pat, users in self.ui.configitems('reposubs'):
            if fnmatch.fnmatch(self.root, pat):
                subs.extend([u.strip() for u in users.split(',')])
        subs.sort()
        return subs

    def seen(self, node):
        pass

    def url(self, path=None):
        return self.ui.config('web', 'baseurl') + (path or self.root)

    def message(self, node, changes):
        sio = templater.stringio()
        seen = self.seen(node)
        if seen:
            seen = self.strip(seen)
            sio.write('Changeset %s merged to %s\n' %
                      (short(node), self.url()))
            sio.write('First seen in %s\n' % self.url(seen))
        else:
            sio.write('Changeset %s new to %s\n' % (short(node), self.url()))
        sio.write('Committed by %s at %s\n' %
                  (changes[1], templater.isodate(changes[2])))
        sio.write('See %s?cmd=changeset;node=%s for full details\n' %
                  (self.url(), short(node)))
        sio.write('\nDescription:\n')
        sio.write(templater.indent(changes[4], '  '))
        msg = email.MIMEText.MIMEText(sio.getvalue(), 'plain')
        firstline = changes[4].lstrip().split('\n', 1)[0].rstrip()
        subject = '%s %s: %s' % (self.root, self.repo.rev(node), firstline)
        if seen:
            subject = '[merge] ' + subject
        if subject.endswith('.'):
            subject = subject[:-1]
        if len(subject) > 67:
            subject = subject[:64] + '...'
        msg['Subject'] = subject
        msg['X-Hg-Repo'] = self.root
        if '@' in changes[1]:
            msg['From'] = changes[1]
        else:
            msg['From'] = self.ui.config('email', 'from')
        msg['Message-Id'] = '<hg.%s.%s.%s@%s>' % (hex(node),
                                                  int(time.time()),
                                                  hash(self.repo.root),
                                                  socket.getfqdn())
        return msg

    def node(self, node):
        mail = self.ui.sendmail()
        changes = self.repo.changelog.read(node)
        fromaddr = self.ui.config('email', 'from', changes[1])
        msg = self.message(node, changes)
        subs = self.subscribers()
        msg['To'] = ', '.join(subs)
        msgtext = msg.as_string(0)
        mail.sendmail(templater.email(fromaddr),
                      [templater.email(s) for s in subs],
                      msgtext)


def hook(ui, repo, hooktype, node=None, **kwargs):
    n = notifier(ui, repo)
    n.node(bin(node))