wabe 0.6.9 → 0.6.11

package/src/file/hookUploadFile.ts

@@ -0,0 +1,50 @@
+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)