# Copyright (c) 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Status generators to have easy views of lkgr_finder's internals."""


import textwrap
import urllib

from infra.services.lkgr_finder.lkgr_lib import STATUS


class StatusGeneratorStub(object):  # pragma: no cover

  def master_cb(self, master):
    pass

  def builder_cb(self, builder):
    pass

  def revision_cb(self, revision):
    pass

  def build_cb(self, master, builder, status, build_num=None):
    pass

  def lkgr_cb(self, revision):
    pass


class DebugStatusGenerator(StatusGeneratorStub):  # pragma: no cover

  def master_cb(self, master):
    print master

  def builder_cb(self, builder):
    print '  %s' % builder

  def revision_cb(self, revision):
    print '    %s' % revision

  def build_cb(self, master, builder, status, build_num=None):
    print '%s : %s : %s : %s' % (master, builder, build_num, status)

  def lkgr_cb(self, revision):
    print 'LKGR %s' % revision


class HTMLStatusGenerator(StatusGeneratorStub):  # pragma: no cover

  def __init__(self, viewvc, config):
    self.masters = []
    self.rows = []
    self.config = config
    self.viewvc = viewvc

  def master_cb(self, master):
    self.masters.append((master, []))

  def builder_cb(self, builder):
    self.masters[-1][1].append(builder)

  def revision_cb(self, revision):
    row = [
        revision,
        '<td class="revision"><a href="%s" target="_blank">%s</a></td>\n' % (
            self.viewvc % urllib.quote(revision), revision)]
    self.rows.append(row)

  def build_cb(self, master, builder, status, build_num=None):
    stat_txt = STATUS.tostr(status)
    cell = '  <td class="%s">' % stat_txt
    if build_num is not None:
      master_config = self.config['masters'][master]
      if 'luci_project' in master_config and 'luci_bucket' in master_config:
        build_url = 'ci.chromium.org/p/%s/builders/%s/%s/%s' % (
            master_config['luci_project'], master_config['luci_bucket'],
            builder, build_num)
      else:
        build_url = 'build.chromium.org/p/%s/builders/%s/builds/%s' % (
            master, builder, build_num)
      cell += '<a href="http://%s" target="_blank">X</a>' % (
          urllib.quote(build_url))
    cell += '</td>\n'
    self.rows[-1].append(cell)

  def lkgr_cb(self, revision):
    row = self.rows[-1]
    row[1] = row[1].replace('class="revision"', 'class="lkgr"', 1)
    for i in range(2, len(row)):
      row[i] = row[i].replace('class="success"', 'class="lkgr"', 1)

  def generate(self):
    html_chunks = [textwrap.dedent("""
        <html>
        <head>
        <style type="text/css">
        table { border-collapse: collapse; }
        th { font-size: xx-small; }
        td, th { text-align: center; }
        .header { border: 1px solid black; }
        .revision { padding-left: 5px; padding-right: 5px; }
        .revision { border-left: 1px solid black; border-right: 1px solid black; }
        .success { background-color: #8d4; }
        .failure { background-color: #e88; }
        .running { background-color: #fe1; }
        .unknown { background-color: #ddd; }
        .lkgr { background-color: #4af; }
        .roll { border-top: 2px solid black; }
        </style>
        </head>
        <body><table>
        """)]
    master_headers = ['<tr class="header"><th></th>\n']
    builder_headers = ['<tr class="header">']
    builder_headers.append('<th>chromium revision</th>\n')
    for master, builders in self.masters:
      master_config = self.config['masters'][master]
      if ('luci_project' in master_config and
          'luci_builder_ui_group' in master_config):
        master_url = 'ci.chromium.org/p/%s/g/%s' % (
            master_config['luci_project'],
            master_config['luci_builder_ui_group'])
      else:
        master_url = 'build.chromium.org/p/%s' % master
      hdr = '  <th colspan="%d" class="header">' % len(builders)
      hdr += '<a href="%s" target="_blank">%s</a></th>\n' % (
          'http://%s' % urllib.quote(master_url), master)
      master_headers.append(hdr)
      for builder in builders:
        if 'luci_project' in master_config and 'luci_bucket' in master_config:
          builder_url = 'ci.chromium.org/p/%s/builders/%s/%s' % (
              master_config['luci_project'], master_config['luci_bucket'],
              builder)
        else:
          builder_url = 'build.chromium.org/p/%s/builders/%s' % (
              master, builder)
        hdr = '  <th><a href="%s" target="_blank">%s</a></th>\n' % (
            'http://%s' % urllib.quote(builder_url), builder)
        builder_headers.append(hdr)
    master_headers.append('</tr>\n')
    builder_headers.append('</tr>\n')
    html_chunks.extend(master_headers)
    html_chunks.extend(builder_headers)
    for row in self.rows:
      html_chunks.extend(row[1:])
      html_chunks.append('</tr>\n')
    html_chunks.append('</table></body></html>\n')
    return ''.join(html_chunks)