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

// 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:_fe_analyzer_shared/src/scanner/token.dart';
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/dart/ast/visitor.dart';
import 'package:analyzer_plugin/utilities/change_builder/change_builder_dart.dart';
import 'package:analyzer_plugin/utilities/fixes/fixes.dart';
import 'package:analyzer_plugin/utilities/range_factory.dart';

class InlineTypedef extends CorrectionProducer {
  String name;

  @override
  List<Object> get fixArguments => [name];

  @override
  FixKind get fixKind => DartFixKind.INLINE_TYPEDEF;

  @override
  Future<void> compute(DartChangeBuilder builder) async {
    //
    // Extract the information needed to build the edit.
    //
    TypeAnnotation returnType;
    TypeParameterList typeParameters;
    List<FormalParameter> parameters;
    if (node is FunctionTypeAlias) {
      var typedef = node as FunctionTypeAlias;
      returnType = typedef.returnType;
      name = typedef.name.name;
      typeParameters = typedef.typeParameters;
      parameters = typedef.parameters.parameters;
    } else if (node is GenericTypeAlias) {
      var typedef = node as GenericTypeAlias;
      if (typedef.typeParameters != null) {
        return;
      }
      var functionType = typedef.functionType;
      returnType = functionType.returnType;
      name = typedef.name.name;
      typeParameters = functionType.typeParameters;
      parameters = functionType.parameters.parameters;
    } else {
      return;
    }
    // TODO(brianwilkerson) Handle parts.
    var finder = _ReferenceFinder(name);
    resolvedResult.unit.accept(finder);
    if (finder.count != 1) {
      return;
    }
    //
    // Build the edit.
    //
    await builder.addFileEdit(file, (DartFileEditBuilder builder) {
      builder.addDeletion(utils.getLinesRange(range.node(node)));
      builder.addReplacement(range.node(finder.reference),
          (DartEditBuilder builder) {
        if (returnType != null) {
          builder.write(utils.getNodeText(returnType));
          builder.write(' ');
        }
        builder.write('Function');
        if (typeParameters != null) {
          builder.write(utils.getNodeText(typeParameters));
        }
        String groupEnd;
        builder.write('(');
        for (int i = 0; i < parameters.length; i++) {
          var parameter = parameters[i];
          if (i > 0) {
            // This intentionally drops any trailing comma in order to improve
            // formatting.
            builder.write(', ');
          }
          if (parameter is DefaultFormalParameter) {
            if (groupEnd == null) {
              if (parameter.isNamed) {
                groupEnd = '}';
                builder.write('{');
              } else {
                groupEnd = ']';
                builder.write('[');
              }
            }
            parameter = (parameter as DefaultFormalParameter).parameter;
          }
          if (parameter is FunctionTypedFormalParameter) {
            builder.write(utils.getNodeText(parameter));
          } else if (parameter is SimpleFormalParameter) {
            if (parameter.metadata.isNotEmpty) {
              builder
                  .write(utils.getRangeText(range.nodes(parameter.metadata)));
            }
            if (parameter.requiredKeyword != null) {
              builder.write('required ');
            }
            if (parameter.covariantKeyword != null) {
              builder.write('covariant ');
            }
            var keyword = parameter.keyword;
            if (keyword != null && keyword.type != Keyword.VAR) {
              builder.write(keyword.lexeme);
            }
            if (parameter.type == null) {
              builder.write('dynamic');
            } else {
              builder.write(utils.getNodeText(parameter.type));
            }
            if (parameter.isNamed) {
              builder.write(' ');
              builder.write(parameter.identifier.name);
            }
          }
        }
        if (groupEnd != null) {
          builder.write(groupEnd);
        }
        builder.write(')');
      });
    });
  }
}

class _ReferenceFinder extends RecursiveAstVisitor {
  final String typeName;

  TypeName reference;

  int count = 0;

  _ReferenceFinder(this.typeName);

  @override
  void visitTypeName(TypeName node) {
    if (node.name.name == typeName) {
      reference ??= node;
      count++;
    }
    super.visitTypeName(node);
  }
}