feat: add security identity management support

src/main/java/com/coveo/pushapiclient/AliasMapping.java

@@ -0,0 +1,12 @@
+package com.coveo.pushapiclient;
+
+import java.util.Map;
+
+public class AliasMapping extends IdentityModel {
+    public String provider;
+
+    public AliasMapping(String provider, String name, SecurityIdentityType type, Map<String, String> additionalInfo) {
+        super(name, type, additionalInfo);
+        this.provider = provider;
+    }
+}

src/main/java/com/coveo/pushapiclient/IdentityModel.java

@@ -0,0 +1,15 @@
+package com.coveo.pushapiclient;
+
+import java.util.Map;
+
+public class IdentityModel {
+    public Map<String, String> additionalInfo;
+    public String name;
+    public SecurityIdentityType type;
+
+    public IdentityModel(String name, SecurityIdentityType type, Map<String, String> additionalInfo) {
+        this.name = name;
+        this.type = type;
+        this.additionalInfo = additionalInfo;
+    }
+}

src/main/java/com/coveo/pushapiclient/PlatformClient.java

@@ -42,24 +42,75 @@ public HttpResponse<String> createSource(String name, SourceVisibility sourceVis
         return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
-    public void createOrUpdateSecurityIdentity(String securityProviderId, Object securityIdentityModel) {
-        // TODO
+    public HttpResponse<String> createOrUpdateSecurityIdentity(String securityProviderId, SecurityIdentityModel securityIdentityModel) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+        URI uri = URI.create(this.getBaseProviderURL(securityProviderId) + "/permissions");
+
+        String json = new Gson().toJson(securityIdentityModel);
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .PUT(HttpRequest.BodyPublishers.ofString(json))
+                .uri(uri)
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
-    public void createOrUpdateSecurityIdentityAlias(String securityProviderId, Object securityIdentityAlias) {
-        // TODO
+    public HttpResponse<String> createOrUpdateSecurityIdentityAlias(String securityProviderId, SecurityIdentityAliasModel securityIdentityAlias) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+        URI uri = URI.create(this.getBaseProviderURL(securityProviderId) + "/mappings");
+
+        String json = new Gson().toJson(securityIdentityAlias);
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .PUT(HttpRequest.BodyPublishers.ofString(json))
+                .uri(uri)
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
-    public void deleteSecurityIdentity(String securityProviderId, Object securityIdentityToDelete) {
-        // TODO
+    public HttpResponse<String> deleteSecurityIdentity(String securityProviderId, SecurityIdentityDelete securityIdentityToDelete) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+        URI uri = URI.create(this.getBaseProviderURL(securityProviderId) + "/permissions");
+
+        String json = new Gson().toJson(securityIdentityToDelete);
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .method("DELETE", HttpRequest.BodyPublishers.ofString(json))
+                .uri(uri)
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
-    public void deleteOldSecurityIdentities(String securityProviderId, Object batchDelete) {
-        // TODO
+    public HttpResponse<String> deleteOldSecurityIdentities(String securityProviderId, SecurityIdentityDeleteOptions batchDelete) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+        URI uri = URI.create(this.getBaseProviderURL(securityProviderId) + String.format("/permissions/olderthan?orderingId=%s&queueDelay=%s", batchDelete.orderingId, batchDelete.queueDelay));
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .DELETE()
+                .uri(uri)
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
-    public void manageSecurityIdentities(String securityProviderId, Object batchConfig) {
-        // TODO
+    public HttpResponse<String> manageSecurityIdentities(String securityProviderId, SecurityIdentityBatchConfig batchConfig) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+        URI uri = URI.create(this.getBaseProviderURL(securityProviderId) + String.format("/permissions/batch?fileId=%s&orderingId=%s", batchConfig.fileId, batchConfig.orderingId));
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .PUT(HttpRequest.BodyPublishers.ofString(""))
+                .uri(uri)
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
     }
 
     public void pushDocument(String sourceId, Document doc) {
@@ -90,6 +141,14 @@ private String getBasePlatformURL() {
         return String.format("https://platform.cloud.coveo.com/rest/organizations/%s", this.organizationId);
     }
 
+    private String getBasePushURL() {
+        return String.format("https://api.cloud.coveo.com/push/v1/organizations/%s", this.organizationId);
+    }
+
+    private String getBaseProviderURL(String providerId) {
+        return String.format("%s/providers/%s", this.getBasePushURL(), providerId);
+    }
+
     private String[] getHeaders(String[]... headers) {
         String[] out = new String[]{};
         for (String[] header : headers) {

src/main/java/com/coveo/pushapiclient/SecurityIdentityAliasModel.java

@@ -0,0 +1,10 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityAliasModel extends SecurityIdentityModelBase {
+    public AliasMapping[] mappings;
+
+    public SecurityIdentityAliasModel(AliasMapping[] mappings, IdentityModel identity, IdentityModel[] wellKnowns) {
+        super(identity, wellKnowns);
+        this.mappings = mappings;
+    }
+}

src/main/java/com/coveo/pushapiclient/SecurityIdentityBatchConfig.java

@@ -0,0 +1,11 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityBatchConfig {
+    public String fileId;
+    public Long orderingId;
+
+    public SecurityIdentityBatchConfig(String fileId, Long orderingId) {
+        this.fileId = fileId;
+        this.orderingId = orderingId;
+    }
+}

src/main/java/com/coveo/pushapiclient/SecurityIdentityDelete.java

@@ -0,0 +1,9 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityDelete {
+    public IdentityModel identity;
+
+    public SecurityIdentityDelete(IdentityModel identity) {
+        this.identity = identity;
+    }
+}

src/main/java/com/coveo/pushapiclient/SecurityIdentityDeleteOptions.java

@@ -0,0 +1,11 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityDeleteOptions {
+    public Integer queueDelay;
+    public Long orderingId;
+
+    public SecurityIdentityDeleteOptions(Integer queueDelay, Long orderingId) {
+        this.queueDelay = queueDelay;
+        this.orderingId = orderingId;
+    }
+}

src/main/java/com/coveo/pushapiclient/SecurityIdentityModel.java

@@ -0,0 +1,10 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityModel extends SecurityIdentityModelBase {
+    public IdentityModel[] members;
+
+    public SecurityIdentityModel(IdentityModel[] members, IdentityModel identity, IdentityModel[] wellKnowns) {
+        super(identity, wellKnowns);
+        this.members = members;
+    }
+}

src/main/java/com/coveo/pushapiclient/SecurityIdentityModelBase.java

@@ -0,0 +1,11 @@
+package com.coveo.pushapiclient;
+
+public class SecurityIdentityModelBase {
+    public IdentityModel identity;
+    public IdentityModel[] wellKnowns;
+
+    public SecurityIdentityModelBase(IdentityModel identity, IdentityModel[] wellKnowns) {
+        this.identity = identity;
+        this.wellKnowns = wellKnowns;
+    }
+}

src/main/java/com/coveo/pushapiclient/Source.java

@@ -13,4 +13,24 @@ public Source(String apiKey, String organizationId) {
     public HttpResponse<String> create(String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
         return this.platformClient.createSource(name, sourceVisibility);
     }
+
+    public HttpResponse<String> createOrUpdateSecurityIdentity(String securityProviderId, SecurityIdentityModel securityIdentityModel) throws IOException, InterruptedException {
+        return this.platformClient.createOrUpdateSecurityIdentity(securityProviderId, securityIdentityModel);
+    }
+
+    public HttpResponse<String> createOrUpdateSecurityIdentityAlias(String securityProviderId, SecurityIdentityAliasModel securityIdentityAliasModel) throws IOException, InterruptedException {
+        return this.platformClient.createOrUpdateSecurityIdentityAlias(securityProviderId, securityIdentityAliasModel);
+    }
+
+    public HttpResponse<String> deleteSecurityIdentity(String securityProviderId, SecurityIdentityDelete securityIdentityDelete) throws IOException, InterruptedException {
+        return this.platformClient.deleteSecurityIdentity(securityProviderId, securityIdentityDelete);
+    }
+
+    public HttpResponse<String> deleteOldSecurityIdentities(String securityProviderId, SecurityIdentityDeleteOptions batchDelete) throws IOException, InterruptedException {
+        return this.platformClient.deleteOldSecurityIdentities(securityProviderId, batchDelete);
+    }
+
+    public HttpResponse<String> manageSecurityIdentities(String securityProviderId, SecurityIdentityBatchConfig batchConfig) throws IOException, InterruptedException {
+        return this.platformClient.manageSecurityIdentities(securityProviderId, batchConfig);
+    }
 }

src/main/java/com/coveo/testlocally/TestingLocally.java

@@ -1,13 +1,13 @@
 package com.coveo.testlocally;
 
-import com.coveo.pushapiclient.DocumentBuilder;
-import com.coveo.pushapiclient.Source;
-import com.coveo.pushapiclient.SourceVisibility;
+import com.coveo.pushapiclient.*;
+import com.google.gson.Gson;
 import io.github.cdimascio.dotenv.Dotenv;
 
 import java.io.IOException;
 import java.net.http.HttpResponse;
 import java.util.Date;
+import java.util.HashMap;
 
 public class TestingLocally {
     public static void main(String[] args) {
@@ -18,12 +18,51 @@ public static void main(String[] args) {
         try {
             HttpResponse res = source.create("testlocaljava", SourceVisibility.SECURED);
             System.out.println(String.format("Status: %s %s", res.statusCode(), res.body().toString()));
-        } catch (IOException e) {
-            e.printStackTrace();
-        } catch (InterruptedException e) {
+        } catch (IOException | InterruptedException e) {
             e.printStackTrace();
         }
 
 
+        IdentityModel identityModel = new IdentityModel("the_name", SecurityIdentityType.USER, new HashMap() {
+        });
+
+        IdentityModel[] identityModels = {
+                identityModel,
+        };
+
+        AliasMapping[] aliasMapping = {new AliasMapping("the_provider_name", "the_name", SecurityIdentityType.USER, new HashMap<>())};
+
+        try {
+            SecurityIdentityModel securityIdentityModel = new SecurityIdentityModel(identityModels, identityModel, identityModels);
+            String jsonSecurityIdentityModel = new Gson().toJson(securityIdentityModel);
+            System.out.println(jsonSecurityIdentityModel);
+            HttpResponse<String> resCreateOrUpdateSecurityIdentity = source.createOrUpdateSecurityIdentity("the_provider_id", securityIdentityModel);
+            System.out.println(resCreateOrUpdateSecurityIdentity.body());
+
+
+            SecurityIdentityAliasModel securityIdentityAliasModel = new SecurityIdentityAliasModel(aliasMapping, identityModel, identityModels);
+            String jsonSecurityIdentityAliasModel = new Gson().toJson(securityIdentityAliasModel);
+            System.out.println(jsonSecurityIdentityAliasModel);
+            HttpResponse<String> resCreateOrUpdateSecurityIdentityAlias = source.createOrUpdateSecurityIdentityAlias("the_provider_id_mappings", securityIdentityAliasModel);
+            System.out.println(resCreateOrUpdateSecurityIdentityAlias.body());
+
+
+            SecurityIdentityDelete securityIdentityDelete = new SecurityIdentityDelete(identityModel);
+            String jsonSecurityIdentityDelete = new Gson().toJson(securityIdentityDelete);
+            System.out.println(jsonSecurityIdentityDelete);
+            HttpResponse<String> resDeleteSecurityIdentity = source.deleteSecurityIdentity("the_provider_id_delete", securityIdentityDelete);
+            System.out.println(resDeleteSecurityIdentity.body());
+
+            SecurityIdentityDeleteOptions batchDelete = new SecurityIdentityDeleteOptions(123, 987l);
+            HttpResponse<String> resDeleteOlderThan = source.deleteOldSecurityIdentities("the_provider_id_delete_older_than", batchDelete);
+            System.out.println(resDeleteOlderThan.body());
+
+            SecurityIdentityBatchConfig batchConfig = new SecurityIdentityBatchConfig("the_file_id", 123l);
+            HttpResponse<String> resBatchConfig = source.manageSecurityIdentities("the_provider_id_batch_config", batchConfig);
+            System.out.println(resBatchConfig.body());
+
+        } catch (IOException | InterruptedException e) {
+            e.printStackTrace();
+        }
     }
 }