From 903c2beb33052fd72bceb79a59e8a9a849e8d72c Mon Sep 17 00:00:00 2001
From: Andrei Nechaev <anechaev@nuxeo.com>
Date: Wed, 19 Jan 2022 12:17:51 +0300
Subject: [PATCH] INSIGHT-1286: dedup drop api

---
 .../java/org/nuxeo/ai/sdk/rest/api/DedupCaller.java  | 12 ++++++++++++
 .../java/org/nuxeo/ai/sdk/rest/api/Resource.java     |  5 +++++
 .../main/java/org/nuxeo/ai/sdk/rest/client/API.java  |  4 +++-
 .../org/nuxeo/ai/sdk/rest/api/TestDedupCaller.java   |  7 +++++++
 .../test/resources/mappings/insight_dedup_drop.json  |  9 +++++++++
 5 files changed, 36 insertions(+), 1 deletion(-)
 create mode 100644 insight-rest-client/src/test/resources/mappings/insight_dedup_drop.json

diff --git a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/DedupCaller.java b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/DedupCaller.java
index d48315a..a1ff015 100644
--- a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/DedupCaller.java
+++ b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/DedupCaller.java
@@ -73,6 +73,12 @@ public DedupCaller(InsightClient client, API.Dedup type) {
         this.type = type;
     }
 
+    @Nullable
+    @Override
+    public <T> T call() throws IOException {
+        return call(emptyMap());
+    }
+
     @Nullable
     @Override
     public <T> T call(Map<String, Serializable> parameters) throws IOException {
@@ -94,6 +100,8 @@ public <T> T call(Map<String, Serializable> parameters, Serializable payload) th
             return (T) handleRecalculateTuples(parameters);
         case DELETE:
             return (T) handleDelete(parameters);
+        case DROP:
+            return (T) handleDrop();
         default:
             throw new InvalidEndpointException("No such endpoint " + this.type.name());
         }
@@ -166,6 +174,10 @@ private Boolean handleDelete(Map<String, Serializable> parameters) {
         return client.delete(this.type.toPath(DELETE, client.getProjectId(), parameters), "{}", Response::isSuccessful);
     }
 
+    private Boolean handleDrop() {
+        return client.delete(this.type.toPath(DELETE, client.getProjectId()), "{}", Response::isSuccessful);
+    }
+
     protected ResponseHandler<List<String>> handleResponse(String docId, String xpath) {
         return response -> {
             if (!response.isSuccessful()) {
diff --git a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/Resource.java b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/Resource.java
index 8fa6150..d3ad09d 100644
--- a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/Resource.java
+++ b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/Resource.java
@@ -30,6 +30,11 @@
  */
 public interface Resource {
 
+    @Nullable
+    default <T> T call() throws IOException {
+        throw new UnsupportedOperationException();
+    }
+
     @Nullable
     <T> T call(Map<String, Serializable> parameters) throws IOException;
 
diff --git a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/client/API.java b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/client/API.java
index de9856d..66aeca4 100644
--- a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/client/API.java
+++ b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/client/API.java
@@ -131,7 +131,7 @@ public String toPath(@Nonnull String project, @Nullable String id, @Nullable Str
      * Endpoints available for Deduplication
      */
     public enum Dedup implements Endpoint {
-        INDEX, FIND, ALL, RECALCULATETUPLES, DELETE;
+        INDEX, FIND, ALL, RECALCULATETUPLES, DELETE, DROP;
 
         public static final String API_DEDUP = "ai/dedup/";
 
@@ -183,6 +183,8 @@ public String toPath(HttpMethod method, @Nonnull String project, Map<String, Ser
                     url += "?xpath=" + xpath;
                 }
                 return url;
+            case DROP:
+                return API_DEDUP + project + "/drop";
             default:
                 throw new UnsupportedPathException("Invalid API call for " + this.name());
             }
diff --git a/insight-rest-client/src/test/java/org/nuxeo/ai/sdk/rest/api/TestDedupCaller.java b/insight-rest-client/src/test/java/org/nuxeo/ai/sdk/rest/api/TestDedupCaller.java
index 80fb885..6d23259 100644
--- a/insight-rest-client/src/test/java/org/nuxeo/ai/sdk/rest/api/TestDedupCaller.java
+++ b/insight-rest-client/src/test/java/org/nuxeo/ai/sdk/rest/api/TestDedupCaller.java
@@ -145,6 +145,13 @@ public void iCanDeleteFromIndex() throws IOException {
         assertThat(result).isTrue();
     }
 
+    @Test
+    public void shouldCallDrop() throws IOException {
+        InsightClient client = getInsightClient();
+        Boolean result = client.api(Dedup.DROP).call();
+        assertThat(result).isTrue();
+    }
+
     private TensorInstances createTensor(String docId) {
         return new TensorInstances(docId, Collections.emptyList());
     }
diff --git a/insight-rest-client/src/test/resources/mappings/insight_dedup_drop.json b/insight-rest-client/src/test/resources/mappings/insight_dedup_drop.json
new file mode 100644
index 0000000..0fa1aee
--- /dev/null
+++ b/insight-rest-client/src/test/resources/mappings/insight_dedup_drop.json
@@ -0,0 +1,9 @@
+{
+  "request": {
+    "method": "DELETE",
+    "url": "/api/v1/ai/dedup/test/drop"
+  },
+  "response": {
+    "status": 200
+  }
+}