private void OnInfo(Script script, string message)
{
if (message.Length != 0)
{
StdInOut.LogScriptInfo(script.SourceName, message);
}
}
static void Main(string[] args)
{
Environment.ExitCode = 1;
if (args.Length == 0)
{
Debugger.Debug();
return;
}
var options = new Options(args);
options.ParseOption("help", @"^[-/]*(help|\?)$", false, "help : Provide help");
options.ParseOption("debug", @"^debug$", false, "debug : Breaks on errors");
options.ParseOption("noexit", @"^noexit$", false, "noexit : The console remains open at the end");
options.ParseOption("noinfo", @"^noinfo$", false, "noinfo : Do not display the text from WScript.Echo");
options.ParseOption("args", @"^args=.*|a=.*$", new string[0], "args,t=value1,value2,... :", "Lists of arguments to send to the script (Comma to separate values).");
options.ParseOption("out", @"^out=.*|o=.*$", (string)null, "out,o=filepath : ", "Log file. To add the current datetime : {yyyyMMdd-HHmmss}");
options.ParseOption("filter", @"^filter=.*|f=.*$", ".*", "filter,f=value : ", "Pattern to filter procedures");
options.ParseOption("params", @"^params=.*|p=.*$", new string[0], "params,p=value1,value2,... : ", "Lists of params to run each script with (Tag = \"@param\").");
options.ParseOption("threads", @"^threads=.*|t=.*$", 1, "threads,t=n : ", "Number of script to execute in parallel.");
options.AddExample(@"vbsc noexit args=firefox,chrome ""c:\scripts\*.vbs\"" ""c:\scripts\*.vbs\""");
options.AddExample(@"vbsc o=""c:\scripts\result-{DATETIME}.log"" ""c:\scripts\*.vbs""");
options.AddExample(@"vbsc t=4 ""c:\scripts\*.vbs""");
if (options.Files.Length > 0)
{
var dir = Path.GetDirectoryName(options.Files[0]);
if (!string.IsNullOrEmpty(dir))
{
Directory.SetCurrentDirectory(dir);
}
}
if ((bool)options["help"])
{
//Display help
Console.WriteLine(options.ToString());
}
else
{
//Run scripts
try {
if (RunScripts(options))
{
Environment.ExitCode = 0;
}
} catch (Exception ex) {
StdInOut.LogException(ex, args);
}
}
//Wait for a key pressed if noexit is present
if ((bool)options["noexit"])
{
Console.ReadKey();
}
}
private void OnError(ScriptError error)
{
this.Results.Add(error);
error.Script.Succeed = false;
StdInOut.LogScriptError(error.SourceName, error.ToString());
if (_debug)
{
Debugger.Debug(_compiler);
}
}
static bool RunScripts(Options options)
{
StdInOut.HideInfo = (bool)options["noinfo"];
var starttime = DateTime.Now;
StdInOut.LogStart(starttime);
//Check log path
var logpath = (string)options["out"];
if (logpath != null)
{
logpath = logpath.Replace("{DATETIME}", DateTime.Now.ToString("yyyyMMdd-HHmmss"));
if (logpath.IndexOf("{ID}") != -1)
{
var id = 0;
var new_logpath = string.Empty;
while (true)
{
new_logpath = logpath.Replace("{ID}", (++id).ToString());
if (!File.Exists(new_logpath))
{
break;
}
}
logpath = new_logpath;
}
if (logpath.IndexOfAny(Path.GetInvalidPathChars()) != -1)
{
StdInOut.LogError("Invalide log file path.", "Argument: " + (string)options["out"]);
return(false);
}
}
string[] list_scripts_path;
try {
list_scripts_path = Utils.ExpandFilePaths(options.Files, "vbs");
} catch (FileNotFoundException ex) {
StdInOut.LogError(ex.Message, "Argument: " + ex.FileName);
return(false);
}
var runner = new MultiScriptRunner((int)options["threads"]);
var results = runner.Run(list_scripts_path, (string[])options["args"], (string[])options["params"], (string)options["filter"], (bool)options["debug"]);
//Print final result
StdInOut.LogResults(results, starttime, DateTime.Now);
//Save log file
if (!string.IsNullOrEmpty(logpath))
{
StdInOut.SaveTo(logpath);
}
return(results.Exists((r) => !r.Succeed) == false);
}
private void RunScripts()
{
var runner = new ScriptRunner(_debug);
while (_queue_scripts.Count != 0)
{
var script = _queue_scripts.Dequeue();
try {
runner.Execute(script, _procedure_pattern);
} catch (Exception ex) {
runner.Results.Add(new ScriptError(script, null, ex.Message));
StdInOut.LogScriptException(script, ex);
}
}
lock (_list_results)
_list_results.AddRange(runner.Results);
}
