irkerhook.py

   1 # -*- coding: utf-8 -*-
   2 # vim:fileencoding=utf-8 ai ts=4 sts=4 et sw=4
   3 # Copyright 2012 Neil Muller
   4 # GPL 2+ - see COPYING for details
   5
   6 from bzrlib.config import Option
   7 import socket
   8 import sys
   9 import json
  10
  11 IRKER_PORT = 6659
  12
  13
  14 class IrkerSender(object):
  15     """An irker message sender."""
  16
  17     def __init__(self, branch, revision_id, config):
  18         self.config = config
  19         self.branch = branch
  20         self._revision_id = revision_id
  21         self.revision = None
  22         self.revno = None
  23
  24     def _setup_revision_and_revno(self):
  25         self.revision = self.branch.repository.get_revision(self._revision_id)
  26         self.revno = self.branch.revision_id_to_revno(self._revision_id)
  27
  28     def _format(self):
  29         """Munge the commit info into an irc message"""
  30         delta = self.branch.repository.get_revision_delta(self._revision_id)
  31         files = []
  32         [files.append(f) for (f, _, _) in delta.added]
  33         [files.append(f) for (f, _, _) in delta.removed]
  34         [files.append(f) for (_, f, _, _, _, _) in delta.renamed]
  35         [files.append(f) for (f, _, _, _, _) in delta.modified]
  36
  37         fields = {
  38             'project': self.project(),
  39             'committer': self.revision.committer,
  40             'repo': self.branch.nick,
  41             'rev': '%d' % self.revno,
  42             'files': ' '.join(files),
  43             'logmsg': self.revision.get_summary(),
  44         }
  45         fields.update(self.colours())
  46         text = ('%(bold)s%(project)s:%(reset)s '
  47                 '%(green)s%(committer)s%(reset)s '
  48                 '%(repo)s * %(bold)s%(rev)s%(reset)s / '
  49                 ' %(bold)s%(files)s%(reset)s: %(logmsg)s ' % fields)
  50         return text
  51
  52     def colours(self):
  53         """Utility function to handle the colours"""
  54         colour_style = self.config.get('irker_colours')
  55         colours = {
  56                 'bold': '',
  57                 'green': '',
  58                 'blue': '',
  59                 'red': '',
  60                 'yellow': '',
  61                 'brown': '',
  62                 'magenta': '',
  63                 'cyan': '',
  64                 'reset': '',
  65                 }
  66         # Vaues taken from irker's irkerhook.py
  67         if colour_style == 'ANSI':
  68             colours = {
  69                     'bold': '\x1b[1m',
  70                     'green': '\x1b[1;32m',
  71                     'blue': '\x1b[1;34m',
  72                     'red':  '\x1b[1;31m',
  73                     'yellow': '\x1b[1;33m',
  74                     'brown': '\x1b[33m',
  75                     'magenta': '\x1b[35m',
  76                     'cyan': '\x1b[36m',
  77                     'reset': '\x1b[0m',
  78                     }
  79         elif colour_style == 'mIRC':
  80             colours = {
  81                     'bold': '\x02',
  82                     'green': '\x0303',
  83                     'blue': '\x0302',
  84                     'red': '\x0305',
  85                     'yellow': '\x0307',
  86                     'brown': '\x0305',
  87                     'magenta': '\x0306',
  88                     'cyan': '\x0310',
  89                     'reset': '\x0F',
  90                     }
  91         return colours
  92
  93     def project(self):
  94         project = self.config.get('irker_project')
  95         if project is None:
  96             project = 'No Project name set'
  97         return project
  98
  99     def send(self):
 100         """Send the info to irkerd.
 101         """
 102         self.branch.lock_read()
 103         self.branch.repository.lock_read()
 104         server = self.config.get('irker_server')
 105         if not server:
 106             server = 'localhost'
 107         port = self.config.get('irker_port')
 108         if not port:
 109             port = IRKER_PORT
 110         try:
 111             # Do this after we have locked, to make things faster.
 112             self._setup_revision_and_revno()
 113             channels = self.config.get('irker_channels')
 114             if channels:
 115                 channels = channels.split(',')
 116             msg = unicode(self._format())
 117             # We rely on the server truncating, rather than trying to be
 118             # too fancy here
 119             message = json.dumps({"to": channels, "privmsg": msg})
 120             if channels:
 121                 # We assume tcp, since I'm lazy, so we just grab that bit
 122                 # of irker's code
 123                 try:
 124                     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 125                     sock.connect((server, port))
 126                     sock.sendall(message + "\n")
 127                 except socket.error, e:
 128                     sys.stderr.write("%s\n" % e)
 129                 finally:
 130                     sock.close()
 131         finally:
 132             self.branch.repository.unlock()
 133             self.branch.unlock()
 134
 135 opt_irker_channels = Option('irker_channels',
 136     help='Channel(s) to post commit messages to.')
 137 opt_irker_colours = Option('irker_colours',
 138     help='Colour option for irker.')
 139 opt_irker_project = Option('irker_project',
 140     help='Project name to use.')
 141 opt_irker_server = Option('irker_server',
 142     help='host for the irkerd server (default localhost).')
 143 opt_irker_port = Option('irker_port',
 144     help='port for the irkerd server (default %d)' % IRKER_PORT)