refactoring

Author: Ian-Nara

src/main/java/com/uid2/admin/vertx/service/ServiceLinkService.java

@@ -19,6 +19,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.util.*;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 
 import static com.uid2.admin.vertx.Endpoints.*;
@@ -118,6 +119,12 @@ private void handleServiceLinkAdd(RoutingContext rc) {
                 return;
             }
 
+            String linkIdRegex = serviceProvider.getService(serviceId).getLinkIdRegex();
+            if (!validateLinkIdAgainstServiceRegex(linkId, linkIdRegex)) {
+                ResponseUtil.error(rc, 400, "link_id " + linkId + " does not match service_id " + serviceId + " link_id_regex: " + linkIdRegex);
+                return;
+            }
+
             Set<Role> serviceRoles = serviceProvider.getService(serviceId).getRoles();
             final Set<Role> roles;
             try {
@@ -270,4 +277,11 @@ private Set<Role> validateRoles(JsonArray rolesToValidate, Set<Role> serviceRole
         }
         return roles;
     }
+
+    private boolean validateLinkIdAgainstServiceRegex(String linkId, String serviceRegex) {
+        if (serviceRegex == null) {
+            return true;
+        }
+        return Pattern.matches(serviceRegex, linkId);
+    }
 }

src/test/java/com/uid2/admin/vertx/ServiceLinkServiceTest.java

@@ -13,6 +13,7 @@
 import io.vertx.junit5.VertxTestContext;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.MethodSource;
 import org.junit.jupiter.params.provider.ValueSource;
 
 import java.util.ArrayList;
@@ -584,4 +585,48 @@ void deleteServiceLink_invalidServiceId_returnsError(Vertx vertx, VertxTestConte
             testContext.completeNow();
         });
     }
+
+    @ParameterizedTest
+    @MethodSource("linkIdRegexCases")
+    void addServiceLink_linkIdRegex_validation(String linkIdRegex, String linkId, boolean expectSuccess,
+                                               Vertx vertx, VertxTestContext testContext) {
+        fakeAuth(Role.MAINTAINER);
+
+        setSites(new Site(123, "name1", false));
+        setServices(new Service(1, 123, "name1", Set.of(Role.MAINTAINER), linkIdRegex));
+
+        JsonObject jo = new JsonObject();
+        jo.put("link_id", linkId);
+        jo.put("service_id", 1);
+        jo.put("site_id", 123);
+        jo.put("name", "name1");
+        jo.put("roles", JsonArray.of(Role.MAINTAINER));
+
+        post(vertx, testContext, "api/service_link/add", jo.encode(), response -> {
+            if (expectSuccess) {
+                assertEquals(200, response.statusCode());
+                ServiceLink expected = new ServiceLink(linkId, 1, 123, "name1", Set.of(Role.MAINTAINER));
+                checkServiceLinkJson(expected, response.bodyAsJsonObject());
+                verify(serviceLinkStoreWriter, times(1)).upload(List.of(expected), null);
+            } else {
+                assertEquals(400, response.statusCode());
+                String expectedMsg = "link_id " + linkId + " does not match service_id 1 link_id_regex: " + linkIdRegex;
+                assertEquals(expectedMsg, response.bodyAsJsonObject().getString("message"));
+                verify(serviceLinkStoreWriter, never()).upload(any(), any());
+            }
+
+            verify(serviceStoreWriter, never()).upload(null, null);
+            testContext.completeNow();
+        });
+    }
+
+    private static java.util.stream.Stream<org.junit.jupiter.params.provider.Arguments> linkIdRegexCases() {
+        return java.util.stream.Stream.of(
+                org.junit.jupiter.params.provider.Arguments.of("link[0-9]+", "invalidLink", false),
+                org.junit.jupiter.params.provider.Arguments.of("link[0-9]+", "link42", true),
+                org.junit.jupiter.params.provider.Arguments.of("^[A-Z0-9]{6,20}$", "XY12345", true), // snowflake valid
+                org.junit.jupiter.params.provider.Arguments.of("^[A-Z0-9]{6,20}$", "UID2ENVIRONMENT", true), // snowflake valid
+                org.junit.jupiter.params.provider.Arguments.of("^[A-Z0-9]{6,20}$", "xy12345", false) // snowflake invalid
+        );
+    }
 }