PackageMapUriResolver for Dart server.

pkg/analysis_server/lib/src/analysis_server.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 analysis.server;
 
 import 'dart:async';
 import 'dart:collection';
 
 import 'package:analysis_server/src/analysis_logger.dart';
 import 'package:analysis_server/src/channel.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/context_directory_manager.dart';
 import 'package:analysis_server/src/domain_analysis.dart';
 import 'package:analysis_server/src/operation/operation_analysis.dart';
 import 'package:analysis_server/src/operation/operation.dart';
 import 'package:analysis_server/src/operation/operation_queue.dart';
 import 'package:analysis_server/src/protocol.dart';
 import 'package:analysis_server/src/resource.dart';
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/error.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/generated/sdk.dart';
 import 'package:analyzer/src/generated/sdk_io.dart';
 import 'package:analyzer/src/generated/source_io.dart';
 import 'package:analyzer/src/generated/java_engine.dart';
+import 'package:analysis_server/src/package_uri_resolver.dart';
 
 
 /**
  * An instance of [DirectoryBasedDartSdk] that is shared between
  * [AnalysisServer] instances to improve performance.
  */
 final DirectoryBasedDartSdk SHARED_SDK = DirectoryBasedDartSdk.defaultSdk;
 
 class AnalysisServerContextDirectoryManager extends ContextDirectoryManager {
   final AnalysisServer analysisServer;
 
   AnalysisServerContextDirectoryManager(this.analysisServer, ResourceProvider resourceProvider)
       : super(resourceProvider);
 
   @override
   void addContext(Folder folder, File pubspecFile) {
     ContextDirectory contextDirectory = new ContextDirectory(
-        analysisServer.defaultSdk, folder, pubspecFile);
+        analysisServer.defaultSdk, resourceProvider, folder, pubspecFile);
     analysisServer.folderMap[folder] = contextDirectory;
     analysisServer.schedulePerformAnalysisOperation(contextDirectory.context);
   }
 
   @override
   void applyChangesToContext(Folder contextFolder, ChangeSet changeSet) {
     AnalysisContext context = analysisServer.folderMap[contextFolder].context;
     context.applyChanges(changeSet);
     analysisServer.schedulePerformAnalysisOperation(context);
   }
@@ skipping 392 matching lines @@
  * Instances of [ContextDirectory] represents a [Folder] associated with an
  * analysis context.  The folder may or may not contain a Pub `pubspec.yaml`.
  *
  * TODO(scheglov) implement complete projects/contexts semantics.
  *
  * This class is intentionally simplified to serve as a base to start working
  * on services while work on complete semantics is being done in parallel.
  */
 class ContextDirectory {
   /**
+   * The root [ResourceProvider] of this [ContextDirectory].
+   */
+  final ResourceProvider _resourceProvider;
+
+  /**
    * The root [Folder] of this [ContextDirectory].
    */
   final Folder _folder;
 
   /**
    * The `pubspec.yaml` file in [_folder], or null if there isn't one.
    */
   File _pubspecFile;
 
   /**
    * The [AnalysisContext] of this [_folder].
    */
   AnalysisContext _context;
 
-  ContextDirectory(DartSdk sdk, this._folder, this._pubspecFile) {
+  ContextDirectory(DartSdk sdk, this._resourceProvider, this._folder,
+      this._pubspecFile) {
     // create AnalysisContext
     _context = AnalysisEngine.instance.createAnalysisContext();
     // TODO(scheglov) replace FileUriResolver with an Resource based resolver
     // TODO(scheglov) create packages resolver
-    _context.sourceFactory = new SourceFactory([
-      new DartUriResolver(sdk),
-      new FileUriResolver(),
-      // new PackageUriResolver(),
-    ]);
+    List<UriResolver> resolvers = <UriResolver>[new DartUriResolver(sdk),
+        new FileUriResolver(),// new PackageUriResolver(),

[Brian Wilkerson, 2014/06/17 21:38:23]

Remove the commented out code?

[scheglov, 2014/06/17 22:50:33]

Done.

+    ];
+    {
+      UriResolver packageResolver = _createPackageUriResolver();
+      if (packageResolver != null) {
+        resolvers.add(packageResolver);
+      }
+    }
+    _context.sourceFactory = new SourceFactory(resolvers);
   }
 
   /**
    * Return the [AnalysisContext] of this folder.
    */
   AnalysisContext get context => _context;
+
+  /**
+   * Returns an [UriResolver] for `package` URIs, or `null` if not a Pub
+   * application.
+   */
+  UriResolver _createPackageUriResolver() {
+    // prepare 'packages' folder
+    Resource packagesResource = _folder.getChild('packages');
+    if (packagesResource is! Folder) {
+      return null;
+    }
+    Folder packagesFolder = packagesResource;
+    // list packages
+    Map<String, Folder> packageMap = <String, Folder>{};
+    List<Resource> children = packagesFolder.getChildren();
+    for (Resource child in children) {
+      if (child is Folder) {
+        packageMap[child.shortName] = child;
+      }
+    }
+    // done
+    return new PackageMapUriResolver(_resourceProvider, packageMap);
+  }
 }
 
 
 /**
  * An enumeration of the services provided by the server domain.
  */
 class ServerService extends Enum2<ServerService> {
   static const ServerService STATUS = const ServerService('STATUS', 0);
 
   static const List<ServerService> VALUES = const [STATUS];
 
   const ServerService(String name, int ordinal) : super(name, ordinal);
 }