Backup_Repos/komga
1
0
Fork 0
mirror of https://github.com/gotson/komga.git synced 2026-05-08 04:22:28 +02:00
Code Issues Projects Releases Wiki Activity

feat(api): reAnalyze library

Browse source
This commit is contained in:
Gauthier Roebroeck 2020-01-06 11:17:02 +08:00
parent 94faf157d6
commit fa65e94ae6
2 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
komga/src/main/kotlin/org/gotson/komga/domain/persistence/BookRepository.kt
View file

@ -27,6 +27,8 @@ interface BookRepository : JpaRepository<Book, Long>, JpaSpecificationExecutor<B
@QueryHints(QueryHint(name = CACHEABLE, value = "true")) @QueryHints(QueryHint(name = CACHEABLE, value = "true"))
fun findBySeriesLibraryIn(seriesLibrary: Collection<Library>, pageable: Pageable): Page<Book> fun findBySeriesLibraryIn(seriesLibrary: Collection<Library>, pageable: Pageable): Page<Book>
fun findBySeriesLibraryIn(seriesLibrary: Collection<Library>): List<Book>
fun findByUrl(url: URL): Book? fun findByUrl(url: URL): Book?
fun findAllByMediaStatus(status: Media.Status): List<Book> fun findAllByMediaStatus(status: Media.Status): List<Book>
fun findAllByMediaThumbnailIsNull(): List<Book> fun findAllByMediaThumbnailIsNull(): List<Book>

20
komga/src/main/kotlin/org/gotson/komga/interfaces/web/rest/LibraryController.kt
View file

@ -5,7 +5,9 @@ import org.gotson.komga.domain.model.DirectoryNotFoundException
import org.gotson.komga.domain.model.DuplicateNameException import org.gotson.komga.domain.model.DuplicateNameException
import org.gotson.komga.domain.model.Library import org.gotson.komga.domain.model.Library
import org.gotson.komga.domain.model.PathContainedInPath import org.gotson.komga.domain.model.PathContainedInPath
import org.gotson.komga.domain.persistence.BookRepository
import org.gotson.komga.domain.persistence.LibraryRepository import org.gotson.komga.domain.persistence.LibraryRepository
import org.gotson.komga.domain.service.AsyncOrchestrator
import org.gotson.komga.domain.service.LibraryLifecycle import org.gotson.komga.domain.service.LibraryLifecycle
import org.gotson.komga.infrastructure.security.KomgaPrincipal import org.gotson.komga.infrastructure.security.KomgaPrincipal
import org.springframework.data.domain.Sort import org.springframework.data.domain.Sort
@ -24,6 +26,7 @@ import org.springframework.web.bind.annotation.ResponseStatus
import org.springframework.web.bind.annotation.RestController import org.springframework.web.bind.annotation.RestController
import org.springframework.web.server.ResponseStatusException import org.springframework.web.server.ResponseStatusException
import java.io.FileNotFoundException import java.io.FileNotFoundException
import java.util.concurrent.RejectedExecutionException
import javax.validation.Valid import javax.validation.Valid
import javax.validation.constraints.NotBlank import javax.validation.constraints.NotBlank
@ -33,7 +36,9 @@ private val logger = KotlinLogging.logger {}
@RequestMapping("api/v1/libraries", produces = [MediaType.APPLICATION_JSON_VALUE]) @RequestMapping("api/v1/libraries", produces = [MediaType.APPLICATION_JSON_VALUE])
class LibraryController( class LibraryController(
private val libraryLifecycle: LibraryLifecycle, private val libraryLifecycle: LibraryLifecycle,
private val libraryRepository: LibraryRepository private val libraryRepository: LibraryRepository,
private val bookRepository: BookRepository,
private val asyncOrchestrator: AsyncOrchestrator
) { ) {
@GetMapping @GetMapping
@ -83,6 +88,19 @@ class LibraryController(
libraryLifecycle.deleteLibrary(it) libraryLifecycle.deleteLibrary(it)
} ?: throw ResponseStatusException(HttpStatus.NOT_FOUND) } ?: throw ResponseStatusException(HttpStatus.NOT_FOUND)
} }
@PostMapping("{libraryId}/analyze")
@PreAuthorize("hasRole('ROLE_ADMIN')")
@ResponseStatus(HttpStatus.ACCEPTED)
fun analyze(@PathVariable libraryId: Long) {
libraryRepository.findByIdOrNull(libraryId)?.let { library ->
try {
asyncOrchestrator.reAnalyzeBooks(bookRepository.findBySeriesLibraryIn(listOf(library)))
} catch (e: RejectedExecutionException) {
throw ResponseStatusException(HttpStatus.SERVICE_UNAVAILABLE, "Another book analysis task is already running")
}
} ?: throw ResponseStatusException(HttpStatus.NOT_FOUND)
}
} }
data class LibraryCreationDto( data class LibraryCreationDto(