import lib

import SimpleXMLRPCServer

class Handler:

    def __init__(self, recv_pool, send_pool):

        self.recv_pool = recv_pool

        self.send_pool = send_pool

        self.getMsg = self.sendMsg

    def _listMethods(self):

        return ['setup', 'play', 'stop', 'close', 'getMsg']

    def _methodHelp(self, method):

        if method == 'setup':

            return "Setup the Media: setup( filename, mux, vcodec, vbitrate,"\

                   " fps, acodec, abitrate, width, height, port, options"

        elif method == 'play':

            return "Play the Media: play()"

        elif method == 'stop':

            return "Stop the Media: stop()"

        elif method == 'close':

            return "Close the connection: close()"

        elif method == 'getMsg':

            return "Return the first message in the pool: getMsg()"

        else:

            # By convention, return empty

            # string if no help is available

            return ""

    def setup(self, filename, mux, vcodec, vbitrate,\

            fps, acodec, abitrate, width, height, port, options):

        msg = "%s %s %s %s %s %s %s" % (filename, mux, vcodec, vbitrate,\

                                        fps, acodec, abitrate, width, height, port)

        if len(options) > 0:

            for opt in options:

                msg += " %s" % opt

        self.recv_pool.append("SETUP")

        self.recv_pool.append(msg)

        return self.sendMsg()

    def play(self):

        self.recv_pool.append("PLAY")

        return self.sendMsg()

    def stop(self):

        self.recv_pool.append("STOP")

        return self.sendMsg()

    def close(self):

        self.recv_pool.append("CLOSE")

        return self.sendMsg()

    def sendMsg(self):

        if self.send_pool != []:

            return self.send_pool.pop(0)

        else:

            return ""

class Server:

    def __init__(self, config):

        self.host = 'localhost'

        self.port = int(config.get("Comm", "port"))

        self.finish = 0

        self.recv_pool = []

        self.send_pool = []

        self.handler = Handler(self.recv_pool, self.send_pool)

        self.xmlrpc = SimpleXMLRPCServer.SimpleXMLRPCServer((self.host, self.port))

        self.xmlrpc.register_instance(self.handler)

    def getMsg(self, size):

        if self.recv_pool != []:

            return self.recv_pool.pop(0)

        else:

            return ""

    def sendMsg(self, msg):

        self.send_pool.append(msg)

    def Ack(self, command):

        msg = "[%s] Command %s received" % (lib.now(), command)

        self.sendMsg(msg + "\n")

    def getRequest(self):

        self.xmlrpc.handle_request()

        return (0, "RPC Client")

    def disconnect_client(self, connection):

        connection = 0

    def stop(self):

        self.xmlrpc.server_close()