Added compliance to snippet dto

ingsis-group-6 · Jul 14, 2023 · 281bb60 · 281bb60
1 parent 634c42b
commit 281bb60
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/src/main/kotlin/ingsis/snippetmanager/domains/snippet/controller/SnippetController.kt b/src/main/kotlin/ingsis/snippetmanager/domains/snippet/controller/SnippetController.kt
@@ -84,6 +84,14 @@ class SnippetController {
         return ResponseEntity(snippetService.getSnippetsByUserIdAndSnippetId(userId, ids), HttpStatus.OK)
     }
 
+    @GetMapping("/snippet/shared_with_me")
+    @ResponseBody
+    fun getSharedWithMeSnippets(@RequestHeader("Authorization") token: String, principal: Principal): ResponseEntity<List<SnippetDTO>> {
+        val ids = ShareSnippetService.getSharedWithMeSnippetsIds(token)
+        val userId = principal.name
+        return ResponseEntity(snippetService.getSnippetsFromIdList(ids), HttpStatus.OK)
+    }
+
     @GetMapping("/snippet/all")
     @ResponseBody
     fun getAllSnippet(@RequestHeader("Authorization") token: String, principal: Principal): ResponseEntity<List<SnippetDTO>> {

diff --git a/src/main/kotlin/ingsis/snippetmanager/domains/snippet/repository/SnippetRepository.kt b/src/main/kotlin/ingsis/snippetmanager/domains/snippet/repository/SnippetRepository.kt
@@ -15,4 +15,7 @@ interface SnippetRepository : JpaRepository<Snippet, UUID> {
 
     @Query("SELECT s FROM Snippet s WHERE s.ownerId = :userId OR s.id IN :snippets")
     fun findAllByUserIdAndSnippetId(userId: String, snippets: List<UUID>): List<Snippet>
+
+    @Query("SELECT s FROM Snippet s WHERE s.id IN :snippets")
+    fun findAllInIdList(snippets: List<UUID>): List<Snippet>
 }
diff --git a/src/main/kotlin/ingsis/snippetmanager/domains/snippet/service/SnippetService.kt b/src/main/kotlin/ingsis/snippetmanager/domains/snippet/service/SnippetService.kt
@@ -17,4 +17,5 @@ interface SnippetService {
     fun getSnippetsByUserIdAndSnippetId(userId: String, snippets: List<UUID>): List<SnippetDTO>
     fun validateOwnership(userId: String, snippetId: UUID)
     fun setSnippetCompliance(snippetId: UUID, compliance: ComplianceState)
+    fun getSnippetsFromIdList(ids: List<UUID>): List<SnippetDTO>
 }
diff --git a/src/main/kotlin/ingsis/snippetmanager/domains/snippet/service/SnippetServiceImpl.kt b/src/main/kotlin/ingsis/snippetmanager/domains/snippet/service/SnippetServiceImpl.kt
@@ -99,4 +99,8 @@ class SnippetServiceImpl: SnippetService {
             throw HTTPError("Snippet not found", HttpStatus.NOT_FOUND)
         }
     }
+
+    override fun getSnippetsFromIdList(ids: List<UUID>): List<SnippetDTO> {
+        return this.snippetRepository.findAllInIdList(ids).map { SnippetDTO(it) }
+    }
 }