private void SetupEnvironmentFunctions(Engine mispEngine)
{
mispEngine.AddFunction("save-environment", "", (context, arguments) =>
{
Engine.ReportSerializationError = (str) => Write(str + "\n");
var file = new System.IO.StreamWriter(arguments[0].ToString());
mispEngine.SerializeEnvironment(file, context.Scope);
file.Close();
return(true);
}, Arguments.Arg("file"));
mispEngine.AddFunction("load-environment", "", (context, arguments) =>
{
var newContext = new Context();
var newEngine = new Engine();
PrepareEnvironment(newEngine);
if (arguments[2] != null)
{
newEngine.Evaluate(newContext, arguments[2], true, true);
if (newContext.evaluationState == EvaluationState.UnwindingError)
{
Write("Prime error:\n");
Write(MISP.Console.PrettyPrint2(newContext.errorObject, 0));
return(false);
}
}
var result = newEngine.EvaluateString(newContext,
System.IO.File.ReadAllText(arguments[0].ToString()), arguments[0].ToString()) as ScriptObject;
if (newContext.evaluationState != EvaluationState.UnwindingError)
{
var environment = AddEnvironment(arguments[1].ToString(), newEngine, newContext);
if (environment.name == this.environment.name)
{
this.environment = environment;
}
SetupEnvironmentFunctions(newEngine);
var scope = new Scope();
foreach (var prop in result.ListProperties())
{
scope.PushVariable(prop as String, result.GetLocalProperty(prop as String));
}
newContext.ReplaceScope(scope);
Write("Loaded.\n");
return(true);
}
else
{
Write("Error:\n");
Write(MISP.Console.PrettyPrint2(newContext.errorObject, 0));
return(false);
}
}, Arguments.Arg("file"), Arguments.Arg("name"), Arguments.Optional(Arguments.Lazy("prime")));
}
public void BindScript(Engine scriptEngine)
{
scriptEngine.AddFunction("clear-click-events", "Clear all click events.",
(context, arguments) =>
{
clickBindings.Clear();
return null;
});
scriptEngine.AddFunction("on-click", "Create an on-click event handler for an entity.",
(context, arguments) =>
{
var objectID = AutoBind.UIntArgument(arguments[0]);
var body = arguments[1];
var lambda = Function.MakeFunction("on-click lambda", new ScriptList(), "",
body as ScriptObject, context.Scope, true);
clickBindings.Upsert(objectID, lambda);
return true;
},
Arguments.Arg("object-id", "id of the clicked object."),
Arguments.Lazy("code"));
}
public void PrepareEnvironment(Engine engine)
{
SetupStandardConsoleFunctions(engine);
if (SetupHost != null)
{
SetupHost(engine);
}
engine.AddFunction("run-file", "Load and run a file.",
(_context, arguments) =>
{
var text = System.IO.File.ReadAllText(ScriptObject.AsString(arguments[0]));
return(engine.EvaluateString(_context, text, ScriptObject.AsString(arguments[0]), false));
},
Arguments.Arg("name"));
}
private void SetupEnvironmentFunctions(Engine mispEngine)
{
mispEngine.AddFunction("save-environment", "", (context, arguments) =>
{
Engine.ReportSerializationError = (str) => Write(str + "\n");
var file = new System.IO.StreamWriter(arguments[0].ToString());
mispEngine.SerializeEnvironment(file, context.Scope);
file.Close();
return true;
}, Arguments.Arg("file"));
mispEngine.AddFunction("load-environment", "", (context, arguments) =>
{
var newContext = new Context();
var newEngine = new Engine();
PrepareEnvironment(newEngine);
if (arguments[2] != null)
{
newEngine.Evaluate(newContext, arguments[2], true, true);
if (newContext.evaluationState == EvaluationState.UnwindingError)
{
Write("Prime error:\n");
Write(MISP.Console.PrettyPrint2(newContext.errorObject, 0));
return false;
}
}
var result = newEngine.EvaluateString(newContext,
System.IO.File.ReadAllText(arguments[0].ToString()), arguments[0].ToString()) as ScriptObject;
if (newContext.evaluationState != EvaluationState.UnwindingError)
{
var environment = AddEnvironment(arguments[1].ToString(), newEngine, newContext);
if (environment.name == this.environment.name)
this.environment = environment;
SetupEnvironmentFunctions(newEngine);
var scope = new Scope();
foreach (var prop in result.ListProperties())
scope.PushVariable(prop as String, result.GetLocalProperty(prop as String));
newContext.ReplaceScope(scope);
Write("Loaded.\n");
return true;
}
else
{
Write("Error:\n");
Write(MISP.Console.PrettyPrint2(newContext.errorObject, 0));
return false;
}
}, Arguments.Arg("file"), Arguments.Arg("name"), Arguments.Optional(Arguments.Lazy("prime")));
}
public void SetupStandardConsoleFunctions(Engine mispEngine)
{
mispEngine.AddFunction("print", "Print something.",
(context, arguments) =>
{
noEcho = true;
foreach (var item in arguments[0] as ScriptList)
Write(UnescapeString(MISP.Console.PrettyPrint2(item, 0)));
return null;
}, Arguments.Optional(Arguments.Repeat("value")));
mispEngine.AddFunction("printf", "Print a string with formatting.",
(context, arguments) =>
{
noEcho = true;
var s = String.Format(AutoBind.StringArgument(arguments[0]),
AutoBind.ListArgument(arguments[1]).ToArray());
Write(UnescapeString(s));
return null;
},
Arguments.Arg("format-string"),
Arguments.Optional(Arguments.Repeat("value")));
mispEngine.AddFunction("emitfl", "Emit a list of functions",
(context, arguments) =>
{
noEcho = true;
if (arguments[0] == null) foreach (var item in mispEngine.functions) EmitFunctionListItem(item.Value);
else if (arguments[0] is String)
{
var regex = new System.Text.RegularExpressions.Regex(arguments[0] as String);
foreach (var item in mispEngine.functions)
if (regex.IsMatch(item.Value.gsp("@name"))) EmitFunctionListItem(item.Value);
}
else
{
foreach (var item in AutoBind.ListArgument(arguments[0]))
if (item is ScriptObject) EmitFunctionListItem(item as ScriptObject);
}
return null;
}, Arguments.Optional("list"));
mispEngine.AddFunction("module", "Load a module from disc",
(context, arguments) =>
{
return NetModule.LoadModule(mispEngine, AutoBind.StringArgument(arguments[0]),
AutoBind.StringArgument(arguments[1]));
}, Arguments.Arg("file"), Arguments.Arg("module"));
}
public void PrepareEnvironment(Engine engine)
{
SetupStandardConsoleFunctions(engine);
if (SetupHost != null) SetupHost(engine);
engine.AddFunction("run-file", "Load and run a file.",
(_context, arguments) =>
{
var text = System.IO.File.ReadAllText(ScriptObject.AsString(arguments[0]));
return engine.EvaluateString(_context, text, ScriptObject.AsString(arguments[0]), false);
},
Arguments.Arg("name"));
}
void IModule.BindScript(Engine scriptEngine)
{
scriptEngine.AddFunction("bind-key", "Bind an event handler to a key.", (context, arguments) =>
{
var keyNames = AutoBind.StringArgument(arguments[0]);
var bindingType = AutoBind.StringArgument(arguments[1]).ToUpperInvariant();
var body = arguments[2];
var lambda = Function.MakeFunction("key-binding lambda", new ScriptList(), "",
body as ScriptObject, context.Scope, true);
BindingType parsedBindType;
if (!Enum.TryParse(bindingType, out parsedBindType))
{
context.RaiseNewError("Unknown binding type", context.currentNode);
return null;
}
foreach (var c in keyNames.Split(new char[] { ' ' }, StringSplitOptions.RemoveEmptyEntries))
{
Keys key;
if (Enum.TryParse(c, out key))
{
if (parsedBindType == BindingType.HELD)
keysWithHeldEvent.Add(key);
if (!eventBindings.ContainsKey(key))
eventBindings.Add(key, new List<Tuple<BindingType, ScriptObject>>());
eventBindings[key].Add(new Tuple<BindingType, ScriptObject>(parsedBindType, lambda));
}
}
return true;
},
Arguments.Arg("key-names", "A string containing key-names separated by spaces."),
Arguments.Arg("binding-type", "Any of DOWN, UP, or HELD."),
Arguments.Lazy("code"));
scriptEngine.AddFunction("unbind-key", "Clear event handlers associated with a key/event combo",
(context, arguments) =>
{
var keyNames = AutoBind.StringArgument(arguments[0]);
var bindingType = AutoBind.StringArgument(arguments[1]).ToUpperInvariant();
BindingType parsedBindType;
if (!Enum.TryParse(bindingType, out parsedBindType))
{
context.RaiseNewError("Unknown binding type", context.currentNode);
return null;
}
foreach (var c in keyNames.Split(new char[] { ' ' }, StringSplitOptions.RemoveEmptyEntries))
{
Keys key;
if (Enum.TryParse(c, out key))
{
if (parsedBindType == BindingType.HELD)
keysWithHeldEvent.Remove(key);
if (eventBindings.ContainsKey(key))
eventBindings[key].RemoveAll(p => p.Item1 != parsedBindType);
}
}
return true;
},
Arguments.Arg("key-names", "A string containing key-names separated by spaces."),
Arguments.Arg("binding-type", "Any of DOWN, UP, or HELD."));
scriptEngine.AddFunction("on-click", "Create an on-click event handler for an entity.",
(context, arguments) =>
{
var objectID = AutoBind.UIntArgument(arguments[0]);
var body = arguments[1];
if (!(body is ScriptObject)) return null;
clickBindings.Upsert(objectID, body as ScriptObject);
return true;
},
Arguments.Arg("object-id", "id of the clicked object."),
Arguments.Arg("lambda"));
}
public void SetupStandardConsoleFunctions(Engine mispEngine)
{
mispEngine.AddFunction("print", "Print something.",
(context, arguments) =>
{
noEcho = true;
foreach (var item in arguments[0] as ScriptList)
{
Write(UnescapeString(MISP.Console.PrettyPrint2(item, 0)));
}
return(null);
}, Arguments.Optional(Arguments.Repeat("value")));
mispEngine.AddFunction("printf", "Print a string with formatting.",
(context, arguments) =>
{
noEcho = true;
var s = String.Format(AutoBind.StringArgument(arguments[0]),
AutoBind.ListArgument(arguments[1]).ToArray());
Write(UnescapeString(s));
return(null);
},
Arguments.Arg("format-string"),
Arguments.Optional(Arguments.Repeat("value")));
mispEngine.AddFunction("emitfl", "Emit a list of functions",
(context, arguments) =>
{
noEcho = true;
if (arguments[0] == null)
{
foreach (var item in mispEngine.functions)
{
EmitFunctionListItem(item.Value);
}
}
else if (arguments[0] is String)
{
var regex = new System.Text.RegularExpressions.Regex(arguments[0] as String);
foreach (var item in mispEngine.functions)
{
if (regex.IsMatch(item.Value.gsp("@name")))
{
EmitFunctionListItem(item.Value);
}
}
}
else
{
foreach (var item in AutoBind.ListArgument(arguments[0]))
{
if (item is ScriptObject)
{
EmitFunctionListItem(item as ScriptObject);
}
}
}
return(null);
}, Arguments.Optional("list"));
mispEngine.AddFunction("module", "Load a module from disc",
(context, arguments) =>
{
return(NetModule.LoadModule(mispEngine, AutoBind.StringArgument(arguments[0]),
AutoBind.StringArgument(arguments[1])));
}, Arguments.Arg("file"), Arguments.Arg("module"));
}