wabe 0.6.9 → 0.6.10

package/src/file/hookReadFile.ts

@@ -0,0 +1,70 @@
+import type { HookObject } from '../hooks/HookObject'
+
+const getFile = async (hookObject: HookObject<any, any>) => {
+	const schema = hookObject.context.wabe.config.schema?.classes?.find(
+		(currentClass) => currentClass.name === hookObject.className,
+	)
+
+	if (!schema) return
+
+	const urlCacheInSeconds =
+		hookObject.context.wabe.config.file?.urlCacheInSeconds || 3600 * 24
+
+	// After read we get the file URL and we update the field url with an URL.
+	// For security purpose we recommend to use a presigned URL
+	await Promise.all(
+		Object.entries(schema.fields)
+			.filter(([_, value]) => value.type === 'File')
+			.map(async ([fieldName]) => {
+				const fileInfo = hookObject.object?.[fieldName]
+
+				if (!fileInfo) return
+
+				const fileName = fileInfo.name as string
+
+				if (!fileName && fileInfo.url) return fileInfo.url
+
+				const fileUrlGeneratedAt = new Date(fileInfo.urlGeneratedAt)
+
+				if (
+					fileUrlGeneratedAt &&
+					fileUrlGeneratedAt.getTime() + urlCacheInSeconds * 1000 > Date.now()
+				)
+					return
+
+				if (!hookObject.context.wabe.controllers.file)
+					throw new Error('No file adapter found')
+
+				const fileUrlFromBucket =
+					await hookObject.context.wabe.controllers.file?.readFile(fileName)
+
+				const newUrl = fileUrlFromBucket || fileInfo.url
+				const newUrlGeneratedAt = new Date()
+
+				// Mutate the object returned to the caller so AfterRead effects are visible immediately
+				// @ts-expect-error
+				hookObject.object[fieldName] = {
+					...fileInfo,
+					urlGeneratedAt: newUrlGeneratedAt,
+					url: newUrl,
+				}
+
+				return hookObject.context.wabe.controllers.database.updateObject({
+					className: hookObject.className,
+					context: hookObject.context,
+					id: hookObject.object?.id || '',
+					data: {
+						[fieldName]: {
+							...fileInfo,
+							urlGeneratedAt: newUrlGeneratedAt,
+							url: newUrl,
+						},
+					},
+					_skipHooks: true,
+				})
+			}),
+	)
+}
+
+export const defaultAfterReadFile = (hookObject: HookObject<any, any>) =>
+	getFile(hookObject)

package/src/file/hookUploadFile.ts

@@ -0,0 +1,53 @@
+import type { HookObject } from '../hooks/HookObject'
+
+const handleFile = async (hookObject: HookObject<any, any>) => {
+	const newData = hookObject.getNewData()
+
+	const schema = hookObject.context.wabe.config.schema?.classes?.find(
+		(currentClass) => currentClass.name === hookObject.className,
+	)
+
+	if (!schema) return
+
+	const beforeUpload = hookObject.context.wabe.config.file?.beforeUpload
+
+	await Promise.all(
+		Object.keys(newData).map(async (keyName) => {
+			const file = newData[keyName]?.file as File
+			const url = newData[keyName]?.url as string
+
+			if (!file && !url) return
+
+			if (url) {
+				hookObject.upsertNewData(keyName, {
+					url,
+					isPresignedUrl: false,
+				})
+				return
+			}
+
+			if (schema.fields[keyName]?.type !== 'File' || !(file instanceof File))
+				return
+
+			if (!hookObject.context.wabe.controllers.file)
+				throw new Error('No file adapter found')
+
+			const fileToUpload =
+				(await beforeUpload?.(file, hookObject.context)) || file
+
+			// We upload the file and set the name of the file in the newData
+			await hookObject.context.wabe.controllers.file?.uploadFile(fileToUpload)
+
+			hookObject.upsertNewData(keyName, {
+				name: file.name,
+				isPresignedUrl: true,
+			})
+		}),
+	)
+}
+
+export const defaultBeforeCreateUpload = (hookObject: HookObject<any, any>) =>
+	handleFile(hookObject)
+
+export const defaultBeforeUpdateUpload = (hookObject: HookObject<any, any>) =>
+	handleFile(hookObject)