Add to_base64 function #354

metafacture · May 28, 2024 · fd91f47 · fd91f47
1 parent 2d4f976
commit fd91f47
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/metafix/src/main/java/org/metafacture/metafix/FixMethod.java b/metafix/src/main/java/org/metafacture/metafix/FixMethod.java
@@ -41,6 +41,7 @@
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 import java.util.stream.Collectors;
+import java.util.Base64;
 
 public enum FixMethod implements FixFunction { // checkstyle-disable-line ClassDataAbstractionCoupling|ClassFanOutComplexity
 
@@ -626,6 +627,13 @@ public void apply(final Metafix metafix, final Record record, final List<String>
             );
         }
     },
+    to_base64 {
+        @Override
+        public void apply(final Metafix metafix, final Record record, final List<String> params,
+                final Map<String, String> options) {
+            record.transform(params.get(0), s -> Base64.getEncoder().encodeToString(s.getBytes()));
+        }
+    },
     to_json {
         @Override
         public void apply(final Metafix metafix, final Record record, final List<String> params, final Map<String, String> options) {

diff --git a/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java b/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
@@ -4062,4 +4062,24 @@ public void shouldUriEncodePathSegmentWithoutSafeChars() {
         );
     }
 
+    @Test
+    public void shouldTransformStringToBase64() {
+        MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
+                "to_base64('data.title')"),
+                i -> {
+                    i.startRecord("1");
+                    i.startEntity("data");
+                    i.literal("title", "this-is-a-test");
+                    i.endEntity();
+                    i.endRecord();
+                },
+                o -> {
+                    o.get().startRecord("1");
+                    o.get().startEntity("data");
+                    o.get().literal("title", "dGhpcy1pcy1hLXRlc3Q=");
+                    o.get().endEntity();
+                    o.get().endRecord();
+                });
+    }
+
 }