Merge pull request #1236 from navikt/deleted-document-404

Deleted document 404
navikt · Oct 8, 2024 · a985a5c · a985a5c
2 parents 9ec1507 + 21d8012
commit a985a5c
Show file tree

Hide file tree

Showing 6 changed files with 109 additions and 104 deletions.
diff --git a/src/main/kotlin/no/nav/klage/dokument/exceptions/DokumentExceptions.kt b/src/main/kotlin/no/nav/klage/dokument/exceptions/DokumentExceptions.kt
@@ -6,4 +6,6 @@ class DokumentValidationException(msg: String) : RuntimeException(msg)
 
 class SmartDocumentValidationException(msg: String, val errors: List<DocumentValidationResponse>) : RuntimeException(msg)
 
-class SvarbrevPreviewException(msg: String) : RuntimeException(msg)
+class SvarbrevPreviewException(msg: String) : RuntimeException(msg)
+
+class DocumentDoesNotExistException(msg: String) : RuntimeException(msg)
diff --git a/src/main/kotlin/no/nav/klage/dokument/service/DokumentUnderArbeidService.kt b/src/main/kotlin/no/nav/klage/dokument/service/DokumentUnderArbeidService.kt
diff --git a/src/main/kotlin/no/nav/klage/dokument/service/InnholdsfortegnelseService.kt b/src/main/kotlin/no/nav/klage/dokument/service/InnholdsfortegnelseService.kt
@@ -2,10 +2,7 @@ package no.nav.klage.dokument.service
 
 import no.nav.klage.dokument.api.mapper.DokumentMapper
 import no.nav.klage.dokument.clients.kabaljsontopdf.domain.InnholdsfortegnelseRequest
-import no.nav.klage.dokument.domain.dokumenterunderarbeid.DokumentUnderArbeidAsHoveddokument
-import no.nav.klage.dokument.domain.dokumenterunderarbeid.DokumentUnderArbeidAsVedlegg
-import no.nav.klage.dokument.domain.dokumenterunderarbeid.Innholdsfortegnelse
-import no.nav.klage.dokument.domain.dokumenterunderarbeid.JournalfoertDokumentUnderArbeidAsVedlegg
+import no.nav.klage.dokument.domain.dokumenterunderarbeid.*
 import no.nav.klage.dokument.repositories.DokumentUnderArbeidRepository
 import no.nav.klage.dokument.repositories.InnholdsfortegnelseRepository
 import no.nav.klage.kodeverk.DokumentType
@@ -41,13 +38,13 @@ class InnholdsfortegnelseService(
     }
 
     fun saveInnholdsfortegnelse(
-        dokumentUnderArbeidId: UUID,
+        dokumentUnderArbeid: DokumentUnderArbeid,
         fnr: String,
     ) {
         logger.debug("Received saveInnholdsfortegnelse")
 
         val content = getInnholdsfortegnelseAsPdf(
-            dokumentUnderArbeidId = dokumentUnderArbeidId,
+            dokumentUnderArbeid = dokumentUnderArbeid,
             fnr = fnr
         )
 
@@ -59,28 +56,26 @@ class InnholdsfortegnelseService(
         innholdsfortegnelseRepository.save(
             Innholdsfortegnelse(
                 mellomlagerId = mellomlagerId,
-                hoveddokumentId = dokumentUnderArbeidId,
+                hoveddokumentId = dokumentUnderArbeid.id,
                 created = LocalDateTime.now(),
                 modified = LocalDateTime.now(),
             )
         )
     }
 
     @Suppress("UNCHECKED_CAST")
-    fun getInnholdsfortegnelseAsPdf(dokumentUnderArbeidId: UUID, fnr: String): ByteArray {
+    fun getInnholdsfortegnelseAsPdf(dokumentUnderArbeid: DokumentUnderArbeid, fnr: String): ByteArray {
         logger.debug("Received getInnholdsfortegnelseAsPdf")
 
-        val document = dokumentUnderArbeidRepository.findById(dokumentUnderArbeidId).get()
-
-        if (document is DokumentUnderArbeidAsVedlegg) {
+        if (dokumentUnderArbeid is DokumentUnderArbeidAsVedlegg) {
             throw IllegalArgumentException("must be hoveddokument")
         }
 
-        document as DokumentUnderArbeidAsHoveddokument
+        dokumentUnderArbeid as DokumentUnderArbeidAsHoveddokument
 
-        val vedlegg = dokumentUnderArbeidCommonService.findVedleggByParentId(dokumentUnderArbeidId)
+        val vedlegg = dokumentUnderArbeidCommonService.findVedleggByParentId(dokumentUnderArbeid.id)
 
-        if (document.dokumentType in listOf(DokumentType.BREV, DokumentType.VEDTAK, DokumentType.BESLUTNING)) {
+        if (dokumentUnderArbeid.dokumentType in listOf(DokumentType.BREV, DokumentType.VEDTAK, DokumentType.BESLUTNING)) {
             if (vedlegg.any { it !is JournalfoertDokumentUnderArbeidAsVedlegg }) {
                 error("All documents must be JournalfoertDokumentUnderArbeidAsVedlegg")
             }
@@ -98,8 +93,8 @@ class InnholdsfortegnelseService(
                 InnholdsfortegnelseRequest(
                     documents = dokumentMapper.getSortedDokumentViewListForInnholdsfortegnelse(
                         allDokumenterUnderArbeid = vedlegg,
-                        behandling = behandlingService.getBehandlingForReadWithoutCheckForAccess(document.behandlingId),
-                        hoveddokument = document,
+                        behandling = behandlingService.getBehandlingForReadWithoutCheckForAccess(dokumentUnderArbeid.behandlingId),
+                        hoveddokument = dokumentUnderArbeid,
                         journalpostList = journalpostList,
                     )
                 )

diff --git a/src/main/kotlin/no/nav/klage/dokument/service/SmartDocumentService.kt b/src/main/kotlin/no/nav/klage/dokument/service/SmartDocumentService.kt
@@ -14,7 +14,6 @@ import no.nav.klage.dokument.domain.dokumenterunderarbeid.Language
 import no.nav.klage.dokument.domain.dokumenterunderarbeid.SmartdokumentUnderArbeidAsHoveddokument
 import no.nav.klage.dokument.domain.dokumenterunderarbeid.SmartdokumentUnderArbeidAsVedlegg
 import no.nav.klage.dokument.gateway.DefaultKabalSmartEditorApiGateway
-import no.nav.klage.dokument.repositories.DokumentUnderArbeidRepository
 import no.nav.klage.dokument.repositories.SmartdokumentUnderArbeidAsHoveddokumentRepository
 import no.nav.klage.dokument.repositories.SmartdokumentUnderArbeidAsVedleggRepository
 import no.nav.klage.kodeverk.DokumentType
@@ -44,7 +43,6 @@ class SmartDocumentService(
     private val kafkaInternalEventService: KafkaInternalEventService,
     private val smartDokumentUnderArbeidAsHoveddokumentRepository: SmartdokumentUnderArbeidAsHoveddokumentRepository,
     private val smartDokumentUnderArbeidAsVedleggRepository: SmartdokumentUnderArbeidAsVedleggRepository,
-    private val dokumentUnderArbeidRepository: DokumentUnderArbeidRepository,
     private val behandlingService: BehandlingService,
     private val applicationEventPublisher: ApplicationEventPublisher,
     private val dokumentMapper: DokumentMapper,
@@ -277,7 +275,7 @@ class SmartDocumentService(
         commentInput: CommentInput,
     ): CommentOutput {
         dokumentUnderArbeidService.validateWriteAccessToDocument(documentId)
-        val document = dokumentUnderArbeidRepository.findById(documentId).get()
+        val document = dokumentUnderArbeidService.getDokumentUnderArbeid(documentId)
 
         val innloggetIdent = innloggetSaksbehandlerService.getInnloggetIdent()
 
@@ -320,7 +318,7 @@ class SmartDocumentService(
         modifyCommentInput: ModifyCommentInput,
     ): CommentOutput {
         dokumentUnderArbeidService.validateWriteAccessToDocument(documentId)
-        val document = dokumentUnderArbeidRepository.findById(documentId).get()
+        val document = dokumentUnderArbeidService.getDokumentUnderArbeid(documentId)
         val innloggetIdent = innloggetSaksbehandlerService.getInnloggetIdent()
 
         val smartEditorId =
@@ -377,7 +375,7 @@ class SmartDocumentService(
         commentInput: CommentInput,
     ): CommentOutput {
         dokumentUnderArbeidService.validateWriteAccessToDocument(documentId)
-        val document = dokumentUnderArbeidRepository.findById(documentId).get()
+        val document = dokumentUnderArbeidService.getDokumentUnderArbeid(documentId)
         val innloggetIdent = innloggetSaksbehandlerService.getInnloggetIdent()
 
         val smartEditorId =
@@ -435,7 +433,7 @@ class SmartDocumentService(
         commentId: UUID,
     ) {
         dokumentUnderArbeidService.validateWriteAccessToDocument(documentId)
-        val document = dokumentUnderArbeidRepository.findById(documentId).get()
+        val document = dokumentUnderArbeidService.getDokumentUnderArbeid(documentId)
         val innloggetIdent = innloggetSaksbehandlerService.getInnloggetIdent()
 
         val smartEditorId =
@@ -526,7 +524,7 @@ class SmartDocumentService(
     }
 
     private fun getSmartEditorId(dokumentId: UUID, readOnly: Boolean): UUID {
-        val dokumentUnderArbeid = dokumentUnderArbeidRepository.findById(dokumentId).get()
+        val dokumentUnderArbeid = dokumentUnderArbeidService.getDokumentUnderArbeid(dokumentId)
 
         if (dokumentUnderArbeid !is DokumentUnderArbeidAsSmartdokument) {
             throw RuntimeException("dokument is not smartdokument")
@@ -547,12 +545,10 @@ class SmartDocumentService(
             behandlingId = behandlingId
         )
         return dokumentMapper.mapToDokumentView(
-            dokumentUnderArbeid = getDokumentUnderArbeid(dokumentId),
+            dokumentUnderArbeid = dokumentUnderArbeidService.getDokumentUnderArbeid(dokumentId),
             journalpost = null,
             smartEditorDocument = smartEditorDocument,
             behandling = behandling,
         )
     }
-
-    private fun getDokumentUnderArbeid(dokumentId: UUID) = dokumentUnderArbeidRepository.findById(dokumentId).get()
 }
diff --git a/src/main/kotlin/no/nav/klage/oppgave/config/ProblemHandlingControllerAdvice.kt b/src/main/kotlin/no/nav/klage/oppgave/config/ProblemHandlingControllerAdvice.kt
@@ -1,9 +1,6 @@
 package no.nav.klage.oppgave.config
 
-import no.nav.klage.dokument.exceptions.AttachmentTooLargeException
-import no.nav.klage.dokument.exceptions.DokumentValidationException
-import no.nav.klage.dokument.exceptions.SmartDocumentValidationException
-import no.nav.klage.dokument.exceptions.SvarbrevPreviewException
+import no.nav.klage.dokument.exceptions.*
 import no.nav.klage.oppgave.exceptions.*
 import no.nav.klage.oppgave.util.getSecureLogger
 import org.springframework.http.HttpStatus
@@ -183,6 +180,13 @@ class ProblemHandlingControllerAdvice : ResponseEntityExceptionHandler() {
     ): ProblemDetail =
         create(HttpStatus.BAD_REQUEST, ex)
 
+    @ExceptionHandler
+    fun handleDocumentDoesNotExistException(
+        ex: DocumentDoesNotExistException,
+        request: NativeWebRequest
+    ): ProblemDetail =
+        create(HttpStatus.NOT_FOUND, ex)
+
     @ExceptionHandler(IllegalUpdateException::class)
     fun handleIllegalUpdateException(
         ex: IllegalUpdateException,