use of com.serotonin.m2m2.rt.script.JsonImportExclusion in project ma-core-public by infiniteautomation.
the class CompiledScriptExecutor method execute.
/**
* Execute the script on the common engine
* @param script
* @param context
* @param additionalContext
* @param runtime
* @param dataTypeId
* @param timestamp
* @param permissions
* @param scriptWriter
* @return
* @throws ScriptException
* @throws ResultTypeException
*/
public static PointValueTime execute(CompiledScript script, Map<String, IDataPointValueSource> context, Map<String, Object> additionalContext, long runtime, int dataTypeId, long timestamp, ScriptPermissions permissions, PrintWriter scriptWriter, ScriptLog log, ScriptPointValueSetter setter, List<JsonImportExclusion> importExclusions, boolean testRun) throws ScriptException, ResultTypeException {
// StopWatch stopWatch = new Log4JStopWatch();
// stopWatch.start();
ensureInit();
// Create the wrapper object context.
ScriptEngine engine = script.getEngine();
// Prepare the Engine
Bindings engineScope = prepareEngine(engine, context, additionalContext, runtime, timestamp, permissions, scriptWriter, log, setter, importExclusions, testRun);
// Execute.
Object result;
try {
result = script.eval(engineScope);
} catch (ScriptException e) {
throw prettyScriptMessage(e);
}
PointValueTime value = getResult(engine, result, dataTypeId, timestamp);
// stopWatch.stop("execute()");
return value;
}
use of com.serotonin.m2m2.rt.script.JsonImportExclusion in project ma-core-public by infiniteautomation.
the class EmailHandlerRT method sendEmail.
private static void sendEmail(EventInstance evt, NotificationType notificationType, Set<String> addresses, String alias, boolean includeSystemInfo, int pointValueCount, boolean includeLogs, String handlerXid, String customTemplate, List<IntStringPair> additionalContext, String script, SetCallback setCallback, ScriptPermissions permissions) {
if (evt.getEventType().isSystemMessage()) {
if (((SystemEventType) evt.getEventType()).getSystemEventType().equals(SystemEventType.TYPE_EMAIL_SEND_FAILURE)) {
// Don't send email notifications about email send failures.
LOG.info("Not sending email for event raised due to email failure");
return;
}
}
Translations translations = Common.getTranslations();
if (StringUtils.isBlank(alias)) {
// Just set the subject to the message
alias = evt.getMessage().translate(translations);
// Strip out the HTML and the  
alias = StringEscapeUtils.unescapeHtml4(alias);
// Since we have <br/> in the code and that isn't proper HTML we need to remove it by hand
alias = alias.replace("<br/>", "\n");
}
// end if alias was blank
// Determine the subject to use.
TranslatableMessage subjectMsg;
TranslatableMessage notifTypeMsg = new TranslatableMessage(notificationType.getKey());
if (StringUtils.isBlank(alias)) {
// Make these more descriptive
if (evt.getId() == Common.NEW_ID)
subjectMsg = new TranslatableMessage("ftl.subject.default", notifTypeMsg);
else
subjectMsg = new TranslatableMessage("ftl.subject.default.id", notifTypeMsg, evt.getId());
} else {
if (evt.getId() == Common.NEW_ID)
subjectMsg = new TranslatableMessage("ftl.subject.alias", alias, notifTypeMsg);
else
subjectMsg = new TranslatableMessage("ftl.subject.alias.id", alias, notifTypeMsg, evt.getId());
}
String alarmLevel = AlarmLevels.getAlarmLevelMessage(evt.getAlarmLevel()).translate(translations);
String subject = alarmLevel + " - " + subjectMsg.translate(translations);
// Trim the subject if its too long
if (subject.length() > 200)
subject = subject.substring(0, 200);
try {
String[] toAddrs = addresses.toArray(new String[0]);
UsedImagesDirective inlineImages = new UsedImagesDirective();
// Send the email.
Map<String, Object> model = new HashMap<String, Object>();
model.put("evt", evt);
if (evt.getContext() != null)
model.putAll(evt.getContext());
model.put("img", inlineImages);
model.put("instanceDescription", SystemSettingsDao.getValue(SystemSettingsDao.INSTANCE_DESCRIPTION));
if (includeSystemInfo) {
// Get the Work Items
List<WorkItemModel> highPriorityWorkItems = Common.backgroundProcessing.getHighPriorityServiceItems();
model.put("highPriorityWorkItems", highPriorityWorkItems);
List<WorkItemModel> mediumPriorityWorkItems = Common.backgroundProcessing.getMediumPriorityServiceQueueItems();
model.put("mediumPriorityWorkItems", mediumPriorityWorkItems);
List<WorkItemModel> lowPriorityWorkItems = Common.backgroundProcessing.getLowPriorityServiceQueueItems();
model.put("lowPriorityWorkItems", lowPriorityWorkItems);
model.put("threadList", getThreadsList());
}
int type = SystemSettingsDao.getIntValue(SystemSettingsDao.EMAIL_CONTENT_TYPE);
// If we are a point event then add the value
if (evt.getEventType() instanceof DataPointEventType) {
DataPointVO dp = (DataPointVO) evt.getContext().get("point");
if (dp != null) {
DataPointRT rt = Common.runtimeManager.getDataPoint(dp.getId());
if (rt != null) {
List<PointValueTime> pointValues = null;
if (pointValueCount > 0)
pointValues = rt.getLatestPointValues(pointValueCount);
if ((pointValues != null) && (pointValues.size() > 0)) {
if (type == MangoEmailContent.CONTENT_TYPE_HTML || type == MangoEmailContent.CONTENT_TYPE_BOTH) {
List<RenderedPointValueTime> renderedPointValues = new ArrayList<RenderedPointValueTime>();
for (PointValueTime pvt : pointValues) {
RenderedPointValueTime rpvt = new RenderedPointValueTime();
rpvt.setValue(Functions.getHtmlText(rt.getVO(), pvt));
rpvt.setTime(Functions.getFullSecondTime(pvt.getTime()));
renderedPointValues.add(rpvt);
}
model.put("renderedHtmlPointValues", renderedPointValues);
}
if (type == MangoEmailContent.CONTENT_TYPE_TEXT || type == MangoEmailContent.CONTENT_TYPE_BOTH) {
List<RenderedPointValueTime> renderedPointValues = new ArrayList<RenderedPointValueTime>();
for (PointValueTime pvt : pointValues) {
RenderedPointValueTime rpvt = new RenderedPointValueTime();
rpvt.setValue(Functions.getRenderedText(rt.getVO(), pvt));
rpvt.setTime(Functions.getFullSecondTime(pvt.getTime()));
renderedPointValues.add(rpvt);
}
model.put("renderedPointValues", renderedPointValues);
}
}
}
}
}
// Build the additional context for the email model
if (additionalContext == null || pointValueCount <= 0)
model.put("additionalContext", new HashMap<>(0));
else {
Map<String, Map<String, Object>> context = new HashMap<>();
for (IntStringPair pair : additionalContext) {
Map<String, Object> point = new HashMap<String, Object>();
DataPointRT rt = Common.runtimeManager.getDataPoint(pair.getKey());
List<PointValueTime> pointValues;
List<RenderedPointValueTime> renderedPointValues;
DataPointVO dpvo;
if (rt != null) {
dpvo = rt.getVO();
pointValues = rt.getLatestPointValues(pointValueCount);
renderedPointValues = new ArrayList<RenderedPointValueTime>();
if (pointValues != null && pointValues.size() > 0)
for (PointValueTime pvt : pointValues) {
RenderedPointValueTime rpvt = new RenderedPointValueTime();
rpvt.setValue(Functions.getRenderedText(rt.getVO(), pvt));
rpvt.setTime(Functions.getFullSecondTime(pvt.getTime()));
renderedPointValues.add(rpvt);
}
} else {
dpvo = DataPointDao.instance.get(pair.getKey());
if (dpvo == null)
continue;
pointValues = Common.databaseProxy.newPointValueDao().getLatestPointValues(pair.getKey(), pointValueCount);
renderedPointValues = new ArrayList<RenderedPointValueTime>();
for (PointValueTime pvt : pointValues) {
RenderedPointValueTime rpvt = new RenderedPointValueTime();
rpvt.setValue(Functions.getRenderedText(dpvo, pvt));
rpvt.setTime(Functions.getFullSecondTime(pvt.getTime()));
renderedPointValues.add(rpvt);
}
}
point.put("values", renderedPointValues);
point.put("deviceName", dpvo.getDeviceName());
point.put("name", dpvo.getName());
point.put("contextKey", pair.getValue());
context.put(pair.getValue(), point);
}
model.put("additionalContext", context);
}
if (!StringUtils.isEmpty(script)) {
// Okay, a script is defined, let's pass it the model so that it may add to it
Map<String, Object> modelContext = new HashMap<String, Object>();
modelContext.put("model", model);
Map<String, IDataPointValueSource> context = new HashMap<String, IDataPointValueSource>();
for (IntStringPair pair : additionalContext) {
DataPointRT dprt = Common.runtimeManager.getDataPoint(pair.getKey());
if (dprt == null) {
DataPointVO targetVo = DataPointDao.instance.getDataPoint(pair.getKey(), false);
if (targetVo == null) {
LOG.warn("Additional context point with ID: " + pair.getKey() + " and context name " + pair.getValue() + " could not be found.");
// Not worth aborting the email, just warn it
continue;
}
if (targetVo.getDefaultCacheSize() == 0)
targetVo.setDefaultCacheSize(1);
dprt = new DataPointRT(targetVo, targetVo.getPointLocator().createRuntime(), DataSourceDao.instance.getDataSource(targetVo.getDataSourceId()), null);
dprt.resetValues();
}
context.put(pair.getValue(), dprt);
}
List<JsonImportExclusion> importExclusions = new ArrayList<JsonImportExclusion>(1);
importExclusions.add(new JsonImportExclusion("xid", handlerXid) {
@Override
public String getImporterType() {
return ConfigurationExportData.EVENT_HANDLERS;
}
});
try {
CompiledScript compiledScript = CompiledScriptExecutor.compile(script);
CompiledScriptExecutor.execute(compiledScript, context, modelContext, Common.timer.currentTimeMillis(), DataTypes.ALPHANUMERIC, evt.isActive() || !evt.isRtnApplicable() ? evt.getActiveTimestamp() : evt.getRtnTimestamp(), permissions, SetPointHandlerRT.NULL_WRITER, new ScriptLog(SetPointHandlerRT.NULL_WRITER, LogLevel.FATAL), setCallback, importExclusions, false);
} catch (ScriptPermissionsException | ScriptException | ResultTypeException e) {
LOG.error("Exception running email handler script: " + e.getMessage(), e);
}
}
MangoEmailContent content;
if (StringUtils.isEmpty(customTemplate))
content = new MangoEmailContent(notificationType.getFile(), model, translations, subject, Common.UTF8);
else
content = new MangoEmailContent(handlerXid, customTemplate, model, translations, subject);
PostEmailRunnable[] postEmail = null;
if (includeLogs) {
final File logZip = getZippedLogfile(content, new File(Common.getLogsDir(), "ma.log"));
// Setup To delete the temp files from zip
if (logZip != null) {
// See that the temp file(s) gets deleted after the email is sent.
PostEmailRunnable deleteTempFile = new PostEmailRunnable() {
@Override
public void run() {
if (!logZip.delete())
LOG.warn("Temp file " + logZip.getPath() + " not deleted");
// Set our state to email failed if necessary
// TODO Create an Event to notify of Failed Emails...
// if(!this.isSuccess()){}
}
};
postEmail = new PostEmailRunnable[] { deleteTempFile };
}
}
for (String s : inlineImages.getImageList()) content.addInline(new EmailInline.FileInline(s, Common.getWebPath(s)));
EmailWorkItem.queueEmail(toAddrs, content, postEmail);
} catch (Exception e) {
LOG.error("", e);
}
}
use of com.serotonin.m2m2.rt.script.JsonImportExclusion in project ma-core-public by infiniteautomation.
the class ScriptExecutor method prepareEngine.
/**
* Prepare the engine by creating the context
* @param engine
* @param context
* @param additionalContext
* @param runtime
* @param permissions
* @param scriptWriter
* @return
*/
protected static Bindings prepareEngine(ScriptEngine engine, Map<String, IDataPointValueSource> context, Map<String, Object> additionalContext, long runtime, long timestamp, ScriptPermissions permissions, PrintWriter scriptWriter, ScriptLog log, ScriptPointValueSetter setter, List<JsonImportExclusion> importExclusions, boolean testRun) {
ScriptUtils.prepareEngine(engine);
ScriptUtils.wrapperContext(engine, new WrapperContext(runtime, timestamp));
Bindings engineScope = engine.getBindings(ScriptContext.ENGINE_SCOPE);
// TODO Bubble PointValueSetter back up to top
if (permissions != null)
ScriptUtils.prepareUtilities(permissions, engine, engineScope, setter, importExclusions, testRun);
if (additionalContext != null) {
Set<Entry<String, Object>> entries = additionalContext.entrySet();
for (Entry<String, Object> entry : entries) engineScope.put(entry.getKey(), entry.getValue());
}
// Put the context variables into the engine with engine scope.
for (String varName : context.keySet()) {
IDataPointValueSource point = context.get(varName);
engineScope.put(varName, ScriptUtils.wrapPoint(engine, point, setter));
}
engineScope.put(ScriptUtils.POINTS_MAP_KEY, context);
// Set the print writer if necessary
if (scriptWriter != null) {
engine.getContext().setWriter(scriptWriter);
engineScope.put(ScriptLog.CONTEXT_KEY, log);
}
return engineScope;
}
Aggregations