From 181fcd089a256259a23dd81d0805525360d30bf6 Mon Sep 17 00:00:00 2001
From: Olivier Lamothe <olamothe@coveo.com>
Date: Wed, 9 Jun 2021 13:57:14 -0400
Subject: [PATCH 1/3] feat: create platform client class

---
 .env.sample                                   |   3 +
 .gitignore                                    |   1 +
 .idea/saveactions_settings.xml                |   1 +
 pom.xml                                       |   5 +
 .../coveo/pushapiclient/PlatformClient.java   | 114 ++++++++++++++++++
 .../java/com/coveo/pushapiclient/Source.java  |  16 +++
 .../coveo/pushapiclient/SourceVisibility.java |  19 +++
 .../com/coveo/testlocally/TestingLocally.java |  19 ++-
 8 files changed, 177 insertions(+), 1 deletion(-)
 create mode 100644 .env.sample
 create mode 100644 src/main/java/com/coveo/pushapiclient/PlatformClient.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/Source.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SourceVisibility.java

diff --git a/.env.sample b/.env.sample
new file mode 100644
index 00000000..856c5819
--- /dev/null
+++ b/.env.sample
@@ -0,0 +1,3 @@
+API_KEY=[REPLACE]
+ORG_ID=[REPLACE]
+SOURCE_ID=[REPLACE]
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
index b83d2226..5dc70ab9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 /target/
+.env
\ No newline at end of file
diff --git a/.idea/saveactions_settings.xml b/.idea/saveactions_settings.xml
index 6025467f..8a7a18ad 100644
--- a/.idea/saveactions_settings.xml
+++ b/.idea/saveactions_settings.xml
@@ -6,6 +6,7 @@
         <option value="activate" />
         <option value="organizeImports" />
         <option value="reformat" />
+        <option value="rearrange" />
       </set>
     </option>
     <option name="configurationPath" value="" />
diff --git a/pom.xml b/pom.xml
index 74d63750..63bef1d5 100644
--- a/pom.xml
+++ b/pom.xml
@@ -13,6 +13,11 @@
             <artifactId>gson</artifactId>
             <version>2.8.7</version>
         </dependency>
+        <dependency>
+            <groupId>io.github.cdimascio</groupId>
+            <artifactId>dotenv-java</artifactId>
+            <version>2.2.0</version>
+        </dependency>
         <dependency>
             <groupId>joda-time</groupId>
             <artifactId>joda-time</artifactId>
diff --git a/src/main/java/com/coveo/pushapiclient/PlatformClient.java b/src/main/java/com/coveo/pushapiclient/PlatformClient.java
new file mode 100644
index 00000000..bd368570
--- /dev/null
+++ b/src/main/java/com/coveo/pushapiclient/PlatformClient.java
@@ -0,0 +1,114 @@
+package com.coveo.pushapiclient;
+
+import com.google.gson.Gson;
+import com.google.gson.reflect.TypeToken;
+
+import java.io.IOException;
+import java.net.URI;
+import java.net.http.HttpClient;
+import java.net.http.HttpRequest;
+import java.net.http.HttpResponse;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.stream.Stream;
+
+public class PlatformClient {
+    private String apiKey;
+    private String organizationId;
+    private HttpClient httpClient;
+
+    public PlatformClient(String apiKey, String organizationId) {
+        this.apiKey = apiKey;
+        this.organizationId = organizationId;
+        this.httpClient = HttpClient.newHttpClient();
+    }
+
+    public HttpResponse<String> createSource(String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
+        String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
+
+        String json = this.toJSON(new HashMap<>() {{
+            put("sourceType", "PUSH");
+            put("pushEnabled", true);
+            put("name", name);
+            put("sourceVisibility", sourceVisibility);
+        }});
+
+        HttpRequest request = HttpRequest.newBuilder()
+                .headers(headers)
+                .POST(HttpRequest.BodyPublishers.ofString(json))
+                .uri(URI.create(this.getBaseSourceURL()))
+                .build();
+
+        return this.httpClient.send(request, HttpResponse.BodyHandlers.ofString());
+    }
+
+    public void createOrUpdateSecurityIdentity(String securityProviderId, Object securityIdentityModel) {
+        // TODO
+    }
+
+    public void createOrUpdateSecurityIdentityAlias(String securityProviderId, Object securityIdentityAlias) {
+        // TODO
+    }
+
+    public void deleteSecurityIdentity(String securityProviderId, Object securityIdentityToDelete) {
+        // TODO
+    }
+
+    public void deleteOldSecurityIdentities(String securityProviderId, Object batchDelete) {
+        // TODO
+    }
+
+    public void manageSecurityIdentities(String securityProviderId, Object batchConfig) {
+        // TODO
+    }
+
+    public void pushDocument(String sourceId, Document doc) {
+        // TODO
+    }
+
+    public void deleteDocument(String sourceId, String documentId, Boolean deleteChildren) {
+        // TODO
+    }
+
+    public void createFileContainer() {
+        // TODO
+    }
+
+    public void uploadContentToFileContainer(String sourceId, Object fileContainer) {
+        // TODO
+    }
+
+    public void pushFileContainerContent(String sourceId, Object fileContainer) {
+        // TODO
+    }
+
+    private String getBaseSourceURL() {
+        return String.format("%s/sources", this.getBasePlatformURL());
+    }
+
+    private String getBasePlatformURL() {
+        return String.format("https://platform.cloud.coveo.com/rest/organizations/%s", this.organizationId);
+    }
+
+    private String[] getHeaders(String[]... headers) {
+        String[] out = new String[]{};
+        for (String[] header : headers) {
+            out = Stream.concat(Arrays.stream(out), Arrays.stream(header))
+                    .toArray(String[]::new);
+        }
+        return out;
+    }
+
+    private String[] getAuthorizationHeader() {
+        return new String[]{"Authorization", String.format("Bearer %s", this.apiKey)};
+    }
+
+    private String[] getContentTypeApplicationJSONHeader() {
+        return new String[]{"Content-Type", "application/json", "Accept", "application/json"};
+    }
+
+    private String toJSON(HashMap<String, Object> hashMap) {
+        return new Gson().toJson(hashMap, new TypeToken<HashMap<String, Object>>() {
+        }.getType());
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/Source.java b/src/main/java/com/coveo/pushapiclient/Source.java
new file mode 100644
index 00000000..656c9472
--- /dev/null
+++ b/src/main/java/com/coveo/pushapiclient/Source.java
@@ -0,0 +1,16 @@
+package com.coveo.pushapiclient;
+
+import java.io.IOException;
+import java.net.http.HttpResponse;
+
+public class Source {
+    PlatformClient platformClient;
+
+    public Source(String apiKey, String organizationId) {
+        this.platformClient = new PlatformClient(apiKey, organizationId);
+    }
+
+    public HttpResponse<String> create(String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
+        return this.platformClient.createSource(name, sourceVisibility);
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SourceVisibility.java b/src/main/java/com/coveo/pushapiclient/SourceVisibility.java
new file mode 100644
index 00000000..84da38d2
--- /dev/null
+++ b/src/main/java/com/coveo/pushapiclient/SourceVisibility.java
@@ -0,0 +1,19 @@
+package com.coveo.pushapiclient;
+
+public enum SourceVisibility {
+    PRIVATE {
+        public String toString() {
+            return "PRIVATE";
+        }
+    },
+    SECURED {
+        public String toString() {
+            return "SECURED";
+        }
+    },
+    SHARED {
+        public String toString() {
+            return "SHARED";
+        }
+    }
+}
diff --git a/src/main/java/com/coveo/testlocally/TestingLocally.java b/src/main/java/com/coveo/testlocally/TestingLocally.java
index 6d5f9bce..a3455ceb 100644
--- a/src/main/java/com/coveo/testlocally/TestingLocally.java
+++ b/src/main/java/com/coveo/testlocally/TestingLocally.java
@@ -1,12 +1,29 @@
 package com.coveo.testlocally;
 
 import com.coveo.pushapiclient.DocumentBuilder;
+import com.coveo.pushapiclient.Source;
+import com.coveo.pushapiclient.SourceVisibility;
+import io.github.cdimascio.dotenv.Dotenv;
 
+import java.io.IOException;
+import java.net.http.HttpResponse;
 import java.util.Date;
 
 public class TestingLocally {
     public static void main(String[] args) {
+        Dotenv dotenv = Dotenv.load();
+
         DocumentBuilder doc = new DocumentBuilder("https://perdu.com", "the title").withData("this is searchable").withDate(new Date());
-        System.out.println(doc.marshal());
+        Source source = new Source(dotenv.get("API_KEY"), dotenv.get("ORG_ID"));
+        try {
+            HttpResponse res = source.create(dotenv.get("SOURCE_ID"), SourceVisibility.SECURED);
+            System.out.println(String.format("Status: %s %s", res.statusCode(), res.body().toString()));
+        } catch (IOException e) {
+            e.printStackTrace();
+        } catch (InterruptedException e) {
+            e.printStackTrace();
+        }
+
+
     }
 }

From 2ee091ea2008c123cfd163b80c28324ff090dd49 Mon Sep 17 00:00:00 2001
From: Olivier Lamothe <olamothe@coveo.com>
Date: Wed, 9 Jun 2021 13:59:32 -0400
Subject: [PATCH 2/3] fix source name

---
 src/main/java/com/coveo/testlocally/TestingLocally.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/coveo/testlocally/TestingLocally.java b/src/main/java/com/coveo/testlocally/TestingLocally.java
index a3455ceb..f72a694a 100644
--- a/src/main/java/com/coveo/testlocally/TestingLocally.java
+++ b/src/main/java/com/coveo/testlocally/TestingLocally.java
@@ -16,7 +16,7 @@ public static void main(String[] args) {
         DocumentBuilder doc = new DocumentBuilder("https://perdu.com", "the title").withData("this is searchable").withDate(new Date());
         Source source = new Source(dotenv.get("API_KEY"), dotenv.get("ORG_ID"));
         try {
-            HttpResponse res = source.create(dotenv.get("SOURCE_ID"), SourceVisibility.SECURED);
+            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();

From af766095a633e2ef2c0e378dbd5eecc81ae5c8c5 Mon Sep 17 00:00:00 2001
From: Olivier Lamothe <olamothe@coveo.com>
Date: Mon, 28 Jun 2021 11:03:31 -0400
Subject: [PATCH 3/3] feat: add security identity management support

---
 .../com/coveo/pushapiclient/AliasMapping.java | 12 +++
 .../coveo/pushapiclient/IdentityModel.java    | 15 ++++
 .../coveo/pushapiclient/PlatformClient.java   | 79 ++++++++++++++++---
 .../SecurityIdentityAliasModel.java           | 10 +++
 .../SecurityIdentityBatchConfig.java          | 11 +++
 .../pushapiclient/SecurityIdentityDelete.java |  9 +++
 .../SecurityIdentityDeleteOptions.java        | 11 +++
 .../pushapiclient/SecurityIdentityModel.java  | 10 +++
 .../SecurityIdentityModelBase.java            | 11 +++
 .../java/com/coveo/pushapiclient/Source.java  | 20 +++++
 .../com/coveo/testlocally/TestingLocally.java | 51 ++++++++++--
 11 files changed, 223 insertions(+), 16 deletions(-)
 create mode 100644 src/main/java/com/coveo/pushapiclient/AliasMapping.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/IdentityModel.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityAliasModel.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityBatchConfig.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityDelete.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityDeleteOptions.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityModel.java
 create mode 100644 src/main/java/com/coveo/pushapiclient/SecurityIdentityModelBase.java

diff --git a/src/main/java/com/coveo/pushapiclient/AliasMapping.java b/src/main/java/com/coveo/pushapiclient/AliasMapping.java
new file mode 100644
index 00000000..b2be0f92
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/IdentityModel.java b/src/main/java/com/coveo/pushapiclient/IdentityModel.java
new file mode 100644
index 00000000..beb8c8fb
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/PlatformClient.java b/src/main/java/com/coveo/pushapiclient/PlatformClient.java
index bd368570..2e2493f6 100644
--- a/src/main/java/com/coveo/pushapiclient/PlatformClient.java
+++ b/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("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/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("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/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) {
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityAliasModel.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityAliasModel.java
new file mode 100644
index 00000000..39d7dbcf
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityBatchConfig.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityBatchConfig.java
new file mode 100644
index 00000000..8c445ac5
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityDelete.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityDelete.java
new file mode 100644
index 00000000..3de358c6
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityDeleteOptions.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityDeleteOptions.java
new file mode 100644
index 00000000..a46e8aab
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityModel.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityModel.java
new file mode 100644
index 00000000..14d2d817
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/SecurityIdentityModelBase.java b/src/main/java/com/coveo/pushapiclient/SecurityIdentityModelBase.java
new file mode 100644
index 00000000..aa55c961
--- /dev/null
+++ b/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;
+    }
+}
diff --git a/src/main/java/com/coveo/pushapiclient/Source.java b/src/main/java/com/coveo/pushapiclient/Source.java
index 656c9472..ffc4a3c0 100644
--- a/src/main/java/com/coveo/pushapiclient/Source.java
+++ b/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);
+    }
 }
diff --git a/src/main/java/com/coveo/testlocally/TestingLocally.java b/src/main/java/com/coveo/testlocally/TestingLocally.java
index f72a694a..cdf33fba 100644
--- a/src/main/java/com/coveo/testlocally/TestingLocally.java
+++ b/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();
+        }
     }
 }