private void ConvertFunction(IInspectionResult result, VBAParser.FunctionStmtContext functionContext)
{
var rewriter = _state.GetRewriter(result.Target);
var asTypeContext = ParserRuleContextHelper.GetChild <VBAParser.AsTypeClauseContext>(functionContext);
if (asTypeContext != null)
{
rewriter.Remove(asTypeContext);
rewriter.Remove(functionContext.children.ElementAt(functionContext.children.IndexOf(asTypeContext) - 1) as ParserRuleContext);
}
if (result.Target.TypeHint != null)
{
rewriter.Remove(ParserRuleContextHelper.GetDescendent <VBAParser.TypeHintContext>(functionContext));
}
rewriter.Replace(functionContext.FUNCTION(), Tokens.Sub);
rewriter.Replace(functionContext.END_FUNCTION(), "End Sub");
foreach (var returnStatement in GetReturnStatements(result.Target))
{
rewriter.Remove(returnStatement);
}
}
public ProcedureNode(VBAParser.FunctionStmtContext context, string scope, string localScope)
: this(context, scope, localScope, VBProcedureKind.Function, context.visibility(), context.ambiguousIdentifier(), () => context.asTypeClause())
{
_argsListContext = context.argList();
_staticNode = context.STATIC();
_keyword = context.FUNCTION();
_asTypeClauseContext = context.asTypeClause();
}
private string GetSignature(VBAParser.FunctionStmtContext context)
{
if (context == null)
{
return(null);
}
var @static = context.STATIC() == null ? string.Empty : context.STATIC().GetText() + ' ';
var keyword = context.FUNCTION().GetText() + ' ';
var args = context.argList() == null ? "()" : context.argList().GetText() + ' ';
var asTypeClause = context.asTypeClause() == null ? string.Empty : context.asTypeClause().GetText();
var visibility = context.visibility() == null ? string.Empty : context.visibility().GetText() + ' ';
return(visibility + @static + keyword + context.functionName().identifier().GetText() + args + asTypeClause);
}
private static void ConvertFunctionDeclaration(VBAParser.FunctionStmtContext functionContext, IModuleRewriter rewriter)
{
rewriter.Replace(functionContext.FUNCTION(), Tokens.Sub);
rewriter.Replace(functionContext.END_FUNCTION(), "End Sub");
}
