// 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.

/// Helper program that tests the equivalence between old and new inferrer data
/// on a dart program or directory of dart programs.

import 'dart:async';
import 'dart:io';
import 'package:args/args.dart';
import 'package:compiler/src/filenames.dart';
import 'package:compiler/src/inferrer/inferrer_engine.dart';
import 'package:compiler/src/resolution/class_hierarchy.dart';
import '../equivalence/id_equivalence_helper.dart';
import 'inference_test_helper.dart';
import '../equivalence/id_equivalence.dart';

main(List<String> args) {
  mainInternal(args);
}

Future<bool> mainInternal(List<String> args,
    {bool whiteList(Uri uri, Id id)}) async {
  ArgParser argParser = new ArgParser(allowTrailingOptions: true);
  argParser.addFlag('verbose', negatable: true, defaultsTo: false);
  argParser.addFlag('colors', negatable: true);
  ArgResults argResults = argParser.parse(args);
  if (argResults.options.contains('colors')) {
    useColors = true;
  }
  bool verbose = argResults['verbose'];

  useOptimizedMixins = true;
  InferrerEngineImpl.useSorterForTesting = true;

  bool success = true;
  for (String arg in argResults.rest) {
    Uri uri = Uri.base.resolve(nativeToUriPath(arg));
    List<Uri> uris = <Uri>[];
    if (FileSystemEntity.isDirectorySync(arg)) {
      for (FileSystemEntity file in new Directory.fromUri(uri).listSync()) {
        if (file is File && file.path.endsWith('.dart')) {
          uris.add(file.uri);
        }
      }
    } else {
      uris.add(uri);
    }
    for (Uri uri in uris) {
      StringBuffer sb = new StringBuffer();
      ZoneSpecification specification = new ZoneSpecification(
          print: (self, parent, zone, line) => sb.writeln(line));

      try {
        print('--$uri------------------------------------------------------');
        bool isSuccess = await runZoned(() {
          return testUri(uri, verbose: verbose, whiteList: whiteList);
        }, zoneSpecification: specification);
        if (!isSuccess) {
          success = false;
          print('  skipped due to compile-time errors');
        }
      } catch (e, s) {
        success = false;
        print(sb);
        print('Failed: $e\n$s');
      }
    }
  }

  return success;
}

Future<bool> testUri(Uri uri,
    {bool verbose: false, bool whiteList(Uri uri, Id id)}) {
  return compareData(
      uri, const {}, computeMemberAstTypeMasks, computeMemberIrTypeMasks,
      options: [stopAfterTypeInference],
      forUserLibrariesOnly: false,
      skipUnprocessedMembers: true,
      skipFailedCompilations: true,
      verbose: verbose,
      whiteList: whiteList);
}