#!/usr/bin/env python
import logging
import os
from tornado.escape import json_decode
from tornado.ioloop import IOLoop
from tornado.options import define, options
from tornado.web import Application, RequestHandler, URLSpec
from candolint import uimodules
from candolint.models import database, Project, Change, Check
rel = lambda *x: os.path.abspath(os.path.join(os.path.dirname(__file__), *x))
define('listen', metavar='IP', default='127.0.0.1')
define('port', metavar='PORT', default=8033, type=int)
define('xheaders', metavar='True|False', default=False, type=bool)
define('debug', metavar='True|False', default=False, type=bool)
class BaseHandler(RequestHandler):
    def prepare(self):
        database.connect()
        super(BaseHandler, self).prepare()
    def on_finish(self):
        if not database.is_closed():
            database.close()
class IndexHandler(BaseHandler):
    def get(self):
        checks = (Check
                  .select(Check, Project, Change)
                  .join(Project)
                  .switch(Check)
                  .join(Change)
                  .group_by(Check.project))
        self.render('index.html', checks=checks)
class ProjectHandler(BaseHandler):
    def get(self, domain, user, name):
        project = Project.select().where(
            Project.domain == domain,
            Project.user == (user if user != '-' else None),
            Project.name == name).get()
        checks = (Check
                  .select(Check, Change)
                  .join(Change)
                  .where(Check.project == project)
                  .limit(10))
        self.render('project.html', project=project, checks=checks)
class CheckHandler(BaseHandler):
    def get(self, domain, user, name, check_num):
        project = Project.select().where(
            Project.domain == domain,
            Project.user == (user if user != '-' else None),
            Project.name == name).get()
        if check_num == 'latest':
            check = Check.get(project=project)
        else:
            check = Check.get(project=project, ordinal=check_num)
        lines = json_decode(check.lines)
        self.render('check.html', project=project, check=check, lines=lines)
class CandolintViewer(Application):
    def __init__(self):
        handlers = [
            URLSpec(r'/', IndexHandler),
            URLSpec(r'/([.a-z0-9_-]+)/([^/]+)/([^/]+)', ProjectHandler),
            URLSpec(r'/([.a-z0-9_-]+)/([^/]+)/([^/]+)/([\d]+|latest)', CheckHandler),
        ]
        settings = dict(
            static_path=rel('static'),
            template_path=rel('templates'),
            ui_modules=uimodules,
            debug=options.debug
        )
        super(CandolintViewer, self).__init__(handlers, **settings)
        if options.debug:
            logging.getLogger('peewee').setLevel(logging.DEBUG)
        database.init(rel('database.sqlite'))
    def listen(self, port, address='', **kwargs):
        name = self.__class__.__name__
        logging.info('%s is serving on %s:%d', name, address, port)
        super(CandolintViewer, self).listen(port, address, **kwargs)
def main():
    options.parse_command_line()
    application = CandolintViewer()
    application.listen(options.port, address=options.listen, xheaders=options.xheaders)
    IOLoop.instance().start()
if __name__ == '__main__':
    main()