from __future__ import absolute_import, division
import traceback
from peewee import DoesNotExist
from tornado.escape import json_decode
from tornado.web import HTTPError, RequestHandler
from tornado.web import ErrorHandler as BaseErrorHandler
from candolint.models import database, Project, Change, Check
def get_or_404(query, *args, **kwargs):
    try:
        return query.get(*args, **kwargs)
    except DoesNotExist:
        raise HTTPError(404)
def get_project_or_404(domain, user, name):
    pq = Project.select().where(
        Project.domain == domain,
        Project.user == (user if user != '-' else None),
        Project.name == name)
    return get_or_404(pq)
class BaseHandler(RequestHandler):
    def prepare(self):
        database.connect()
        super(BaseHandler, self).prepare()
    def on_finish(self):
        if not database.is_closed():
            database.close()
    def write_error(self, status_code, **kwargs):
        data = {
            'code': status_code,
            'message': self._reason,
            'debug_message': ''
        }
        if 'exc_info' in kwargs:
            if self.settings.get('serve_traceback'):
                fexc = traceback.format_exception(*kwargs['exc_info'])
                data['debug_message'] = '\n'.join(fexc)
        if status_code == 404:
            self.render('404.html', **data)
        else:
            self.render('500.html', **data)
class IndexHandler(BaseHandler):
    def get(self):
        checks = (Check
                  .select(Project, Change, *Check.get_light_fields())
                  .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 = get_project_or_404(domain, user, name)
        checks = (Check
                  .select(Change, *Check.get_light_fields())
                  .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, format_='html'):
        project = get_project_or_404(domain, user, name)
        check_query = (Check
                       .select(Check, Change)
                       .join(Change)
                       .where(Check.project == project))
        if check_num != 'latest':
            check_query = check_query.where(Check.ordinal == check_num)
        check = get_or_404(check_query)
        lines = json_decode(check.lines)
        if format_ == 'raw':
            self.get_raw(project, check, lines)
        else:
            self.get_html(project, check, lines)
    def get_raw(self, project, check, lines):
        self.set_header('Content-Type', 'text/plain; charset=utf-8')
        for line in lines:
            self.write(line['text'])
            self.write('\n')
    def get_html(self, project, check, lines):
        adapter = project.get_adapter()
        for line in lines:
            if 'filename' in line:
                line['link'] = adapter.get_line_url(check.change, line)
        self.render('check.html', project=project, check=check, lines=lines, adapter=adapter)
class StatusHandler(BaseHandler):
    def get(self, domain, user, name):
        project = get_project_or_404(domain, user, name)
        check = get_or_404(Check, project=project)
        parts = [('#555', 30, 14.5, 'lint')]
        if not check.success:
            parts.append(('#777', 62, 30.5, 'unknown'))
        elif check.errors or check.warnings:
            if check.errors:
                msg = self.locale.translate('{} error', '{} errors', check.errors)
                text = msg.format(check.errors)
                width = 7 + 6 * len(text) + 7
                parts.append(('#da314b', width, width // 2 - 0.5, text))
            if check.warnings:
                msg = self.locale.translate('{} warning', '{} warnings', check.warnings)
                text = msg.format(check.warnings)
                width = 7 + 6 * len(text) + 9
                parts.append(('#faa732', width, width // 2 + 0.5, text))
        else:
            parts.append(('#8cc14c', 40, 19.5, 'none'))
        width = sum(p[1] for p in parts)
        self.set_header('Content-Type', 'image/svg+xml; charset=utf-8')
        self.render('status.svg', width=width, parts=parts, height=20)
class ErrorHandler(BaseHandler, BaseErrorHandler):
    pass