| // Copyright (c) 2020, 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/dart/abstract_producer.dart'; |
| import 'package:analysis_server/src/services/correction/fix.dart'; |
| import 'package:analyzer/dart/ast/ast.dart'; |
| import 'package:analyzer_plugin/utilities/change_builder/change_builder_core.dart'; |
| import 'package:analyzer_plugin/utilities/fixes/fixes.dart'; |
| |
| class ReplaceReturnTypeFuture extends CorrectionProducer { |
| @override |
| FixKind get fixKind => DartFixKind.REPLACE_RETURN_TYPE_FUTURE; |
| |
| @override |
| Future<void> compute(ChangeBuilder builder) async { |
| // prepare the existing type |
| var typeName = node.thisOrAncestorOfType<TypeAnnotation>(); |
| var typeProvider = this.typeProvider; |
| await builder.addDartFileEdit(file, (builder) { |
| builder.replaceTypeWithFuture(typeName, typeProvider); |
| }); |
| } |
| |
| /// Return an instance of this class. Used as a tear-off in `FixProcessor`. |
| static ReplaceReturnTypeFuture newInstance() => ReplaceReturnTypeFuture(); |
| } |