internal static Options CreateVBScriptParserOptions(ILog log)
{
var options = new Options(log);
options.CaseSensitive = false;
options.LineWrap = TokenComparison.VBScriptLineWrap(
log,
options);
return(options);
}
public static void MatchVBScriptFunction_T2(ILog log)
{
const string input = "ignorethis obj.Func(1)";
const string funcStr = "obj.Func";
var statementList = new StatementList(log);
var mainComp = TokenComparison.CreateVBScriptFunctionOrVar(null, null, null, null, log);
statementList.Add(new StorePosAsVariable(log, "BeginPos"));
statementList.Add(mainComp);
Action <RunState> InitRunState = (runState) => {
{
var funcIndex = input.IndexOf(funcStr);
IAssertion sub = new PositionAssertion(true, funcIndex, funcIndex + funcStr.Length);
sub.Name = "sub";
runState.AddAssertion("sub", sub);
}
{
var funcIndex = input.IndexOf(funcStr);
IAssertion isEndOrArgList = new PositionAssertion(
true, funcIndex + funcStr.Length, funcIndex + funcStr.Length + 3
);
isEndOrArgList.Name = "IsEndOrArgumentList";
runState.AddAssertion("IsEndOrArgumentList", isEndOrArgList);
}
};
Func <RunState, string, int, string> ReplaceFunc = (runState, _input, pos) => {
var beginPos = (int)runState.GetVariable("BeginPos");
var str = input.Substring(beginPos, pos - beginPos);
return(str);
};
var parser = new Parser(log);
int numMatches;
parser.Extract(input, null, statementList, null, null, out numMatches, ReplaceFunc, InitRunState);
System.Diagnostics.Debug.Assert(numMatches == 1);
}
public static IComparisonWithAdvance CreateVBScriptSub(
Options options,
string name = null,
ILog log = null)
{
// This requires more than one statement
StatementList statementList = new StatementList(log);
statementList.Name = name;
// Function end: . , ", or whitespace
CharComparison isDot = new CharComparison(log, options, '.');
CharComparison isQuote = new CharComparison(log, options, '\"');
CharDelegateComparison isWhitespace = new CharDelegateComparison(log, Char.IsWhiteSpace);
OrComparison funcNameEnd = new OrComparison(log);
funcNameEnd.Add(isDot);
funcNameEnd.Add(isQuote);
funcNameEnd.Add(isWhitespace);
// Identifier but not including classes/namespaces
var identifier = TokenComparison.CreateIdentifier(
options,
end: funcNameEnd,
exclusion: null,
name: null,
log: log);
// List of identifiers seperated by . which will include classes/namespaces
var sub = TokenComparison.CreateListOfTokens(
options,
token: identifier,
segment: isDot,
name: null,
log: null);
statementList.Add(sub);
return(statementList);
}
// E.g.
// class.Function _ \r\n (p1, p2, class.Function2( _\r\n p1,p2 ) )
// or
// class.FunctionWithNoArgs
// or
// variableName
public static IComparisonWithAdvance CreateVBScriptFunctionOrVar(
Options options,
IComparisonWithAdvance vbScriptConcatCommaOrWhitespaceOrEnd,
IComparison vbScriptKeywords,
string name = null,
ILog log = null)
{
if (options == null)
{
options = CreateVBScriptParserOptions(log);
}
if (vbScriptConcatCommaOrWhitespaceOrEnd == null)
{
vbScriptConcatCommaOrWhitespaceOrEnd = CreateVBScriptConcatCommaOrWhitespaceOrEnd(log);
}
if (vbScriptKeywords == null)
{
vbScriptKeywords = TokenComparison.VBScriptKeywords(log, options);
}
if (options.SkipLineWrapOperation == null)
{
Parser.ThrowParseError("The line wrap comparison has not been specified. This is mandatory.");
}
// This requires more than one statement
StatementList statementList = new StatementList(log);
statementList.Name = name;
// Function
CharComparison isDot = new CharComparison(log, options, '.');
OrComparison isDotOrParenOrEnd = new OrComparison(log);
isDotOrParenOrEnd.Add(isDot);
isDotOrParenOrEnd.Add(vbScriptConcatCommaOrWhitespaceOrEnd);
CharComparison openParen = new CharComparison(log, options, '(');
isDotOrParenOrEnd.Add(openParen);
var identifier = TokenComparison.CreateIdentifier(
options,
end: isDotOrParenOrEnd,
exclusion: vbScriptKeywords,
name: "identifier",
log: log);
// List of identifiers seperated by . which will include classes/namespaces
var sub = TokenComparison.CreateListOfTokens(
options,
token: identifier,
segment: isDot,
name: "sub",
log: log);
statementList.Add(sub);
// Skip whitespace
{
var skip = SkipWhitespace(log);
skip.Name = "SkipWhitespace1";
statementList.Add(skip);
}
// Skip the line wrap character
statementList.Add(options.SkipLineWrapOperation);
// Function parenthesis and arguments
CharComparison close = new CharComparison(log, options, ')');
// Either the end (and don't advance past it), or function argument list
OrComparison isEndOrArgumentList = new OrComparison(log);
isEndOrArgumentList.Name = "IsEndOrArgumentList";
isEndOrArgumentList.Add(new CompareNoAdvance(log, vbScriptConcatCommaOrWhitespaceOrEnd));
isEndOrArgumentList.Add(new NestedOpenCloseComparison(log, openParen, close));
statementList.Add(isEndOrArgumentList);
return(statementList);
}
// From "response.write blah"
// To "response.write(blah)"
public static string AddParenthesisToFunctionCalls(
ILog log,
string input,
out int numMatches,
Action <Action <string, IAssertion> > fCreateAssertList = null)
{
// The parser object that will do the work
Parser parser = new Parser(log);
//// Parser options
// This create a special line wrap
Options parserOptions = TokenComparison.CreateVBScriptParserOptions(log);
CharDelegateComparison isWhitespace = new CharDelegateComparison(log, Char.IsWhiteSpace);
IOperation skipWhitespace = TokenComparison.SkipWhitespace(log);
// State
List <State> stateList = VBScriptState(log, parserOptions);
// The main statement list
StatementList mainStatements = new StatementList(log);
// Capturing
Dictionary <string, Capture> capturing = new Dictionary <string, Capture>();
// Function name capture
Capture funcNameCapture = new Capture(log);
funcNameCapture.Name = "FunctionName";
capturing.Add("funcName", funcNameCapture);
mainStatements.Add(funcNameCapture);
// Function name capture statements
var vbScriptKeywords = TokenComparison.VBScriptKeywords(log, parserOptions);
StatementList funcNameCaptureStatements = new StatementList(log);
funcNameCapture.Comparison = funcNameCaptureStatements;
funcNameCaptureStatements.Exclusion = vbScriptKeywords;
//// Starts with whitespace or is the start of the input string
StartOfInputStringComparison isStartOfString = new StartOfInputStringComparison(log);
OrComparison startsWithWhitespaceOrIsStartOfInputString = new OrComparison(log);
startsWithWhitespaceOrIsStartOfInputString.Add(isStartOfString);
startsWithWhitespaceOrIsStartOfInputString.Add(isWhitespace);
funcNameCaptureStatements.Add(startsWithWhitespaceOrIsStartOfInputString);
// Skip whitespace
funcNameCaptureStatements.Add(skipWhitespace);
//// Function/sub name.
//// For example: class1.Func or class1.Func"string arg"
funcNameCaptureStatements.Add(TokenComparison.CreateVBScriptSub(parserOptions, name: null, log: log));
//// Arguments list
// Skip whitespace and don't capture it
mainStatements.Add(skipWhitespace);
// Capture
Capture argListCapture = new Capture(log);
capturing.Add("funcArgs", argListCapture);
mainStatements.Add(argListCapture);
// Function argument list capture statements
StatementList argListCaptureStatements = new StatementList(log);
argListCapture.Comparison = argListCaptureStatements;
// Skip the line wrap (and capture it)
argListCaptureStatements.Add(parserOptions.SkipLineWrapOperation);
// Concatenated single arguments
// Delimited by whitespace, comma & or +
var vbScriptConcatCommaOrWhitespaceOrEnd =
TokenComparison.CreateVBScriptConcatCommaOrWhitespaceOrEnd(log, parserOptions);
// Numbers
var numberComparison =
TokenComparison.CreateNumber(parserOptions, vbScriptConcatCommaOrWhitespaceOrEnd, "ArgsNumber", log: log);
// VB script quoted strings
var quotedText = TokenComparison.CreateVBScriptQuotedString(parserOptions, name: "Args quoted text", log: log);
// VB script function which could include arguments, or a variable name
var VbScriptFunctionOrVar = TokenComparison.CreateVBScriptFunctionOrVar(
parserOptions,
vbScriptConcatCommaOrWhitespaceOrEnd,
vbScriptKeywords,
"VbScriptFunctionOrVar",
log: log);
// Individual arguments
// Types
OrComparison individualArgumentTypes = new OrComparison(log);
individualArgumentTypes.Add(numberComparison);
individualArgumentTypes.Add(quotedText);
individualArgumentTypes.Add(VbScriptFunctionOrVar);
// List delimiter for the individual arguments
OrComparison vbScriptConcactOrComma = new OrComparison(log);
{
var matchAmpersand = new CharComparison(log, parserOptions, '&');
matchAmpersand.Name = "ArgsStringConcatenation";
vbScriptConcactOrComma.Add(matchAmpersand);
}
vbScriptConcactOrComma.Add(new CharComparison(log, parserOptions, '+'));
var isComma = new CharComparison(log, parserOptions, ',');
vbScriptConcactOrComma.Add(isComma);
DelimitedListComparison individualArgumentList = new DelimitedListComparison(
log, parserOptions, individualArgumentTypes, seperator: vbScriptConcactOrComma
);
individualArgumentList.MinAmount = 1;
individualArgumentList.ItemTrim = skipWhitespace;
// The argument list - comma seperated list of function arguments
DelimitedListComparison argumentList = new DelimitedListComparison(
log, parserOptions, individualArgumentList, seperator: isComma
);
argumentList.Name = "ArgumentList";
argumentList.MinAmount = 1;
argumentList.ItemTrim = skipWhitespace;
argListCaptureStatements.Add(argumentList);
Action <RunState> InitRunState = null;
if (fCreateAssertList != null)
{
InitRunState = (runState) => {
fCreateAssertList(runState.AddAssertion);
};
}
const string replaceWith = "'funcName'('funcArgs')";
string replaced = parser.Replace(
input, replaceWith, mainStatements, capturing, stateList, out numMatches, null, InitRunState
);
return(replaced);
}
//TODO this is not fully tested. needs a new feature to allow repeats
// - "repeated comparison"
// - also doesn't work inside quotes. needs a state
// I think state ment list should have a state member
// <script ..>function blah end function </script>
public static string WrapFunctionsInScriptBlock(
ILog log,
string input)
{
// The parser object that will do the work
Parser parser = new Parser(log);
//// Parser options
Options parserOptions = new Options(log);
parserOptions.CaseSensitive = false;
// Line wrap
parserOptions.LineWrap = TokenComparison.VBScriptLineWrap(
log,
parserOptions);
// State
List <State> stateList = VBScriptState(log, parserOptions);
// The main statement list
StatementList mainStatements = new StatementList(log);
// Begins with whitespace (don't capture the space)
CharDelegateComparison isWhitespace = new CharDelegateComparison(log, Char.IsWhiteSpace);
mainStatements.Add(isWhitespace);
// Skip whitespace (don't capture the space)
mainStatements.Add(TokenComparison.SkipWhitespace(log));
// Capturing
Dictionary <string, Capture> capturing = new Dictionary <string, Capture>();
// The capture
Capture capture = new Capture(log);
capture.Name = "capture";
capturing.Add("functionDefinition", capture);
mainStatements.Add(capture);
// Capture statements
StatementList captureStatements = new StatementList(log);
capture.Comparison = captureStatements;
// Function or sub
OrComparison functionOrSub = new OrComparison(log);
functionOrSub.Add(new StringComparison(log, parserOptions, "function"));
functionOrSub.Add(new StringComparison(log, parserOptions, "sub"));
captureStatements.Add(functionOrSub);
// Whitespace
captureStatements.Add(isWhitespace);
// Line wrap
captureStatements.Add(parserOptions.SkipLineWrapOperation);
//// Name of function/sub
// End is either whitespace or (
OrComparison whiteSpaceOrOpenParen = new OrComparison(log);
whiteSpaceOrOpenParen.Add(isWhitespace);
whiteSpaceOrOpenParen.Add(new CharComparison(log, parserOptions, '('));
var funcName = TokenComparison.CreateIdentifier(
parserOptions,
end: whiteSpaceOrOpenParen,
exclusion: null,
name: null,
log: log);
captureStatements.Add(funcName);
//// Find the end of the sub/function
StatementList functionEnd = new StatementList(log);
// Whitespace
functionEnd.Add(isWhitespace);
// 'end'
functionEnd.Add(new StringComparison(log, parserOptions, "end"));
// Whitespace
functionEnd.Add(isWhitespace);
// Skip the line wrap
functionEnd.Add(parserOptions.SkipLineWrapOperation);
// Function or sub
functionEnd.Add(functionOrSub);
// Whitespace or %>
OrComparison whitespaceOrPageDirective = new OrComparison(log);
whitespaceOrPageDirective.Add(isWhitespace);
whitespaceOrPageDirective.Add(new CompareNoAdvance(log, new StringComparison(log, parserOptions, "%>")));
functionEnd.Add(whitespaceOrPageDirective);
// Skip till find the function name
captureStatements.Add(new AdvanceUntilComparison(log, functionEnd));
//const string replaceWith = "'funcName'('funcArgs')";
string replaceWith = string.Format(
"{0}<script language=\"VB\" runat=\"Server\">{0}'functionDefinition'{0}</script>{0}",
Environment.NewLine);
string replaced = parser.Replace(input, replaceWith, mainStatements, capturing, stateList);
return(replaced);
}
