Added a new API for updating resources, tools and forwards

orpiske · orpiske · commit 401fd35f2060 · 2025-04-04T17:02:52.000+02:00
This closes issue wanaku-ai#162
diff --git a/core/core-persistence/core-persistence-api/src/main/java/ai/wanaku/core/persistence/api/WanakuRepository.java b/core/core-persistence/core-persistence-api/src/main/java/ai/wanaku/core/persistence/api/WanakuRepository.java
@@ -74,4 +74,13 @@ public interface WanakuRepository<A, B extends WanakuEntity, C> {
     default List<A> convertToModels(List<B> entities) {
         return entities.stream().map(entity -> convertToModel(entity)).collect(Collectors.toList());
     }
+
+
+    /**
+     * Updates a model to the repository.
+     *
+     * @param model the model to update
+     * @return true if the record was updated
+     */
+    boolean update(C id, A model);
 }
diff --git a/core/core-persistence/core-persistence-file/src/main/java/ai/wanaku/core/persistence/file/AbstractFileRepository.java b/core/core-persistence/core-persistence-file/src/main/java/ai/wanaku/core/persistence/file/AbstractFileRepository.java
@@ -116,4 +116,17 @@ public A findById(K id) {
     }
 
     abstract Class<T> getEntityClass();
+
+    @Override
+    public boolean update(K id, A model) {
+        final A byId = findById(id);
+        if (byId == null) {
+            return false;
+        }
+
+        deleteById(id);
+        persist(model);
+
+        return true;
+    }
 }
diff --git a/core/core-persistence/core-persistence-mongodb/src/main/java/ai/wanaku/core/persistence/mongodb/AbstractMongoDBRepository.java b/core/core-persistence/core-persistence-mongodb/src/main/java/ai/wanaku/core/persistence/mongodb/AbstractMongoDBRepository.java
@@ -8,6 +8,7 @@
 import com.mongodb.client.MongoCursor;
 import com.mongodb.client.model.Filters;
 import com.mongodb.client.result.DeleteResult;
+import com.mongodb.client.result.UpdateResult;
 import org.bson.Document;
 
 import java.util.ArrayList;
@@ -25,12 +26,17 @@ public AbstractMongoDBRepository(MongoClient mongoClient, WanakuMarshallerServic
 
     @Override
     public void persist(A model) {
+        Document document = toDocument(model);
+
+        mongoCollection().insertOne(document);
+    }
+
+    private Document toDocument(A model) {
         T entity = convertToEntity(model);
         String marshalled = wanakuMarshallerService.marshal(entity);
         Document document = Document.parse(marshalled);
         document.append("_id", entity.getId());
-
-        mongoCollection().insertOne(document);
+        return document;
     }
 
     @Override
@@ -73,4 +79,12 @@ protected MongoCollection<Document> mongoCollection() {
     }
 
     abstract Class<T> getEntityClass();
+
+    @Override
+    public boolean update(K id, A model) {
+        Document document = toDocument(model);
+        final UpdateResult id1 = mongoCollection().updateOne(Filters.eq("_id", id), document);
+
+        return id1.getModifiedCount() > 1;
+    }
 }
diff --git a/routers/wanaku-router/src/main/webui/openapi.json b/routers/wanaku-router/src/main/webui/openapi.json
@@ -339,6 +339,34 @@
         "tags" : [ "Forwards Resource" ]
       }
     },
+    "/api/v1/forwards/update" : {
+      "post" : {
+        "requestBody" : {
+          "content" : {
+            "application/json" : {
+              "schema" : {
+                "$ref" : "#/components/schemas/ForwardReference"
+              }
+            }
+          },
+          "required" : true
+        },
+        "responses" : {
+          "200" : {
+            "description" : "OK"
+          },
+          "500" : {
+            "description" : "Wanaku error",
+            "content" : { }
+          },
+          "400" : {
+            "description" : "Bad Request"
+          }
+        },
+        "summary" : "Update",
+        "tags" : [ "Forwards Resource" ]
+      }
+    },
     "/api/v1/management/info/version" : {
       "get" : {
         "responses" : {
@@ -563,6 +591,34 @@
         "tags" : [ "Resources Resource" ]
       }
     },
+    "/api/v1/resources/update" : {
+      "post" : {
+        "requestBody" : {
+          "content" : {
+            "application/json" : {
+              "schema" : {
+                "$ref" : "#/components/schemas/ResourceReference"
+              }
+            }
+          },
+          "required" : true
+        },
+        "responses" : {
+          "200" : {
+            "description" : "OK"
+          },
+          "500" : {
+            "description" : "Wanaku error",
+            "content" : { }
+          },
+          "400" : {
+            "description" : "Bad Request"
+          }
+        },
+        "summary" : "Update",
+        "tags" : [ "Resources Resource" ]
+      }
+    },
     "/api/v1/tools/add" : {
       "post" : {
         "requestBody" : {
@@ -634,6 +690,34 @@
         "summary" : "Remove",
         "tags" : [ "Tools Resource" ]
       }
+    },
+    "/api/v1/tools/update" : {
+      "post" : {
+        "requestBody" : {
+          "content" : {
+            "application/json" : {
+              "schema" : {
+                "$ref" : "#/components/schemas/ToolReference"
+              }
+            }
+          },
+          "required" : true
+        },
+        "responses" : {
+          "200" : {
+            "description" : "OK"
+          },
+          "500" : {
+            "description" : "Wanaku error",
+            "content" : { }
+          },
+          "400" : {
+            "description" : "Bad Request"
+          }
+        },
+        "summary" : "Update",
+        "tags" : [ "Tools Resource" ]
+      }
     }
   },
   "info" : {
diff --git a/routers/wanaku-router/src/main/webui/openapi.yaml b/routers/wanaku-router/src/main/webui/openapi.yaml
@@ -220,6 +220,25 @@ paths:
       summary: Remove Forward
       tags:
       - Forwards Resource
+  /api/v1/forwards/update:
+    post:
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: "#/components/schemas/ForwardReference"
+        required: true
+      responses:
+        "200":
+          description: OK
+        "500":
+          description: Wanaku error
+          content: {}
+        "400":
+          description: Bad Request
+      summary: Update
+      tags:
+      - Forwards Resource
   /api/v1/management/info/version:
     get:
       responses:
@@ -374,6 +393,25 @@ paths:
       summary: Remove
       tags:
       - Resources Resource
+  /api/v1/resources/update:
+    post:
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: "#/components/schemas/ResourceReference"
+        required: true
+      responses:
+        "200":
+          description: OK
+        "500":
+          description: Wanaku error
+          content: {}
+        "400":
+          description: Bad Request
+      summary: Update
+      tags:
+      - Resources Resource
   /api/v1/tools/add:
     post:
       requestBody:
@@ -424,6 +462,25 @@ paths:
       summary: Remove
       tags:
       - Tools Resource
+  /api/v1/tools/update:
+    post:
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: "#/components/schemas/ToolReference"
+        required: true
+      responses:
+        "200":
+          description: OK
+        "500":
+          description: Wanaku error
+          content: {}
+        "400":
+          description: Bad Request
+      summary: Update
+      tags:
+      - Tools Resource
 info:
   title: wanaku-router API
   version: 0.0.4-SNAPSHOT
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/forwards/ForwardsBean.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/forwards/ForwardsBean.java
@@ -157,4 +157,8 @@ public List<ForwardReference> listForwards() {
 
         return services.stream().toList();
     }
+
+    public void update(ForwardReference resource) {
+        forwardReferenceRepository.update(resource.getName(), resource);
+    }
 }
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/forwards/ForwardsResource.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/forwards/ForwardsResource.java
@@ -46,4 +46,12 @@ public Response removeForward(ForwardReference reference) {
     public RestResponse<WanakuResponse<List<ForwardReference>>> listForwards() {
         return RestResponse.ok(new WanakuResponse<>(forwardsBean.listForwards()));
     }
+
+    @Path("/update")
+    @POST
+    @Consumes(MediaType.APPLICATION_JSON)
+    public Response update(ForwardReference resource) throws WanakuException {
+        forwardsBean.update(resource);
+        return Response.ok().build();
+    }
 }
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/resources/ResourcesBean.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/resources/ResourcesBean.java
@@ -64,4 +64,8 @@ public void remove(String name) {
         }
 
     }
+
+    public void update(ResourceReference resource) {
+        resourceReferenceRepository.update(resource.getName(), resource);
+    }
 }
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/resources/ResourcesResource.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/resources/ResourcesResource.java
@@ -58,4 +58,12 @@ public Response remove(@QueryParam("resource") String resource) throws WanakuExc
         resourcesBean.remove(resource);
         return Response.ok().build();
     }
+
+    @Path("/update")
+    @POST
+    @Consumes(MediaType.APPLICATION_JSON)
+    public Response update(ResourceReference resource) throws WanakuException {
+        resourcesBean.update(resource);
+        return Response.ok().build();
+    }
 }
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/tools/ToolsBean.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/tools/ToolsBean.java
@@ -67,4 +67,8 @@ public void remove(String name) {
             toolReferenceRepository.deleteById(name);
         }
     }
+
+    public void update(ToolReference resource) {
+        toolReferenceRepository.update(resource.getName(), resource);
+    }
 }
diff --git a/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/tools/ToolsResource.java b/servers/wanaku-server-quarkus/src/main/java/ai/wanaku/server/quarkus/api/v1/tools/ToolsResource.java
@@ -59,4 +59,12 @@ public Response remove(@QueryParam("tool") String tool) throws WanakuException {
         toolsBean.remove(tool);
         return Response.ok().build();
     }
+
+    @Path("/update")
+    @POST
+    @Consumes(MediaType.APPLICATION_JSON)
+    public Response update(ToolReference resource) throws WanakuException {
+        toolsBean.update(resource);
+        return Response.ok().build();
+    }
 }
diff --git a/ui/src/api/wanaku-router-api.ts b/ui/src/api/wanaku-router-api.ts

Original file line number	Diff line number	Diff line change
`@@ -74,4 +74,13 @@ public interface WanakuRepository<A, B extends WanakuEntity, C> {`
`74`	`74`	`default List<A> convertToModels(List<B> entities) {`
`75`	`75`	`return entities.stream().map(entity -> convertToModel(entity)).collect(Collectors.toList());`
`76`	`76`	`}`
	`77`	`+`
	`78`	`+`
	`79`	`+ /**`
	`80`	`+ * Updates a model to the repository.`
	`81`	`+ *`
	`82`	`+ * @param model the model to update`
	`83`	`+ * @return true if the record was updated`
	`84`	`+ */`
	`85`	`+ boolean update(C id, A model);`
`77`	`86`	`}`
Original file line number	Diff line number	Diff line change
`@@ -157,4 +157,8 @@ public List<ForwardReference> listForwards() {`
`157`	`157`
`158`	`158`	`return services.stream().toList();`
`159`	`159`	`}`
	`160`	`+`
	`161`	`+ public void update(ForwardReference resource) {`
	`162`	`+ forwardReferenceRepository.update(resource.getName(), resource);`
	`163`	`+ }`
`160`	`164`	`}`
Original file line number	Diff line number	Diff line change
`@@ -46,4 +46,12 @@ public Response removeForward(ForwardReference reference) {`
`46`	`46`	`public RestResponse<WanakuResponse<List<ForwardReference>>> listForwards() {`
`47`	`47`	`return RestResponse.ok(new WanakuResponse<>(forwardsBean.listForwards()));`
`48`	`48`	`}`
	`49`	`+`
	`50`	`+ @Path("/update")`
	`51`	`+ @POST`
	`52`	`+ @Consumes(MediaType.APPLICATION_JSON)`
	`53`	`+ public Response update(ForwardReference resource) throws WanakuException {`
	`54`	`+ forwardsBean.update(resource);`
	`55`	`+ return Response.ok().build();`
	`56`	`+ }`
`49`	`57`	`}`
Original file line number	Diff line number	Diff line change
`@@ -64,4 +64,8 @@ public void remove(String name) {`
`64`	`64`	`}`
`65`	`65`
`66`	`66`	`}`
	`67`	`+`
	`68`	`+ public void update(ResourceReference resource) {`
	`69`	`+ resourceReferenceRepository.update(resource.getName(), resource);`
	`70`	`+ }`
`67`	`71`	`}`
Original file line number	Diff line number	Diff line change
`@@ -58,4 +58,12 @@ public Response remove(@QueryParam("resource") String resource) throws WanakuExc`
`58`	`58`	`resourcesBean.remove(resource);`
`59`	`59`	`return Response.ok().build();`
`60`	`60`	`}`
	`61`	`+`
	`62`	`+ @Path("/update")`
	`63`	`+ @POST`
	`64`	`+ @Consumes(MediaType.APPLICATION_JSON)`
	`65`	`+ public Response update(ResourceReference resource) throws WanakuException {`
	`66`	`+ resourcesBean.update(resource);`
	`67`	`+ return Response.ok().build();`
	`68`	`+ }`
`61`	`69`	`}`
Original file line number	Diff line number	Diff line change
`@@ -67,4 +67,8 @@ public void remove(String name) {`
`67`	`67`	`toolReferenceRepository.deleteById(name);`
`68`	`68`	`}`
`69`	`69`	`}`
	`70`	`+`
	`71`	`+ public void update(ToolReference resource) {`
	`72`	`+ toolReferenceRepository.update(resource.getName(), resource);`
	`73`	`+ }`
`70`	`74`	`}`
Original file line number	Diff line number	Diff line change
`@@ -59,4 +59,12 @@ public Response remove(@QueryParam("tool") String tool) throws WanakuException {`
`59`	`59`	`toolsBean.remove(tool);`
`60`	`60`	`return Response.ok().build();`
`61`	`61`	`}`
	`62`	`+`
	`63`	`+ @Path("/update")`
	`64`	`+ @POST`
	`65`	`+ @Consumes(MediaType.APPLICATION_JSON)`
	`66`	`+ public Response update(ToolReference resource) throws WanakuException {`
	`67`	`+ toolsBean.update(resource);`
	`68`	`+ return Response.ok().build();`
	`69`	`+ }`
`62`	`70`	`}`