import os
import sys
import rcm

class rcm_protocol:
    def __init__(self,rcm_server=None):
	if(rcm_server):	self.rcm_server=rcm_server
	
    def config(self,build_platform=''):
        conf=rcm.rcm_config()
        if(build_platform):
            (check,url)=self.rcm_server.get_checksum(build_platform)
            conf.set_version(check,url)
        queueList = self.rcm_server.get_queue()
        for q in queueList:
            conf.add_queue(q)
	for vnc_id,menu_entry in self.rcm_server.pconfig.get_vnc_menu().iteritems():
	    conf.add_vnc(vnc_id,menu_entry)
        conf.serialize()


    def version(self,build_platform=''):
        print "get version"
        if (self.client_sendfunc):
            return self.client_sendfunc("version "+build_platform)


    def loginlist(self,subnet=''):
#	import socket
        self.rcm_server.subnet = subnet
#        fullhostname = socket.getfqdn()
        self.rcm_server.fill_sessions_hash()
        s=rcm.rcm_sessions()
        for sid, ses in self.rcm_server.sessions.items():
            s.array.append(self.rcm_server.sessions[sid])
        sys.stdout.write(rcm.serverOutputString)
	sys.stdout.write(s.get_string())
	sys.stdout.flush()

    def list(self,subnet=''):
        self.rcm_server.subnet = subnet
        
        self.rcm_server.load_sessions()
        s=rcm.rcm_sessions()
        for sid in self.rcm_server.sids['run']:
            s.array.append(self.rcm_server.sessions[sid])
        sys.stdout.write(rcm.serverOutputString)
	sys.stdout.write(s.get_string())
	sys.stdout.flush()
    
    def new(self,geometry='',queue='',sessionname='',subnet='',vncpassword='',vncpassword_crypted='',vnc_id=''):
        print "create new vnc display session"
	if(subnet): self.rcm_server.subnet = subnet
	if(queue): self.rcm_server.queue = queue
	if(sessionname): self.rcm_server.sessionname = sessionname
	if(vncpassword): self.rcm_server.vncpassword = vncpassword
	if(vncpassword_crypted): self.rcm_server.vncpassword_crypted = vncpassword_crypted
	
	self.rcm_server.substitutions['RCM_GEOMETRY'] = geometry	
	self.rcm_server.set_vnc_setup(vnc_id)
        self.rcm_server.execute_new()
    
    
    def kill(self,session_id=''):
	self.rcm_server.load_sessions()
	if(session_id):
           if session_id in self.rcm_server.sids['run']:
                jid=self.rcm_server.sessions[session_id].hash['jobid']
                self.rcm_server.kill_job(jid)
                file='%s/%s.session' % (self.rcm_server.get_rcmdirs()[0],session_id)
                c=rcm.rcm_session(fromfile=file)
                c.hash['state']='killing'
                c.serialize(file)
           else:
		sys.stderr.write("Not running session: %s\n" % (session_id))
		sys.stderr.flush()
                sys.exit(1)