#!/usr/bin/env python

import codecs
import datetime
import json
import os
import subprocess
import sys
import traceback

import jenkins
import jinja2
from jinja2 import Environment, FileSystemLoader

JENKINS_SERVER = ('localhost', 8080)
LOGS_BUCKET = 'oss-fuzz-build-logs'

SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))


class Result(object):
  """Result."""

  def __init__(self, name, output):
    self.name = name
    self.output = output


def get_build_results(server):
  """Return successes, failures."""
  successes = []
  failures = []
  unstable = []

  for job in server.get_jobs(1):
    try:
      name = job['fullname']
      if not name.startswith('projects/'):
        continue

      print name
      project = name[len('projects/'):]

      info = server.get_job_info(name)
      last_build_number = info['lastCompletedBuild']['number']
      build_result = Result(
          project,
          server.get_build_console_output(name, last_build_number))

      last_failed_builder_number = None
      last_unstable_build_number = None
      if info['lastFailedBuild']:
        last_failed_builder_number = info['lastFailedBuild']['number']
      if info['lastUnstableBuild']:
        last_unstable_build_number = info['lastUnstableBuild']['number']

      if last_build_number == last_failed_builder_number:
        failures.append(build_result)
      elif last_build_number == last_unstable_build_number:
        unstable.append(build_result)
      else:
        successes.append(build_result)
    except Exception as _:
      traceback.print_exc()

  return successes, failures, unstable


def upload_status(successes, failures, unstable):
  """Upload main status page."""
  env = Environment(loader=FileSystemLoader(os.path.join(SCRIPT_DIR,
                                                         'templates')))
  failures = [f.name for f in failures]
  successes = [s.name for s in successes]
  unstable = [p.name for p in unstable]

  data = {
      'projects': failures + successes + unstable,
      'failures': failures,
      'successes': successes,
      'unstable': unstable,
      'last_updated': datetime.datetime.utcnow().ctime()
  }

  with open('status.html', 'w') as f:
    f.write(env.get_template('status_template.html').render(data))

  subprocess.check_output(['gsutil', 'cp', 'status.html', 'gs://' +
                           LOGS_BUCKET], stderr=subprocess.STDOUT)

  with open('status.json', 'w') as f:
    f.write(json.dumps(data))

  subprocess.check_output(['gsutil', 'cp', 'status.json', 'gs://' +
                           LOGS_BUCKET], stderr=subprocess.STDOUT)


def upload_build_logs(results):
  """Upload individual build logs."""
  for result in results:
    with codecs.open('latest.txt', 'w', encoding='utf-8') as f:
      f.write(result.output)

    subprocess.check_output(['gsutil', 'cp', 'latest.txt',
                             'gs://%s/build_logs/%s/' %
                             (LOGS_BUCKET, result.name)],
                             stderr=subprocess.STDOUT)


def main():
  jenkins_login = get_jenkins_login()
  server = jenkins.Jenkins('http://%s:%d' % JENKINS_SERVER,
                           username=jenkins_login[0], password=jenkins_login[1])

  successes, failures, unstable = get_build_results(server)
  upload_status(successes, failures, unstable)
  upload_build_logs(successes + failures + unstable)


def get_jenkins_login():
  """Returns (username, password) for jenkins."""
  username = os.getenv('JENKINS_USER')
  password = os.getenv('JENKINS_PASS')
  return username, password


if __name__ == '__main__':
  main()