pkg/analyzer_plugin/lib/src/utilities/fixes/fixes.dart

// Copyright (c) 2017, 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:analyzer/dart/analysis/results.dart';
import 'package:analyzer/diagnostic/diagnostic.dart';
import 'package:analyzer/file_system/file_system.dart';
import 'package:analyzer_plugin/protocol/protocol_generated.dart';
import 'package:analyzer_plugin/utilities/analyzer_converter.dart';
import 'package:analyzer_plugin/utilities/fixes/fixes.dart';

/// A concrete implementation of [DartFixesRequest].
class DartFixesRequestImpl implements DartFixesRequest {
  @override
  final ResourceProvider resourceProvider;

  @override
  final int offset;

  @override
  final List<Diagnostic> errorsToFix;

  @override
  final ResolvedUnitResult result;

  /// Initialize a newly create request with the given data.
  DartFixesRequestImpl(
    this.resourceProvider,
    this.offset,
    this.errorsToFix,
    this.result,
  );
}

/// A concrete implementation of [FixCollector].
class FixCollectorImpl implements FixCollector {
  /// The list of fixes that have been collected.
  final Map<Diagnostic, List<PrioritizedSourceChange>> fixMap =
      <Diagnostic, List<PrioritizedSourceChange>>{};

  /// Return the fixes that have been collected up to this point.
  List<AnalysisErrorFixes> get fixes {
    var fixes = <AnalysisErrorFixes>[];
    var converter = AnalyzerConverter();
    for (var error in fixMap.keys) {
      fixes.add(
        AnalysisErrorFixes(
          converter.convertAnalysisError(error),
          fixes: fixMap[error],
        ),
      );
    }
    return fixes;
  }

  @override
  void addFix(Diagnostic diagnostic, PrioritizedSourceChange change) {
    fixMap
        .putIfAbsent(diagnostic, () => <PrioritizedSourceChange>[])
        .add(change);
  }
}