import turbogears as tg
import fedora
from turbogears import controllers, expose, flash
from israwhidebroken.model import Test, Tree, TestResult, hub
from sqlobject import AND
from turbogears import identity, redirect
from cherrypy import request, response
# from israwhidebroken import json
# import logging
# log = logging.getLogger("israwhidebroken.controllers")

def to_int(fstr):
    '''Returns fstr converted to int, or None if fstr is None'''
    return fstr and int(float(fstr))

class Root(controllers.RootController):
    # Main index page
    @expose(template="israwhidebroken.templates.index")
    def index(self, c=None):
        if not c:
            newest_tree = Tree.select(orderBy='-compose_id')[0]
            c = newest_tree.compose_id
        # FIXME BLUH WIDGETIFY THIS
        # TODO need id of prev/next compose for navigation/browsing
        return dict(c=c,
                    in_qa='qa' in identity.current.groups,
                    admin='qa-admin' in identity.current.groups,
                    tests=Test.select(),
                    trees=Tree.select(Tree.q.compose_id == c))

    # Helpers for JSON calls
    def _tree_select(self, *args, **kw):
        qlist = []
        for field in 'id', 'arch', 'compose_id', 'tree_time', 'repodata_time':
            if field in kw:
                val = kw[field]
                if field != 'arch':
                    val = to_int(val)
                qlist.append('%s == %s' % (field, Tree.sqlrepr(val)))
        print qlist
        return Tree.select(' AND '.join(qlist))

    # JSON RPC calls
    @expose(allow_json=True)
    def get_tests(self):
        tests = Test.select()
        return dict(tests=list(tests))

    @expose(allow_json=True)
    def get_trees(self, *args, **kw):
        tree_results = self._tree_select(*args, **kw)
        return dict(trees=list(tree_results))

    # XXX get_results()?

    @identity.require(identity.in_group("qa"))
    @expose(allow_json=True)
    def add_tree(self, arch, compose_id, tree_time=None, repodata_time=None):
        # check to see if tree already exists
        tree_result = self._tree_select(arch=arch, compose_id=compose_id,
                           tree_time=tree_time, repodata_time=repodata_time)
        if tree_result.count():
            return dict(exc='ValueError', tg_flash='Tree exists')
        t = Tree(arch=arch,
                 compose_id=to_int(compose_id),
                 tree_time=to_int(tree_time),
                 repodata_time=to_int(repodata_time))
        hub.commit()
        return dict(tree=t)

    @identity.require(identity.in_group("qa"))
    @expose(allow_json=True)
    def update_tree(self, treeid, arch=None, compose_id=None, tree_time=None, repodata_time=None):
        tr = list(Tree.select(Tree.q.id == treeid))
        if len(tr) == 0:
            return dict(exc='ValueError', tg_flash='Tree %u not found' % treeid)
        tree = tr[0]
        if arch:
            tree.arch = arch
        if tree_time:
            tree.tree_time = to_int(tree_time)
        if repodata_time:
            tree.repodata_time = to_int(repodata_time)
        if compose_id:
            tree.compose_id = to_int(compose_id)
        hub.commit()
        return dict(tree=tree)

    @identity.require(identity.in_group("qa"))
    @expose(allow_json=True)
    def add_result(self, treeid, testid, result, overwrite=False):
        if not overwrite:
            # TODO return an exception if there's already a result here
            pass
        tr = TestResult(tree=to_int(treeid),
                        test=to_int(testid),
                        result=to_int(result),
                        userid=identity.current.user.id,
                        bug_id=None)
        hub.commit() # XXX necessary?
        return dict(id=tr.id)

    @identity.require(identity.in_group("qa"))
    @expose(allow_json=True)
    def delete_result(self, id):
        return dict(youwantedtodelete=id, thisisastub=True)

    # XXX: delete_tree()? (would require qa-admin)

    # Identity stuff (login/logout)
    @expose(template="israwhidebroken.templates.login")
    @expose(allow_json=True)
    def login(self, forward_url=None, *args, **kw):

        if forward_url:
            if isinstance(forward_url, list):
                forward_url = forward_url.pop(0)
            else:
                del request.params['forward_url']

        # If the login was successful...
        if not identity.current.anonymous and identity.was_login_attempted() \
                and not identity.get_identity_errors():
            if 'json' == fedora.tg.util.request_format():
                return dict(user=identity.current.user)
            flash("Welcome, %s" % identity.current.user_name)
            redirect(tg.url(forward_url or '/', kw))

        if identity.was_login_attempted():
            msg = _("The credentials you supplied were not correct or "
                   "did not grant access to this resource.")
        elif identity.get_identity_errors():
            msg = _("You must provide your credentials before accessing "
                   "this resource.")
        else:
            msg = _("Please log in.")
            if not forward_url:
                forward_url = request.headers.get("Referer", "/")

        response.status = 401
        return dict(logging_in=True, message=msg,
            forward_url=forward_url, previous_url=request.path_info,
            original_parameters=request.params)

    @expose(allow_json=True)
    def logout(self):
        identity.current.logout()
        if 'json' in fedora.tg.util.request_format():
            return dict()
        redirect("/")