import * as Actions from "~/common/actions"; import * as Store from "~/common/store"; import * as Credentials from "~/common/credentials"; import * as Strings from "~/common/strings"; import * as Validations from "~/common/validations"; import * as Events from "~/common/custom-events"; import * as Logging from "~/common/logging"; import * as Environment from "~/common/environment"; import * as Window from "~/common/window"; import { encode, isBlurhashValid } from "blurhash"; import { v4 as uuid } from "uuid"; const STAGING_DEAL_BUCKET = "stage-deal"; export const fileKey = ({ lastModified, name }) => `${lastModified}-${name}`; const loadImage = async (src) => new Promise((resolve, reject) => { const img = new Image(); img.crossOrigin = "Anonymous"; img.onload = () => resolve(img); img.onerror = (...args) => reject(args); img.src = src; }); const getImageData = (image) => { let ratio = Math.min(100 / image.height, 100 / image.width); image.height = image.height * ratio; image.width = image.width * ratio; const canvas = document.createElement("canvas"); canvas.width = image.width; canvas.height = image.height; const context = canvas.getContext("2d"); context.scale(ratio, ratio); context.drawImage(image, 0, 0); return context.getImageData(0, 0, image.width, image.height); }; const encodeImageToBlurhash = async (imageUrl) => { const image = await loadImage(imageUrl); const imageData = getImageData(image); return encode(imageData.data, imageData.width, imageData.height, 4, 4); }; // NOTE(jim): We're speaking to a different server now. const getCookie = (name) => { var match = document.cookie.match(new RegExp("(^| )" + name + "=([^;]+)")); if (match) return match[2]; }; export const uploadLink = async ({ url, slate }) => { Events.dispatchMessage({ message: "Uploading link...", status: "INFO" }); let createResponse = await Actions.createLink({ url, slate }); if (Events.hasError(createResponse)) { return; } const { added, skipped } = createResponse.data; if (added) { Events.dispatchMessage({ message: "Link added", status: "INFO" }); } else if (skipped) { Events.dispatchMessage({ message: "You've already saved this link", }); return; } }; export const uploadFiles = async ({ context, files, slate, keys, numFailed = 0 }) => { if (!files || !files.length) { context._handleRegisterLoadingFinished({ keys }); return; } const resolvedFiles = []; for (let i = 0; i < files.length; i++) { const currentFileKey = fileKey(files[i]); if (Store.checkCancelled(currentFileKey)) { continue; } // NOTE(jim): With so many failures, probably good to wait a few seconds. await Window.delay(3000); // NOTE(jim): Sends XHR request. let response; try { response = await upload({ file: files[i], context, }); } catch (e) { Logging.error(e); } if (!response || response.error) { continue; } resolvedFiles.push(response); } if (!resolvedFiles.length) { context._handleRegisterLoadingFinished({ keys }); return; } //NOTE(martina): this commented out portion is only for if parallel uploading // let responses = await Promise.allSettled(resolvedFiles); // let succeeded = responses // .filter((res) => { // return res.status === "fulfilled" && res.value && !res.value.error; // }) // .map((res) => res.value); let createResponse = await Actions.createFile({ files: resolvedFiles, slate }); if (Events.hasError(createResponse)) { context._handleRegisterLoadingFinished({ keys }); return; } const { added, skipped } = createResponse.data; let message = Strings.formatAsUploadMessage(added, skipped + numFailed, slate); Events.dispatchMessage({ message, status: !added ? null : "INFO" }); context._handleRegisterLoadingFinished({ keys }); }; //NOTE(migration): check that upload works still and file.name export const upload = async ({ file, context, bucketName }) => { const currentFileKey = fileKey(file); let formData = new FormData(); const HEIC2ANY = require("heic2any"); // NOTE(jim): You must provide a file from an type="file" input field. if (!file) { return null; } const isZipFile = file.type.startsWith("application/zip") || file.type.startsWith("application/x-zip-compressed"); const isUnityFile = await Validations.isUnityFile(file); // TODO(jim): Put this somewhere else to handle conversion cases. if (file.type.startsWith("image/heic")) { const converted = await HEIC2ANY({ blob: file, toType: "image/png", quality: 1, }); //TODO(martina): figure out how to cancel an await if upload has been cancelled formData.append("data", converted); } else { formData.append("data", file); } if (Store.checkCancelled(currentFileKey)) { return; } const _privateUploadMethod = (path, file) => new Promise((resolve) => { const XHR = new XMLHttpRequest(); window.addEventListener(`cancel-${currentFileKey}`, () => { XHR.abort(); }); XHR.open("post", path, true); XHR.setRequestHeader("authorization", getCookie(Credentials.session.key)); XHR.onerror = (event) => { Logging.error(event); XHR.abort(); }; // NOTE(jim): UPLOADS ONLY. XHR.upload.addEventListener( "progress", (event) => { if (!context) { return; } if (event.lengthComputable) { Logging.log("FILE UPLOAD PROGRESS", event); context.setState({ fileLoading: { ...context.state.fileLoading, [currentFileKey]: { name: file.name, loaded: event.loaded, total: event.total, }, }, }); } }, false ); window.removeEventListener(`cancel-${currentFileKey}`, () => XHR.abort()); XHR.onloadend = (event) => { Logging.log("FILE UPLOAD END", event); try { return resolve(JSON.parse(event.target.response)); } catch (e) { return resolve({ error: "SERVER_UPLOAD_ERROR", }); } }; XHR.send(formData); }); const storageDealRoute = `${Environment.URI_SHOVEL}/api/deal/`; const generalRoute = `${Environment.URI_SHOVEL}/api/data/`; const zipUploadRoute = `${Environment.URI_SHOVEL}/api/data/zip/`; if (!storageDealRoute || !generalRoute || !zipUploadRoute) { Events.dispatchMessage({ message: "We could not find our upload server." }); return { decorator: "NO_UPLOAD_RESOURCE_URI_ATTACHED", error: true, }; } let res; if (isZipFile && isUnityFile) { res = await _privateUploadMethod(`${zipUploadRoute}${file.name}`, file); } else if (bucketName && bucketName === STAGING_DEAL_BUCKET) { res = await _privateUploadMethod(`${storageDealRoute}${file.name}`, file); } else { res = await _privateUploadMethod(`${generalRoute}${file.name}`, file); } if (!res?.data || res.error) { if (context) { await context.setState({ fileLoading: { ...context.state.fileLoading, [`${file.lastModified}-${file.name}`]: { name: file.name, failed: true, }, }, }); } Events.dispatchMessage({ message: "Some of your files could not be uploaded" }); return !res ? { decorator: "NO_RESPONSE_FROM_SERVER", error: true } : res; } let item = res.data.data; if (item.type.startsWith("image/")) { let url = Strings.getURLfromCID(item.cid); try { let blurhash = await encodeImageToBlurhash(url); if (isBlurhashValid(blurhash).result) { item.blurhash = blurhash; } } catch (e) { Logging.error(e); } } return item; }; export const formatPastedImages = ({ clipboardItems }) => { let files = []; let fileLoading = {}; for (let i = 0; i < clipboardItems.length; i++) { // Note(Amine): skip content if it's not an image if (clipboardItems[i].type.indexOf("image") === -1) continue; const file = clipboardItems[i].getAsFile(); files.push(file); fileLoading[`${file.lastModified}-${file.name}`] = { name: file.name, loaded: 0, total: file.size, }; } return { fileLoading, toUpload: files }; }; export const formatDroppedFiles = async ({ dataTransfer }) => { // NOTE(jim): If this is true, then drag and drop came from a slate object. const data = dataTransfer.getData("slate-object-drag-data"); if (data) { return; } const files = []; let fileLoading = {}; if (dataTransfer.items && dataTransfer.items.length) { for (var i = 0; i < dataTransfer.items.length; i++) { const data = dataTransfer.items[i]; let file = null; if (data.kind === "file") { file = data.getAsFile(); } else if (data.kind == "string" && data.type == "text/uri-list") { try { const dataAsString = new Promise((resolve) => data.getAsString((d) => resolve(d))); const resp = await fetch(await dataAsString); const blob = resp.blob(); file = new File(blob, `data-${uuid()}`); file.name = `data-${uuid()}`; } catch (e) { Events.dispatchMessage({ message: "File type not supported. Please try a different file", }); return { error: true }; } } files.push(file); fileLoading[`${file.lastModified}-${file.name}`] = { name: file.name, loaded: 0, total: file.size, }; } } if (!files.length) { Events.dispatchMessage({ message: "File type not supported. Please try a different file" }); } return { fileLoading, files, numFailed: dataTransfer.items.length - files.length }; }; export const formatUploadedFiles = ({ files }) => { let toUpload = []; let fileLoading = {}; for (let i = 0; i < files.length; i++) { let file = files[i]; if (!file) { continue; } toUpload.push(file); fileLoading[fileKey(file)] = { name: file.name, loaded: 0, total: file.size, }; } if (!toUpload.length) { Events.dispatchMessage({ message: "We could not find any files to upload." }); return false; } return { toUpload, fileLoading, numFailed: files.length - toUpload.length }; };