From 9014b018fe3a13dba4cad2c9b737890e211da0e1 Mon Sep 17 00:00:00 2001
From: Pascal Christoph <christoph@hbz-nrw.de>
Date: Thu, 14 Nov 2024 15:50:05 +0100
Subject: [PATCH] Add option "url_safe" to "to_base64" function (#384)

- add test
- update README
---
 README.md                                     |  6 ++++-
 .../org/metafacture/metafix/FixMethod.java    |  8 ++++++-
 .../metafix/MetafixMethodTest.java            | 22 +++++++++++++++++++
 3 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index c86640c6..64a20ded 100644
--- a/README.md
+++ b/README.md
@@ -905,8 +905,12 @@ to_json("<sourceField>"[, pretty: "<boolean>"][, error_string: "<errorValue>"])
 
 Replaces the value with its Base64 encoding.
 
+Options:
+
+-`url_safe`: Whether to encode a URL. (Default: `false`)
+
 ```perl
-to_base64("<sourceField>")
+to_base64("<sourceField>"[, url_safe: "<boolean>"])
 ```
 
 [Example in Playground](https://metafacture.org/playground/?example=to_base64)
diff --git a/metafix/src/main/java/org/metafacture/metafix/FixMethod.java b/metafix/src/main/java/org/metafacture/metafix/FixMethod.java
index 644e9cc3..50cbcaa3 100644
--- a/metafix/src/main/java/org/metafacture/metafix/FixMethod.java
+++ b/metafix/src/main/java/org/metafacture/metafix/FixMethod.java
@@ -680,7 +680,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()));
+            final boolean urlSafe = getBoolean(options, "url_safe");
+            if (!urlSafe) {
+                record.transform(params.get(0), s -> Base64.getEncoder().encodeToString(s.getBytes()));
+            }
+            else {
+                record.transform(params.get(0), s -> Base64.getUrlEncoder().encodeToString(s.getBytes()));
+            }
         }
     },
     to_json {
diff --git a/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java b/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
index 8b167257..9f58ecce 100644
--- a/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
+++ b/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
@@ -4084,6 +4084,28 @@ public void shouldTransformStringToBase64() {
         );
     }
 
+    @Test
+    public void shouldTransformUrlToBase64() {
+        MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
+                "to_base64('data.title', url_safe:'true')"
+            ),
+            i -> {
+                i.startRecord("1");
+                i.startEntity("data");
+                i.literal("title", "https://www.youtube.com/watch?v=daLgsPSvD9A");
+                i.endEntity();
+                i.endRecord();
+            },
+            o -> {
+                o.get().startRecord("1");
+                o.get().startEntity("data");
+                o.get().literal("title", "aHR0cHM6Ly93d3cueW91dHViZS5jb20vd2F0Y2g_dj1kYUxnc1BTdkQ5QQ==");
+                o.get().endEntity();
+                o.get().endRecord();
+            }
+        );
+    }
+
     @Test // checkstyle-disable-line JavaNCSS
     public void shouldCreateVariableFromLiteralValue() {
         MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(