from git_taskrepo.testinfo import SemiStrictParser, ParserError
import string
import os
import commands

# Keys from testinfo to populate with
keys = ("test_archs",
        "releases",
        "runfor",
        "types",
        "bugs",
        )

singles = ("test_name",
           "test_description",
           "owner",
           )

class TaskRepoException(Exception):
    pass

class TRX(TaskRepoException):
    pass

class TRX_Parse(TRX):
    pass

class TRX_TestInfo(TRX):
    pass

def only_ascii(s):
    return filter(lambda x: x in string.printable, s)

def _update_taskrepo(taskrepo, taskname, testinfo):
    with taskrepo:
        cur = taskrepo.cursor()
        # Do we already have an entry for this task?
        cur.execute("SELECT id FROM tasks WHERE name=?", (taskname,))
        result = cur.fetchone()
        if result:
            taskid = result[0]
        else:
            cur.execute("INSERT INTO tasks(name) VALUES (?)", (taskname,))
            taskid = cur.lastrowid;

        # Clear old values
        cur.execute("DELETE FROM key_value_inc WHERE task_id=?", (taskid,))
        cur.execute("DELETE FROM key_value_exc WHERE task_id=?", (taskid,))

        # Populate with new values
        for key in keys:
            for value in getattr(testinfo, key):
                if value and type(value) == type(str()) and value[0] == '-':
                    table = "key_value_exc"
                    value = value[1:]
                else:
                    table = "key_value_inc"
                cur.execute("INSERT INTO %s VALUES (?,?,?)" % table, (taskid, key, value))
        for key in singles:
            value = only_ascii(getattr(testinfo, key))
            cur.execute("INSERT INTO key_value_inc VALUES (?,?,?)", (taskid, key, value))

def update_taskrepo(repo, taskrepo, taskpath):
    if os.path.exists(os.path.join(taskpath, "Makefile")):
        (status, output) = commands.getstatusoutput("make -C %s -q testinfo.desc" % taskpath)
        if status == 1:
            os.system("make -C %s testinfo.desc" % taskpath)
    if os.path.exists(os.path.join(taskpath, "testinfo.desc")):
        taskname = taskpath.split(repo.working_tree_dir)[1]
        try:
            testinfo = parse_testinfo(os.path.join(taskpath, "testinfo.desc"))
        except ParserError:
            raise
        _update_taskrepo(taskrepo, taskname, testinfo)
    else:
        raise TRX_TestInfo('No testinfo.desc')

def parse_testinfo(filename):
    p = SemiStrictParser(True)
    p.parse(open(filename).readlines())
    return p.info