/* This file is part of the Notesnook project (https://notesnook.com/) Copyright (C) 2023 Streetwriters (Private) Limited This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . */ import RNFetchBlob from "react-native-blob-util"; import { useAttachmentStore } from "../../stores/use-attachment-store"; import { DatabaseLogger, db } from "../database"; import { cacheDir } from "./utils"; import { isImage, isDocument } from "@notesnook/core/utils/filename"; import { Platform } from "react-native"; import { IOS_APPGROUPID } from "../../utils/constants"; export async function uploadFile(filename, data, cancelToken) { if (!data) return false; let { url, headers } = data; DatabaseLogger.info(`Preparing to upload file: ${filename}`); try { let res = await fetch(url, { method: "PUT", headers }); if (!res.ok) throw new Error(`${res.status}: Unable to resolve upload url`); const uploadUrl = await res.text(); if (!uploadUrl) throw new Error("Unable to resolve attachment upload url"); let uploadFilePath = `${cacheDir}/${filename}`; const iosAppGroup = Platform.OS === "ios" ? await RNFetchBlob.fs.pathForAppGroup(IOS_APPGROUPID) : null; const appGroupPath = `${iosAppGroup}/${filename}`; let exists = await RNFetchBlob.fs.exists(uploadFilePath); if (!exists && Platform.OS === "ios") { uploadFilePath = appGroupPath; } DatabaseLogger.info(`Starting upload: ${filename}`); let request = RNFetchBlob.config({ IOSBackgroundTask: !globalThis["IS_SHARE_EXTENSION"] }) .fetch( "PUT", uploadUrl, { "content-type": "" }, RNFetchBlob.wrap(uploadFilePath) ) .uploadProgress((sent, total) => { useAttachmentStore .getState() .setProgress(sent, total, filename, 0, "upload"); DatabaseLogger.info(`uploading file: ${sent}/${total}`); }); cancelToken.cancel = request.cancel; let response = await request; let status = response.info().status; let text = await response.text(); let result = status >= 200 && status < 300 && text.length === 0; useAttachmentStore.getState().remove(filename); if (result) { let attachment = db.attachments.attachment(filename); if (!attachment) return result; if ( !isImage(attachment.metadata.type) && !isDocument(attachment.metadata?.type) ) { RNFetchBlob.fs.unlink(`${cacheDir}/${filename}`).catch(console.log); } } DatabaseLogger.info(`File uploaded: ${filename}`); return result; } catch (e) { useAttachmentStore.getState().remove(filename); DatabaseLogger.info(`File upload error: ${filename}, ${e}`); DatabaseLogger.error(e); return false; } }