#!/usr/bin/python

 import os

 import lib

 import sys

 import imp

 import ConfigParser

 import logging as log

 log.basicConfig(level=log.DEBUG,

                 format="%(asctime)s %(levelname)-8s %(message)s",

                 datefmt='%Y-%m-%d %H:%M:%S')

 config = ConfigParser.ConfigParser()

 config.read("stream.conf")

 def load_module(pathlist, name):

     fp, path, desc = imp.find_module(name, pathlist)

     try:

         module = imp.load_module(name, fp, path, desc)

         return module

     finally:

         if fp:

             fp.close()

 media_plugin = config.get("Media", "engine")

 media_plugin_module = load_module(["./plugins/media"], media_plugin)

 media = media_plugin_module.Media(config)

 comm_plugin = config.get("Comm", "engine")

 comm_plugin_module = load_module(["./plugins/comm"], comm_plugin)

 server = comm_plugin_module.Server(config)

 log.info("Starting GMyth-Stream server")

 '''

 PROTOCOL DESCRIPTION

 =====================

 COMMAND OPTIONS

 -> SETUP DESCRIPTION

 |-> used to setup transcoding and streaming parameters

 |-> must be used before any "PLAY" command

 |-> e.g:

 file://file_name mux vcodec vbitrate fps acodec abitrate width height options

 dvd://title_number mux vcodec vbitrate fps acodec abitrate width height options

 -> PLAY DESCRIPTION

  |-> used to start transcoding and streaming of file

  |-> must be used just if SETUP was used before

  |-> after it, _must_ send STOP

 -> STOP DESCRIPTION

  |-> used to stop transcoding and streaming process

  |-> must be used just if PLAY was used before

  |-> must be used after PLAY

 -> QUIT DESCRIPTION

  |-> used to quit the main loop (quit program)

 '''

 nextport = 0

 setup = (False, "STOPPED")

 def do_setup(server, filename, mux, vcodec, vbitrate, fps, acodec, abitrate,

              width, height, *options):

     global nextport

     global setup

     if setup[1] != "PLAYING":

         nextport += 1

         ret = media.setup(filename, mux, vcodec, vbitrate, fps, acodec,

                           abitrate, width, height, nextport, options)

         if ret[0]:

             server.sendOk()

         else:

             server.sendNotOk(ret[1])

         setup = (True, setup[1])

     else: server.sendNotOk("You must STOP before SETingUP again")

     return True

 def do_play(server):

     global setup

     if setup[0] and setup[1] == "STOPPED":

         setup = (setup[0], "PLAYING")

         ret = media.play()

         if ret[0]:

             server.sendOk("%d" % nextport)

         else:

             server.sendNotOk(ret[1])

     else:

         if setup[1] == "STOPPED":

             server.sendNotOk("You must SETUP before PLAYing")

         else:

             server.sendNotOk("You must STOP before PLAYing again")

     return True

 def do_stop(server):

     global setup

     media.stop()

     setup = (False, "STOPPED")

     server.sendOk()

     return True

 def do_list(server, *directory):

     file_list = []

     for j in directory:

         lib.list_media_files(j, file_list)

     server.sendOk(file_list)

     return True

 def do_quit(server):

     server.finish = 1

     media.stop()

     server.sendOk()

     return True

 mapping = {

     "SETUP": do_setup,

     "PLAY": do_play,

     "STOP": do_stop,

     "LIST": do_list,

     "QUIT": do_quit,

     }

 def dispatch(server, msg):

     pieces = msg.split()

     if len(pieces) < 1:

         log.error("Invalid client command format: %r" % msg)

         server.sendNotOk("Invalid Format")

         return False

     cmd = pieces[0]

     f = mapping.get(cmd, None)

     if not f:

         log.error("Unknow command: %r" % msg)

         server.sendNotOk("Unknow Command")

         return False

     try:

         return f(server, *pieces[1:])

     except Exception, e:

         log.error("Could not execute %r: %s" % (msg, e))

         server.sendNotOk(str(e))

         return False

 while not server.finish:

     conn, client, port = server.getRequest()

     if nextport == 0:

         nextport = port

     while not server.finish:

         msg = server.getMsg()

         if not msg:

             break

         log.info("Client %s sent command: %r" % (client, msg))

         dispatch(server, msg)

     log.info("Closing connection with %s" % (client,))

     server.disconnect_client(conn)

     try:

         os.wait()

     except Exception, e:

         log.error(e)

 server.stop()

 log.info("Server stopped. Closing...")