New assist to add/edit hide at import for ambiguous import

pkg/analysis_server/lib/src/services/correction/dart/ambiguous_import_fix.dart

@@ -0,0 +1,336 @@
+// Copyright (c) 2025, 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.
+
+import 'package:analysis_server/src/services/correction/fix.dart';
+import 'package:analysis_server_plugin/edit/dart/correction_producer.dart';
+import 'package:analyzer/dart/analysis/results.dart';
+import 'package:analyzer/dart/ast/ast.dart';
+import 'package:analyzer/dart/element/element2.dart';
+import 'package:analyzer/source/source_range.dart';
+import 'package:analyzer/src/dart/ast/extensions.dart';
+import 'package:analyzer/src/utilities/extensions/results.dart';
+import 'package:analyzer_plugin/utilities/change_builder/change_builder_core.dart';
+import 'package:analyzer_plugin/utilities/fixes/fixes.dart';
+import 'package:analyzer_plugin/utilities/range_factory.dart';
+import 'package:collection/collection.dart';
+
+class AmbiguousImportFix extends MultiCorrectionProducer {
+  AmbiguousImportFix({required super.context});
+
+  @override
+  Future<List<ResolvedCorrectionProducer>> get producers async {
+    var node = this.node;
+    Element2? element;
+    String? prefix;
+    if (node is NamedType) {
+      element = node.element2;
+      prefix = node.importPrefix?.name.lexeme;
+    } else if (node is SimpleIdentifier) {
+      element = node.element;
+      if (node.parent case PrefixedIdentifier(prefix: var currentPrefix)) {
+        prefix = currentPrefix.name;
+      }
+    }
+    if (element is! MultiplyDefinedElement2) {
+      return const [];
+    }
+    var conflictingElements = element.conflictingElements2;
+    var name = element.name3;
+    if (name == null || name.isEmpty) {
+      return const [];
+    }
+
+    var (unit, importDirectives, uris) = _getImportDirectives(
+      libraryResult,
+      unitResult,
+      conflictingElements,
+      name,
+      prefix,
+    );
+
+    // If we have multiple imports of the same library, then we won't fix it.
+    if (uris.length != uris.toSet().length) {
+      return const [];
+    }
+
+    if (unit == null || importDirectives.isEmpty || uris.isEmpty) {
+      return const [];
+    }
+
+    var producers = <ResolvedCorrectionProducer>[];
+    var thisContext = CorrectionProducerContext.createResolved(
+      libraryResult: libraryResult,
+      unitResult: unit,
+      applyingBulkFixes: applyingBulkFixes,
+      dartFixContext: context.dartFixContext,
+    );
+
+    for (var uri in uris) {
+      var directives =
+          importDirectives
+              .whereNot((directive) => directive.uri.stringValue == uri)
+              .toList();
+      producers.add(
+        _ImportAddHide(name, uri, prefix, directives, context: thisContext),
+      );
+      producers.add(
+        _ImportRemoveShow(name, uri, prefix, directives, context: thisContext),
+      );
+    }
+    return producers;
+  }
+
+  /// Returns [ImportDirective]s that import the given [conflictingElements]
+  /// into [unitResult] and the set of uris (String) that represent each of the
+  /// import directives.
+  ///
+  /// The uris and the import directives are both returned so that we can
+  /// run the fix for a certain uri on all of the other import directives.
+  ///
+  /// The resulting [ResolvedUnitResult?] is the unit that contains the import
+  /// directives. Usually this is the unit that contains the conflicting
+  /// element, but it could be a parent unit if the conflicting element is
+  /// a part file and the relevant imports are in an upstream file in the
+  /// part hierarchy (enhanced parts).
+  (ResolvedUnitResult?, List<ImportDirective>, List<String>)
+  _getImportDirectives(
+    ResolvedLibraryResult libraryResult,
+    ResolvedUnitResult? unitResult,
+    List<Element2> conflictingElements,
+    String name,
+    String? prefix,
+  ) {
+    // The uris of all import directives that import the conflicting elements.
+    var uris = <String>[];
+    // The import directives that import the conflicting elements.
+    var importDirectives = <ImportDirective>[];
+
+    // Search in each unit up the chain for related imports.
+    while (unitResult is ResolvedUnitResult) {
+      for (var conflictingElement in conflictingElements) {
+        // Find all ImportDirective that import this library in this unit
+        // and have the same prefix.
+        for (var directive
+            in unitResult.unit.directives.whereType<ImportDirective>()) {
+          var libraryImport = directive.libraryImport;
+          if (libraryImport == null) {
+            continue;
+          }
+
+          // If the prefix is different, then this directive is not relevant.
+          if (directive.prefix?.name != prefix) {
+            continue;
+          }
+
+          // If this library is imported directly or if the directive exports
+          // the library for this element.
+          var element =
+              prefix != null
+                  ? libraryImport.namespace.getPrefixed2(prefix, name)
+                  : libraryImport.namespace.get2(name);
+          if (element == conflictingElement) {
+            var uri = directive.uri.stringValue;
+            if (uri != null) {
+              uris.add(uri);
+              importDirectives.add(directive);
+            }
+          }
+        }
+      }
+
+      if (importDirectives.isNotEmpty) {
+        break;
+      }
+
+      // We continue up the chain.
+      unitResult = libraryResult.parentUnitOf(unitResult);
+    }
+
+    return (unitResult, importDirectives, uris);
+  }
+}
+
+class _ImportAddHide extends ResolvedCorrectionProducer {
+  final List<ImportDirective> importDirectives;
+  final String uri;
+  final String? prefix;
+  final String _elementName;
+
+  _ImportAddHide(
+    this._elementName,
+    this.uri,
+    this.prefix,
+    this.importDirectives, {
+    required super.context,
+  });
+
+  @override
+  CorrectionApplicability get applicability =>
+          // TODO(applicability): comment on why.
+          CorrectionApplicability
+          .singleLocation;
+
+  @override
+  List<String> get fixArguments {
+    var prefix = '';
+    if (!this.prefix.isEmptyOrNull) {
+      prefix = ' as ${this.prefix}';
+    }
+    return [_elementName, uri, prefix];
+  }
+
+  @override
+  FixKind get fixKind => DartFixKind.IMPORT_LIBRARY_HIDE;
+
+  @override
+  Future<void> compute(ChangeBuilder builder) async {
+    if (_elementName.isEmpty || uri.isEmpty) {
+      return;
+    }
+
+    var hideCombinators =
+        <({ImportDirective directive, List<HideCombinator> hideList})>[];
+
+    for (var directive in importDirectives) {
+      var show = directive.combinators.whereType<ShowCombinator>().firstOrNull;
+      // If there is an import with a show combinator, then we don't want to
+      // deal with this case here.
+      if (show != null) {
+        return;
+      }
+      var hide = directive.combinators.whereType<HideCombinator>().toList();
+      hideCombinators.add((directive: directive, hideList: hide));
+    }
+
+    await builder.addDartFileEdit(file, (builder) {
+      for (var (:directive, :hideList) in hideCombinators) {
+        for (var hide in hideList) {
+          var allNames = [
+            ...hide.hiddenNames.map((name) => name.name),
+            _elementName,
+          ];
+          if (_sortCombinators) {
+            allNames.sort();
+          }
+          var combinator = 'hide ${allNames.join(', ')}';
+          builder.addSimpleReplacement(range.node(hide), combinator);
+        }
+        if (hideList.isEmpty) {
+          var hideCombinator = ' hide $_elementName';
+          builder.addSimpleInsertion(directive.end - 1, hideCombinator);
+        }
+      }
+    });
+  }
+}
+
+class _ImportRemoveShow extends ResolvedCorrectionProducer {
+  final List<ImportDirective> importDirectives;
+  final String _elementName;
+  final String uri;
+  final String? prefix;
+
+  _ImportRemoveShow(
+    this._elementName,
+    this.uri,
+    this.prefix,
+    this.importDirectives, {
+    required super.context,
+  });
+
+  @override
+  CorrectionApplicability get applicability =>
+          // TODO(applicability): comment on why.
+          CorrectionApplicability
+          .singleLocation;
+
+  @override
+  List<String> get fixArguments {
+    var prefix = '';
+    if (!this.prefix.isEmptyOrNull) {
+      prefix = ' as ${this.prefix}';
+    }
+    return [_elementName, uri, prefix];
+  }
+
+  @override
+  FixKind get fixKind => DartFixKind.IMPORT_LIBRARY_REMOVE_SHOW;
+
+  @override
+  Future<void> compute(ChangeBuilder builder) async {
+    if (_elementName.isEmpty || uri.isEmpty) {
+      return;
+    }
+
+    var showCombinators =
+        <
+          ({
+            ImportDirective directive,
+            List<ShowCombinator> showList,
+            List<HideCombinator> hideList,
+          })
+        >[];
+
+    for (var directive in importDirectives) {
+      var show = directive.combinators.whereType<ShowCombinator>().toList();
+      var hide = directive.combinators.whereType<HideCombinator>().toList();
+      // If there is no show combinator, then we don't want to deal with this
+      // case here.
+      if (show.isEmpty) {
+        return;
+      }
+      showCombinators.add((
+        directive: directive,
+        showList: show,
+        hideList: hide,
+      ));
+    }
+
+    await builder.addDartFileEdit(file, (builder) {
+      for (var (:directive, :showList, :hideList) in showCombinators) {
+        var noShow = true;
+        for (var show in showList) {
+          var allNames = [
+            ...show.shownNames
+                .map((name) => name.name)
+                .where((name) => name != _elementName),
+          ];
+          if (_sortCombinators) {
+            allNames.sort();
+          }
+          if (allNames.isEmpty) {
+            builder.addDeletion(SourceRange(show.offset - 1, show.length + 1));
+          } else {
+            noShow = false;
+            var combinator = 'show ${allNames.join(', ')}';
+            var range = SourceRange(show.offset, show.length);
+            builder.addSimpleReplacement(range, combinator);
+          }
+        }
+        if (noShow) {
+          if (hideList.isEmpty) {
+            var hideCombinator = ' hide $_elementName';
+            builder.addSimpleInsertion(directive.end - 1, hideCombinator);
+          }
+          for (var hide in hideList) {
+            var allNames = [
+              ...hide.hiddenNames.map((name) => name.name),
+              _elementName,
+            ];
+            if (_sortCombinators) {
+              allNames.sort();
+            }
+            var combinator = 'hide ${allNames.join(', ')}';
+            builder.addSimpleReplacement(range.node(hide), combinator);
+          }
+        }
+      }
+    });
+  }
+}
+
+extension on ResolvedCorrectionProducer {
+  bool get _sortCombinators =>
+      getCodeStyleOptions(unitResult.file).sortCombinators;
+}

pkg/analysis_server/lib/src/services/correction/error_fix_status.yaml

@@ -156,12 +156,11 @@ CompileTimeErrorCode.AMBIGUOUS_EXTENSION_MEMBER_ACCESS_THREE_OR_MORE:
 CompileTimeErrorCode.AMBIGUOUS_EXTENSION_MEMBER_ACCESS_TWO:
   status: hasFix
 CompileTimeErrorCode.AMBIGUOUS_IMPORT:
-  status: needsFix
+  status: hasFix
   notes: |-
-    1. For each imported name, add a fix to hide the name.
-    2. For each imported name, add a fix to add a prefix. We wouldn't be able to
-       add the prefix everywhere, but could add it wherever the name was already
-       unambiguous.
+    For each imported name, add a fix to add a prefix. We wouldn't be able to
+    add the prefix everywhere, but could add it wherever the name was already
+    unambiguous.
 CompileTimeErrorCode.AMBIGUOUS_SET_OR_MAP_LITERAL_BOTH:
   status: noFix
   notes: |-

pkg/analysis_server/lib/src/services/correction/fix.dart

@@ -849,6 +849,11 @@ abstract final class DartFixKind {
     DartFixKindPriority.standard + 5,
     "Update library '{0}' import",
   );
+  static const IMPORT_LIBRARY_HIDE = FixKind(
+    'dart.fix.import.libraryHide',
+    DartFixKindPriority.standard,
+    "Hide others to use '{0}' from '{1}'{2}",
+  );
   static const IMPORT_LIBRARY_PREFIX = FixKind(
     'dart.fix.import.libraryPrefix',
     DartFixKindPriority.standard + 5,
@@ -914,6 +919,11 @@ abstract final class DartFixKind {
     DartFixKindPriority.standard + 1,
     "Import library '{0}' with 'show'",
   );
+  static const IMPORT_LIBRARY_REMOVE_SHOW = FixKind(
+    'dart.fix.import.libraryRemoveShow',
+    DartFixKindPriority.standard - 1,
+    "Remove show to use '{0}' from '{1}'{2}",
+  );
   static const IMPORT_LIBRARY_SDK = FixKind(
     'dart.fix.import.librarySdk',
     DartFixKindPriority.standard + 4,
@@ -924,16 +934,16 @@ abstract final class DartFixKind {
     DartFixKindPriority.standard + 4,
     "Import library '{0}' with prefix '{1}'",
   );
-  static const IMPORT_LIBRARY_SDK_PREFIXED_SHOW = FixKind(
-    'dart.fix.import.librarySdkPrefixedShow',
-    DartFixKindPriority.standard + 4,
-    "Import library '{0}' with prefix '{1}' and 'show'",
-  );
   static const IMPORT_LIBRARY_SDK_SHOW = FixKind(
     'dart.fix.import.librarySdkShow',
     DartFixKindPriority.standard + 4,
     "Import library '{0}' with 'show'",
   );
+  static const IMPORT_LIBRARY_SDK_PREFIXED_SHOW = FixKind(
+    'dart.fix.import.librarySdkPrefixedShow',
+    DartFixKindPriority.standard + 4,
+    "Import library '{0}' with prefix '{1}' and 'show'",
+  );
   static const INLINE_INVOCATION = FixKind(
     'dart.fix.inlineInvocation',
     DartFixKindPriority.standard - 20,