locating runtime files automatically (fixes #96)

lib/src/dependency_graph.dart

 // Copyright (c) 2015, 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.
 
 /// Tracks the shape of the import/export graph and dependencies between files.
 library dev_compiler.src.dependency_graph;
 
 import 'dart:collection' show HashSet;
 
 import 'package:analyzer/analyzer.dart' show parseDirectives;
 import 'package:analyzer/src/generated/ast.dart'
     show
         AstNode,
         CompilationUnit,
         ExportDirective,
         Identifier,
         ImportDirective,
         LibraryDirective,
         PartDirective,
         PartOfDirective;
 import 'package:analyzer/src/generated/engine.dart'
     show ParseDartTask, AnalysisContext;
 import 'package:analyzer/src/generated/source.dart' show Source, SourceKind;
 import 'package:html5lib/dom.dart' show Document, Node;
 import 'package:html5lib/parser.dart' as html;
-import 'package:logging/logging.dart' show Level;
+import 'package:logging/logging.dart' show Logger, Level;
 import 'package:path/path.dart' as path;
 import 'package:source_span/source_span.dart' show SourceSpan;
 
 import 'info.dart';
 import 'options.dart';
 import 'report.dart';
 import 'utils.dart';
 
 /// Holds references to all source nodes in the import graph. This is mainly
 /// used as a level of indirection to ensure that each source has a canonical
 /// representation.
 class SourceGraph {
   /// All nodes in the source graph. Used to get a canonical representation for
   /// any node.
   final Map<Uri, SourceNode> nodes = {};
 
+  /// Resources included by default on any application.
+  final runtimeDeps = new Set<ResourceSourceNode>();
+
   /// Analyzer used to resolve source files.
   final AnalysisContext _context;
   final CheckerReporter _reporter;
   final CompilerOptions _options;
 
-  SourceGraph(this._context, this._reporter, this._options);
+  SourceGraph(this._context, this._reporter, this._options) {
+    var dir = _options.runtimeDir;
+    if (dir == null) {
+      _log.severe('Runtime dir could not be determined automatically, '
+          'please specify the --runtime-dir flag on the command line.');
+      return;
+    }
+    var prefix = path.absolute(dir);
+    var files =
+        _options.serverMode ? runtimeFilesForServerMode : defaultRuntimeFiles;
+    for (var file in files) {
+      runtimeDeps.add(nodeFromUri(path.toUri(path.join(prefix, file))));
+    }
+  }
 
   /// Node associated with a resolved [uri].
   SourceNode nodeFromUri(Uri uri) {
     var uriString = Uri.encodeFull('$uri');
     return nodes.putIfAbsent(uri, () {
       var source = _context.sourceFactory.forUri(uriString);
       var extension = path.extension(uriString);
       if (extension == '.html') {
         return new HtmlSourceNode(uri, source, this);
       } else if (extension == '.dart' || uriString.startsWith('dart:')) {
@@ skipping 58 matching lines @@
 
   String toString() {
     var simpleUri = uri.scheme == 'file' ? path.relative(uri.path) : "$uri";
     return '[$runtimeType: $simpleUri]';
   }
 }
 
 /// A node representing an entry HTML source file.
 class HtmlSourceNode extends SourceNode {
   /// Resources included by default on any application.
-  final runtimeDeps = new Set<ResourceSourceNode>();
+  final runtimeDeps;
 
   /// Libraries referred to via script tags.
   Set<DartSourceNode> scripts = new Set<DartSourceNode>();
 
   @override
   Iterable<SourceNode> get allDeps => [scripts, runtimeDeps].expand((e) => e);
 
   @override
   Iterable<SourceNode> get depsWithoutParts => allDeps;
 
   /// Parsed document, updated whenever [update] is invoked.
   Document document;
 
-  HtmlSourceNode(uri, source, graph) : super(uri, source) {
-    var prefix = 'package:dev_compiler/runtime';
-    var files = ['harmony_feature_check.js', 'dart_runtime.js', 'dart_core.js'];
-    if (graph._options.serverMode) {
-      files.addAll(const ['messages_widget.js', 'messages.css']);
-    }
-    files.forEach((file) {
-      runtimeDeps.add(graph.nodeFromUri(Uri.parse('$prefix/$file')));
-    });
-  }
+  HtmlSourceNode(Uri uri, Source source, SourceGraph graph)
+      : runtimeDeps = graph.runtimeDeps,
+        super(uri, source);
 
   void update(SourceGraph graph) {
     super.update(graph);
     if (needsRebuild) {
       graph._reporter.clearHtml(uri);
       document = html.parse(source.contents.data, generateSpans: true);
       var newScripts = new Set<DartSourceNode>();
       var tags = document.querySelectorAll('script[type="application/dart"]');
       for (var script in tags) {
         var src = script.attributes['src'];
@@ skipping 267 matching lines @@
   helper(start);
 }
 
 bool _same(Set a, Set b) => a.length == b.length && a.containsAll(b);
 
 /// An error message discovered while parsing the dependencies between files.
 class DependencyGraphError extends MessageWithSpan {
   const DependencyGraphError(String message, SourceSpan span)
       : super(message, Level.SEVERE, span);
 }
+
+/// Runtime files added to all applications when running the compiler in the
+/// command line.
+const defaultRuntimeFiles = const [
+  'harmony_feature_check.js',
+  'dart_runtime.js',
+  'dart_core.js'
+];
+
+/// Runtime files added to applications when running in server mode.
+final runtimeFilesForServerMode = new List<String>.from(defaultRuntimeFiles)
+  ..addAll(const ['messages_widget.js', 'messages.css']);
+
+final _log = new Logger('dev_compiler.dependency_graph');