pkg/analyzer2dart/lib/src/driver.dart

// Copyright (c) 2014, 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.

library analyzer2dart.driver;

import 'package:analyzer/file_system/file_system.dart';
import 'package:analyzer/src/generated/element.dart';
import 'package:analyzer/src/generated/engine.dart';
import 'package:analyzer/src/generated/sdk.dart';
import 'package:analyzer/src/generated/source_io.dart';

import 'package:compiler/compiler.dart';

import 'closed_world.dart';
import 'tree_shaker.dart';

/**
 * Top level driver for Analyzer2Dart.
 */
class Driver {
  final ResourceProvider resourceProvider;
  final AnalysisContext context;
  final CompilerOutputProvider outputProvider;

  Driver(this.resourceProvider, DartSdk sdk, this.outputProvider)
      : context = AnalysisEngine.instance.createAnalysisContext() {
    // Set up the source factory.
    // TODO(paulberry): do we want to use ExplicitPackageUriResolver?
    List<UriResolver> uriResolvers = [
        new FileUriResolver(),
        new DartUriResolver(sdk) /* ,
        new PackageUriResolver(packagesDirectories) */
    ];
    context.sourceFactory = new SourceFactory(uriResolvers);
  }

  /**
   * Compute the closed world that is reachable from an entry point.
   */
  ClosedWorld computeWorld(FunctionElement entryPointElement) {
    InternalAnalysisContext analysisContext = context;
    TreeShaker treeShaker =
        new TreeShaker(analysisContext.typeProvider, entryPointElement);
    return treeShaker.shake();
  }

  /**
   * Given a source, resolve it and return its entry point.
   */
  FunctionElement resolveEntryPoint(Source source) {
    // Get the library element associated with the source.
    LibraryElement libraryElement = context.computeLibraryElement(source);

    // Get the resolved AST for main
    FunctionElement entryPointElement = libraryElement.entryPoint;
    if (entryPointElement == null) {
      throw new Exception('No main()!');
    }
    return entryPointElement;
  }

  /**
   * Add the given file as the root of analysis, and return the corresponding
   * source.
   */
  Source setRoot(String path) {
    File file = resourceProvider.getResource(path);
    Source source = file.createSource();
    // add the Source
    ChangeSet changeSet = new ChangeSet();
    changeSet.addedSource(source);
    context.applyChanges(changeSet);
    // return the Source
    return source;
  }
}