# Copyright 2018 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.

import json
import urllib
import urllib2


TEST_RESULTS_SERVER = 'http://test-results.appspot.com'
TOOL_USER_AGENT = 'flakiness_cli/0.1'


def _Request(path, params=None):
  """Request json data from the test results server."""
  url = TEST_RESULTS_SERVER + path
  if params:
    url += '?' + urllib.urlencode(params)
  request = urllib2.Request(url, headers={'User-Agent': TOOL_USER_AGENT})
  response = urllib2.urlopen(request)
  return json.load(response)


def GetBuilders():
  """Retrieve json with list of all known masters, builders, and test_types."""
  return _Request('/data/builders')


def GetTestResults(master, builder, test_type):
  """Retrieve test results json for a given master, builder, and test_type."""
  return _Request('/testfile', {
      'name': 'results.json',
      'master': master,
      'builder': builder,
      'testtype': test_type})