Uncomment and test fixes for importing libraries.

pkg/analysis_services/lib/src/correction/fix.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.
 
 // This code was auto-generated, is not intended to be edited, and is subject to
 // significant change. Please see the README file for more information.
 
 library services.src.correction.fix;
 
+import 'dart:collection';
+
 import 'package:analysis_services/correction/change.dart';
 import 'package:analysis_services/correction/fix.dart';
 import 'package:analysis_services/search/hierarchy.dart';
 import 'package:analysis_services/search/search_engine.dart';
 import 'package:analysis_services/src/correction/levenshtein.dart';
 import 'package:analysis_services/src/correction/name_suggestion.dart';
 import 'package:analysis_services/src/correction/source_buffer.dart';
 import 'package:analysis_services/src/correction/source_range.dart' as rf;
 import 'package:analysis_services/src/correction/strings.dart';
 import 'package:analysis_services/src/correction/util.dart';
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/element.dart';
+import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/error.dart';
 import 'package:analyzer/src/generated/java_core.dart';
 import 'package:analyzer/src/generated/parser.dart';
 import 'package:analyzer/src/generated/scanner.dart';
+import 'package:analyzer/src/generated/sdk.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/generated/utilities_dart.dart';
+import 'package:path/path.dart';
 
 
 /**
  * A predicate is a one-argument function that returns a boolean value.
  */
 typedef bool Predicate<E>(E argument);
 
 
 /**
  * The computer for Dart fixes.
  */
 class FixProcessor {
   static const int MAX_LEVENSHTEIN_DISTANCE = 3;
 
   final SearchEngine searchEngine;
   final Source source;
   final String file;
   final CompilationUnit unit;
   final AnalysisError error;
   CompilationUnitElement unitElement;
   LibraryElement unitLibraryElement;
+  String unitLibraryFile;
+  String unitLibraryFolder;
 
   final List<Edit> edits = <Edit>[];
   final Map<String, LinkedPositionGroup> linkedPositionGroups = <String,
       LinkedPositionGroup>{};
   Position endPosition = null;
   final List<Fix> fixes = <Fix>[];
 
   CorrectionUtils utils;
   int errorOffset;
   int errorLength;
   int errorEnd;
   AstNode node;
   AstNode coveredNode;
 
 
   FixProcessor(this.searchEngine, this.source, this.file, this.unit, this.error)
       {
     unitElement = unit.element;
     unitLibraryElement = unitElement.library;
+    unitLibraryFile = unitLibraryElement.source.fullName;
+    unitLibraryFolder = dirname(unitLibraryFile);
   }
 
   DartType get coreTypeBool => _getCoreType("bool");
 
   /**
    * Returns the EOL to use for this [CompilationUnit].
    */
   String get eol => utils.endOfLine;
 
   List<Fix> compute() {
@@ skipping 646 matching lines @@
 //            new CreateFileCorrectionProposal(
 //                newFile,
 //                source,
 //                FixKind.CREATE_PART,
 //                [uriString]));
 //      }
 //    }
   }
 
   void _addFix_importLibrary(FixKind kind, String importPath) {
-    // TODO(scheglov) implement
-//    CompilationUnitElement libraryUnitElement =
-//        _unitLibraryElement.definingCompilationUnit;
-//    CompilationUnit libraryUnit = libraryUnitElement.node;
-//    // prepare new import location
-//    int offset = 0;
-//    String prefix;
-//    String suffix;
-//    {
-//      // if no directives
-//      prefix = "";
-//      suffix = eol;
-//      CorrectionUtils libraryUtils = new CorrectionUtils(libraryUnit);
-//      // after last directive in library
-//      for (Directive directive in libraryUnit.directives) {
-//        if (directive is LibraryDirective || directive is ImportDirective) {
-//          offset = directive.end;
-//          prefix = eol;
-//          suffix = "";
-//        }
-//      }
-//      // if still beginning of file, skip shebang and line comments
-//      if (offset == 0) {
-//        CorrectionUtils_InsertDesc desc = libraryUtils.insertDescTop;
-//        offset = desc.offset;
-//        prefix = desc.prefix;
-//        suffix = "${desc.suffix}${eol}";
-//      }
-//    }
-//    // insert new import
-//    String importSource = "${prefix}import '${importPath}';${suffix}";
-//    _addInsertEdit(offset, importSource);
-//    // add proposal
-//    _addUnitCorrectionProposal2(libraryUnitElement.source, kind, [importPath]);
+    CompilationUnitElement libraryUnitElement =
+        unitLibraryElement.definingCompilationUnit;
+    CompilationUnit libraryUnit = libraryUnitElement.node;
+    // prepare new import location
+    int offset = 0;
+    String prefix;
+    String suffix;
+    {
+      // if no directives
+      prefix = "";
+      suffix = eol;
+      CorrectionUtils libraryUtils = new CorrectionUtils(libraryUnit);
+      // after last directive in library
+      for (Directive directive in libraryUnit.directives) {
+        if (directive is LibraryDirective || directive is ImportDirective) {
+          offset = directive.end;
+          prefix = eol;
+          suffix = "";
+        }
+      }
+      // if still beginning of file, skip shebang and line comments
+      if (offset == 0) {
+        CorrectionUtils_InsertDesc desc = libraryUtils.insertDescTop;
+        offset = desc.offset;
+        prefix = desc.prefix;
+        suffix = "${desc.suffix}${eol}";
+      }
+    }
+    // insert new import
+    String importSource = "${prefix}import '${importPath}';${suffix}";
+    _addInsertEdit(offset, importSource);
+    // add proposal
+    _addFix(kind, [importPath], fixFile: libraryUnitElement.source.fullName);
   }
 
   void _addFix_importLibrary_withElement(String name, ElementKind kind) {
-    // TODO(scheglov) implement
-//    // ignore if private
-//    if (name.startsWith("_")) {
-//      return;
-//    }
-//    // may be there is an existing import, but it is with prefix and we don't use this prefix
-//    for (ImportElement imp in _unitLibraryElement.imports) {
-//      // prepare element
-//      LibraryElement libraryElement = imp.importedLibrary;
-//      Element element =
-//          CorrectionUtils.getExportedElement(libraryElement, name);
-//      if (element == null) {
-//        continue;
-//      }
-//      if (element is PropertyAccessorElement) {
-//        element = (element as PropertyAccessorElement).variable;
-//      }
-//      if (element.kind != kind) {
-//        continue;
-//      }
-//      // may be apply prefix
-//      PrefixElement prefix = imp.prefix;
-//      if (prefix != null) {
-//        SourceRange range = SourceRangeFactory.rangeStartLength(node, 0);
-//        _addReplaceEdit(range, "${prefix.displayName}.");
-//        _addFix(
-//            FixKind.IMPORT_LIBRARY_PREFIX,
-//            [libraryElement.displayName, prefix.displayName]);
-//        continue;
-//      }
-//      // may be update "show" directive
-//      List<NamespaceCombinator> combinators = imp.combinators;
-//      if (combinators.length == 1 && combinators[0] is ShowElementCombinator) {
-//        ShowElementCombinator showCombinator =
-//            combinators[0] as ShowElementCombinator;
-//        // prepare new set of names to show
-//        Set<String> showNames = new Set<String>();
-//        showNames.addAll(showCombinator.shownNames);
-//        showNames.add(name);
-//        // prepare library name - unit name or 'dart:name' for SDK library
-//        String libraryName = libraryElement.definingCompilationUnit.displayName;
-//        if (libraryElement.isInSdk) {
-//          libraryName = imp.uri;
-//        }
-//        // update library
-//        String newShowCode = "show ${StringUtils.join(showNames, ", ")}";
-//        // TODO(scheglov)
-//        _addReplaceEdit(
-//            SourceRangeFactory.rangeShowCombinator(showCombinator),
-//            newShowCode);
-//        _addUnitCorrectionProposal2(
-//            _unitLibraryElement.source,
-//            FixKind.IMPORT_LIBRARY_SHOW,
-//            [libraryName]);
-//        // we support only one import without prefix
-//        return;
-//      }
-//    }
-//    // check SDK libraries
-//    AnalysisContext context = _unitLibraryElement.context;
-//    {
-//      DartSdk sdk = context.sourceFactory.dartSdk;
-//      List<SdkLibrary> sdkLibraries = sdk.sdkLibraries;
-//      for (SdkLibrary sdkLibrary in sdkLibraries) {
-//        SourceFactory sdkSourceFactory = context.sourceFactory;
-//        String libraryUri = sdkLibrary.shortName;
-//        Source librarySource = sdkSourceFactory.resolveUri(null, libraryUri);
-//        // prepare LibraryElement
-//        LibraryElement libraryElement =
-//            context.getLibraryElement(librarySource);
-//        if (libraryElement == null) {
-//          continue;
-//        }
-//        // prepare exported Element
-//        Element element =
-//            CorrectionUtils.getExportedElement(libraryElement, name);
-//        if (element == null) {
-//          continue;
-//        }
-//        if (element is PropertyAccessorElement) {
-//          element = (element as PropertyAccessorElement).variable;
-//        }
-//        if (element.kind != kind) {
-//          continue;
-//        }
-//        // add import
-//        _addFix_importLibrary(FixKind.IMPORT_LIBRARY_SDK, libraryUri);
-//      }
-//    }
-//    // check project libraries
-//    {
-//      List<Source> librarySources = context.librarySources;
-//      for (Source librarySource in librarySources) {
-//        // we don't need SDK libraries here
-//        if (librarySource.isInSystemLibrary) {
-//          continue;
-//        }
-//        // prepare LibraryElement
-//        LibraryElement libraryElement =
-//            context.getLibraryElement(librarySource);
-//        if (libraryElement == null) {
-//          continue;
-//        }
-//        // prepare exported Element
-//        Element element =
-//            CorrectionUtils.getExportedElement(libraryElement, name);
-//        if (element == null) {
-//          continue;
-//        }
-//        if (element.kind != kind) {
-//          continue;
-//        }
-//        // prepare "library" file
-//        JavaFile libraryFile = getSourceFile(librarySource);
-//        if (libraryFile == null) {
-//          continue;
-//        }
-//        // may be "package:" URI
-//        {
-//          Uri libraryPackageUri = _findPackageUri(context, libraryFile);
-//          if (libraryPackageUri != null) {
-//            _addFix_importLibrary(
-//                FixKind.IMPORT_LIBRARY_PROJECT,
-//                libraryPackageUri.toString());
-//            continue;
-//          }
-//        }
-//        // relative URI
-//        String relative =
-//            URIUtils.computeRelativePath(
-//                _unitLibraryFolder.getAbsolutePath(),
-//                libraryFile.getAbsolutePath());
-//        _addFix_importLibrary(
-//            FixKind.IMPORT_LIBRARY_PROJECT,
-//            relative);
-//      }
-//    }
+    // ignore if private
+    if (name.startsWith("_")) {
+      return;
+    }
+
+    // may be there is an existing import,
+    // but it is with prefix and we don't use this prefix
+    for (ImportElement imp in unitLibraryElement.imports) {
+      // prepare element
+      LibraryElement libraryElement = imp.importedLibrary;
+      Element element = getExportedElement(libraryElement, name);
+      if (element == null) {
+        continue;
+      }
+      if (element is PropertyAccessorElement) {
+        element = (element as PropertyAccessorElement).variable;
+      }
+      if (element.kind != kind) {
+        continue;
+      }
+      // may be apply prefix
+      PrefixElement prefix = imp.prefix;
+      if (prefix != null) {
+        SourceRange range = rf.rangeStartLength(node, 0);
+        _addReplaceEdit(range, "${prefix.displayName}.");
+        _addFix(
+            FixKind.IMPORT_LIBRARY_PREFIX,
+            [libraryElement.displayName, prefix.displayName]);
+        continue;
+      }
+      // may be update "show" directive
+      List<NamespaceCombinator> combinators = imp.combinators;
+      if (combinators.length == 1 && combinators[0] is ShowElementCombinator) {
+        ShowElementCombinator showCombinator =
+            combinators[0] as ShowElementCombinator;
+        // prepare new set of names to show
+        Set<String> showNames = new SplayTreeSet<String>();
+        showNames.addAll(showCombinator.shownNames);
+        showNames.add(name);
+        // prepare library name - unit name or 'dart:name' for SDK library
+        String libraryName = libraryElement.definingCompilationUnit.displayName;
+        if (libraryElement.isInSdk) {
+          libraryName = imp.uri;
+        }
+        // update library
+        String newShowCode = "show ${StringUtils.join(showNames, ", ")}";
+        _addReplaceEdit(rf.rangeOffsetEnd(showCombinator), newShowCode);
+        _addFix(
+            FixKind.IMPORT_LIBRARY_SHOW,
+            [libraryName],
+            fixFile: unitLibraryFile);
+        // we support only one import without prefix
+        return;
+      }
+    }
+    // check SDK libraries
+    AnalysisContext context = unitLibraryElement.context;
+    {
+      DartSdk sdk = context.sourceFactory.dartSdk;
+      List<SdkLibrary> sdkLibraries = sdk.sdkLibraries;
+      for (SdkLibrary sdkLibrary in sdkLibraries) {
+        SourceFactory sdkSourceFactory = context.sourceFactory;
+        String libraryUri = sdkLibrary.shortName;
+        Source librarySource = sdkSourceFactory.resolveUri(null, libraryUri);
+        // prepare LibraryElement
+        LibraryElement libraryElement =
+            context.getLibraryElement(librarySource);
+        if (libraryElement == null) {
+          continue;
+        }
+        // prepare exported Element
+        Element element = getExportedElement(libraryElement, name);
+        if (element == null) {
+          continue;
+        }
+        if (element is PropertyAccessorElement) {
+          element = (element as PropertyAccessorElement).variable;
+        }
+        if (element.kind != kind) {
+          continue;
+        }
+        // add import
+        _addFix_importLibrary(FixKind.IMPORT_LIBRARY_SDK, libraryUri);
+      }
+    }
+    // check project libraries
+    {
+      List<Source> librarySources = context.librarySources;
+      for (Source librarySource in librarySources) {
+        // we don't need SDK libraries here
+        if (librarySource.isInSystemLibrary) {
+          continue;
+        }
+        // prepare LibraryElement
+        LibraryElement libraryElement =
+            context.getLibraryElement(librarySource);
+        if (libraryElement == null) {
+          continue;
+        }
+        // prepare exported Element
+        Element element = getExportedElement(libraryElement, name);
+        if (element == null) {
+          continue;
+        }
+        if (element is PropertyAccessorElement) {
+          element = (element as PropertyAccessorElement).variable;
+        }
+        if (element.kind != kind) {
+          continue;
+        }
+        // prepare "library" file
+        String libraryFile = librarySource.fullName;
+        // may be "package:" URI
+        // TODO(scheglov) this code does not have a test
+        {
+          String libraryPackageUri = _findPackageUri(context, libraryFile);
+          if (libraryPackageUri != null) {
+            _addFix_importLibrary(
+                FixKind.IMPORT_LIBRARY_PROJECT,
+                libraryPackageUri);
+            continue;
+          }
+        }
+        // relative URI
+        String relateFile = relative(libraryFile, from: unitLibraryFolder);
+        _addFix_importLibrary(FixKind.IMPORT_LIBRARY_PROJECT, relateFile);
+      }
+    }
   }
 
   void _addFix_importLibrary_withFunction() {
-    // TODO(scheglov) implement
-//    if (node is SimpleIdentifier && node.parent is MethodInvocation) {
-//      MethodInvocation invocation = node.parent as MethodInvocation;
-//      if (invocation.realTarget == null &&
-//          identical(invocation.methodName, node)) {
-//        String name = (node as SimpleIdentifier).name;
-//        _addFix_importLibrary_withElement(name, ElementKind.FUNCTION);
-//      }
-//    }
+    if (node is SimpleIdentifier && node.parent is MethodInvocation) {
+      MethodInvocation invocation = node.parent as MethodInvocation;
+      if (invocation.realTarget == null && invocation.methodName == node) {
+        String name = (node as SimpleIdentifier).name;
+        _addFix_importLibrary_withElement(name, ElementKind.FUNCTION);
+      }
+    }
   }
 
   void _addFix_importLibrary_withTopLevelVariable() {
-    // TODO(scheglov) implement
-//    if (node is SimpleIdentifier) {
-//      String name = (node as SimpleIdentifier).name;
-//      _addFix_importLibrary_withElement(name, ElementKind.TOP_LEVEL_VARIABLE);
-//    }
+    if (node is SimpleIdentifier) {
+      String name = (node as SimpleIdentifier).name;
+      _addFix_importLibrary_withElement(name, ElementKind.TOP_LEVEL_VARIABLE);
+    }
   }
 
   void _addFix_importLibrary_withType() {
-    // TODO(scheglov) implement
-//    if (_mayBeTypeIdentifier(node)) {
-//      String typeName = (node as SimpleIdentifier).name;
-//      _addFix_importLibrary_withElement(typeName, ElementKind.CLASS);
-//    }
+    if (_mayBeTypeIdentifier(node)) {
+      String typeName = (node as SimpleIdentifier).name;
+      _addFix_importLibrary_withElement(typeName, ElementKind.CLASS);
+    }
   }
 
   void _addFix_insertSemicolon() {
     if (error.message.contains("';'")) {
       int insertOffset = error.offset + error.length;
       _addInsertEdit(insertOffset, ";");
       _addFix(FixKind.INSERT_SEMICOLON, []);
     }
   }
 
@@ skipping 823 matching lines @@
         if (operatorType == TokenType.AMPERSAND_AMPERSAND ||
             operatorType == TokenType.BAR_BAR) {
           return coreTypeBool;
         }
       }
     }
     // we don't know
     return null;
   }
 
+  /**
+   * Inserts the given [SourceBuilder] at its offset.
+   */
+  void _insertBuilder(SourceBuilder builder) {
+    String text = builder.toString();
+    _addInsertEdit(builder.offset, text);
+    // add linked positions
+    builder.linkedPositionGroups.forEach((LinkedPositionGroup group) {
+      LinkedPositionGroup fixGroup = _getLinkedPosition(group.id);
+      group.positions.forEach((Position position) {
+        fixGroup.addPosition(position);
+      });
+      group.proposals.forEach((String proposal) {
+        fixGroup.addProposal(proposal);
+      });
+    });
+  }
+
 //  void _addLinkedPositionProposal(String group,
 //      LinkedPositionProposal proposal) {
 //    List<LinkedPositionProposal> nodeProposals = linkedPositionProposals[group];
 //    if (nodeProposals == null) {
 //      nodeProposals = <LinkedPositionProposal>[];
 //      linkedPositionProposals[group] = nodeProposals;
 //    }
 //    nodeProposals.add(proposal);
 //  }
 
 //  /**
 //   * Returns `true` if the given [ClassMember] is a part of a static method or
 //   * a field initializer.
 //   */
 //  bool _inStaticMemberContext2(ClassMember member) {
 //    if (member is MethodDeclaration) {
 //      return member.isStatic;
 //    }
 //    // field initializer cannot reference "this"
 //    if (member is FieldDeclaration) {
 //      return true;
 //    }
 //    return false;
 //  }
 
-  /**
-   * Inserts the given [SourceBuilder] at its offset.
-   */
-  void _insertBuilder(SourceBuilder builder) {
-    String text = builder.toString();
-    _addInsertEdit(builder.offset, text);
-    // add linked positions
-    builder.linkedPositionGroups.forEach((LinkedPositionGroup group) {
-      LinkedPositionGroup fixGroup = _getLinkedPosition(group.id);
-      group.positions.forEach((Position position) {
-        fixGroup.addPosition(position);
-      });
-      group.proposals.forEach((String proposal) {
-        fixGroup.addProposal(proposal);
-      });
-    });
-  }
-
   _ConstructorLocation
       _prepareNewConstructorLocation(ClassDeclaration classDeclaration) {
     List<ClassMember> members = classDeclaration.members;
     // find the last field/constructor
     ClassMember lastFieldOrConstructor = null;
     for (ClassMember member in members) {
       if (member is FieldDeclaration || member is ConstructorDeclaration) {
         lastFieldOrConstructor = member;
       } else {
         break;
@@ skipping 49 matching lines @@
       ClassElement element = type.element as ClassElement;
       sb.addProposal(element.name);
       _addSuperTypeProposals(sb, alreadyAdded, element.supertype);
       for (InterfaceType interfaceType in element.interfaces) {
         _addSuperTypeProposals(sb, alreadyAdded, interfaceType);
       }
     }
   }
 
   /**
+   * Attempts to convert the given absolute path into a "package" URI.
+   *
+   * [context] - the [AnalysisContext] to work in.
+   * [path] - the absolute path, not `null`.
+   *
+   * Returns the "package" URI, may be `null`.
+   */
+  static String _findPackageUri(AnalysisContext context, String path) {
+//    Source fileSource = new FileBasedSource.con1(path);
+    Source fileSource = new NonExistingSource(path, UriKind.FILE_URI);
+    Uri uri = context.sourceFactory.restoreUri(fileSource);
+    if (uri == null) {
+      return null;
+    }
+    return uri.toString();
+  }
+
+  /**
    * @return the suggestions for given [Type] and [DartExpression], not empty.
    */
   static List<String> _getArgumentNameSuggestions(Set<String> excluded,
       DartType type, Expression expression, int index) {
     List<String> suggestions =
         getVariableNameSuggestionsForExpression(type, expression, excluded);
     if (suggestions.length != 0) {
       return suggestions;
     }
     return <String>["arg${index}"];
@@ skipping 51 matching lines @@
 /**
  * Describes the location for a newly created [ConstructorDeclaration].
  */
 class _ConstructorLocation {
   final String _prefix;
   final int _offset;
   final String _suffix;
 
   _ConstructorLocation(this._prefix, this._offset, this._suffix);
 }