pkg/analyzer/tool/summary/build_sdk_summaries.dart

import 'dart:io';

import 'package:analyzer/dart/element/element.dart';
import 'package:analyzer/src/generated/engine.dart';
import 'package:analyzer/src/generated/java_io.dart';
import 'package:analyzer/src/generated/sdk.dart';
import 'package:analyzer/src/generated/sdk_io.dart';
import 'package:analyzer/src/generated/source.dart';
import 'package:analyzer/src/summary/format.dart';
import 'package:analyzer/src/summary/summarize_elements.dart';
import 'package:path/path.dart';

main(List<String> args) {
  if (args.length < 1 || args.length > 2) {
    _printUsage();
    exitCode = 1;
    return;
  }
  //
  // Prepare output file path.
  //
  String outputDirectoryPath = args[0];
  if (!FileSystemEntity.isDirectorySync(outputDirectoryPath)) {
    print("'$outputDirectoryPath' is not a directory.");
    _printUsage();
    exitCode = 1;
    return;
  }
  //
  // Prepare SDK path.
  //
  String sdkPath;
  if (args.length == 2) {
    sdkPath = args[1];
    if (!FileSystemEntity.isDirectorySync('$sdkPath/lib')) {
      print("'$sdkPath/lib' does not exist.");
      _printUsage();
      exitCode = 1;
      return;
    }
  } else {
    sdkPath = DirectoryBasedDartSdk.defaultSdkDirectory.getAbsolutePath();
  }
  //
  // Build spec and strong summaries.
  //
  _buildSummary(sdkPath, outputDirectoryPath, false);
  _buildSummary(sdkPath, outputDirectoryPath, true);
}

/**
 * The name of the SDK summaries builder application.
 */
const BINARY_NAME = "build_sdk_summaries";

/**
 * Build a strong or spec mode summary for the Dart SDK at [sdkPath].
 */
void _buildSummary(
    String sdkPath, String outputDirectoryPath, bool strongMode) {
  print('Generating ${strongMode ? 'strong' : 'spec'} mode summary.');
  Stopwatch sw = new Stopwatch()..start();
  //
  // Prepare SDK.
  //
  DirectoryBasedDartSdk sdk = new DirectoryBasedDartSdk(new JavaFile(sdkPath));
  AnalysisContext context = sdk.context;
  context.analysisOptions = new AnalysisOptionsImpl()..strongMode = strongMode;
  //
  // Serialize each SDK library.
  //
  List<String> linkedLibraryUris = <String>[];
  List<LinkedLibraryBuilder> linkedLibraries = <LinkedLibraryBuilder>[];
  List<String> unlinkedUnitUris = <String>[];
  List<UnlinkedUnitBuilder> unlinkedUnits = <UnlinkedUnitBuilder>[];
  for (SdkLibrary lib in sdk.sdkLibraries) {
    Source librarySource = sdk.mapDartUri(lib.shortName);
    LibraryElement libraryElement =
        context.computeLibraryElement(librarySource);
    LibrarySerializationResult libraryResult =
        serializeLibrary(libraryElement, context.typeProvider, strongMode);
    linkedLibraryUris.add(lib.shortName);
    linkedLibraries.add(libraryResult.linked);
    unlinkedUnitUris.addAll(libraryResult.unitUris);
    unlinkedUnits.addAll(libraryResult.unlinkedUnits);
  }
  //
  // Write the whole SDK bundle.
  //
  SdkBundleBuilder sdkBundle = new SdkBundleBuilder(
      linkedLibraryUris: linkedLibraryUris,
      linkedLibraries: linkedLibraries,
      unlinkedUnitUris: unlinkedUnitUris,
      unlinkedUnits: unlinkedUnits);
  String outputFilePath =
      join(outputDirectoryPath, strongMode ? 'strong.sum' : 'spec.sum');
  File file = new File(outputFilePath);
  file.writeAsBytesSync(sdkBundle.toBuffer(), mode: FileMode.WRITE_ONLY);
  //
  // Done.
  //
  print('\tDone in ${sw.elapsedMilliseconds} ms.');
}

/**
 * Print information about how to use the SDK summaries builder.
 */
void _printUsage() {
  print('Usage: $BINARY_NAME output_directory_path [sdk_path]');
  print('Build files spec.sum and strong.sum in the output directory.');
}