Skip to content

Commit dd7c4d1

Browse files
committed
I didn't need to DI the ArgListContext through two classes...
1 parent a0a7d1d commit dd7c4d1

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

RetailCoder.VBE/Inspections/ProcedureShouldBeFunctionInspection.cs

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,7 @@ public IEnumerable<CodeInspectionResultBase> GetInspectionResults(RubberduckPars
2828
new QualifiedContext<VBAParser.ArgListContext>(context.ModuleName,
2929
context.Context as VBAParser.ArgListContext),
3030
new QualifiedContext<VBAParser.SubStmtContext>(context.ModuleName,
31-
context.Context.Parent as VBAParser.SubStmtContext),
32-
new QualifiedContext<VBAParser.ArgContext>(context.ModuleName,
33-
((VBAParser.ArgListContext) context.Context).arg()
34-
.First(a => a.BYREF() != null || (a.BYREF() == null && a.BYVAL() == null)))));
31+
context.Context.Parent as VBAParser.SubStmtContext)));
3532
}
3633
}
3734
}

RetailCoder.VBE/Inspections/ProcedureShouldBeFunctionInspectionResult.cs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,15 @@ public class ProcedureShouldBeFunctionInspectionResult : CodeInspectionResultBas
1212
{
1313
private readonly IEnumerable<CodeInspectionQuickFix> _quickFixes;
1414

15-
public ProcedureShouldBeFunctionInspectionResult(IInspection inspection, RubberduckParserState state, QualifiedContext<VBAParser.ArgListContext> argListQualifiedContext, QualifiedContext<VBAParser.SubStmtContext> subStmtQualifiedContext, QualifiedContext<VBAParser.ArgContext> argQualifiedContext)
15+
public ProcedureShouldBeFunctionInspectionResult(IInspection inspection, RubberduckParserState state, QualifiedContext<VBAParser.ArgListContext> argListQualifiedContext, QualifiedContext<VBAParser.SubStmtContext> subStmtQualifiedContext)
1616
: base(inspection,
1717
string.Format(inspection.Description, subStmtQualifiedContext.Context.ambiguousIdentifier().GetText()),
1818
subStmtQualifiedContext.ModuleName,
1919
subStmtQualifiedContext.Context.ambiguousIdentifier())
2020
{
2121
_quickFixes = new[]
2222
{
23-
new ChangeProcedureToFunction(state, argListQualifiedContext, subStmtQualifiedContext, argQualifiedContext, QualifiedSelection),
23+
new ChangeProcedureToFunction(state, argListQualifiedContext, subStmtQualifiedContext, QualifiedSelection),
2424
};
2525
}
2626

@@ -37,14 +37,15 @@ public class ChangeProcedureToFunction : CodeInspectionQuickFix
3737
public ChangeProcedureToFunction(RubberduckParserState state,
3838
QualifiedContext<VBAParser.ArgListContext> argListQualifiedContext,
3939
QualifiedContext<VBAParser.SubStmtContext> subStmtQualifiedContext,
40-
QualifiedContext<VBAParser.ArgContext> argQualifiedContext,
4140
QualifiedSelection selection)
4241
: base(subStmtQualifiedContext.Context, selection, InspectionsUI.ProcedureShouldBeFunctionInspectionQuickFix)
4342
{
4443
_state = state;
4544
_argListQualifiedContext = argListQualifiedContext;
4645
_subStmtQualifiedContext = subStmtQualifiedContext;
47-
_argQualifiedContext = argQualifiedContext;
46+
_argQualifiedContext = new QualifiedContext<VBAParser.ArgContext>(_argListQualifiedContext.ModuleName,
47+
_argListQualifiedContext.Context.arg()
48+
.First(a => a.BYREF() != null || (a.BYREF() == null && a.BYVAL() == null)));
4849
}
4950

5051
public override void Fix()

0 commit comments

Comments
 (0)