| | | 1 | | using System.Dynamic; |
| | | 2 | | using System.Text.RegularExpressions; |
| | | 3 | | using Elsa.Expressions.Models; |
| | | 4 | | using Elsa.Extensions; |
| | | 5 | | using Elsa.Expressions.JavaScript.Extensions; |
| | | 6 | | using Elsa.Expressions.JavaScript.Helpers; |
| | | 7 | | using Elsa.Expressions.JavaScript.Notifications; |
| | | 8 | | using Elsa.Expressions.JavaScript.Options; |
| | | 9 | | using Elsa.Mediator.Contracts; |
| | | 10 | | using Elsa.Workflows.Activities; |
| | | 11 | | using JetBrains.Annotations; |
| | | 12 | | using Jint.Native; |
| | | 13 | | using Microsoft.Extensions.Options; |
| | | 14 | | |
| | | 15 | | namespace Elsa.Expressions.JavaScript.Handlers; |
| | | 16 | | |
| | | 17 | | /// <summary> |
| | | 18 | | /// A handler that configures the Jint engine with workflow variables. |
| | | 19 | | /// </summary> |
| | | 20 | | [UsedImplicitly] |
| | 909 | 21 | | public partial class ConfigureEngineWithVariables(IOptions<JintOptions> options) : INotificationHandler<EvaluatingJavaSc |
| | | 22 | | { |
| | 564 | 23 | | private bool IsEnabled => options.Value is { DisableWrappers: false, DisableVariableCopying: false }; |
| | | 24 | | |
| | | 25 | | /// <inheritdoc /> |
| | | 26 | | public Task HandleAsync(EvaluatingJavaScript notification, CancellationToken cancellationToken) |
| | | 27 | | { |
| | 285 | 28 | | if (!IsEnabled) |
| | 0 | 29 | | return Task.CompletedTask; |
| | | 30 | | |
| | 285 | 31 | | CopyVariablesIntoEngine(notification); |
| | 285 | 32 | | return Task.CompletedTask; |
| | | 33 | | } |
| | | 34 | | |
| | | 35 | | public Task HandleAsync(EvaluatedJavaScript notification, CancellationToken cancellationToken) |
| | | 36 | | { |
| | 279 | 37 | | if (!IsEnabled) |
| | 0 | 38 | | return Task.CompletedTask; |
| | | 39 | | |
| | 279 | 40 | | CopyVariablesIntoWorkflowExecutionContext(notification); |
| | 279 | 41 | | return Task.CompletedTask; |
| | | 42 | | } |
| | | 43 | | |
| | | 44 | | private void CopyVariablesIntoWorkflowExecutionContext(EvaluatedJavaScript notification) |
| | | 45 | | { |
| | 279 | 46 | | var context = notification.Context; |
| | 279 | 47 | | var engine = notification.Engine; |
| | 279 | 48 | | var variablesContainer = (IDictionary<string, object?>)engine.GetValue("variables").ToObject()!; |
| | | 49 | | |
| | | 50 | | // Only the variables the expression actually referenced were copied in, so an expression that never |
| | | 51 | | // mentions "variables." has nothing to copy back and does not need the input names resolved at all. |
| | 279 | 52 | | if (variablesContainer.Count == 0) |
| | 263 | 53 | | return; |
| | | 54 | | |
| | 16 | 55 | | var inputNames = GetInputNames(context).FilterInvalidVariableNames().ToHashSet(StringComparer.Ordinal); |
| | | 56 | | |
| | 64 | 57 | | foreach (var (variableName, variableValue) in variablesContainer) |
| | | 58 | | { |
| | 16 | 59 | | if (inputNames.Contains(variableName)) |
| | | 60 | | continue; |
| | | 61 | | |
| | 16 | 62 | | var processedValue = variableValue is JsObject jsValue ? jsValue.ToObject() : variableValue ?? context.GetVa |
| | 16 | 63 | | context.SetVariable(variableName, processedValue); |
| | | 64 | | } |
| | 16 | 65 | | } |
| | | 66 | | |
| | | 67 | | private void CopyVariablesIntoEngine(EvaluatingJavaScript notification) |
| | | 68 | | { |
| | 285 | 69 | | var engine = notification.Engine; |
| | 285 | 70 | | var context = notification.Context; |
| | 285 | 71 | | var expression = notification.Expression; |
| | 285 | 72 | | var variableNames = GetUsedVariableNames(context, expression).ToList(); |
| | 285 | 73 | | var variablesContainer = (IDictionary<string, object?>)new ExpandoObject(); |
| | | 74 | | |
| | 590 | 75 | | foreach (var variableName in variableNames) |
| | | 76 | | { |
| | 10 | 77 | | var variableValue = context.GetVariableInScope(variableName); |
| | 10 | 78 | | variableValue = ObjectConverterHelper.ProcessVariableValue(engine, variableValue); |
| | 10 | 79 | | variablesContainer[variableName] = variableValue; |
| | | 80 | | } |
| | | 81 | | |
| | 285 | 82 | | engine.SetValue("variables", variablesContainer); |
| | 285 | 83 | | } |
| | | 84 | | |
| | | 85 | | private IEnumerable<string> GetUsedVariableNames(ExpressionExecutionContext context, string expression) |
| | | 86 | | { |
| | 285 | 87 | | var variableNames = context.GetVariableNamesInScope().FilterInvalidVariableNames(); |
| | | 88 | | |
| | 285 | 89 | | var variableNamesInScript = ExtractVariableNamesRegex().Matches(expression) |
| | 11 | 90 | | .Select(m => m.Groups[1].Value) |
| | 285 | 91 | | .ToList(); |
| | | 92 | | |
| | 443 | 93 | | return variableNames.Where(x => variableNamesInScript.Contains(x)); |
| | | 94 | | } |
| | | 95 | | |
| | | 96 | | private IEnumerable<string> GetInputNames(ExpressionExecutionContext context) |
| | | 97 | | { |
| | 16 | 98 | | var activityExecutionContext = context.TryGetActivityExecutionContext(out var aec) ? aec : null; |
| | | 99 | | |
| | 45 | 100 | | while (activityExecutionContext != null) |
| | | 101 | | { |
| | 29 | 102 | | if (activityExecutionContext.Activity is Workflow workflow) |
| | | 103 | | { |
| | 14 | 104 | | var inputDefinitions = workflow.Inputs; |
| | | 105 | | |
| | 28 | 106 | | foreach (var inputDefinition in inputDefinitions) |
| | 0 | 107 | | yield return inputDefinition.Name; |
| | | 108 | | } |
| | | 109 | | |
| | 58 | 110 | | foreach (var syntheticProperty in activityExecutionContext.Activity.SyntheticProperties) |
| | | 111 | | { |
| | 0 | 112 | | yield return syntheticProperty.Key; |
| | | 113 | | } |
| | | 114 | | |
| | 29 | 115 | | activityExecutionContext = activityExecutionContext.ParentActivityExecutionContext; |
| | | 116 | | } |
| | 16 | 117 | | } |
| | | 118 | | |
| | | 119 | | [GeneratedRegex(@"variables\.(\w+)(?:\.\w+)*")] |
| | | 120 | | private static partial Regex ExtractVariableNamesRegex(); |
| | | 121 | | } |