pkg/analysis_server_plugin/lib/edit/fix/dart_fix_context.dart

// Copyright (c) 2024, 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_plugin/edit/fix/fix_context.dart';
import 'package:analysis_server_plugin/src/correction/change_workspace.dart';
import 'package:analyzer/dart/analysis/results.dart';
import 'package:analyzer/dart/element/element.dart';
import 'package:analyzer/dart/element/element2.dart';
import 'package:analyzer/error/error.dart';
import 'package:analyzer/instrumentation/service.dart';
import 'package:analyzer/src/dart/analysis/driver_based_analysis_context.dart';
import 'package:analyzer/src/dart/analysis/file_state_filter.dart';
import 'package:analyzer/src/services/top_level_declarations.dart';

/// An object used to provide context information for Dart fix contributors.
///
/// Clients may not extend, implement or mix-in this class.
class DartFixContext implements FixContext {
  /// Whether fixes were triggered automatically (for example by a save
  /// operation).
  ///
  /// Some fixes may be excluded when running automatically. For example
  /// removing unused imports or parameters is less acceptable while the code is
  /// incomplete and being worked on than when manually executing fixes ready
  /// for committing.
  final bool autoTriggered;

  /// The instrumentation service used to report errors that prevent a fix from
  /// being composed.
  final InstrumentationService instrumentationService;

  /// The resolution result in which the fix operates.
  final ResolvedUnitResult resolvedResult;

  /// The workspace in which the fix contributor operates.
  final ChangeWorkspace workspace;

  @override
  final AnalysisError error;

  DartFixContext({
    required this.instrumentationService,
    required this.workspace,
    required this.resolvedResult,
    required this.error,
    this.autoTriggered = false,
  });

  /// Returns the mapping from each library (that is available to this context)
  /// to a top-level declaration that is exported (not necessary declared) by
  /// this library, and has the requested base name.
  ///
  /// For getters and setters the corresponding top-level variable is returned.
  Future<Map<LibraryElement, Element>> getTopLevelDeclarations(
      String name) async {
    return TopLevelDeclarations(resolvedResult).withName(name);
  }

  /// Returns libraries with extensions that declare non-static public
  /// extension members with the [memberName].
  // TODO(srawlins): The documentation above is wrong; `memberName` is unused.
  Stream<LibraryElement> librariesWithExtensions(String memberName) async* {
    var analysisContext = resolvedResult.session.analysisContext;
    var analysisDriver = (analysisContext as DriverBasedAnalysisContext).driver;
    await analysisDriver.discoverAvailableFiles();

    var fsState = analysisDriver.fsState;
    var filter = FileStateFilter(
      fsState.getFileForPath(resolvedResult.path),
    );

    for (var file in fsState.knownFiles.toList()) {
      if (!filter.shouldInclude(file)) {
        continue;
      }

      var elementResult = await analysisDriver.getLibraryByUri(file.uriStr);
      if (elementResult is! LibraryElementResult) {
        continue;
      }

      yield elementResult.element;
    }
  }

  /// Returns libraries with extensions that declare non-static public
  /// extension members with the [memberName].
  // TODO(srawlins): The documentation above is wrong; `memberName` is unused.
  Stream<LibraryElement2> librariesWithExtensions2(String memberName) async* {
    var analysisContext = resolvedResult.session.analysisContext;
    var analysisDriver = (analysisContext as DriverBasedAnalysisContext).driver;
    await analysisDriver.discoverAvailableFiles();

    var fsState = analysisDriver.fsState;
    var filter = FileStateFilter(
      fsState.getFileForPath(resolvedResult.path),
    );

    for (var file in fsState.knownFiles.toList()) {
      if (!filter.shouldInclude(file)) {
        continue;
      }

      var elementResult = await analysisDriver.getLibraryByUri(file.uriStr);
      if (elementResult is! LibraryElementResult) {
        continue;
      }

      yield elementResult.element2;
    }
  }
}