#!/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...")