Start filling out analyzer2dart

pkg/analyzer2dart/bin/analyzer2dart.dart

Index: pkg/analyzer2dart/bin/analyzer2dart.dart
diff --git a/pkg/analyzer2dart/bin/analyzer2dart.dart b/pkg/analyzer2dart/bin/analyzer2dart.dart
index 52d167482da33495f7a33630f515da24e797f1cd..3f85ef7afd4b82cfb2ceab2a4a2eddaef0ec6d23 100644
--- a/pkg/analyzer2dart/bin/analyzer2dart.dart
+++ b/pkg/analyzer2dart/bin/analyzer2dart.dart
@@ -6,17 +6,120 @@
 library analyzer2dart.cmdline;
 
 import 'package:analyzer/analyzer.dart';
-import 'package:compiler/implementation/compiler.dart';
+import 'package:analyzer/src/generated/engine.dart';
+import 'package:analyzer/src/generated/sdk_io.dart';
+import 'package:analyzer/src/generated/source_io.dart';
+import 'package:analyzer/src/generated/java_io.dart';
+import 'package:analyzer/src/generated/element.dart';
+
+class TreeShakingVisitor extends RecursiveAstVisitor {
+  final TreeShaker treeShaker;
+
+  TreeShakingVisitor(this.treeShaker);
+
+  @override
+  void visitFunctionDeclaration(FunctionDeclaration node) {
+    print('Visiting function ${node.name.name}');
+    super.visitFunctionDeclaration(node);
+  }
+
+  @override
+  void visitMethodInvocation(MethodInvocation node) {
+    print('Visiting invocation of ${node.methodName.name}');
+    Element staticElement = node.methodName.staticElement;
+    if (staticElement != null) {
+      // TODO(paulberry): deal with the case where staticElement is
+      // not necessarily the exact target.  (Dart2js calls this a
+      // "dynamic invocation").  We need a notion of "selector".  Maybe
+      // we can use Dart2js selectors.
+      treeShaker.add(staticElement);
+    } else {
+      // TODO(paulberry): deal with this case.
+    }
+    super.visitMethodInvocation(node);
+  }
+
+}
+
+class CpsGeneratingVisitor extends RecursiveAstVisitor {
+  // TODO(johnniwinther)
+}
+
+class ClosedWorld {
+  // TODO(paulberry): is it a problem to hold on to all the AST's for the
+  // duration of tree shaking & CPS generation?
+  Map<Element, AstNode> elements = <Element, AstNode>{};
+  ClosedWorld();
+}
+
+class TreeShaker {
+  List<Element> _queue = <Element>[];
+  Set<Element> _alreadyEnqueued = new Set<Element>();

[scheglov, 2014/08/28 16:52:49]

new HashSet<Element>() ?

+  ClosedWorld _world = new ClosedWorld();
+
+  void add(Element e) {
+    if (!_alreadyEnqueued.contains(e)) {

[scheglov, 2014/08/28 16:52:49]

You could use if (!_alreadyEnqueued.add(e)) instead.

+      _queue.add(e);
+      _alreadyEnqueued.add(e);
+    }
+  }
+
+  ClosedWorld shake(AnalysisContext context) {
+    while (_queue.isNotEmpty) {
+      Element e = _queue.removeAt(0);

[scheglov, 2014/08/28 16:52:49]

List.removeLast() is much faster.
So, if the order is not important, we could use it instead.

+      print('Tree shaker handling $e');
+      CompilationUnit compilationUnit =
+          context.getResolvedCompilationUnit(e.source, e.library);
+      AstNode identifier =
+          new NodeLocator.con1(e.nameOffset).searchWithin(compilationUnit);
+      FunctionDeclaration declaration =
+          identifier.getAncestor((node) => node is FunctionDeclaration);
+      _world.elements[e] = declaration;
+      declaration.accept(new TreeShakingVisitor(this));
+    }
+    print('Tree shaking done');
+    return _world;
+  }
+}
 
 void main(List<String> args) {
-  // TODO(brianwilkerson,paulberry): Run the analyzer `args[0]` and provide
-  // access to the element model/ast of the `main` method.
+  // Create the analysis context
+  AnalysisContext context = AnalysisEngine.instance.createAnalysisContext();
+
+  // Set up the source factory.
+  // TODO(paulberry): do we want to use ExplicitPackageUriResolver?
+  List<UriResolver> uriResolvers = [
+      new FileUriResolver(),
+      new DartUriResolver(DirectoryBasedDartSdk.defaultSdk) /* ,
+      new PackageUriResolver(packagesDirectories) */
+  ];
+  context.sourceFactory = new SourceFactory(uriResolvers);
+
+  // Tell the analysis server about the root
+  JavaFile javaFile = new JavaFile(args[0]); // TODO(paulberry): hacky
+  Source source = new FileBasedSource.con1(javaFile);
+  ChangeSet changeSet = new ChangeSet();
+  changeSet.addedSources.add(source);
+  context.applyChanges(changeSet);
+
+  // 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()!');
+  }
 
   // TODO(brianwilkerson,paulberry,johnniwinther): Perform tree-growing by
   // visiting the ast and feeding the dependencies into a work queue (enqueuer).
+  TreeShaker treeShaker = new TreeShaker();
+  treeShaker.add(entryPointElement);
+  ClosedWorld world = treeShaker.shake(context);
 
   // TODO(brianwilkerson,paulberry,johnniwinther): Convert the ast into cps by
   // visiting the ast and invoking the ir builder.
+  new CpsGeneratingVisitor();
 
   // TODO(johnniwinther): Convert the analyzer element model into the dart2js
   // element model to fit the needs of the cps encoding above.