119 lines
4.1 KiB
Kotlin
119 lines
4.1 KiB
Kotlin
package com.cubetiqs.web.modules.uploader
|
|
|
|
import com.cubetiqs.web.util.RouteConstants
|
|
import io.swagger.v3.oas.annotations.Operation
|
|
import io.swagger.v3.oas.annotations.Parameter
|
|
import io.swagger.v3.oas.annotations.tags.Tag
|
|
import org.springdoc.core.converters.models.PageableAsQueryParam
|
|
import org.springframework.beans.factory.annotation.Autowired
|
|
import org.springframework.data.domain.Page
|
|
import org.springframework.data.domain.Pageable
|
|
import org.springframework.http.MediaType
|
|
import org.springframework.util.FileCopyUtils
|
|
import org.springframework.web.bind.annotation.*
|
|
import org.springframework.web.multipart.MultipartFile
|
|
import java.util.*
|
|
import javax.servlet.http.HttpServletResponse
|
|
|
|
@UploaderModule
|
|
@Tag(name = "Uploader Controller")
|
|
@RestController
|
|
@RequestMapping(RouteConstants.INDEX + "uploader")
|
|
class UploaderController @Autowired constructor(
|
|
private val repository: UploaderRepository,
|
|
) {
|
|
@GetMapping
|
|
@PageableAsQueryParam
|
|
@Operation(summary = "Get all files")
|
|
fun getAll(
|
|
@Parameter(hidden = true)
|
|
pageable: Pageable?,
|
|
): Page<UploaderEntity> {
|
|
return repository.findAll(pageable ?: Pageable.unpaged())
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.OK)
|
|
@GetMapping("/{id}")
|
|
@Operation(summary = "Get a file by id")
|
|
fun get(
|
|
@PathVariable id: String,
|
|
): UploaderEntity {
|
|
val entity = repository.findById(UUID.fromString(id)).orElseThrow {
|
|
throw IllegalArgumentException("File not found")
|
|
}
|
|
return repository.save(entity)
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.OK)
|
|
@GetMapping("/{id}/stream", produces = [MediaType.APPLICATION_OCTET_STREAM_VALUE])
|
|
@Operation(summary = "Get file stream by id")
|
|
fun stream(
|
|
@PathVariable id: String,
|
|
@RequestParam(required = false, value = "download") download: Boolean?,
|
|
response: HttpServletResponse,
|
|
) {
|
|
val entity = repository.findById(UUID.fromString(id)).orElseThrow {
|
|
throw IllegalArgumentException("User not found")
|
|
}
|
|
|
|
if (!entity.isFileExists()) {
|
|
throw IllegalArgumentException("File not found")
|
|
}
|
|
|
|
val file = entity.getFile() ?: throw IllegalArgumentException("File source not found")
|
|
val disposition = if (download == true) {
|
|
"attachment"
|
|
} else {
|
|
"inline"
|
|
}
|
|
|
|
response.setHeader("Content-Disposition", "$disposition; filename=\"${entity.filename}\"")
|
|
response.contentType = entity.contentType
|
|
response.setContentLengthLong(file.length())
|
|
|
|
FileCopyUtils.copy(file.readBytes(), response.outputStream)
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.CREATED)
|
|
@PostMapping(consumes = [MediaType.MULTIPART_FORM_DATA_VALUE])
|
|
@Operation(summary = "Upload a file")
|
|
fun create(
|
|
@RequestPart file: MultipartFile,
|
|
): UploaderEntity {
|
|
val entity = UploaderEntity.fromFile(file)
|
|
return repository.save(entity)
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.OK)
|
|
@PutMapping("/{id}")
|
|
@Operation(summary = "Update a file by id")
|
|
fun update(
|
|
@PathVariable id: String,
|
|
@RequestBody body: UploaderEntity
|
|
): UploaderEntity {
|
|
val entity = repository.findById(UUID.fromString(id)).orElseThrow {
|
|
throw IllegalArgumentException("File not found")
|
|
}
|
|
body.id = entity.id
|
|
return repository.save(body)
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.NO_CONTENT)
|
|
@DeleteMapping("/{id}")
|
|
@Operation(summary = "Delete a file by id")
|
|
fun delete(
|
|
@PathVariable id: String,
|
|
) {
|
|
val entity = repository.findById(UUID.fromString(id)).orElseThrow {
|
|
throw IllegalArgumentException("File not found")
|
|
}
|
|
repository.delete(entity)
|
|
}
|
|
|
|
@ResponseStatus(value = org.springframework.http.HttpStatus.NO_CONTENT)
|
|
@DeleteMapping
|
|
@Operation(summary = "Delete all files")
|
|
fun deleteAll() {
|
|
repository.deleteAll()
|
|
}
|
|
} |