protected override CodeAction GetAction(RefactoringContext context, IfElseStatement ifElseStatement)
{
if (!ifElseStatement.IfToken.Contains(context.Location))
{
return(null);
}
Expression c, e1, e2;
AstNode rs;
if (!ConvertIfStatementToReturnStatementAction.GetMatch(ifElseStatement, out c, out e1, out e2, out rs))
{
return(null);
}
return(new CodeAction(
context.TranslateString("Replace with 'return'"),
script => {
script.Replace(ifElseStatement, new ReturnStatement(
CreateCondition(c, e1, e2)
));
if (rs != null)
{
script.Remove(rs);
}
},
ifElseStatement
));
}
public override void VisitIfElseStatement(IfElseStatement ifElseStatement)
{
base.VisitIfElseStatement(ifElseStatement);
if (ifElseStatement.Parent is IfElseStatement)
{
return;
}
Expression c, e1, e2;
AstNode rs;
if (!ConvertIfStatementToReturnStatementAction.GetMatch(ifElseStatement, out c, out e1, out e2, out rs))
{
return;
}
if (ConvertIfStatementToConditionalTernaryExpressionIssue.IsComplexCondition(c) ||
ConvertIfStatementToConditionalTernaryExpressionIssue.IsComplexExpression(e1) ||
ConvertIfStatementToConditionalTernaryExpressionIssue.IsComplexExpression(e2))
{
return;
}
AddIssue(new CodeIssue(
ifElseStatement.IfToken,
ctx.TranslateString("Convert to 'return' statement")
)
{
IssueMarker = IssueMarker.DottedLine, ActionProvider = { typeof(ConvertIfStatementToReturnStatementAction) }
});
}