jack1/python/sessionmanager.py

#!/usr/bin/env python

import libjack
import state
import subprocess
from optparse import OptionParser
from ConfigParser import SafeConfigParser
import os

try:
    import dbus.service
    import gobject
    have_dbus = True
except:
    have_dbus = False


defaults = { "jackclientname": "sessionmanager", "sessiondir": "~/jackSessions" }

class SessionManager( object ):
    def __init__( self ):
        self.config = SafeConfigParser( defaults )
        self.config.read( os.path.expanduser( "~/.jacksessionrc" ) )
        self.jackname = self.config.get( "DEFAULT", "jackclientname" )
        self.cl = libjack.JackClient( self.jackname )
        if self.config.has_section( "infra" ):
            self.infra_clients = {}
            for inf in self.config.items( "infra" ):
                self.infra_clients[inf[0]] = inf[1]
        else:
            self.infra_clients = { "a2j": "a2jmidid" }
        self.implicit_clients = [ "system" ]

        self.sessiondir = os.path.expanduser( self.config.get( "DEFAULT", "sessiondir" ) ) + "/" 
        if not os.path.exists( self.sessiondir ):
            print "Sessiondir %s does not exist. Creating it..."%self.sessiondir
            os.mkdir( self.sessiondir )

    def list_projects( self ):
        files = os.listdir( self.sessiondir )
        files = filter( lambda x: os.path.isdir( os.path.join( self.sessiondir, x ) ), files )
        return files

    def load_session( self, name ):
        if not os.path.exists( self.sessiondir+name+"/session.xml" ):
            print "Session %s does not exist"%name
            return -1

        sd = state.SessionDom( self.sessiondir+name+"/session.xml" )

        g=self.cl.get_graph()

        children = []

        for ic in sd.get_infra_clients():
            if not ic[0] in g.clients.keys():
                children.append( subprocess.Popen( ic[1], shell=True ) )


        print sd.get_client_names()

	if opt.renames:
	    g.ensure_clientnames( sd.get_reg_client_names() )
	    # get graph again... renaming isnt prefect yet.
	    g=self.cl.get_graph()

	# fixup names, doing this unconditionally, because
	# a client rename might have failed.
	sd.fixup_client_names( g )

	# now we have mangled all the names, lets reserve them.
	for (uuid, clientname) in sd.get_uuid_client_pairs():
	    print "reserving name %s"%clientname
	    g.reserve_name( uuid, clientname )

        # build up list of port connections
        conns = []
        for p in sd.get_port_names():
            for c in sd.get_connections_for_port( p ):
                conns.append( (p,c) )
        print conns

        # now fire up the processes
        for cname in sd.get_reg_client_names():
            cmd = sd.get_commandline_for_client( cname )
            children.append( subprocess.Popen( cmd, shell=True ) )

        avail_ports = g.get_port_list()
        # wait for ports to appear, and connect em.

        while len(conns) > 0:
            p = self.cl.port_queue.get()
            print p[0],p[1]
            pname = libjack.port_name(p[0])
            for c1 in conns:
                if c1[0]==pname:
                    if c1[1] in avail_ports:
                        self.cl.connect( pname, c1[1] )

                        conns.remove( c1 )
                        if (c1[1],c1[0]) in conns:
                            conns.remove( (c1[1],c1[0]) )

                    break

                if c1[1]==pname:
                    if c1[0] in avail_ports:
                        self.cl.connect( pname, c1[0] )

                        conns.remove( c1 )
                        if (c1[1],c1[0]) in conns:
                            conns.remove( (c1[1],c1[0]) )
                
                    break

            avail_ports.append( pname )
            

        print "session restored..."
        return 0

    def quit_session( self, name ):
	self.save_session( name, True )


    def save_session( self, name, quit=False ): 
        if os.path.exists( self.sessiondir+name ):
            print "session %s already exists"%name
            return -1
        os.mkdir( self.sessiondir+name )
        g=self.cl.get_graph()
	if quit:
	    notify = self.cl.session_save_and_quit( self.sessiondir+name+"/" )
	else:
	    notify = self.cl.session_save( self.sessiondir+name+"/" )

        for n in notify:
            c = g.get_client( n.clientname )
            c.set_commandline( n.commandline )
	    c.set_uuid( n.uuid )

        sd = state.SessionDom()

        for c in g.clients.values():
            if c.get_commandline() == "":
                if not c.name in self.infra_clients.keys()+self.implicit_clients:
                    g.remove_client( c.name )
                elif c.name in self.implicit_clients:
                    g.remove_client_only( c.name )
                else:
                    c.set_infra( self.infra_clients[c.name] )


        for i in g.clients.values():
            sd.add_client(i)

        f = file( self.sessiondir+name+"/session.xml", "w" )
        f.write( sd.get_xml() )
        f.close()

        print sd.get_xml()
        return 0

    def exit( self ):
        self.cl.close()

if have_dbus:
    class DbusSM( dbus.service.Object ):
        def __init__( self, sm ):
            self.sm = sm
            dbus.service.Object.__init__( self, None, 
                    "/org/jackaudio/sessionmanager",
                    dbus.service.BusName( "org.jackaudio.sessionmanager", bus=dbus.SessionBus() ) )
        @dbus.service.method( dbus_interface="org.jackaudio.sessionmanager",
                in_signature="s", out_signature="i" )
        def save_as( self, name ):
            return self.sm.save_session( name )

        @dbus.service.method( dbus_interface="org.jackaudio.sessionmanager",
                in_signature="s", out_signature="i" )
        def quit_as( self, name ):
            return self.sm.quit_session( name )

        @dbus.service.method( dbus_interface="org.jackaudio.sessionmanager",
                in_signature="s", out_signature="i" )
        def load( self, name ):
            return self.sm.load_session( name )

        @dbus.service.method( dbus_interface="org.jackaudio.sessionmanager",
                in_signature="", out_signature="as" )
        def list( self ):
            return self.sm.list_projects()

        @dbus.service.method( dbus_interface="org.jackaudio.sessionmanager",
                in_signature="", out_signature="" )
        def daemon_quit( self ):
            loop.quit() 

oparser = OptionParser()
oparser.add_option( "--nodbus", action="store_false", dest="dbus", default=have_dbus,
                    help="Dont use DBUS to issue commands to a running instance" )
oparser.add_option( "--daemon", action="store_true", dest="daemon", default=False,
                    help="Start in daemon mode, and listen for dbus commands" )
#oparser.add_option( "--save", action="store_true", dest="save", default=False,
#                    help="Tell SessionManger to save." )
oparser.add_option( "--saveas", action="store", type="string", dest="saveas",
                    help="Save Session As <name>" )

#oparser.add_option( "--quit", action="store_true", dest="quit", default=False,
#                    help="Tell SessionManager to Save And Quit" )

oparser.add_option( "--list", action="store_true", dest="list", default=False,
                    help="List Projects" )
oparser.add_option( "--quitdaemon", action="store_true", dest="quitdaemon", default=False,
                    help="Tell SessionManager Daemon to Exit" )
oparser.add_option( "--quitas", action="store", dest="quitas", type="string",
                    help="SaveAs And Quit" )
oparser.add_option( "--load", action="store", dest="load", type="string",
                    help="Load Session with <name>" )
oparser.add_option( "--renames", action="store_true", dest="renames", default=False,
                    help="Allow renaming offending clients" )

(opt,args) = oparser.parse_args()

if not opt.dbus:
    sm = SessionManager()
    try:
	if opt.saveas:
	    sm.save_session( opt.saveas )

	if opt.load:
	    sm.load_session( opt.load )

	if opt.quitas:
	    sm.quit_session( opt.quitas ) 
    except:
	sm.exit()
	raise

    sm.exit()
else:
    if opt.daemon:
	try:
	    sm = SessionManager()
	    from dbus.mainloop.glib import DBusGMainLoop
	    DBusGMainLoop(set_as_default=True)
	    dbsm = DbusSM( sm )
	    loop = gobject.MainLoop()
	    loop.run()
	except:
	    sm.exit()
	    raise

        sm.exit()
    else:
        session_bus = dbus.SessionBus()
        sm_proxy = session_bus.get_object( "org.jackaudio.sessionmanager", "/org/jackaudio/sessionmanager" )
        sm_iface = dbus.Interface( sm_proxy, "org.jackaudio.sessionmanager" )
        if opt.saveas:
            sm_iface.save_as( opt.saveas )
        if opt.quitas:
            sm_iface.quit_as( opt.quitas )
        if opt.load:
            sm_iface.load( opt.load )
        if opt.list:
            projects = sm_iface.list()
            for i in projects:
                print i

        if opt.quitdaemon:
            sm_iface.quit()