/b-gtk/fix-viz

#!/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.commands import open_display

from bzrlib.plugins.gtk.notify import NotifyPopupMenu

Gtk = open_display()

import cgi

import sys

import dbus

import dbus.service

from gi.repository import GObject

from gi.repository import Notify

from bzrlib.bzrdir import BzrDir

from bzrlib.osutils import format_date

from bzrlib.transport import get_transport

menu = NotifyPopupMenu()

try:

    from gi.repository import AppIndicator3

except ImportError:

    icon = Gtk.StatusIcon.new_from_icon_name("bzr-panel")

    icon.connect('popup-menu', menu.display)

    icon.set_visible(False)

    hide_icon = lambda: icon.set_visible(False)

    show_icon = lambda: icon.set_visible(True)

else:

    indicator = AppIndicator3.Indicator.new(

        "bzr-gtk-notify", "bzr-panel", AppIndicator3.IndicatorCategory.OTHER)

    indicator.set_status (AppIndicator3.IndicatorStatus.PASSIVE)

    indicator.set_attention_icon("bzr-panel")

    indicator.set_menu(menu)

    hide_icon = lambda: indicator.set_status (

        AppIndicator3.IndicatorStatus.PASSIVE)

    show_icon = lambda: indicator.set_status (

        AppIndicator3.IndicatorStatus.ATTENTION)

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 = Notify.Notification.new(summary, body, None)

        def start_viz(notification=None, action=None, data=None):

            """Start the viz program."""

            from bzrlib.plugins.gtk.commands import start_viz_window

            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()

        if "actions" in Notify.get_server_caps():

            nw.add_action("inspect", "Inspect", start_viz, None)

            nw.add_action("branch", "Branch", start_branch, None)

        show_icon()

        GObject.timeout_add(5000, hide_icon)

        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")

Notify.init("bzr-notify")

if sys.argv[-1] == 'test':

    # Exit before main loop.

    sys.exit(0)

Gtk.main()