Enable users to delete their images

README.md

@@ -48,6 +48,6 @@
 
 - `POST   /images/upload`             Upload image
 - `GET    /images/{id}`               Get image link
+- `DELETE /images/{id}`         Delete my image
 - `GET    /images/my`           My uploaded images
 - `GET    /images`              Home / search images      **_Public_**
-- `DELETE /images/{id}`         Delete my image

src/main/java/dev/barboud/taglens/controller/ImageController.java

@@ -44,11 +44,12 @@ public ResponseEntity<Map<String, String>> getImageLink(
     }
 
 
-//
-//    @DeleteMapping
-//    public ResponseEntity<Void> deleteImage(@RequestParam String key) {
-//        imageService.deleteImage(key);
-//        return ResponseEntity.noContent().build();
-//    }
+    @DeleteMapping("/{id}")
+    public ResponseEntity<Void> deleteImage(
+            @PathVariable UUID id,
+            @CookieValue("session_id") UUID sessionId) {
+        imageService.deleteImage(id, sessionId);
+        return ResponseEntity.noContent().build();
+    }
 
 }

src/main/java/dev/barboud/taglens/repository/ImageRepository.java

@@ -52,4 +52,12 @@ public Optional<Image> findImageById(UUID imageId) {
         return results.stream().findFirst();
     }
 
+    public void deleteImage(UUID imageId) {
+        String sql = """
+        DELETE FROM images
+        WHERE id = ?
+        """;
+        jdbcTemplate.update(sql, imageId);
+    }
+
 }

src/main/java/dev/barboud/taglens/service/ImageService.java

@@ -88,4 +88,17 @@ public String getImageLink(UUID imageId, UUID sessionId) {
 
         return storageService.getLink(image.getStorageKey(), 15);
     }
+
+    public void deleteImage(UUID imageId, UUID sessionId) {
+        Session session = sessionRepository.findValidSession(sessionId)
+                .orElseThrow(() -> new RuntimeException("Unauthorized"));
+        Image image = imageRepository.findImageById(imageId)
+                .orElseThrow(() -> new RuntimeException("Not found"));
+        if (!image.getUserId().equals(session.getUserId())) {
+            throw new RuntimeException("Not found");
+        }
+
+        storageService.delete(image.getStorageKey());
+        imageRepository.deleteImage(imageId);
+    }
 }