067d9069cb
* Use $queryRaw instead of $queryRawUnsafe * Replace comment with destructuring * Return all upload fees as BigInt
90 lines
3.0 KiB
JavaScript
90 lines
3.0 KiB
JavaScript
import { USER_ID, IMAGE_PIXELS_MAX, UPLOAD_SIZE_MAX, UPLOAD_SIZE_MAX_AVATAR, UPLOAD_TYPES_ALLOW, AWS_S3_URL_REGEXP, AVATAR_TYPES_ALLOW } from '@/lib/constants'
|
|
import { createPresignedPost } from '@/api/s3'
|
|
import { GqlAuthenticationError, GqlInputError } from '@/lib/error'
|
|
import { msatsToSats } from '@/lib/format'
|
|
|
|
export default {
|
|
Query: {
|
|
uploadFees: async (parent, { s3Keys }, { models, me }) => {
|
|
return uploadFees(s3Keys, { models, me })
|
|
}
|
|
},
|
|
Mutation: {
|
|
getSignedPOST: async (parent, { type, size, width, height, avatar }, { models, me }) => {
|
|
if (UPLOAD_TYPES_ALLOW.indexOf(type) === -1) {
|
|
throw new GqlInputError(`upload must be ${UPLOAD_TYPES_ALLOW.map(t => t.replace(/^(image|video)\//, '')).join(', ')}`)
|
|
}
|
|
|
|
if (size > UPLOAD_SIZE_MAX) {
|
|
throw new GqlInputError(`upload must be less than ${UPLOAD_SIZE_MAX / (1024 ** 2)} megabytes`)
|
|
}
|
|
|
|
if (avatar) {
|
|
if (AVATAR_TYPES_ALLOW.indexOf(type) === -1) {
|
|
throw new GqlInputError(`avatar must be ${AVATAR_TYPES_ALLOW.map(t => t.replace('image/', '')).join(', ')}`)
|
|
}
|
|
|
|
if (size > UPLOAD_SIZE_MAX_AVATAR) {
|
|
throw new GqlInputError(`avatar must be less than ${UPLOAD_SIZE_MAX_AVATAR / (1024 ** 2)} megabytes`)
|
|
}
|
|
}
|
|
|
|
// width and height is 0 for videos
|
|
if (width * height > IMAGE_PIXELS_MAX) {
|
|
throw new GqlInputError(`image must be less than ${IMAGE_PIXELS_MAX} pixels`)
|
|
}
|
|
|
|
const fileParams = {
|
|
type,
|
|
size,
|
|
width,
|
|
height,
|
|
userId: me?.id || USER_ID.anon,
|
|
paid: false
|
|
}
|
|
|
|
if (avatar) {
|
|
if (!me) throw new GqlAuthenticationError()
|
|
fileParams.paid = undefined
|
|
}
|
|
|
|
const upload = await models.upload.create({ data: { ...fileParams } })
|
|
return createPresignedPost({ key: String(upload.id), type, size })
|
|
}
|
|
}
|
|
}
|
|
|
|
export function uploadIdsFromText (text) {
|
|
if (!text) return []
|
|
return [...new Set([...text.matchAll(AWS_S3_URL_REGEXP)].map(m => Number(m[1])))]
|
|
}
|
|
|
|
export async function uploadFees (s3Keys, { models, me }) {
|
|
const [{
|
|
bytes24h,
|
|
bytesUnpaid,
|
|
nUnpaid,
|
|
uploadFeesMsats
|
|
}] = await models.$queryRaw`SELECT * FROM upload_fees(${me?.id ?? USER_ID.anon}::INTEGER, ${s3Keys}::INTEGER[])`
|
|
const uploadFees = msatsToSats(uploadFeesMsats)
|
|
const totalFeesMsats = BigInt(nUnpaid) * uploadFeesMsats
|
|
const totalFees = msatsToSats(totalFeesMsats)
|
|
return { bytes24h, bytesUnpaid, nUnpaid, uploadFees, totalFees, totalFeesMsats }
|
|
}
|
|
|
|
export async function throwOnExpiredUploads (uploadIds, { tx }) {
|
|
if (uploadIds.length === 0) return
|
|
|
|
const existingUploads = await tx.upload.findMany({
|
|
where: { id: { in: uploadIds } },
|
|
select: { id: true }
|
|
})
|
|
|
|
const existingIds = new Set(existingUploads.map(upload => upload.id))
|
|
const deletedIds = uploadIds.filter(id => !existingIds.has(id))
|
|
|
|
if (deletedIds.length > 0) {
|
|
throw new Error(`upload(s) ${deletedIds.join(', ')} are expired, consider reuploading.`)
|
|
}
|
|
}
|