#!/usr/bin/env python

# Copyright 2016 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.

import os
import time
from HTMLParser import HTMLParser

from google.appengine.api import urlfetch

import jinja2
import webapp2

JINJA_ENVIRONMENT = jinja2.Environment(
    loader=jinja2.FileSystemLoader(os.path.dirname(__file__)),
    extensions=['jinja2.ext.autoescape'],
    autoescape=True)

BASE_URL = 'https://luci-scheduler.appspot.com/jobs/'
SNAPSHOT_URL = 'https://storage.googleapis.com/fuchsia/jiri/snapshots'

TARGETS = [
    [
        ('topaz'),
        ('fuchsia/topaz-arm64-debug-qemu_kvm', 'topaz-arm64-debug-qemu_kvm'),
        ('fuchsia/topaz-arm64-release-qemu_kvm', 'topaz-arm64-release-qemu_kvm'),
        ('fuchsia/topaz-x64-debug-qemu_kvm', 'topaz-x64-debug-qemu_kvm'),
        ('fuchsia/topaz-x64-release-qemu_kvm', 'topaz-x64-release-qemu_kvm'),
    ],
    [
        ('peridot'),
        ('fuchsia/peridot-arm64-debug-qemu_kvm', 'peridot-arm64-debug-qemu_kvm'),
        ('fuchsia/peridot-arm64-release-qemu_kvm', 'peridot-arm64-release-qemu_kvm'),
        ('fuchsia/peridot-x64-debug-qemu_kvm', 'peridot-x64-debug-qemu_kvm'),
        ('fuchsia/peridot-x64-release-qemu_kvm', 'peridot-x64-release-qemu_kvm'),
    ],
    [
        ('garnet'),
        ('fuchsia/garnet-arm64-debug-qemu_kvm', 'garnet-arm64-debug-qemu_kvm'),
        ('fuchsia/garnet-arm64-release-qemu_kvm', 'garnet-arm64-release-qemu_kvm'),
        ('fuchsia/garnet-x64-debug-qemu_kvm', 'garnet-x64-debug-qemu_kvm'),
        ('fuchsia/garnet-x64-release-qemu_kvm', 'garnet-x64-release-qemu_kvm'),
    ],
    [
        ('zircon'),
        ('fuchsia/zircon-x64-clang-qemu_kvm', 'zircon-x64-clang-qemu_kvm'),
        ('fuchsia/zircon-x64-gcc-qemu_kvm', 'zircon-x64-gcc-qemu_kvm'),
        ('fuchsia/zircon-arm64-clang-qemu_kvm', 'zircon-arm64-clang-qemu_kvm'),
        ('fuchsia/zircon-arm64-gcc-qemu_kvm', 'zircon-arm64-gcc-qemu_kvm'),
    ],
    [
        ('prebuilders'),
        ('fuchsia/web_view-linux', 'web_view-linux'),
        ('fuchsia/ffmpeg-linux', 'ffmpeg-linux'),
        ('fuchsia/jiri-x64-linux', 'jiri-x64-linux'),
        ('fuchsia/jiri-x64-mac', 'jiri-x64-mac'),
    ],
    [
        ('rollers'),
        ('fuchsia/zircon-roller', 'zircon-roller'),
        ('fuchsia/garnet-roller', 'garnet-roller'),
        ('fuchsia/peridot-roller', 'peridot-roller'),
    ]
]


class BuildResult:
    """This is an enum of sorts, except the values match css class names."""
    Pass = "pass"
    Fail = "fail"
    ServerError = "server_error"
    ParserError = "parser_error"


class LuciResultParser(HTMLParser):
    """Parses the HTML of the LUCI scheduler page to get the build results."""

    def __init__(self, success_only=False):
        HTMLParser.__init__(self)
        self.success_only = success_only
        self.parsing_invocations = False
        self.parsing_row = False
        self.stop_parsing = False
        self.result = BuildResult.ParserError

    def handle_starttag(self, tag, attrs):
        if self.stop_parsing: return
        if tag == 'table':
            for k, v in attrs:
                if k == 'id' and v == 'invocations-table':
                    self.parsing_invocations = True
        elif tag == 'tr' and self.parsing_invocations:
            for k, v in attrs:
                if k == 'class' and v == 'danger' and not self.success_only:
                    self.result = BuildResult.Fail
                    self.parsing_invocations = False
                    self.parsing_row = True
                if k == 'class' and v == 'success':
                    self.result = BuildResult.Pass
                    self.parsing_invocations = False
                    self.parsing_row = True
        elif tag == 'a' and self.parsing_row:
            for k, v in attrs:
                if k == 'href':
                    self.link = v
                if k == 'class' and 'label' in v:
                    self.stop_parsing = True


def getBuildResult(target, success_only=False):
    try:
        resp = urlfetch.fetch(BASE_URL + target, deadline=5)
        if resp.status_code != 200:
            return BuildResult.ServerError, '#'
        parser = LuciResultParser(success_only)
        parser.feed(resp.content)
        parser.close()
        return parser.result, parser.link
    except:
        return BuildResult.ServerError, '#'


class MiloResultParser(HTMLParser):
    """Parses the HTML of the Milo steps to get the snapshot link."""

    def __init__(self):
        HTMLParser.__init__(self)
        self.stop_parsing = False
        self.link = None

    def handle_starttag(self, tag, attrs):
        if self.stop_parsing: return
        if tag == 'a':
            for k, v in attrs:
                if k == 'href' and v.startswith(SNAPSHOT_URL):
                    self.link = v
                    self.stop_parsing= True


def getSnapshot(href):
    try:
        resp = urlfetch.fetch(href, deadline=5)
        if resp.status_code != 200:
            return BuildResult.ServerError
        parser = MiloResultParser()
        parser.feed(resp.content)
        parser.close()
        return parser.link
    except:
        return BuildResult.ServerError


class MainPage(webapp2.RequestHandler):
    """The main handler."""

    def get(self):
        template_values = {
            'clock': time.strftime("%H:%M UTC", time.gmtime()),
            'targets': [],
        }
        for t in TARGETS:
            build_jobs = []
            project = t[0]
            job_targets = t[1:]
            for job in job_targets:
                url_suffix = job[0]
                display_name = job[1]
                result, link = getBuildResult(url_suffix)
                result = {
                    'name': display_name,
                    'result': result,
                    'href': link,
                }
                build_jobs.append(result)
            target = {
                'project': project,
                'build_jobs': build_jobs
            }
            template_values['targets'].append(target)

        template = JINJA_ENVIRONMENT.get_template('index.html')
        self.response.write(template.render(template_values))


class SnapshotPage(webapp2.RequestHandler):
    """The snapshot handler."""

    def get(self, target):
        snapshot_found = False
        for t in TARGETS:
            job_targets = t[1:]
            for j in job_targets:
                if target == j[1]:
                    result, link = getBuildResult(j[0], True)
                    self.redirect(getSnapshot(link))
                    snapshot_found = True
        if not snapshot_found:
            self.abort(404)


app = webapp2.WSGIApplication([
        ('/', MainPage),
        (r'/lkgs/([a-z0-9-_]+)', SnapshotPage),
], debug=True)