~jameinel/bzr-gtk/sanitize_commit_messages

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#!/usr/bin/env python

"""Run the bzr tray icon.

This is a background program which will pop up a notification on the users
screen when a commit occurs.
"""

from bzrlib.plugin import load_plugins
load_plugins()

from bzrlib.plugins.gtk import open_display, icon_path

from bzrlib.plugins.gtk.notify import NotifyPopupMenu
gtk = open_display()
menu = NotifyPopupMenu()
icon = gtk.status_icon_new_from_file(icon_path("bzr-icon-64.png"))
icon.connect('popup-menu', menu.display)

import cgi
import dbus
import dbus.service
import pynotify
from bzrlib.bzrdir import BzrDir
from bzrlib import errors
from bzrlib.osutils import format_date
from bzrlib.transport import get_transport
if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
	import dbus.glib
BROADCAST_INTERFACE = "org.bazaarvcs.plugins.dbus.Broadcast"
bus = dbus.SessionBus()

def catch_branch(revision_id, urls):
	# TODO: show all the urls, or perhaps choose the 'best'.
	url = urls[0]
	try:
		if isinstance(revision_id, unicode):
			revision_id = revision_id.encode('utf8')
		transport = get_transport(url)
		a_dir = BzrDir.open_from_transport(transport)
		branch = a_dir.open_branch()
		revno = branch.revision_id_to_revno(revision_id)
		revision = branch.repository.get_revision(revision_id)
		summary = 'New revision %d in %s' % (revno, url)
		body  = 'Committer: %s\n' % revision.committer
		body += 'Date: %s\n' % format_date(revision.timestamp,
			revision.timezone)
		body += '\n'
		body += revision.message
		body = cgi.escape(body)
		nw = pynotify.Notification(summary, body)
		def start_viz(notification=None, action=None, data=None):
			"""Start the viz program."""
			pp = start_viz_window(branch, revision_id)
			pp.show()
		def start_branch(notification=None, action=None, data=None):
			"""Start a Branch dialog"""
			from bzrlib.plugins.gtk.branch import BranchDialog
			bd = BranchDialog(remote_path=url)
			bd.run()
		nw.add_action("inspect", "Inspect", start_viz, None)
		nw.add_action("branch", "Branch", start_branch, None)
		nw.set_timeout(5000)
		nw.show()
	except Exception, e:
		print e
		raise
bus.add_signal_receiver(catch_branch,
						dbus_interface=BROADCAST_INTERFACE,
						signal_name="Revision")
pynotify.init("bzr-notify")
gtk.main()