// Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'dart:async';
import 'logdog_rpc.dart';
import 'cache_new.dart';

/// Gets latest build numbers of completed runs from the last completed runs.
/// This works if one assumes that the bots we are interested in also completes
/// runs regularly.
Future<Map<String, int>> latestBuildNumbers(WithCacheFunction withCache) async {
  var logdog = new LogdogRpc();
  // This queries logdog for all logs that wrote recipe_result, which is the
  // last step of any recipe. The ** searches the last incoming commits that
  // fits the scheme.
  // TODO(mkroghj): Give project as an option to allow for FYI.
  var streams = await logdog.query(
      "chromium",
      "bb/client.dart/**/+/recipes/steps/recipe_result/0/logs/result/0",
      withCache);
  // All logs have the build-number in their path, so we just get it out.
  var builderAndNumberInPathRegExp = new RegExp(r"^.*\/.*\/(.*)\/(\d*)\/\+");
  var map = <String, int>{};
  streams.forEach((stream) {
    var match = builderAndNumberInPathRegExp.firstMatch(stream.path);
    map.putIfAbsent(match.group(1), () => int.parse(match.group(2)));
  });
  return map;
}

/// Gets latest build number for each builder in [builders].
Future<Map<String, int>> latestBuildNumbersForBuilders(
    Iterable<String> builders) {
  var logdog = new LogdogRpc();
  // This queries logdog for all logs that wrote recipe_result, which is the
  // last step of any recipe. The ** searches the last incoming commits that
  // fits the scheme.
  // TODO(mkroghj): Give project as an option to allow for FYI.
  return Future.wait(builders.map((String builder) {
    return logdog
        .query(
            "chromium",
            "bb/client.dart/${builder}/**/+/recipes/steps/recipe_result/0/logs/result/0",
            noCache()(),
            maxResults: 1)
        .then((logdogStream) {
      // All logs have the build-number in their path, so we just get it out.
      return _buildNumberFromLogdogPath(logdogStream.first.path);
    }).catchError((e) => 0);
  })).then((List<int> numbers) => new Map.fromIterables(builders, numbers));
}

Future<List<int>> latestBuildNumbersForBuilder(String name, int maxResults) {
  var logdog = new LogdogRpc();
  // This queries logdog for all logs that wrote recipe_result, which is the
  // last step of any recipe. The ** searches the last incoming commits that
  // fits the scheme.
  // TODO(mkroghj): Give project as an option to allow for FYI.
  return logdog
      .query(
          "chromium",
          "bb/client.dart/${name}/**/+/recipes/steps/recipe_result/0/logs/result/0",
          noCache()(),
          maxResults: maxResults)
      .then((List<LogdogStream> logdogStreams) =>
          logdogStreams.map((logdogStream) {
            return _buildNumberFromLogdogPath(logdogStream.path);
          }).toList());
}

int _buildNumberFromLogdogPath(String path) {
  return int.parse(path.split('/')[3]);
}