dbus.py 3.11 KB
Newer Older
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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
###
### Internal dbus management. This can go away once native gtk+ is in fink or
### macports and we can require their dbus.
###



import os, sys, commands, signal

if sys.platform != "darwin":
	raise ImportError("System platform is not OS/X")

import osx, osx.nsapp
from common.configpaths import gajimpaths


_GTK_BASE = "/Library/Frameworks/GTK+.framework/Versions/Current"


def readEnv():
	gajimpaths.add_from_root(u'dbus.env', u'dbus.env')
	try:
		dbus_env = file(gajimpaths[u'dbus.env'], "r")
	except:
		return False
	try:
		line1 = dbus_env.readline()
		line2 = dbus_env.readline()
		dbus_env.close()
	except:
		print "Invalid dbus.env file"
		return False
	return parseEnv(line1, line2)


def parseEnv(line1, line2):
	try:
		if not line1 or not line2:
			return False
		if (not line1.startswith("DBUS_SESSION_BUS_ADDRESS=") or
		not line2.startswith("DBUS_SESSION_BUS_PID=")):
			return False
		arr = line2.split("=")
		pid = arr[1].strip().strip('"')
		if not osx.checkPID(int(pid), "dbus-daemon"):
			return False
		line1 = line1.strip()
		loc = line1.find("=")
		address = line1[loc + 1:]
		address = address.strip().strip('"')
		return [address, pid]
	except Exception, e:
		print "Invalid dbus.env file", e
		return False
	return None


def setEnv(env):
	os.environ['DBUS_SESSION_BUS_ADDRESS'] = env[0]
	os.environ['DBUS_SESSION_BUS_PID'] = env[1]
	return


def writeEnv(env):
	gajimpaths.add_from_root(u'dbus.env', u'dbus.env')
	try:
		dbus_env = file(gajimpaths[u'dbus.env'], "w+")
		dbus_env.write("DBUS_SESSION_BUS_ADDRESS=\"" + env[0] + "\"\n")
		dbus_env.write("DBUS_SESSION_BUS_PID=\"" + env[1] + "\"\n")
		dbus_env.close()
	except Exception, e:
		print "Failed to write file: %s" % gajimpaths[u'dbus.env']
		print str(e)
	return


def checkUUID():
	if os.path.exists(_GTK_BASE + "/var/lib/dbus/machine-id"):
		return
	ret = commands.getstatusoutput(_GTK_BASE + "/bin/dbus-uuidgen --ensure")
	if ret[0] != 0:
		print "Failed to initialize dbus machine UUID:", ret[1]
	return


def load(start):
	# Look for existing external session and just use it if it exists
	if (('DBUS_SESSION_BUS_ADDRESS' in os.environ) and
		('DBUS_SESSION_BUS_PID' in os.environ) and
		osx.checkPID(int(os.environ['DBUS_SESSION_BUS_PID']), 'dbus-daemon')):
		return True

	# Look for our own internal session
	env = readEnv()
	if env:
		# We have a valid existing dbus session, yay
		setEnv(env)
		return True

	# Initialize the machine's UUID if not done yet
	checkUUID()

	if start:
		# None found, start a new session
		print "Starting new dbus session"
		#cmd = os.path.join(osx.nsapp.getBundlePath(),
		#		  "Contents/Resources/bin/dbus-launch --exit-with-session")
		cmd = _GTK_BASE + "/bin/dbus-launch --exit-with-session"
		ret = commands.getstatusoutput(cmd)
		arr = ret[1].split("\n")
		if len(arr) != 2:
			print "Failed to start internal dbus session:"
			print ret[1]
			return
		env = parseEnv(arr[0].strip(), arr[1].strip())
		if not env:
			print "Failed to start internal dbus session:"
			print ret[1]
			return
		setEnv(env)
		writeEnv(env)
		return True
	return False


def shutdown():
	env = readEnv()
	if not env:
		return
	os.kill(int(env[1]), signal.SIGINT)
	return