blob: dffd207c95d2ae0232986388cc8d6f5fd4580d3a [file] [log] [blame]
// Copyright (c) 2014, 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 'dart:async';
import 'dart:collection';
import 'package:analysis_server/plugin/edit/assist/assist_core.dart';
import 'package:analysis_server/plugin/edit/assist/assist_dart.dart';
import 'package:analysis_server/src/services/correction/assist.dart';
import 'package:analysis_server/src/services/correction/name_suggestion.dart';
import 'package:analysis_server/src/services/correction/selection_analyzer.dart';
import 'package:analysis_server/src/services/correction/statement_analyzer.dart';
import 'package:analysis_server/src/services/correction/util.dart';
import 'package:analysis_server/src/services/search/hierarchy.dart';
import 'package:analysis_server/src/utilities/flutter.dart' as flutter;
import 'package:analyzer/dart/analysis/session.dart';
import 'package:analyzer/dart/ast/ast.dart';
import 'package:analyzer/dart/ast/standard_resolution_map.dart';
import 'package:analyzer/dart/ast/token.dart';
import 'package:analyzer/dart/ast/visitor.dart';
import 'package:analyzer/dart/element/element.dart';
import 'package:analyzer/dart/element/type.dart';
import 'package:analyzer/src/dart/analysis/driver.dart';
import 'package:analyzer/src/dart/analysis/session_helper.dart';
import 'package:analyzer/src/dart/ast/token.dart';
import 'package:analyzer/src/dart/ast/utilities.dart';
import 'package:analyzer/src/generated/java_core.dart';
import 'package:analyzer/src/generated/resolver.dart';
import 'package:analyzer/src/generated/source.dart';
import 'package:analyzer_plugin/protocol/protocol_common.dart' hide Element;
import 'package:analyzer_plugin/utilities/assist/assist.dart'
hide AssistContributor;
import 'package:analyzer_plugin/utilities/change_builder/change_builder_dart.dart';
import 'package:analyzer_plugin/utilities/range_factory.dart';
import 'package:meta/meta.dart';
import 'package:path/path.dart';
typedef _SimpleIdentifierVisitor(SimpleIdentifier node);
/**
* The computer for Dart assists.
*/
class AssistProcessor {
/**
* The analysis driver being used to perform analysis.
*/
AnalysisDriver driver;
/**
* The analysis session to be used to create the change builder.
*/
AnalysisSession session;
/**
* The helper wrapper around the [session].
*/
AnalysisSessionHelper sessionHelper;
Source source;
String file;
CompilationUnit unit;
CompilationUnitElement unitElement;
LibraryElement unitLibraryElement;
int selectionOffset;
int selectionLength;
int selectionEnd;
final List<Assist> assists = <Assist>[];
CorrectionUtils utils;
AstNode node;
TypeProvider _typeProvider;
AssistProcessor(DartAssistContext dartContext) {
driver = dartContext.analysisDriver;
session = driver.currentSession;
sessionHelper = new AnalysisSessionHelper(session);
// source
source = dartContext.source;
file = dartContext.source.fullName;
// unit
unit = dartContext.unit;
unitElement = dartContext.unit.declaredElement;
// library
unitLibraryElement = resolutionMap
.elementDeclaredByCompilationUnit(dartContext.unit)
.library;
// selection
selectionOffset = dartContext.selectionOffset;
selectionLength = dartContext.selectionLength;
selectionEnd = selectionOffset + selectionLength;
}
/**
* Returns the EOL to use for this [CompilationUnit].
*/
String get eol => utils.endOfLine;
TypeProvider get typeProvider {
if (_typeProvider == null) {
_typeProvider = unitElement.context.typeProvider;
}
return _typeProvider;
}
Future<List<Assist>> compute() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
if (!_setupCompute()) {
return assists;
}
await _addProposal_addTypeAnnotation_DeclaredIdentifier();
await _addProposal_addTypeAnnotation_SimpleFormalParameter();
await _addProposal_addTypeAnnotation_VariableDeclaration();
await _addProposal_assignToLocalVariable();
await _addProposal_convertClassToMixin();
await _addProposal_convertIntoFinalField();
await _addProposal_convertIntoGetter();
await _addProposal_convertDocumentationIntoBlock();
await _addProposal_convertDocumentationIntoLine();
await _addProposal_convertToAsyncFunctionBody();
await _addProposal_convertToBlockFunctionBody();
await _addProposal_convertToDoubleQuotedString();
await _addProposal_convertToExpressionFunctionBody();
await _addProposal_convertPartOfToUri();
await _addProposal_convertToForIndexLoop();
await _addProposal_convertToGenericFunctionSyntax();
await _addProposal_convertToIntLiteral();
await _addProposal_convertToIsNot_onIs();
await _addProposal_convertToIsNot_onNot();
await _addProposal_convertToIsNotEmpty();
await _addProposal_convertToFieldParameter();
await _addProposal_convertToNormalParameter();
await _addProposal_convertToSingleQuotedString();
await _addProposal_encapsulateField();
await _addProposal_exchangeOperands();
await _addProposal_flutterConvertToChildren();
await _addProposal_flutterConvertToStatefulWidget();
await _addProposal_flutterMoveWidgetDown();
await _addProposal_flutterMoveWidgetUp();
await _addProposal_flutterRemoveWidget_singleChild();
await _addProposal_flutterRemoveWidget_multipleChildren();
await _addProposal_flutterSwapWithChild();
await _addProposal_flutterSwapWithParent();
await _addProposal_flutterWrapWidget();
await _addProposal_flutterWrapWidgets();
await _addProposal_importAddShow();
await _addProposal_introduceLocalTestedType();
await _addProposal_invertIf();
await _addProposal_joinIfStatementInner();
await _addProposal_joinIfStatementOuter();
await _addProposal_joinVariableDeclaration_onAssignment();
await _addProposal_joinVariableDeclaration_onDeclaration();
await _addProposal_removeTypeAnnotation();
await _addProposal_reparentFlutterList();
await _addProposal_replaceConditionalWithIfElse();
await _addProposal_replaceIfElseWithConditional();
await _addProposal_splitAndCondition();
await _addProposal_splitVariableDeclaration();
await _addProposal_surroundWith();
return assists;
}
Future<List<Assist>> computeAssist(AssistKind assistKind) async {
if (!_setupCompute()) {
return assists;
}
// Calculate only specific assists for edit.dartFix
if (assistKind == DartAssistKind.CONVERT_CLASS_TO_MIXIN) {
await _addProposal_convertClassToMixin();
} else if (assistKind == DartAssistKind.CONVERT_TO_INT_LITERAL) {
await _addProposal_convertToIntLiteral();
}
return assists;
}
FunctionBody getEnclosingFunctionBody() {
// TODO(brianwilkerson) Determine whether there is a reason why this method
// isn't just "return node.getAncestor((node) => node is FunctionBody);"
{
FunctionExpression function =
node.getAncestor((node) => node is FunctionExpression);
if (function != null) {
return function.body;
}
}
{
FunctionDeclaration function =
node.getAncestor((node) => node is FunctionDeclaration);
if (function != null) {
return function.functionExpression.body;
}
}
{
ConstructorDeclaration constructor =
node.getAncestor((node) => node is ConstructorDeclaration);
if (constructor != null) {
return constructor.body;
}
}
{
MethodDeclaration method =
node.getAncestor((node) => node is MethodDeclaration);
if (method != null) {
return method.body;
}
}
return null;
}
void _addAssistFromBuilder(DartChangeBuilder builder, AssistKind kind,
{List args: null}) {
SourceChange change = builder.sourceChange;
if (change.edits.isEmpty) {
_coverageMarker();
return;
}
change.id = kind.id;
change.message = formatList(kind.message, args);
assists.add(new Assist(kind, change));
}
Future<void> _addProposal_addTypeAnnotation_DeclaredIdentifier() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
DeclaredIdentifier declaredIdentifier =
node.getAncestor((n) => n is DeclaredIdentifier);
if (declaredIdentifier == null) {
ForEachStatement forEach = node.getAncestor((n) => n is ForEachStatement);
int offset = node.offset;
if (forEach != null &&
forEach.iterable != null &&
offset < forEach.iterable.offset) {
declaredIdentifier = forEach.loopVariable;
}
}
if (declaredIdentifier == null) {
_coverageMarker();
return;
}
// Ensure that there isn't already a type annotation.
if (declaredIdentifier.type != null) {
_coverageMarker();
return;
}
DartType type = declaredIdentifier.identifier.staticType;
if (type is! InterfaceType && type is! FunctionType) {
_coverageMarker();
return;
}
_configureTargetLocation(node);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
bool validChange = true;
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
Token keyword = declaredIdentifier.keyword;
if (keyword.keyword == Keyword.VAR) {
builder.addReplacement(range.token(keyword), (DartEditBuilder builder) {
validChange = builder.writeType(type);
});
} else {
builder.addInsertion(declaredIdentifier.identifier.offset,
(DartEditBuilder builder) {
validChange = builder.writeType(type);
builder.write(' ');
});
}
});
if (validChange) {
_addAssistFromBuilder(changeBuilder, DartAssistKind.ADD_TYPE_ANNOTATION);
}
}
Future<void> _addProposal_addTypeAnnotation_SimpleFormalParameter() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
AstNode node = this.node;
// should be the name of a simple parameter
if (node is! SimpleIdentifier || node.parent is! SimpleFormalParameter) {
_coverageMarker();
return;
}
SimpleIdentifier name = node;
SimpleFormalParameter parameter = node.parent;
// the parameter should not have a type
if (parameter.type != null) {
_coverageMarker();
return;
}
// prepare the type
DartType type = parameter.declaredElement.type;
// TODO(scheglov) If the parameter is in a method declaration, and if the
// method overrides a method that has a type for the corresponding
// parameter, it would be nice to copy down the type from the overridden
// method.
if (type is! InterfaceType) {
_coverageMarker();
return;
}
// prepare type source
_configureTargetLocation(node);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
bool validChange = true;
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addInsertion(name.offset, (DartEditBuilder builder) {
validChange = builder.writeType(type);
builder.write(' ');
});
});
if (validChange) {
_addAssistFromBuilder(changeBuilder, DartAssistKind.ADD_TYPE_ANNOTATION);
}
}
Future<void> _addProposal_addTypeAnnotation_VariableDeclaration() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
AstNode node = this.node;
// prepare VariableDeclarationList
VariableDeclarationList declarationList =
node.getAncestor((node) => node is VariableDeclarationList);
if (declarationList == null) {
_coverageMarker();
return;
}
// may be has type annotation already
if (declarationList.type != null) {
_coverageMarker();
return;
}
// prepare single VariableDeclaration
List<VariableDeclaration> variables = declarationList.variables;
if (variables.length != 1) {
_coverageMarker();
return;
}
VariableDeclaration variable = variables[0];
// must be not after the name of the variable
if (selectionOffset > variable.name.end) {
_coverageMarker();
return;
}
// we need an initializer to get the type from
Expression initializer = variable.initializer;
if (initializer == null) {
_coverageMarker();
return;
}
DartType type = initializer.staticType;
// prepare type source
if ((type is! InterfaceType || type.isDartCoreNull) &&
type is! FunctionType) {
_coverageMarker();
return;
}
_configureTargetLocation(node);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
bool validChange = true;
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
Token keyword = declarationList.keyword;
if (keyword?.keyword == Keyword.VAR) {
builder.addReplacement(range.token(keyword), (DartEditBuilder builder) {
validChange = builder.writeType(type);
});
} else {
builder.addInsertion(variable.offset, (DartEditBuilder builder) {
validChange = builder.writeType(type);
builder.write(' ');
});
}
});
if (validChange) {
_addAssistFromBuilder(changeBuilder, DartAssistKind.ADD_TYPE_ANNOTATION);
}
}
Future<void> _addProposal_convertToIntLiteral() async {
if (node is! DoubleLiteral) {
_coverageMarker();
return;
}
DoubleLiteral literal = node;
int intValue;
try {
intValue = literal.value?.truncate();
} catch (e) {
// Double cannot be converted to int
}
if (intValue == null || intValue != literal.value) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(new SourceRange(literal.offset, literal.length),
(DartEditBuilder builder) {
builder.write('${intValue}');
});
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_TO_INT_LITERAL);
}
Future<void> _addProposal_assignToLocalVariable() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare enclosing ExpressionStatement
ExpressionStatement expressionStatement;
for (AstNode node = this.node; node != null; node = node.parent) {
if (node is ExpressionStatement) {
expressionStatement = node;
break;
}
if (node is ArgumentList ||
node is AssignmentExpression ||
node is Statement ||
node is ThrowExpression) {
_coverageMarker();
return;
}
}
if (expressionStatement == null) {
_coverageMarker();
return;
}
// prepare expression
Expression expression = expressionStatement.expression;
int offset = expression.offset;
// prepare expression type
DartType type = expression.staticType;
if (type.isVoid) {
_coverageMarker();
return;
}
// prepare excluded names
Set<String> excluded = new Set<String>();
ScopedNameFinder scopedNameFinder = new ScopedNameFinder(offset);
expression.accept(scopedNameFinder);
excluded.addAll(scopedNameFinder.locals.keys.toSet());
List<String> suggestions =
getVariableNameSuggestionsForExpression(type, expression, excluded);
if (suggestions.isNotEmpty) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addInsertion(offset, (DartEditBuilder builder) {
builder.write('var ');
builder.addSimpleLinkedEdit('NAME', suggestions[0],
kind: LinkedEditSuggestionKind.VARIABLE,
suggestions: suggestions);
builder.write(' = ');
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.ASSIGN_TO_LOCAL_VARIABLE);
}
}
Future<void> _addProposal_convertClassToMixin() async {
ClassDeclaration classDeclaration =
node.getAncestor((n) => n is ClassDeclaration);
if (classDeclaration == null) {
return;
}
if (selectionOffset > classDeclaration.name.end ||
selectionEnd < classDeclaration.classKeyword.offset) {
return;
}
if (classDeclaration.members
.any((member) => member is ConstructorDeclaration)) {
return;
}
_SuperclassReferenceFinder finder = new _SuperclassReferenceFinder();
classDeclaration.accept(finder);
List<ClassElement> referencedClasses = finder.referencedClasses;
List<InterfaceType> superclassConstraints = <InterfaceType>[];
List<InterfaceType> interfaces = <InterfaceType>[];
ClassElement classElement = classDeclaration.declaredElement;
for (InterfaceType type in classElement.mixins) {
if (referencedClasses.contains(type.element)) {
superclassConstraints.add(type);
} else {
interfaces.add(type);
}
}
ExtendsClause extendsClause = classDeclaration.extendsClause;
if (extendsClause != null) {
if (referencedClasses.length > superclassConstraints.length) {
superclassConstraints.insert(0, classElement.supertype);
} else {
interfaces.insert(0, classElement.supertype);
}
}
interfaces.addAll(classElement.interfaces);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(
range.startStart(
classDeclaration.abstractKeyword ?? classDeclaration.classKeyword,
classDeclaration.leftBracket), (DartEditBuilder builder) {
builder.write('mixin ');
builder.write(classDeclaration.name.name);
builder.writeTypes(superclassConstraints, prefix: ' on ');
builder.writeTypes(interfaces, prefix: ' implements ');
builder.write(' ');
});
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_CLASS_TO_MIXIN);
}
Future<void> _addProposal_convertDocumentationIntoBlock() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
Comment comment = node.getAncestor((n) => n is Comment);
if (comment == null || !comment.isDocumentation) {
return;
}
var tokens = comment.tokens;
if (tokens.isEmpty ||
tokens.any((Token token) =>
token is! DocumentationCommentToken ||
token.type != TokenType.SINGLE_LINE_COMMENT)) {
return;
}
String prefix = utils.getNodePrefix(comment);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(comment), (DartEditBuilder builder) {
builder.writeln('/**');
for (Token token in comment.tokens) {
builder.write(prefix);
builder.write(' *');
builder.writeln(token.lexeme.substring('///'.length));
}
builder.write(prefix);
builder.write(' */');
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_DOCUMENTATION_INTO_BLOCK);
}
Future<void> _addProposal_convertDocumentationIntoLine() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
Comment comment = node.getAncestor((n) => n is Comment);
if (comment == null ||
!comment.isDocumentation ||
comment.tokens.length != 1) {
_coverageMarker();
return;
}
Token token = comment.tokens.first;
if (token.type != TokenType.MULTI_LINE_COMMENT) {
_coverageMarker();
return;
}
String text = token.lexeme;
List<String> lines = text.split('\n');
String prefix = utils.getNodePrefix(comment);
List<String> newLines = <String>[];
bool firstLine = true;
String linePrefix = '';
for (String line in lines) {
if (firstLine) {
firstLine = false;
String expectedPrefix = '/**';
if (!line.startsWith(expectedPrefix)) {
_coverageMarker();
return;
}
line = line.substring(expectedPrefix.length).trim();
if (line.isNotEmpty) {
newLines.add('/// $line');
linePrefix = eol + prefix;
}
} else {
if (line.startsWith(prefix + ' */')) {
break;
}
String expectedPrefix = prefix + ' * ';
if (!line.startsWith(expectedPrefix)) {
_coverageMarker();
return;
}
line = line.substring(expectedPrefix.length).trim();
newLines.add('$linePrefix/// $line');
linePrefix = eol + prefix;
}
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(comment), (DartEditBuilder builder) {
for (String newLine in newLines) {
builder.write(newLine);
}
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_DOCUMENTATION_INTO_LINE);
}
Future<void> _addProposal_convertIntoFinalField() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// Find the enclosing getter.
MethodDeclaration getter;
for (AstNode n = node; n != null; n = n.parent) {
if (n is MethodDeclaration) {
getter = n;
break;
}
if (n is SimpleIdentifier ||
n is TypeAnnotation ||
n is TypeArgumentList) {
continue;
}
break;
}
if (getter == null || !getter.isGetter) {
return;
}
// Check that there is no corresponding setter.
{
ExecutableElement element = getter.declaredElement;
if (element == null) {
return;
}
Element enclosing = element.enclosingElement;
if (enclosing is ClassElement) {
if (enclosing.getSetter(element.name) != null) {
return;
}
}
}
// Try to find the returned expression.
Expression expression;
{
FunctionBody body = getter.body;
if (body is ExpressionFunctionBody) {
expression = body.expression;
} else if (body is BlockFunctionBody) {
List<Statement> statements = body.block.statements;
if (statements.length == 1) {
Statement statement = statements.first;
if (statement is ReturnStatement) {
expression = statement.expression;
}
}
}
}
// Use the returned expression as the field initializer.
if (expression != null) {
String code = 'final';
if (getter.returnType != null) {
code += ' ' + _getNodeText(getter.returnType);
}
code += ' ' + _getNodeText(getter.name);
if (expression is! NullLiteral) {
code += ' = ' + _getNodeText(expression);
}
code += ';';
SourceRange replacementRange =
range.startEnd(getter.returnType ?? getter.propertyKeyword, getter);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(replacementRange, code);
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_INTO_FINAL_FIELD);
}
}
Future<void> _addProposal_convertIntoGetter() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// Find the enclosing field declaration.
FieldDeclaration fieldDeclaration;
for (AstNode n = node; n != null; n = n.parent) {
if (n is FieldDeclaration) {
fieldDeclaration = n;
break;
}
if (n is SimpleIdentifier ||
n is VariableDeclaration ||
n is VariableDeclarationList ||
n is TypeAnnotation ||
n is TypeArgumentList) {
continue;
}
break;
}
if (fieldDeclaration == null) {
return;
}
// The field must be final and has only one variable.
VariableDeclarationList fieldList = fieldDeclaration.fields;
if (!fieldList.isFinal || fieldList.variables.length != 1) {
return;
}
VariableDeclaration field = fieldList.variables.first;
// Prepare the initializer.
Expression initializer = field.initializer;
if (initializer == null) {
return;
}
// Add proposal.
String code = '';
if (fieldList.type != null) {
code += _getNodeText(fieldList.type) + ' ';
}
code += 'get';
code += ' ' + _getNodeText(field.name);
code += ' => ' + _getNodeText(initializer);
code += ';';
SourceRange replacementRange =
range.startEnd(fieldList.keyword, fieldDeclaration);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(replacementRange, code);
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_INTO_GETTER);
}
Future<void> _addProposal_convertPartOfToUri() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
PartOfDirective directive =
node.getAncestor((node) => node is PartOfDirective);
if (directive == null || directive.libraryName == null) {
return;
}
String libraryPath = unitLibraryElement.source.fullName;
String partPath = unit.declaredElement.source.fullName;
String relativePath = relative(libraryPath, from: dirname(partPath));
String uri = new Uri.file(relativePath).toString();
SourceRange replacementRange = range.node(directive.libraryName);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(replacementRange, "'$uri'");
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_PART_OF_TO_URI);
}
Future<void> _addProposal_convertToAsyncFunctionBody() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
FunctionBody body = getEnclosingFunctionBody();
if (body == null || body.isAsynchronous || body.isGenerator) {
_coverageMarker();
return;
}
// Function bodies can be quite large, e.g. Flutter build() methods.
// It is surprising to see this Quick Assist deep in a function body.
if (body is BlockFunctionBody &&
selectionOffset > body.block.beginToken.end) {
return;
}
if (body is ExpressionFunctionBody &&
selectionOffset > body.beginToken.end) {
return;
}
AstNode parent = body.parent;
if (parent is ConstructorDeclaration) {
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.convertFunctionFromSyncToAsync(body, typeProvider);
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_INTO_ASYNC_BODY);
}
Future<void> _addProposal_convertToBlockFunctionBody() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
FunctionBody body = getEnclosingFunctionBody();
// prepare expression body
if (body is! ExpressionFunctionBody || body.isGenerator) {
_coverageMarker();
return;
}
Expression returnValue = (body as ExpressionFunctionBody).expression;
// Return expressions can be quite large, e.g. Flutter build() methods.
// It is surprising to see this Quick Assist deep in the function body.
if (selectionOffset >= returnValue.offset) {
_coverageMarker();
return;
}
DartType returnValueType = returnValue.staticType;
String returnValueCode = _getNodeText(returnValue);
// prepare prefix
String prefix = utils.getNodePrefix(body.parent);
String indent = utils.getIndent(1);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(body), (DartEditBuilder builder) {
if (body.isAsynchronous) {
builder.write('async ');
}
builder.write('{$eol$prefix$indent');
if (!returnValueType.isVoid && !returnValueType.isBottom) {
builder.write('return ');
}
builder.write(returnValueCode);
builder.write(';');
builder.selectHere();
builder.write('$eol$prefix}');
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_INTO_BLOCK_BODY);
}
Future<void> _addProposal_convertToDoubleQuotedString() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
await _convertQuotes(false, DartAssistKind.CONVERT_TO_DOUBLE_QUOTED_STRING);
}
Future<void> _addProposal_convertToExpressionFunctionBody() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare current body
FunctionBody body = getEnclosingFunctionBody();
if (body is! BlockFunctionBody || body.isGenerator) {
_coverageMarker();
return;
}
// prepare return statement
List<Statement> statements = (body as BlockFunctionBody).block.statements;
if (statements.length != 1) {
_coverageMarker();
return;
}
Statement onlyStatement = statements.first;
// prepare returned expression
Expression returnExpression;
if (onlyStatement is ReturnStatement) {
returnExpression = onlyStatement.expression;
} else if (onlyStatement is ExpressionStatement) {
returnExpression = onlyStatement.expression;
}
if (returnExpression == null) {
_coverageMarker();
return;
}
// Return expressions can be quite large, e.g. Flutter build() methods.
// It is surprising to see this Quick Assist deep in the function body.
if (selectionOffset >= returnExpression.offset) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(body), (DartEditBuilder builder) {
if (body.isAsynchronous) {
builder.write('async ');
}
builder.write('=> ');
builder.write(_getNodeText(returnExpression));
if (body.parent is! FunctionExpression ||
body.parent.parent is FunctionDeclaration) {
builder.write(';');
}
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_INTO_EXPRESSION_BODY);
}
Future<void> _addProposal_convertToFieldParameter() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
if (node == null) {
return;
}
// prepare ConstructorDeclaration
ConstructorDeclaration constructor =
node.getAncestor((node) => node is ConstructorDeclaration);
if (constructor == null) {
return;
}
FormalParameterList parameterList = constructor.parameters;
List<ConstructorInitializer> initializers = constructor.initializers;
// prepare parameter
SimpleFormalParameter parameter;
if (node.parent is SimpleFormalParameter &&
node.parent.parent is FormalParameterList &&
node.parent.parent.parent is ConstructorDeclaration) {
parameter = node.parent;
}
if (node is SimpleIdentifier &&
node.parent is ConstructorFieldInitializer) {
String name = (node as SimpleIdentifier).name;
ConstructorFieldInitializer initializer = node.parent;
if (initializer.expression == node) {
for (FormalParameter formalParameter in parameterList.parameters) {
if (formalParameter is SimpleFormalParameter &&
formalParameter.identifier.name == name) {
parameter = formalParameter;
}
}
}
}
// analyze parameter
if (parameter != null) {
String parameterName = parameter.identifier.name;
ParameterElement parameterElement = parameter.declaredElement;
// check number of references
{
int numOfReferences = 0;
AstVisitor visitor =
new _SimpleIdentifierRecursiveAstVisitor((SimpleIdentifier node) {
if (node.staticElement == parameterElement) {
numOfReferences++;
}
});
for (ConstructorInitializer initializer in initializers) {
initializer.accept(visitor);
}
if (numOfReferences != 1) {
return;
}
}
// find the field initializer
ConstructorFieldInitializer parameterInitializer;
for (ConstructorInitializer initializer in initializers) {
if (initializer is ConstructorFieldInitializer) {
Expression expression = initializer.expression;
if (expression is SimpleIdentifier &&
expression.name == parameterName) {
parameterInitializer = initializer;
}
}
}
if (parameterInitializer == null) {
return;
}
String fieldName = parameterInitializer.fieldName.name;
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// replace parameter
builder.addSimpleReplacement(range.node(parameter), 'this.$fieldName');
// remove initializer
int initializerIndex = initializers.indexOf(parameterInitializer);
if (initializers.length == 1) {
builder
.addDeletion(range.endEnd(parameterList, parameterInitializer));
} else {
if (initializerIndex == 0) {
ConstructorInitializer next = initializers[initializerIndex + 1];
builder.addDeletion(range.startStart(parameterInitializer, next));
} else {
ConstructorInitializer prev = initializers[initializerIndex - 1];
builder.addDeletion(range.endEnd(prev, parameterInitializer));
}
}
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_TO_FIELD_PARAMETER);
}
}
Future<void> _addProposal_convertToForIndexLoop() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// find enclosing ForEachStatement
ForEachStatement forEachStatement =
node.getAncestor((n) => n is ForEachStatement);
if (forEachStatement == null) {
_coverageMarker();
return;
}
if (selectionOffset < forEachStatement.offset ||
forEachStatement.rightParenthesis.end < selectionOffset) {
_coverageMarker();
return;
}
// loop should declare variable
DeclaredIdentifier loopVariable = forEachStatement.loopVariable;
if (loopVariable == null) {
_coverageMarker();
return;
}
// iterable should be VariableElement
String listName;
Expression iterable = forEachStatement.iterable;
if (iterable is SimpleIdentifier &&
iterable.staticElement is VariableElement) {
listName = iterable.name;
} else {
_coverageMarker();
return;
}
// iterable should be List
{
DartType iterableType = iterable.staticType;
InterfaceType listType = typeProvider.listType;
if (iterableType is! InterfaceType ||
iterableType.element != listType.element) {
_coverageMarker();
return;
}
}
// body should be Block
if (forEachStatement.body is! Block) {
_coverageMarker();
return;
}
Block body = forEachStatement.body;
// prepare a name for the index variable
String indexName;
{
Set<String> conflicts =
utils.findPossibleLocalVariableConflicts(forEachStatement.offset);
if (!conflicts.contains('i')) {
indexName = 'i';
} else if (!conflicts.contains('j')) {
indexName = 'j';
} else if (!conflicts.contains('k')) {
indexName = 'k';
} else {
_coverageMarker();
return;
}
}
// prepare environment
String prefix = utils.getNodePrefix(forEachStatement);
String indent = utils.getIndent(1);
int firstBlockLine = utils.getLineContentEnd(body.leftBracket.end);
// add change
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// TODO(brianwilkerson) Create linked positions for the loop variable.
builder.addSimpleReplacement(
range.startEnd(forEachStatement, forEachStatement.rightParenthesis),
'for (int $indexName = 0; $indexName < $listName.length; $indexName++)');
builder.addSimpleInsertion(firstBlockLine,
'$prefix$indent$loopVariable = $listName[$indexName];$eol');
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_INTO_FOR_INDEX);
}
Future<void> _addProposal_convertToGenericFunctionSyntax() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
AstNode node = this.node;
while (node != null) {
if (node is FunctionTypeAlias) {
await _convertFunctionTypeAliasToGenericTypeAlias(node);
return;
} else if (node is FunctionTypedFormalParameter) {
await _convertFunctionTypedFormalParameterToSimpleFormalParameter(node);
return;
} else if (node is FormalParameterList) {
// It would be confusing for this assist to alter a surrounding context
// when the selection is inside a parameter list.
return;
}
node = node.parent;
}
}
Future<void> _addProposal_convertToIsNot_onIs() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// may be child of "is"
AstNode node = this.node;
while (node != null && node is! IsExpression) {
node = node.parent;
}
// prepare "is"
if (node is! IsExpression) {
_coverageMarker();
return;
}
IsExpression isExpression = node as IsExpression;
if (isExpression.notOperator != null) {
_coverageMarker();
return;
}
// prepare enclosing ()
AstNode parent = isExpression.parent;
if (parent is! ParenthesizedExpression) {
_coverageMarker();
return;
}
ParenthesizedExpression parExpression = parent as ParenthesizedExpression;
// prepare enclosing !()
AstNode parent2 = parent.parent;
if (parent2 is! PrefixExpression) {
_coverageMarker();
return;
}
PrefixExpression prefExpression = parent2 as PrefixExpression;
if (prefExpression.operator.type != TokenType.BANG) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
if (getExpressionParentPrecedence(prefExpression) >=
TokenClass.RELATIONAL_OPERATOR.precedence) {
builder.addDeletion(range.token(prefExpression.operator));
} else {
builder.addDeletion(
range.startEnd(prefExpression, parExpression.leftParenthesis));
builder.addDeletion(
range.startEnd(parExpression.rightParenthesis, prefExpression));
}
builder.addSimpleInsertion(isExpression.isOperator.end, '!');
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_INTO_IS_NOT);
}
Future<void> _addProposal_convertToIsNot_onNot() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// may be () in prefix expression
AstNode node = this.node;
if (node is ParenthesizedExpression && node.parent is PrefixExpression) {
node = node.parent;
}
// prepare !()
if (node is! PrefixExpression) {
_coverageMarker();
return;
}
PrefixExpression prefExpression = node as PrefixExpression;
// should be ! operator
if (prefExpression.operator.type != TokenType.BANG) {
_coverageMarker();
return;
}
// prepare !()
Expression operand = prefExpression.operand;
if (operand is! ParenthesizedExpression) {
_coverageMarker();
return;
}
ParenthesizedExpression parExpression = operand as ParenthesizedExpression;
operand = parExpression.expression;
// prepare "is"
if (operand is! IsExpression) {
_coverageMarker();
return;
}
IsExpression isExpression = operand as IsExpression;
if (isExpression.notOperator != null) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
if (getExpressionParentPrecedence(prefExpression) >=
TokenClass.RELATIONAL_OPERATOR.precedence) {
builder.addDeletion(range.token(prefExpression.operator));
} else {
builder.addDeletion(
range.startEnd(prefExpression, parExpression.leftParenthesis));
builder.addDeletion(
range.startEnd(parExpression.rightParenthesis, prefExpression));
}
builder.addSimpleInsertion(isExpression.isOperator.end, '!');
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.CONVERT_INTO_IS_NOT);
}
/**
* Converts "!isEmpty" -> "isNotEmpty" if possible.
*/
Future<void> _addProposal_convertToIsNotEmpty() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare "expr.isEmpty"
AstNode isEmptyAccess = null;
SimpleIdentifier isEmptyIdentifier = null;
if (node is SimpleIdentifier) {
SimpleIdentifier identifier = node as SimpleIdentifier;
AstNode parent = identifier.parent;
// normal case (but rare)
if (parent is PropertyAccess) {
isEmptyIdentifier = parent.propertyName;
isEmptyAccess = parent;
}
// usual case
if (parent is PrefixedIdentifier) {
isEmptyIdentifier = parent.identifier;
isEmptyAccess = parent;
}
}
if (isEmptyIdentifier == null) {
_coverageMarker();
return;
}
// should be "isEmpty"
Element propertyElement = isEmptyIdentifier.staticElement;
if (propertyElement == null || 'isEmpty' != propertyElement.name) {
_coverageMarker();
return;
}
// should have "isNotEmpty"
Element propertyTarget = propertyElement.enclosingElement;
if (propertyTarget == null ||
getChildren(propertyTarget, 'isNotEmpty').isEmpty) {
_coverageMarker();
return;
}
// should be in PrefixExpression
if (isEmptyAccess.parent is! PrefixExpression) {
_coverageMarker();
return;
}
PrefixExpression prefixExpression =
isEmptyAccess.parent as PrefixExpression;
// should be !
if (prefixExpression.operator.type != TokenType.BANG) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addDeletion(
range.startStart(prefixExpression, prefixExpression.operand));
builder.addSimpleReplacement(range.node(isEmptyIdentifier), 'isNotEmpty');
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_INTO_IS_NOT_EMPTY);
}
Future<void> _addProposal_convertToNormalParameter() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
if (node is SimpleIdentifier &&
node.parent is FieldFormalParameter &&
node.parent.parent is FormalParameterList &&
node.parent.parent.parent is ConstructorDeclaration) {
ConstructorDeclaration constructor = node.parent.parent.parent;
FormalParameterList parameterList = node.parent.parent;
FieldFormalParameter parameter = node.parent;
ParameterElement parameterElement = parameter.declaredElement;
String name = (node as SimpleIdentifier).name;
// prepare type
DartType type = parameterElement.type;
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// replace parameter
if (type.isDynamic) {
builder.addSimpleReplacement(range.node(parameter), name);
} else {
builder.addReplacement(range.node(parameter),
(DartEditBuilder builder) {
builder.writeType(type);
builder.write(' ');
builder.write(name);
});
}
// add field initializer
List<ConstructorInitializer> initializers = constructor.initializers;
if (initializers.isEmpty) {
builder.addSimpleInsertion(parameterList.end, ' : $name = $name');
} else {
builder.addSimpleInsertion(initializers.last.end, ', $name = $name');
}
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.CONVERT_TO_NORMAL_PARAMETER);
}
}
Future<void> _addProposal_convertToSingleQuotedString() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
await _convertQuotes(true, DartAssistKind.CONVERT_TO_SINGLE_QUOTED_STRING);
}
Future<void> _addProposal_encapsulateField() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// find FieldDeclaration
FieldDeclaration fieldDeclaration =
node.getAncestor((x) => x is FieldDeclaration);
if (fieldDeclaration == null) {
_coverageMarker();
return;
}
// not interesting for static
if (fieldDeclaration.isStatic) {
_coverageMarker();
return;
}
// has a parse error
VariableDeclarationList variableList = fieldDeclaration.fields;
if (variableList.keyword == null && variableList.type == null) {
_coverageMarker();
return;
}
// not interesting for final
if (variableList.isFinal) {
_coverageMarker();
return;
}
// should have exactly one field
List<VariableDeclaration> fields = variableList.variables;
if (fields.length != 1) {
_coverageMarker();
return;
}
VariableDeclaration field = fields.first;
SimpleIdentifier nameNode = field.name;
FieldElement fieldElement = nameNode.staticElement;
// should have a public name
String name = nameNode.name;
if (Identifier.isPrivateName(name)) {
_coverageMarker();
return;
}
// should be on the name
if (nameNode != node) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// rename field
builder.addSimpleReplacement(range.node(nameNode), '_$name');
// update references in constructors
ClassOrMixinDeclaration classDeclaration = fieldDeclaration.parent;
for (ClassMember member in classDeclaration.members) {
if (member is ConstructorDeclaration) {
for (FormalParameter parameter in member.parameters.parameters) {
ParameterElement parameterElement = parameter.declaredElement;
if (parameterElement is FieldFormalParameterElement &&
parameterElement.field == fieldElement) {
SimpleIdentifier identifier = parameter.identifier;
builder.addSimpleReplacement(range.node(identifier), '_$name');
}
}
}
}
// Write getter and setter.
builder.addInsertion(fieldDeclaration.end, (builder) {
String docCode;
if (fieldDeclaration.documentationComment != null) {
docCode = utils.getNodeText(fieldDeclaration.documentationComment);
}
String typeCode = '';
if (variableList.type != null) {
typeCode = _getNodeText(variableList.type) + ' ';
}
// Write getter.
builder.writeln();
builder.writeln();
if (docCode != null) {
builder.write(' ');
builder.writeln(docCode);
}
builder.write(' ${typeCode}get $name => _$name;');
// Write setter.
builder.writeln();
builder.writeln();
if (docCode != null) {
builder.write(' ');
builder.writeln(docCode);
}
builder.writeln(' set $name($typeCode$name) {');
builder.writeln(' _$name = $name;');
builder.write(' }');
});
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.ENCAPSULATE_FIELD);
}
Future<void> _addProposal_exchangeOperands() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// check that user invokes quick assist on binary expression
if (node is! BinaryExpression) {
_coverageMarker();
return;
}
BinaryExpression binaryExpression = node as BinaryExpression;
// prepare operator position
if (!_isOperatorSelected(
binaryExpression, selectionOffset, selectionLength)) {
_coverageMarker();
return;
}
// add edits
Expression leftOperand = binaryExpression.leftOperand;
Expression rightOperand = binaryExpression.rightOperand;
// find "wide" enclosing binary expression with same operator
while (binaryExpression.parent is BinaryExpression) {
BinaryExpression newBinaryExpression =
binaryExpression.parent as BinaryExpression;
if (newBinaryExpression.operator.type != binaryExpression.operator.type) {
_coverageMarker();
break;
}
binaryExpression = newBinaryExpression;
}
// exchange parts of "wide" expression parts
SourceRange leftRange = range.startEnd(binaryExpression, leftOperand);
SourceRange rightRange = range.startEnd(rightOperand, binaryExpression);
// maybe replace the operator
Token operator = binaryExpression.operator;
// prepare a new operator
String newOperator = null;
TokenType operatorType = operator.type;
if (operatorType == TokenType.LT) {
newOperator = '>';
} else if (operatorType == TokenType.LT_EQ) {
newOperator = '>=';
} else if (operatorType == TokenType.GT) {
newOperator = '<';
} else if (operatorType == TokenType.GT_EQ) {
newOperator = '<=';
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(leftRange, _getRangeText(rightRange));
builder.addSimpleReplacement(rightRange, _getRangeText(leftRange));
// Optionally replace the operator.
if (newOperator != null) {
builder.addSimpleReplacement(range.token(operator), newOperator);
}
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.EXCHANGE_OPERANDS);
}
Future<void> _addProposal_flutterConvertToChildren() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// Find "child: widget" under selection.
NamedExpression namedExp;
{
AstNode node = this.node;
AstNode parent = node?.parent;
AstNode parent2 = parent?.parent;
if (node is SimpleIdentifier &&
parent is Label &&
parent2 is NamedExpression &&
node.name == 'child' &&
node.staticElement != null &&
flutter.isWidgetExpression(parent2.expression)) {
namedExp = parent2;
} else {
_coverageMarker();
return;
}
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
_convertFlutterChildToChildren(namedExp, eol, utils.getNodeText,
utils.getLinePrefix, utils.getIndent, utils.getText, builder);
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.FLUTTER_CONVERT_TO_CHILDREN);
}
Future<void> _addProposal_flutterConvertToStatefulWidget() async {
ClassDeclaration widgetClass =
node.getAncestor((n) => n is ClassDeclaration);
TypeName superclass = widgetClass?.extendsClause?.superclass;
if (widgetClass == null || superclass == null) {
_coverageMarker();
return;
}
// Don't spam, activate only from the `class` keyword to the class body.
if (selectionOffset < widgetClass.classKeyword.offset ||
selectionOffset > widgetClass.leftBracket.end) {
_coverageMarker();
return;
}
// Find the build() method.
MethodDeclaration buildMethod;
for (var member in widgetClass.members) {
if (member is MethodDeclaration &&
member.name.name == 'build' &&
member.parameters != null &&
member.parameters.parameters.length == 1) {
buildMethod = member;
break;
}
}
if (buildMethod == null) {
_coverageMarker();
return;
}
// Must be a StatelessWidget subclasses.
ClassElement widgetClassElement = widgetClass.declaredElement;
if (!flutter.isExactlyStatelessWidgetType(widgetClassElement.supertype)) {
_coverageMarker();
return;
}
String widgetName = widgetClassElement.displayName;
String stateName = widgetName + 'State';
// Find fields assigned in constructors.
var fieldsAssignedInConstructors = new Set<FieldElement>();
for (var member in widgetClass.members) {
if (member is ConstructorDeclaration) {
member.accept(new _SimpleIdentifierRecursiveAstVisitor((node) {
if (node.parent is FieldFormalParameter) {
Element element = node.staticElement;
if (element is FieldFormalParameterElement) {
fieldsAssignedInConstructors.add(element.field);
}
}
if (node.parent is ConstructorFieldInitializer) {
Element element = node.staticElement;
if (element is FieldElement) {
fieldsAssignedInConstructors.add(element);
}
}
if (node.inSetterContext()) {
Element element = node.staticElement;
if (element is PropertyAccessorElement) {
PropertyInducingElement field = element.variable;
if (field is FieldElement) {
fieldsAssignedInConstructors.add(field);
}
}
}
}));
}
}
// Prepare nodes to move.
var nodesToMove = new Set<ClassMember>();
var elementsToMove = new Set<Element>();
for (var member in widgetClass.members) {
if (member is FieldDeclaration && !member.isStatic) {
for (VariableDeclaration fieldNode in member.fields.variables) {
FieldElement fieldElement = fieldNode.declaredElement;
if (!fieldsAssignedInConstructors.contains(fieldElement)) {
nodesToMove.add(member);
elementsToMove.add(fieldElement);
elementsToMove.add(fieldElement.getter);
if (fieldElement.setter != null) {
elementsToMove.add(fieldElement.setter);
}
}
}
}
if (member is MethodDeclaration && !member.isStatic) {
nodesToMove.add(member);
elementsToMove.add(member.declaredElement);
}
}
/// Return the code for the [movedNode] which is suitable to be used
/// inside the `State` class, so that references to the widget fields and
/// methods, that are not moved, are qualified with the corresponding
/// instance `widget.`, or static `MyWidgetClass.` qualifier.
String rewriteWidgetMemberReferences(AstNode movedNode) {
var linesRange = utils.getLinesRange(range.node(movedNode));
var text = utils.getRangeText(linesRange);
// Insert `widget.` before references to the widget instance members.
final List<SourceEdit> edits = [];
movedNode.accept(new _SimpleIdentifierRecursiveAstVisitor((node) {
if (node.inDeclarationContext()) {
return;
}
var element = node.staticElement;
if (element is ExecutableElement &&
element?.enclosingElement == widgetClassElement &&
!elementsToMove.contains(element)) {
var offset = node.offset - linesRange.offset;
var qualifier = element.isStatic ? widgetName : 'widget';
AstNode parent = node.parent;
if (parent is InterpolationExpression &&
parent.leftBracket.type ==
TokenType.STRING_INTERPOLATION_IDENTIFIER) {
edits.add(new SourceEdit(offset, 0, '{$qualifier.'));
edits.add(new SourceEdit(offset + node.length, 0, '}'));
} else {
edits.add(new SourceEdit(offset, 0, '$qualifier.'));
}
}
}));
return SourceEdit.applySequence(text, edits.reversed);
}
var statefulWidgetClass = await sessionHelper.getClass(
flutter.WIDGETS_LIBRARY_URI, 'StatefulWidget');
var stateClass =
await sessionHelper.getClass(flutter.WIDGETS_LIBRARY_URI, 'State');
if (statefulWidgetClass == null || stateClass == null) {
return;
}
var stateType = stateClass.type.instantiate([widgetClassElement.type]);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(superclass), (builder) {
builder.writeType(statefulWidgetClass.type);
});
int replaceOffset = 0;
bool hasBuildMethod = false;
/// Replace code between [replaceOffset] and [replaceEnd] with
/// `createState()`, empty line, or nothing.
void replaceInterval(int replaceEnd,
{bool replaceWithEmptyLine: false,
bool hasEmptyLineBeforeCreateState: false,
bool hasEmptyLineAfterCreateState: true}) {
int replaceLength = replaceEnd - replaceOffset;
builder.addReplacement(
new SourceRange(replaceOffset, replaceLength),
(builder) {
if (hasBuildMethod) {
if (hasEmptyLineBeforeCreateState) {
builder.writeln();
}
builder.writeln(' @override');
builder.writeln(' $stateName createState() {');
builder.writeln(' return new $stateName();');
builder.writeln(' }');
if (hasEmptyLineAfterCreateState) {
builder.writeln();
}
hasBuildMethod = false;
} else if (replaceWithEmptyLine) {
builder.writeln();
}
},
);
replaceOffset = 0;
}
// Remove continuous ranges of lines of nodes being moved.
bool lastToRemoveIsField = false;
int endOfLastNodeToKeep = 0;
for (var node in widgetClass.members) {
if (nodesToMove.contains(node)) {
if (replaceOffset == 0) {
var linesRange = utils.getLinesRange(range.node(node));
replaceOffset = linesRange.offset;
}
if (node == buildMethod) {
hasBuildMethod = true;
}
lastToRemoveIsField = node is FieldDeclaration;
} else {
var linesRange = utils.getLinesRange(range.node(node));
endOfLastNodeToKeep = linesRange.end;
if (replaceOffset != 0) {
replaceInterval(linesRange.offset,
replaceWithEmptyLine:
lastToRemoveIsField && node is! FieldDeclaration);
}
}
}
// Remove nodes at the end of the widget class.
if (replaceOffset != 0) {
// Remove from the last node to keep, so remove empty lines.
if (endOfLastNodeToKeep != 0) {
replaceOffset = endOfLastNodeToKeep;
}
replaceInterval(widgetClass.rightBracket.offset,
hasEmptyLineBeforeCreateState: endOfLastNodeToKeep != 0,
hasEmptyLineAfterCreateState: false);
}
// Create the State subclass.
builder.addInsertion(widgetClass.end, (builder) {
builder.writeln();
builder.writeln();
builder.writeClassDeclaration(stateName, superclass: stateType,
membersWriter: () {
bool writeEmptyLine = false;
for (var member in nodesToMove) {
if (writeEmptyLine) {
builder.writeln();
}
String text = rewriteWidgetMemberReferences(member);
builder.write(text);
// Write empty lines between members, but not before the first.
writeEmptyLine = true;
}
});
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.FLUTTER_CONVERT_TO_STATEFUL_WIDGET);
}
Future<void> _addProposal_flutterMoveWidgetDown() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
var widget = flutter.identifyWidgetExpression(node);
if (widget == null) {
return;
}
AstNode parentList = widget.parent;
if (parentList is ListLiteral) {
List<Expression> parentElements = parentList.elements;
int index = parentElements.indexOf(widget);
if (index != parentElements.length - 1) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
Expression nextWidget = parentElements[index + 1];
var nextRange = range.node(nextWidget);
var nextText = utils.getRangeText(nextRange);
var widgetRange = range.node(widget);
var widgetText = utils.getRangeText(widgetRange);
builder.addSimpleReplacement(nextRange, widgetText);
builder.addSimpleReplacement(widgetRange, nextText);
int lengthDelta = nextRange.length - widgetRange.length;
int newWidgetOffset = nextRange.offset + lengthDelta;
changeBuilder.setSelection(new Position(file, newWidgetOffset));
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.FLUTTER_MOVE_DOWN);
}
}
}
Future<void> _addProposal_flutterMoveWidgetUp() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
var widget = flutter.identifyWidgetExpression(node);
if (widget == null) {
return;
}
AstNode parentList = widget.parent;
if (parentList is ListLiteral) {
List<Expression> parentElements = parentList.elements;
int index = parentElements.indexOf(widget);
if (index > 0) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
Expression previousWidget = parentElements[index - 1];
var previousRange = range.node(previousWidget);
var previousText = utils.getRangeText(previousRange);
var widgetRange = range.node(widget);
var widgetText = utils.getRangeText(widgetRange);
builder.addSimpleReplacement(previousRange, widgetText);
builder.addSimpleReplacement(widgetRange, previousText);
int newWidgetOffset = previousRange.offset;
changeBuilder.setSelection(new Position(file, newWidgetOffset));
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.FLUTTER_MOVE_UP);
}
}
}
Future<void> _addProposal_flutterRemoveWidget_multipleChildren() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
var widgetCreation = flutter.identifyNewExpression(node);
if (widgetCreation == null) {
return;
}
// Prepare the list of our children.
List<Expression> childrenExpressions;
{
var childrenArgument = flutter.findChildrenArgument(widgetCreation);
var childrenExpression = childrenArgument?.expression;
if (childrenExpression is ListLiteral &&
childrenExpression.elements.isNotEmpty) {
childrenExpressions = childrenExpression.elements;
} else {
return;
}
}
// We can inline the list of our children only into another list.
var widgetParentNode = widgetCreation.parent;
if (childrenExpressions.length > 1 && widgetParentNode is! ListLiteral) {
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
var firstChild = childrenExpressions.first;
var lastChild = childrenExpressions.last;
var childText = utils.getRangeText(range.startEnd(firstChild, lastChild));
var indentOld = utils.getLinePrefix(firstChild.offset);
var indentNew = utils.getLinePrefix(widgetCreation.offset);
childText = _replaceSourceIndent(childText, indentOld, indentNew);
builder.addSimpleReplacement(range.node(widgetCreation), childText);
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.FLUTTER_REMOVE_WIDGET);
}
Future<void> _addProposal_flutterRemoveWidget_singleChild() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
var widgetCreation = flutter.identifyNewExpression(node);
if (widgetCreation == null) {
return;
}
var childArgument = flutter.findChildArgument(widgetCreation);
if (childArgument == null) {
return;
}
// child: new ThisWidget(child: ourChild)
// children: [foo, new ThisWidget(child: ourChild), bar]
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
var childExpression = childArgument.expression;
var childText = utils.getNodeText(childExpression);
var indentOld = utils.getLinePrefix(childExpression.offset);
var indentNew = utils.getLinePrefix(widgetCreation.offset);
childText = _replaceSourceIndent(childText, indentOld, indentNew);
builder.addSimpleReplacement(range.node(widgetCreation), childText);
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.FLUTTER_REMOVE_WIDGET);
}
Future<void> _addProposal_flutterSwapWithChild() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
InstanceCreationExpression parent = flutter.identifyNewExpression(node);
if (!flutter.isWidgetCreation(parent)) {
_coverageMarker();
return;
}
NamedExpression childArgument = flutter.findChildArgument(parent);
if (childArgument?.expression is! InstanceCreationExpression ||
!flutter.isWidgetCreation(childArgument.expression)) {
_coverageMarker();
return;
}
InstanceCreationExpression child = childArgument.expression;
await _swapParentAndChild(
parent, child, DartAssistKind.FLUTTER_SWAP_WITH_CHILD);
}
Future<void> _addProposal_flutterSwapWithParent() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
InstanceCreationExpression child = flutter.identifyNewExpression(node);
if (!flutter.isWidgetCreation(child)) {
_coverageMarker();
return;
}
// NamedExpression (child:), ArgumentList, InstanceCreationExpression
AstNode expr = child.parent?.parent?.parent;
if (expr is! InstanceCreationExpression) {
_coverageMarker();
return;
}
InstanceCreationExpression parent = expr;
await _swapParentAndChild(
parent, child, DartAssistKind.FLUTTER_SWAP_WITH_PARENT);
}
Future<void> _addProposal_flutterWrapWidget() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
await _addProposal_flutterWrapWidgetImpl();
await _addProposal_flutterWrapWidgetImpl(
kind: DartAssistKind.FLUTTER_WRAP_CENTER,
parentLibraryUri: flutter.WIDGETS_LIBRARY_URI,
parentClassName: 'Center',
widgetValidator: (expr) {
return !flutter.isExactWidgetTypeCenter(expr.staticType);
});
await _addProposal_flutterWrapWidgetImpl(
kind: DartAssistKind.FLUTTER_WRAP_CONTAINER,
parentLibraryUri: flutter.WIDGETS_LIBRARY_URI,
parentClassName: 'Container',
widgetValidator: (expr) {
return !flutter.isExactWidgetTypeContainer(expr.staticType);
});
await _addProposal_flutterWrapWidgetImpl(
kind: DartAssistKind.FLUTTER_WRAP_PADDING,
parentLibraryUri: flutter.WIDGETS_LIBRARY_URI,
parentClassName: 'Padding',
leadingLines: ['padding: const EdgeInsets.all(8.0),'],
widgetValidator: (expr) {
return !flutter.isExactWidgetTypePadding(expr.staticType);
});
}
Future<void> _addProposal_flutterWrapWidgetImpl(
{AssistKind kind: DartAssistKind.FLUTTER_WRAP_GENERIC,
bool Function(Expression widgetExpr) widgetValidator,
String parentLibraryUri,
String parentClassName,
List<String> leadingLines: const []}) async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
Expression widgetExpr = flutter.identifyWidgetExpression(node);
if (widgetExpr == null) {
_coverageMarker();
return;
}
if (widgetValidator != null && !widgetValidator(widgetExpr)) {
_coverageMarker();
return;
}
String widgetSrc = utils.getNodeText(widgetExpr);
// If the wrapper class is specified, find its element.
ClassElement parentClassElement;
if (parentLibraryUri != null && parentClassName != null) {
parentClassElement =
await sessionHelper.getClass(parentLibraryUri, parentClassName);
if (parentClassElement == null) {
return;
}
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(widgetExpr), (DartEditBuilder builder) {
if (parentClassElement == null) {
builder.addSimpleLinkedEdit('WIDGET', 'widget');
} else {
builder.writeType(parentClassElement.type);
}
builder.write('(');
if (widgetSrc.contains(eol) || leadingLines.isNotEmpty) {
String indentOld = utils.getLinePrefix(widgetExpr.offset);
String indentNew = '$indentOld${utils.getIndent(1)}';
for (var leadingLine in leadingLines) {
builder.write(eol);
builder.write(indentNew);
builder.write(leadingLine);
}
builder.write(eol);
builder.write(indentNew);
widgetSrc = widgetSrc.replaceAll(
new RegExp("^$indentOld", multiLine: true), indentNew);
widgetSrc += ",$eol$indentOld";
}
if (parentClassElement == null) {
builder.addSimpleLinkedEdit('CHILD', 'child');
} else {
builder.write('child');
}
builder.write(': ');
builder.write(widgetSrc);
builder.write(')');
});
});
_addAssistFromBuilder(changeBuilder, kind);
}
Future<void> _addProposal_flutterWrapWidgets() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
var selectionRange = new SourceRange(selectionOffset, selectionLength);
var analyzer = new SelectionAnalyzer(selectionRange);
unit.accept(analyzer);
List<Expression> widgetExpressions = [];
if (analyzer.hasSelectedNodes) {
for (var selectedNode in analyzer.selectedNodes) {
if (!flutter.isWidgetExpression(selectedNode)) {
return;
}
widgetExpressions.add(selectedNode);
}
} else {
var widget = flutter.identifyWidgetExpression(analyzer.coveringNode);
if (widget != null) {
widgetExpressions.add(widget);
}
}
if (widgetExpressions.isEmpty) {
return;
}
var firstWidget = widgetExpressions.first;
var lastWidget = widgetExpressions.last;
var selectedRange = range.startEnd(firstWidget, lastWidget);
String src = utils.getRangeText(selectedRange);
Future<void> addAssist(
{@required AssistKind kind,
@required String parentLibraryUri,
@required String parentClassName}) async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
ClassElement parentClassElement =
await sessionHelper.getClass(parentLibraryUri, parentClassName);
ClassElement widgetClassElement =
await sessionHelper.getClass(flutter.WIDGETS_LIBRARY_URI, 'Widget');
if (parentClassElement == null || widgetClassElement == null) {
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(selectedRange, (DartEditBuilder builder) {
builder.writeType(parentClassElement.type);
builder.write('(');
String indentOld = utils.getLinePrefix(firstWidget.offset);
String indentNew1 = indentOld + utils.getIndent(1);
String indentNew2 = indentOld + utils.getIndent(2);
builder.write(eol);
builder.write(indentNew1);
builder.write('children: <');
builder.writeType(widgetClassElement.type);
builder.write('>[');
builder.write(eol);
String newSrc = _replaceSourceIndent(src, indentOld, indentNew2);
builder.write(indentNew2);
builder.write(newSrc);
builder.write(',');
builder.write(eol);
builder.write(indentNew1);
builder.write('],');
builder.write(eol);
builder.write(indentOld);
builder.write(')');
});
});
_addAssistFromBuilder(changeBuilder, kind);
}
await addAssist(
kind: DartAssistKind.FLUTTER_WRAP_COLUMN,
parentLibraryUri: flutter.WIDGETS_LIBRARY_URI,
parentClassName: 'Column');
await addAssist(
kind: DartAssistKind.FLUTTER_WRAP_ROW,
parentLibraryUri: flutter.WIDGETS_LIBRARY_URI,
parentClassName: 'Row');
}
Future<void> _addProposal_importAddShow() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare ImportDirective
ImportDirective importDirective =
node.getAncestor((node) => node is ImportDirective);
if (importDirective == null) {
_coverageMarker();
return;
}
// there should be no existing combinators
if (importDirective.combinators.isNotEmpty) {
_coverageMarker();
return;
}
// prepare whole import namespace
ImportElement importElement = importDirective.element;
if (importElement == null) {
_coverageMarker();
return;
}
Map<String, Element> namespace = getImportNamespace(importElement);
// prepare names of referenced elements (from this import)
SplayTreeSet<String> referencedNames = new SplayTreeSet<String>();
_SimpleIdentifierRecursiveAstVisitor visitor =
new _SimpleIdentifierRecursiveAstVisitor((SimpleIdentifier node) {
Element element = node.staticElement;
if (element != null && namespace[node.name] == element) {
referencedNames.add(element.displayName);
}
});
unit.accept(visitor);
// ignore if unused
if (referencedNames.isEmpty) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
String showCombinator = ' show ${referencedNames.join(', ')}';
builder.addSimpleInsertion(importDirective.end - 1, showCombinator);
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.IMPORT_ADD_SHOW);
}
Future<void> _addProposal_introduceLocalTestedType() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
AstNode node = this.node;
if (node is IfStatement) {
node = (node as IfStatement).condition;
} else if (node is WhileStatement) {
node = (node as WhileStatement).condition;
}
// prepare IsExpression
if (node is! IsExpression) {
_coverageMarker();
return;
}
IsExpression isExpression = node;
DartType castType = isExpression.type.type;
String castTypeCode = _getNodeText(isExpression.type);
// prepare environment
String indent = utils.getIndent(1);
String prefix;
Block targetBlock;
{
Statement statement = node.getAncestor((n) => n is Statement);
if (statement is IfStatement && statement.thenStatement is Block) {
targetBlock = statement.thenStatement;
} else if (statement is WhileStatement && statement.body is Block) {
targetBlock = statement.body;
} else {
_coverageMarker();
return;
}
prefix = utils.getNodePrefix(statement);
}
// prepare location
int offset;
String statementPrefix;
if (isExpression.notOperator == null) {
offset = targetBlock.leftBracket.end;
statementPrefix = indent;
} else {
offset = targetBlock.rightBracket.end;
statementPrefix = '';
}
// prepare excluded names
Set<String> excluded = new Set<String>();
ScopedNameFinder scopedNameFinder = new ScopedNameFinder(offset);
isExpression.accept(scopedNameFinder);
excluded.addAll(scopedNameFinder.locals.keys.toSet());
// name(s)
List<String> suggestions =
getVariableNameSuggestionsForExpression(castType, null, excluded);
if (suggestions.isNotEmpty) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addInsertion(offset, (DartEditBuilder builder) {
builder.write(eol + prefix + statementPrefix);
builder.write(castTypeCode);
builder.write(' ');
builder.addSimpleLinkedEdit('NAME', suggestions[0],
kind: LinkedEditSuggestionKind.VARIABLE,
suggestions: suggestions);
builder.write(' = ');
builder.write(_getNodeText(isExpression.expression));
builder.write(';');
builder.selectHere();
});
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.INTRODUCE_LOCAL_CAST_TYPE);
}
}
Future<void> _addProposal_invertIf() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
if (node is! IfStatement) {
return;
}
IfStatement ifStatement = node as IfStatement;
Expression condition = ifStatement.condition;
// should have both "then" and "else"
Statement thenStatement = ifStatement.thenStatement;
Statement elseStatement = ifStatement.elseStatement;
if (thenStatement == null || elseStatement == null) {
return;
}
// prepare source
String invertedCondition = utils.invertCondition(condition);
String thenSource = _getNodeText(thenStatement);
String elseSource = _getNodeText(elseStatement);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(range.node(condition), invertedCondition);
builder.addSimpleReplacement(range.node(thenStatement), elseSource);
builder.addSimpleReplacement(range.node(elseStatement), thenSource);
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.INVERT_IF_STATEMENT);
}
Future<void> _addProposal_joinIfStatementInner() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// climb up condition to the (supposedly) "if" statement
AstNode node = this.node;
while (node is Expression) {
node = node.parent;
}
// prepare target "if" statement
if (node is! IfStatement) {
_coverageMarker();
return;
}
IfStatement targetIfStatement = node as IfStatement;
if (targetIfStatement.elseStatement != null) {
_coverageMarker();
return;
}
// prepare inner "if" statement
Statement targetThenStatement = targetIfStatement.thenStatement;
Statement innerStatement = getSingleStatement(targetThenStatement);
if (innerStatement is! IfStatement) {
_coverageMarker();
return;
}
IfStatement innerIfStatement = innerStatement as IfStatement;
if (innerIfStatement.elseStatement != null) {
_coverageMarker();
return;
}
// prepare environment
String prefix = utils.getNodePrefix(targetIfStatement);
// merge conditions
String condition;
{
Expression targetCondition = targetIfStatement.condition;
Expression innerCondition = innerIfStatement.condition;
String targetConditionSource = _getNodeText(targetCondition);
String innerConditionSource = _getNodeText(innerCondition);
if (_shouldWrapParenthesisBeforeAnd(targetCondition)) {
targetConditionSource = '($targetConditionSource)';
}
if (_shouldWrapParenthesisBeforeAnd(innerCondition)) {
innerConditionSource = '($innerConditionSource)';
}
condition = '$targetConditionSource && $innerConditionSource';
}
// replace target "if" statement
Statement innerThenStatement = innerIfStatement.thenStatement;
List<Statement> innerThenStatements = getStatements(innerThenStatement);
SourceRange lineRanges = utils.getLinesRangeStatements(innerThenStatements);
String oldSource = utils.getRangeText(lineRanges);
String newSource = utils.indentSourceLeftRight(oldSource);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(range.node(targetIfStatement),
'if ($condition) {$eol$newSource$prefix}');
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.JOIN_IF_WITH_INNER);
}
Future<void> _addProposal_joinIfStatementOuter() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// climb up condition to the (supposedly) "if" statement
AstNode node = this.node;
while (node is Expression) {
node = node.parent;
}
// prepare target "if" statement
if (node is! IfStatement) {
_coverageMarker();
return;
}
IfStatement targetIfStatement = node as IfStatement;
if (targetIfStatement.elseStatement != null) {
_coverageMarker();
return;
}
// prepare outer "if" statement
AstNode parent = targetIfStatement.parent;
if (parent is Block) {
if ((parent as Block).statements.length != 1) {
_coverageMarker();
return;
}
parent = parent.parent;
}
if (parent is! IfStatement) {
_coverageMarker();
return;
}
IfStatement outerIfStatement = parent as IfStatement;
if (outerIfStatement.elseStatement != null) {
_coverageMarker();
return;
}
// prepare environment
String prefix = utils.getNodePrefix(outerIfStatement);
// merge conditions
Expression targetCondition = targetIfStatement.condition;
Expression outerCondition = outerIfStatement.condition;
String targetConditionSource = _getNodeText(targetCondition);
String outerConditionSource = _getNodeText(outerCondition);
if (_shouldWrapParenthesisBeforeAnd(targetCondition)) {
targetConditionSource = '($targetConditionSource)';
}
if (_shouldWrapParenthesisBeforeAnd(outerCondition)) {
outerConditionSource = '($outerConditionSource)';
}
String condition = '$outerConditionSource && $targetConditionSource';
// replace outer "if" statement
Statement targetThenStatement = targetIfStatement.thenStatement;
List<Statement> targetThenStatements = getStatements(targetThenStatement);
SourceRange lineRanges =
utils.getLinesRangeStatements(targetThenStatements);
String oldSource = utils.getRangeText(lineRanges);
String newSource = utils.indentSourceLeftRight(oldSource);
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(range.node(outerIfStatement),
'if ($condition) {$eol$newSource$prefix}');
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.JOIN_IF_WITH_OUTER);
}
Future<void> _addProposal_joinVariableDeclaration_onAssignment() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// check that node is LHS in assignment
if (node is SimpleIdentifier &&
node.parent is AssignmentExpression &&
(node.parent as AssignmentExpression).leftHandSide == node &&
node.parent.parent is ExpressionStatement) {
} else {
_coverageMarker();
return;
}
AssignmentExpression assignExpression = node.parent as AssignmentExpression;
// check that binary expression is assignment
if (assignExpression.operator.type != TokenType.EQ) {
_coverageMarker();
return;
}
// prepare "declaration" statement
Element element = (node as SimpleIdentifier).staticElement;
if (element == null) {
_coverageMarker();
return;
}
int declOffset = element.nameOffset;
AstNode declNode = new NodeLocator(declOffset).searchWithin(unit);
if (declNode != null &&
declNode.parent is VariableDeclaration &&
(declNode.parent as VariableDeclaration).name == declNode &&
declNode.parent.parent is VariableDeclarationList &&
declNode.parent.parent.parent is VariableDeclarationStatement) {
} else {
_coverageMarker();
return;
}
VariableDeclaration decl = declNode.parent as VariableDeclaration;
VariableDeclarationStatement declStatement =
decl.parent.parent as VariableDeclarationStatement;
// may be has initializer
if (decl.initializer != null) {
_coverageMarker();
return;
}
// check that "declaration" statement declared only one variable
if (declStatement.variables.variables.length != 1) {
_coverageMarker();
return;
}
// check that the "declaration" and "assignment" statements are
// parts of the same Block
ExpressionStatement assignStatement =
node.parent.parent as ExpressionStatement;
if (assignStatement.parent is Block &&
assignStatement.parent == declStatement.parent) {
} else {
_coverageMarker();
return;
}
Block block = assignStatement.parent as Block;
// check that "declaration" and "assignment" statements are adjacent
List<Statement> statements = block.statements;
if (statements.indexOf(assignStatement) ==
statements.indexOf(declStatement) + 1) {
} else {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(
range.endStart(declNode, assignExpression.operator), ' ');
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.JOIN_VARIABLE_DECLARATION);
}
Future<void> _addProposal_joinVariableDeclaration_onDeclaration() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare enclosing VariableDeclarationList
VariableDeclarationList declList =
node.getAncestor((node) => node is VariableDeclarationList);
if (declList != null && declList.variables.length == 1) {
} else {
_coverageMarker();
return;
}
VariableDeclaration decl = declList.variables[0];
// already initialized
if (decl.initializer != null) {
_coverageMarker();
return;
}
// prepare VariableDeclarationStatement in Block
if (declList.parent is VariableDeclarationStatement &&
declList.parent.parent is Block) {
} else {
_coverageMarker();
return;
}
VariableDeclarationStatement declStatement =
declList.parent as VariableDeclarationStatement;
Block block = declStatement.parent as Block;
List<Statement> statements = block.statements;
// prepare assignment
AssignmentExpression assignExpression;
{
// declaration should not be last Statement
int declIndex = statements.indexOf(declStatement);
if (declIndex < statements.length - 1) {
} else {
_coverageMarker();
return;
}
// next Statement should be assignment
Statement assignStatement = statements[declIndex + 1];
if (assignStatement is ExpressionStatement) {
} else {
_coverageMarker();
return;
}
ExpressionStatement expressionStatement =
assignStatement as ExpressionStatement;
// expression should be assignment
if (expressionStatement.expression is AssignmentExpression) {
} else {
_coverageMarker();
return;
}
assignExpression = expressionStatement.expression as AssignmentExpression;
}
// check that pure assignment
if (assignExpression.operator.type != TokenType.EQ) {
_coverageMarker();
return;
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addSimpleReplacement(
range.endStart(decl.name, assignExpression.operator), ' ');
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.JOIN_VARIABLE_DECLARATION);
}
Future<void> _addProposal_removeTypeAnnotation() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
VariableDeclarationList declarationList =
node.getAncestor((n) => n is VariableDeclarationList);
if (declarationList == null) {
_coverageMarker();
return;
}
// we need a type
TypeAnnotation typeNode = declarationList.type;
if (typeNode == null) {
_coverageMarker();
return;
}
// ignore if an incomplete variable declaration
if (declarationList.variables.length == 1 &&
declarationList.variables[0].name.isSynthetic) {
_coverageMarker();
return;
}
// must be not after the name of the variable
VariableDeclaration firstVariable = declarationList.variables[0];
if (selectionOffset > firstVariable.name.end) {
_coverageMarker();
return;
}
// The variable must have an initializer, otherwise there is no other
// source for its type.
if (firstVariable.initializer == null) {
_coverageMarker();
return;
}
Token keyword = declarationList.keyword;
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
SourceRange typeRange = range.startStart(typeNode, firstVariable);
if (keyword != null && keyword.lexeme != 'var') {
builder.addSimpleReplacement(typeRange, '');
} else {
builder.addSimpleReplacement(typeRange, 'var ');
}
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.REMOVE_TYPE_ANNOTATION);
}
Future<void> _addProposal_reparentFlutterList() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
if (node is! ListLiteral) {
return;
}
if ((node as ListLiteral).elements.any((Expression exp) =>
!(exp is InstanceCreationExpression &&
flutter.isWidgetCreation(exp)))) {
_coverageMarker();
return;
}
String literalSrc = utils.getNodeText(node);
int newlineIdx = literalSrc.lastIndexOf(eol);
if (newlineIdx < 0 || newlineIdx == literalSrc.length - 1) {
_coverageMarker();
return; // Lists need to be in multi-line format already.
}
String indentOld = utils.getLinePrefix(node.offset + 1 + newlineIdx);
String indentArg = '$indentOld${utils.getIndent(1)}';
String indentList = '$indentOld${utils.getIndent(2)}';
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
builder.addReplacement(range.node(node), (DartEditBuilder builder) {
builder.write('[');
builder.write(eol);
builder.write(indentArg);
builder.write('new ');
builder.addSimpleLinkedEdit('WIDGET', 'widget');
builder.write('(');
builder.write(eol);
builder.write(indentList);
// Linked editing not needed since arg is always a list.
builder.write('children: ');
builder.write(literalSrc.replaceAll(
new RegExp("^$indentOld", multiLine: true), "$indentList"));
builder.write(',');
builder.write(eol);
builder.write(indentArg);
builder.write('),');
builder.write(eol);
builder.write(indentOld);
builder.write(']');
});
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.FLUTTER_WRAP_GENERIC);
}
Future<void> _addProposal_replaceConditionalWithIfElse() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
ConditionalExpression conditional = null;
// may be on Statement with Conditional
Statement statement = node.getAncestor((node) => node is Statement);
if (statement == null) {
_coverageMarker();
return;
}
// variable declaration
bool inVariable = false;
if (statement is VariableDeclarationStatement) {
VariableDeclarationStatement variableStatement = statement;
for (VariableDeclaration variable
in variableStatement.variables.variables) {
if (variable.initializer is ConditionalExpression) {
conditional = variable.initializer as ConditionalExpression;
inVariable = true;
break;
}
}
}
// assignment
bool inAssignment = false;
if (statement is ExpressionStatement) {
ExpressionStatement exprStmt = statement;
if (exprStmt.expression is AssignmentExpression) {
AssignmentExpression assignment =
exprStmt.expression as AssignmentExpression;
if (assignment.operator.type == TokenType.EQ &&
assignment.rightHandSide is ConditionalExpression) {
conditional = assignment.rightHandSide as ConditionalExpression;
inAssignment = true;
}
}
}
// return
bool inReturn = false;
if (statement is ReturnStatement) {
ReturnStatement returnStatement = statement;
if (returnStatement.expression is ConditionalExpression) {
conditional = returnStatement.expression as ConditionalExpression;
inReturn = true;
}
}
// prepare environment
String indent = utils.getIndent(1);
String prefix = utils.getNodePrefix(statement);
if (inVariable || inAssignment || inReturn) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// Type v = Conditional;
if (inVariable) {
VariableDeclaration variable =
conditional.parent as VariableDeclaration;
builder.addDeletion(range.endEnd(variable.name, conditional));
String conditionSrc = _getNodeText(conditional.condition);
String thenSrc = _getNodeText(conditional.thenExpression);
String elseSrc = _getNodeText(conditional.elseExpression);
String name = variable.name.name;
String src = eol;
src += prefix + 'if ($conditionSrc) {' + eol;
src += prefix + indent + '$name = $thenSrc;' + eol;
src += prefix + '} else {' + eol;
src += prefix + indent + '$name = $elseSrc;' + eol;
src += prefix + '}';
builder.addSimpleReplacement(range.endLength(statement, 0), src);
}
// v = Conditional;
if (inAssignment) {
AssignmentExpression assignment =
conditional.parent as AssignmentExpression;
Expression leftSide = assignment.leftHandSide;
String conditionSrc = _getNodeText(conditional.condition);
String thenSrc = _getNodeText(conditional.thenExpression);
String elseSrc = _getNodeText(conditional.elseExpression);
String name = _getNodeText(leftSide);
String src = '';
src += 'if ($conditionSrc) {' + eol;
src += prefix + indent + '$name = $thenSrc;' + eol;
src += prefix + '} else {' + eol;
src += prefix + indent + '$name = $elseSrc;' + eol;
src += prefix + '}';
builder.addSimpleReplacement(range.node(statement), src);
}
// return Conditional;
if (inReturn) {
String conditionSrc = _getNodeText(conditional.condition);
String thenSrc = _getNodeText(conditional.thenExpression);
String elseSrc = _getNodeText(conditional.elseExpression);
String src = '';
src += 'if ($conditionSrc) {' + eol;
src += prefix + indent + 'return $thenSrc;' + eol;
src += prefix + '} else {' + eol;
src += prefix + indent + 'return $elseSrc;' + eol;
src += prefix + '}';
builder.addSimpleReplacement(range.node(statement), src);
}
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.REPLACE_CONDITIONAL_WITH_IF_ELSE);
}
}
Future<void> _addProposal_replaceIfElseWithConditional() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// should be "if"
if (node is! IfStatement) {
_coverageMarker();
return;
}
IfStatement ifStatement = node as IfStatement;
// single then/else statements
Statement thenStatement = getSingleStatement(ifStatement.thenStatement);
Statement elseStatement = getSingleStatement(ifStatement.elseStatement);
if (thenStatement == null || elseStatement == null) {
_coverageMarker();
return;
}
Expression thenExpression = null;
Expression elseExpression = null;
bool hasReturnStatements = false;
if (thenStatement is ReturnStatement && elseStatement is ReturnStatement) {
hasReturnStatements = true;
thenExpression = thenStatement.expression;
elseExpression = elseStatement.expression;
}
bool hasExpressionStatements = false;
if (thenStatement is ExpressionStatement &&
elseStatement is ExpressionStatement) {
if (thenStatement.expression is AssignmentExpression &&
elseStatement.expression is AssignmentExpression) {
hasExpressionStatements = true;
thenExpression = thenStatement.expression;
elseExpression = elseStatement.expression;
}
}
if (hasReturnStatements || hasExpressionStatements) {
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// returns
if (hasReturnStatements) {
String conditionSrc = _getNodeText(ifStatement.condition);
String theSrc = _getNodeText(thenExpression);
String elseSrc = _getNodeText(elseExpression);
builder.addSimpleReplacement(range.node(ifStatement),
'return $conditionSrc ? $theSrc : $elseSrc;');
}
// assignments -> v = Conditional;
if (hasExpressionStatements) {
AssignmentExpression thenAssignment = thenExpression;
AssignmentExpression elseAssignment = elseExpression;
String thenTarget = _getNodeText(thenAssignment.leftHandSide);
String elseTarget = _getNodeText(elseAssignment.leftHandSide);
if (thenAssignment.operator.type == TokenType.EQ &&
elseAssignment.operator.type == TokenType.EQ &&
thenTarget == elseTarget) {
String conditionSrc = _getNodeText(ifStatement.condition);
String theSrc = _getNodeText(thenAssignment.rightHandSide);
String elseSrc = _getNodeText(elseAssignment.rightHandSide);
builder.addSimpleReplacement(range.node(ifStatement),
'$thenTarget = $conditionSrc ? $theSrc : $elseSrc;');
}
}
});
_addAssistFromBuilder(
changeBuilder, DartAssistKind.REPLACE_IF_ELSE_WITH_CONDITIONAL);
}
}
Future<void> _addProposal_splitAndCondition() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// check that user invokes quick assist on binary expression
if (node is! BinaryExpression) {
_coverageMarker();
return;
}
BinaryExpression binaryExpression = node as BinaryExpression;
// prepare operator position
if (!_isOperatorSelected(
binaryExpression, selectionOffset, selectionLength)) {
_coverageMarker();
return;
}
// should be &&
if (binaryExpression.operator.type != TokenType.AMPERSAND_AMPERSAND) {
_coverageMarker();
return;
}
// prepare "if"
Statement statement = node.getAncestor((node) => node is Statement);
if (statement is! IfStatement) {
_coverageMarker();
return;
}
IfStatement ifStatement = statement as IfStatement;
// no support "else"
if (ifStatement.elseStatement != null) {
_coverageMarker();
return;
}
// check that binary expression is part of first level && condition of "if"
BinaryExpression condition = binaryExpression;
while (condition.parent is BinaryExpression &&
(condition.parent as BinaryExpression).operator.type ==
TokenType.AMPERSAND_AMPERSAND) {
condition = condition.parent as BinaryExpression;
}
if (ifStatement.condition != condition) {
_coverageMarker();
return;
}
// prepare environment
String prefix = utils.getNodePrefix(ifStatement);
String indent = utils.getIndent(1);
// prepare "rightCondition"
String rightConditionSource;
{
SourceRange rightConditionRange =
range.startEnd(binaryExpression.rightOperand, condition);
rightConditionSource = _getRangeText(rightConditionRange);
}
DartChangeBuilder changeBuilder = new DartChangeBuilder(session);
await changeBuilder.addFileEdit(file, (DartFileEditBuilder builder) {
// remove "&& rightCondition"
builder
.addDeletion(range.endEnd(binaryExpression.leftOperand, condition));
// update "then" statement
Statement thenStatement = ifStatement.thenStatement;
if (thenStatement is Block) {
Block thenBlock = thenStatement;
SourceRange thenBlockRange = range.node(thenBlock);
// insert inner "if" with right part of "condition"
int thenBlockInsideOffset = thenBlockRange.offset + 1;
builder.addSimpleInsertion(thenBlockInsideOffset,
'$eol$prefix${indent}if ($rightConditionSource) {');
// insert closing "}" for inner "if"
int thenBlockEnd = thenBlockRange.end;
// insert before outer "then" block "}"
builder.addSimpleInsertion(thenBlockEnd - 1, '$indent}$eol$prefix');
} else {
// insert inner "if" with right part of "condition"
String source = '$eol$prefix${indent}if ($rightConditionSource)';
builder.addSimpleInsertion(
ifStatement.rightParenthesis.offset + 1, source);
}
// indent "then" statements to correspond inner "if"
{
List<Statement> thenStatements = getStatements(thenStatement);
SourceRange linesRange = utils.getLinesRangeStatements(thenStatements);
String thenIndentOld = '$prefix$indent';
String thenIndentNew = '$thenIndentOld$indent';
builder.addSimpleReplacement(
linesRange,
utils.replaceSourceRangeIndent(
linesRange, thenIndentOld, thenIndentNew));
}
});
_addAssistFromBuilder(changeBuilder, DartAssistKind.SPLIT_AND_CONDITION);
}
Future<void> _addProposal_splitVariableDeclaration() async {
// TODO(brianwilkerson) Determine whether this await is necessary.
await null;
// prepare DartVariableStatement, should be part of Block
VariableDeclarationStatement statement =
node.getAncestor((node) => node is VariableDeclarationStatement);
if (statement != null && statement.parent is Block) {
} else {
_coverageMarker