Create JsonJSTransformerProvider for custom scripts

Signed-off-by: Andre Kurait <[email protected]>

transformation/transformationPlugins/jsonMessageTransformers/jsonJSTransformerProvider/build.gradle

@@ -0,0 +1,21 @@
+buildscript {
+    dependencies {
+        classpath 'org.junit.platform:junit-platform-gradle-plugin:1.0.1'
+    }
+}
+
+plugins {
+    id 'io.freefair.lombok'
+}
+
+dependencies {
+    implementation project(':transformation:transformationPlugins:jsonMessageTransformers:jsonMessageTransformerInterface')
+    implementation project(':transformation:transformationPlugins:jsonMessageTransformers:jsonJSTransformer')
+
+    implementation group: 'com.fasterxml.jackson.core', name: 'jackson-core'
+    implementation group: 'com.fasterxml.jackson.core', name: 'jackson-databind'
+}
+
+tasks.named('test') {
+    useJUnitPlatform()
+}

transformation/transformationPlugins/jsonMessageTransformers/jsonJSTransformerProvider/src/main/java/org/opensearch/migrations/transform/JsonJSTransformerProvider.java

@@ -0,0 +1,42 @@
+package org.opensearch.migrations.transform;
+
+import java.util.Map;
+import java.util.function.Function;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.SneakyThrows;
+
+public class JsonJSTransformerProvider implements IJsonTransformerProvider {
+
+    public static final String INITIALIZATION_SCRIPT = "initializationScript";
+    public static final String INVOCATION_SCRIPT = "invocationScript";
+    public static final String BINDINGS_PROVIDER = "bindingsProvider";
+
+    public final static ObjectMapper mapper = new ObjectMapper();
+
+    @SneakyThrows
+    @Override
+    public IJsonTransformer createTransformer(Object jsonConfig) {
+        if (jsonConfig == null || (jsonConfig instanceof String && ((String) jsonConfig).isEmpty())) {
+            throw new IllegalArgumentException("Configuration must contain 'initializationScript', 'invocationScript', and 'bindingsProvider'.");
+        } else if (!(jsonConfig instanceof Map)) {
+            throw new IllegalArgumentException(getConfigUsageStr());
+        }
+
+        var config = (Map<String, Object>) jsonConfig;
+        String initializationScript = (String) config.get(INITIALIZATION_SCRIPT);
+        String invocationScript = (String) config.get(INVOCATION_SCRIPT);
+        Function<Object, Object> bindingsProvider = (Function<Object, Object>) config.get(BINDINGS_PROVIDER);
+
+        if (initializationScript == null || invocationScript == null || bindingsProvider == null) {
+            throw new IllegalArgumentException("'initializationScript', 'invocationScript', and 'bindingsProvider' must be provided.");
+        }
+
+        return new JavascriptTransformer(initializationScript, invocationScript, bindingsProvider);
+    }
+
+    private String getConfigUsageStr() {
+        return this.getClass().getName() + " expects the incoming configuration to be a Map<String, Object>, " +
+                "with values '" + INITIALIZATION_SCRIPT + "', '" + INVOCATION_SCRIPT + "', and '" + BINDINGS_PROVIDER + "'.";
+    }
+}

transformation/transformationPlugins/jsonMessageTransformers/jsonJSTransformerProvider/src/resources/META-INF/services/org.opensearch.migrations.transform.IJsonTransformerProvider

@@ -0,0 +1 @@
+org.opensearch.migrations.transform.JsonJSTransformerProvider

transformation/transformationPlugins/jsonMessageTransformers/jsonTypeMappingsSanitizationTransformerProvider/build.gradle

@@ -11,6 +11,7 @@ plugins {
 dependencies {
     implementation project(':transformation:transformationPlugins:jsonMessageTransformers:jsonMessageTransformerInterface')
     implementation project(':transformation:transformationPlugins:jsonMessageTransformers:jsonTypeMappingsSanitizationTransformer')
+    implementation project(':transformation:transformationPlugins:jsonMessageTransformers:jsonJSTransformerProvider')
 
     implementation group: 'com.fasterxml.jackson.core', name: 'jackson-core'
     implementation group: 'com.fasterxml.jackson.core', name: 'jackson-databind'

transformation/transformationPlugins/jsonMessageTransformers/jsonTypeMappingsSanitizationTransformerProvider/src/main/java/org/opensearch/migrations/transform/TypeMappingSanitizationTransformerProvider.java

@@ -9,7 +9,7 @@
 import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.SneakyThrows;
 
-public class TypeMappingSanitizationTransformerProvider implements IJsonTransformerProvider {
+public class TypeMappingSanitizationTransformerProvider extends JsonJSTransformerProvider {
 
     public static final String FEATURE_FLAGS = "featureFlags";
     public static final String STATIC_MAPPINGS = "staticMappings";
@@ -21,6 +21,7 @@ public class TypeMappingSanitizationTransformerProvider implements IJsonTransfor
 
     @SneakyThrows
     @Override
+    @SuppressWarnings("unchecked")
     public IJsonTransformer createTransformer(Object jsonConfig) {
         try {
             if ((jsonConfig == null) ||