deepcrayon
/
lila
1
0
Fork 0
Code Releases Activity

store picfit image extension in the ID

pull/9705/head
Thibault Duplessis 2021-09-01 18:24:23 +02:00
parent 68603c3821
commit fad5cad609
1 changed files with 41 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
modules/memo/src/main/Picfit.scala
View File

@ -46,9 +46,15 @@ final class PicfitApi(coll: Coll, ws: StandaloneWSClient, config: PicfitConfig)(
def upload(rel: String, uploaded: Uploaded, userId: String): Fu[PicfitImage] =
if (uploaded.fileSize > uploadMaxBytes)
fufail(s"File size must not exceed ${uploadMaxMb}MB.")
else {
else
uploaded.contentType collect {
case "image/png" => "png"
case "image/jpeg" => "jpg"
} match {
case None => fufail(s"Invalid file type: ${uploaded.contentType | "unknown"}")
case Some(extension) => {
val image = PicfitImage(
_id = PicfitImage.Id(lila.common.ThreadLocalRandom nextString 10),
_id = PicfitImage.Id(s"${lila.common.ThreadLocalRandom nextString 10}.$extension"),
user = userId,
rel = rel,
name = sanitizeName(uploaded.filename),
@ -60,6 +66,7 @@ final class PicfitApi(coll: Coll, ws: StandaloneWSClient, config: PicfitConfig)(
deletePrevious(image) >>
coll.insert.one(image) inject image
}
}
private def deletePrevious(image: PicfitImage): Funit =
coll
@ -74,17 +81,12 @@ final class PicfitApi(coll: Coll, ws: StandaloneWSClient, config: PicfitConfig)(
private object picfitServer {
def store(image: PicfitImage, from: Uploaded): Funit = from.contentType collect {
case "image/png" => "png"
case "image/jpeg" => "jpg"
} match {
case None => fufail(s"Invalid file type: ${from.contentType | "unknown"}")
case Some(extension) => {
def store(image: PicfitImage, from: Uploaded): Funit = {
type Part = MultipartFormData.FilePart[Source[ByteString, _]]
import WSBodyWritables._
val part: Part = MultipartFormData.FilePart(
key = "data",
filename = s"${image.id.value}.$extension",
filename = image.id.value,
contentType = from.contentType,
ref = FileIO.fromPath(from.ref.path),
fileSize = from.fileSize
@ -100,7 +102,6 @@ final class PicfitApi(coll: Coll, ws: StandaloneWSClient, config: PicfitConfig)(
}
.monSuccess(_.picfit.uploadTime(image.user))
}
}
def delete(image: PicfitImage): Funit =
ws.url(s"${config.endpointPost}/${image.id}").delete().flatMap {