public override void ApplyQuickFix(ITokenNode tokenNode)
{
var logicalExpression = tokenNode.Parent as IVBBinaryExpression;
if (logicalExpression == null)
{
return;
}
string shortCircuitOperator = null;
if (tokenNode.GetTokenType() == VBTokenType.OR_KEYWORD)
{
shortCircuitOperator = VBTokenType.ORELSE_KEYWORD.TokenRepresentation;
}
else if (tokenNode.GetTokenType() == VBTokenType.AND_KEYWORD)
{
shortCircuitOperator = VBTokenType.ANDALSO_KEYWORD.TokenRepresentation;
}
if (shortCircuitOperator == null)
{
return;
}
logicalExpression.GetPsiServices().Transactions.Execute("UseShortCircuitOperatorsQuickFix",
() => {
var elementFactory = VBElementFactory.GetInstance(tokenNode.GetPsiModule());
var newlogicalExpression = elementFactory.CreateExpression("$0 " + shortCircuitOperator + " $1",
logicalExpression.LeftExpr, logicalExpression.RightExpr);
logicalExpression.ReplaceByExtension(newlogicalExpression);
});
}
public void MakeCallShared(IReference reference)
{
var referenceExpression1 = reference.GetTreeNode() as IReferenceExpression;
if (referenceExpression1 == null || !reference.Resolve().Result.IsExtensionMethod())
{
return;
}
var byExpression = IndexExpressionNavigator.GetByExpression(referenceExpression1);
if (byExpression == null)
{
return;
}
var substitution = reference.Resolve().Result.Substitution;
var referenceExpression2 = byExpression.Expression as IReferenceExpression;
if (referenceExpression2 == null)
{
return;
}
var instance = VBElementFactory.GetInstance(referenceExpression1.GetPsiModule());
var vbExpression = referenceExpression2.QualifierExpression ?? instance.CreateExpression("me", new object[0]);
byExpression.AddArgumentAfter(instance.CreateArgument(vbExpression), null);
//byExpression.SetExpression(instance.CreateReferenceExpression(this.Executer.Method.ShortName, new object[0]));
//((IReferenceExpression)byExpression.Expression).Reference.BindTo((IDeclaredElement)this.Executer.Method);
//if (((IReferenceExpression)byExpression.Expression).Reference.BindTo((IDeclaredElement)this.Workflow.Method, substitution).CheckResolveResult() == ResolveErrorType.OK) return;
//this.Driver.AddConflict((IConflict)ReferenceConflict.CreateError(reference, "{0} can not be transformed correctly"));
}
public override void ApplyQuickFix(ILogicalNotExpression logicalNotExpression)
{
logicalNotExpression.GetPsiServices().Transactions.Execute("UseIsNotOperatorQuickFix",
() => {
// WriteLock is used in our call to ReplaceByExtension
//using (WriteLockCookie.Create()) {
// Confirm IsExpression exists under LogicalNotExpression
var isExpression = logicalNotExpression.Children <IIsExpression>().FirstOrDefault();
if ((isExpression) == null)
{
return;
}
// Transform ILogicalNotExpression into IIsNotExpression
var elementFactory = VBElementFactory.GetInstance(logicalNotExpression.GetPsiModule());
var newIsNotExpression = elementFactory.CreateExpression("$0 IsNot $1", isExpression.LeftExpr, isExpression.RightExpr);
logicalNotExpression.ReplaceByExtension(newIsNotExpression);
});
}
public void MakeCallExtension(IReference reference)
{
var treeNode = reference.GetTreeNode();
if (!treeNode.IsVB9Supported())
{
return;
}
var referenceExpression = treeNode as IReferenceExpression;
if (referenceExpression == null)
{
return;
}
IIndexExpression byExpression = IndexExpressionNavigator.GetByExpression(referenceExpression);
if (byExpression == null)
{
return;
}
IVBArgument vbArgument = null;
int index = 0;
using (var enumerator = byExpression.ArgumentList.Arguments.GetEnumerator()) {
while (enumerator.MoveNext())
{
var current = enumerator.Current;
var matchingParameter = current.GetMatchingParameter();
if (matchingParameter != null && matchingParameter.Element.IndexOf() == 0)
{
vbArgument = current;
break;
}
++index;
}
}
if (vbArgument == null)
{
return;
}
IVBExpression vbExpression = null;
var expressionArgument = vbArgument as IExpressionArgument;
if (expressionArgument != null)
{
vbExpression = expressionArgument.Expression;
}
if (vbExpression == null)
{
return;
}
var indexExpression1 =
(IIndexExpression)VBElementFactory.GetInstance(treeNode.GetPsiModule())
.CreateExpression("$0." + reference.GetName() + "()", new object[] { vbExpression });
indexExpression1.SetArgumentList(byExpression.ArgumentList);
indexExpression1.RemoveArgument(indexExpression1.Arguments[index]);
byExpression.ReplaceBy(indexExpression1);
}