#!/usr/bin/python

# EeePC Sync using unison 
# 
# Frederic Descamps <lefred@inuits.be>
#
# 16 JAN 2009
#

# This program should run on the machine where you want to synchronize the EeePC
# Both machines requires the following :
# - unison installed
# - a folder EeePC on ~/Desktop (you can change that name below via the variable 'folder'
# - having ssh access to the EeePC via a key
#
# on the EeePC you need to have the following file :
# - /etc/avahi/services/eeepc.service :
# <?xml version="1.0" standalone='no'?><!--*-nxml-*-->
# <!DOCTYPE service-group SYSTEM "avahi-service.dtd">
# 
# <!-- $Id: ssh.service 1391 2007-02-07 11:54:54Z lennart $ -->
# 
# <!--
#   This file is part of avahi.
# 
#   avahi is free software; you can redistribute it and/or modify it
#   under the terms of the GNU Lesser General Public License as
#   published by the Free Software Foundation; either version 2 of the
#   License, or (at your option) any later version.
# 
#   avahi is distributed in the hope that it will be useful, but
#   WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
#   General Public License for more details.
# 
#   You should have received a copy of the GNU Lesser General Public
#   License along with avahi; if not, write to the Free Software
#   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
#   02111-1307 USA.
# -->
# 
# <!-- See avahi.service(5) for more information about this configuration file -->
# 
# <service-group>
# 
#   <name replace-wildcards="yes">%h</name>
# 
#   <service>
#     <type>_eeepc._tcp</type>
#   </service>
# 
# </service-group>
#


import dbus, gobject, avahi
from dbus import DBusException
from dbus.mainloop.glib import DBusGMainLoop
import pygtk
pygtk.require('2.0')
import gtk
import os
import sys 

TYPE = '_eeepc._tcp'

global ip
global folder

folder='/Desktop/EeePC'


try:
    import pynotify
    if pynotify.init("EeePC Synchronization Tool"):
        n = pynotify.Notification("EeePC Synchronization Tool Started", "Looking for your EeePC")
        n.set_urgency(pynotify.URGENCY_LOW)
        n.set_timeout(2000)
        n.show()
    else:
        print "there was a problem initializing the pynotify module"
except:
    print "you don't seem to have pynotify installed"



class StatusIcc:

    # activate callback
    def activate( self, widget, data=None):
      self.doit()
  
    # destroyer callback
    def  destroyer(self, widget,response_id, data= None):
           if response_id == gtk.RESPONSE_OK:
               #gtk.main_quit()
               #gobject.MainLoop().quit()
               sys.exit()
           else:
               widget.hide()

    # popup callback
    def popup(self, button, widget, data=None):
        dialog = gtk.MessageDialog(
        parent         = None,
        flags          = gtk.DIALOG_DESTROY_WITH_PARENT,
        type           = gtk.MESSAGE_INFO,
        buttons        = gtk.BUTTONS_OK_CANCEL,
        message_format = "Do you want to close EeePC Sync ?")
        dialog.set_title('EeePC Sync')
        dialog.connect('response', self.destroyer)
        dialog.show()

    def __init__(self):
        # create a new Status Icon
        self.staticon = gtk.StatusIcon()
        self.staticon.set_from_stock(gtk.STOCK_REFRESH)
        self.staticon.set_tooltip("Click to find EeePC & Sync")
        self.staticon.set_blinking(False)
        self.staticon.connect("activate", self.activate)
        self.staticon.connect("popup_menu", self.popup)
        self.staticon.set_visible(True)

        # invoking the main()
        self.doit()
        gtk.main()

    def doit(self):
          def sync_cb(n, action):
    		assert action == "sync"
		global ip
    		global folder
    		print "Syncrhonizing with %s" % (ip)
    		os.system("/usr/bin/unison -ui text -batch  ssh://" + ip + folder +" ~/Desktop/EeePC/")
    		n.close()
    		gtk.main_quit()
    		n = pynotify.Notification("EeePC Synchronized !", "Your EeePC is now synchronized.")
    		n.set_urgency(pynotify.URGENCY_LOW)
    		n.set_timeout(pynotify.EXPIRES_DEFAULT)
    		n.show()
   
    	  def default_cb(n, action):
    		assert action == "default"
    		n.close()
    		gtk.main_quit()

    	  def service_resolved(*args):
    		n = pynotify.Notification("EeePC Synchronization Tool", "Found a EeePC :\n  %s ( %s )" % (args[2], args[7]) )
    		global ip 
    		ip = args[7]
    		n.set_urgency(pynotify.URGENCY_LOW)
    		n.set_timeout(0)
    		n.set_category("device")
    		n.add_action("sync", "Sync", sync_cb)
    		n.add_action("default", "Default Action", default_cb)
    		n.show()
    		gtk.main()

    	  def print_error(*args):
    		print 'error_handler'
    		print args[0]
    
    	  def myhandler(interface, protocol, name, stype, domain, flags):
    		if flags & avahi.LOOKUP_RESULT_LOCAL:
            		# local service, skip
            		pass

        	server.ResolveService(interface, protocol, name, stype, 
            		domain, avahi.PROTO_UNSPEC, dbus.UInt32(0), 
            		reply_handler=service_resolved, error_handler=print_error)

	  loop = DBusGMainLoop()

	  bus = dbus.SystemBus(mainloop=loop)

	  server = dbus.Interface( bus.get_object(avahi.DBUS_NAME, '/'),
        		'org.freedesktop.Avahi.Server')

	  sbrowser = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
        		server.ServiceBrowserNew(avahi.IF_UNSPEC,
        		avahi.PROTO_UNSPEC, TYPE, 'local', dbus.UInt32(0))),
	        	avahi.DBUS_INTERFACE_SERVICE_BROWSER)

	  sbrowser.connect_to_signal("ItemNew", myhandler)

	  gobject.MainLoop().run()



statusicon = StatusIcc()