import os
import subprocess
import fcntl

import lib.utils as utils
import lib.server as server

from select import select

def _setup_mythfilename(self):
    # mythtv:mythtv@192.168.3.110:6543/1002_20070426230000.nuv

    try:
        _mysql = self.args["input"].split("@")[0].split(":")
    except IndexError, e:
        _mysql = ["mythtv", "mythtv"]

    self.log.debug("aqui 1")

    _args = self.args["input"].split("@")[1].split(":")
    gmyth_dict = {}
    gmyth_dict["mysql"] = _mysql
    gmyth_dict["backend"] = _args[0]
    gmyth_dict["port"] = _args[1].split("/", 1)[0]

    self.log.debug("aqui 2 %s" % _args)

    _tmp_file = _args[1].split("/", 1)[1]

    self.log.debug("aqui 2a %s" % _tmp_file)

    if _tmp_file.find("channel") >= 0:
        gmyth_dict["kind"] = "c"
        gmyth_dict["cfile"] = _tmp_file.split("=")[1]
        self.log.debug("aqui 3")
    else:
        gmyth_dict["kind"] = "f"
        gmyth_dict["cfile"] = _tmp_file
        self.log.debug("aqui 4")

    self.log.debug("aqui 5")

    self.args["input"] = "-"
    return gmyth_dict
# _setup_mythfilename

def _setup_mythfile(err):
    size = err.read(20).split("\n")[0].split("Size:")[1]
    flags = fcntl.fcntl (err, fcntl.F_GETFL, 0) | os.O_NONBLOCK
    fcntl.fcntl(err, fcntl.F_SETFL, flags)
    return size
# _setup_mythfile

def _setup_gmythcat(self):
    gmyth_cat = utils.which("gmyth-cat")
    return [ utils.which("gmyth-cat"),
            "-h", self.args["gmyth-cat"]["backend"],
            "-p", self.args["gmyth-cat"]["port"],
            "-" + self.args["gmyth-cat"]["kind"],
            self.args["gmyth-cat"]["cfile"]
           ]
# _setup_gmythcat

def start_myth(self, outfd):
    opts = _setup_gmythcat(self)
    try:
        self.gmyth = subprocess.Popen(opts, stdout=subprocess.PIPE,
                                      stderr=subprocess.PIPE,
                                      close_fds=True)
    except Exception, e:
        self.log.error("Error executing gmyth-cat: %s" % e)
        return False

    if not self._run_mencoder(input=self.gmyth.stdout,
                              output=subprocess.PIPE):
        return False

    if self.args["gmyth-cat"]["kind"] == "f":
        try:
            size = _setup_mythfile(self.gmyth.stderr)
            self.log.debug("Size of file: %s" % size)
        except Exception, e:
            self.log.error("Problems getting size of file: %s" % e)
            return False

    try:
        while self.proc and self.proc.poll() == None:
            r, w, x = select([self.gmyth.stderr, self.proc.stdout],
                             [], [], 0)
            if self.proc.stdout in r:
                d = self.proc.stdout.read(4096)
                outfd.write(d)

            if self.gmyth.stderr in r:
                partial = self.gmyth.stderr.read(50).split("\n")[-2]
                if partial != "":
                    self.status = utils.progress_bar(self.log,
                                                     int(partial),
                                                     int(size), 50)

    except IndexError, e:
        pass
    except Exception, e:
        self.log.error("Problems handling data: %s" % e)
        return False

    self.log.info("Finished sending data")
    return True
# _start_myth()