mirror of
https://github.com/streetwriters/notesnook.git
synced 2025-12-22 14:39:34 +01:00
120 lines
3.3 KiB
JavaScript
120 lines
3.3 KiB
JavaScript
/*
|
|
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 <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import hosts from "../utils/constants";
|
|
import TokenManager from "../api/token-manager";
|
|
|
|
export default class FileStorage {
|
|
constructor(fs, storage) {
|
|
this.fs = fs;
|
|
this.tokenManager = new TokenManager(storage);
|
|
this._queue = [];
|
|
}
|
|
|
|
async downloadFile(groupId, filename, chunkSize, metadata) {
|
|
const url = `${hosts.API_HOST}/s3?name=${filename}`;
|
|
const token = await this.tokenManager.getAccessToken();
|
|
const { execute, cancel } = this.fs.downloadFile(filename, {
|
|
metadata,
|
|
url,
|
|
chunkSize,
|
|
headers: { Authorization: `Bearer ${token}` }
|
|
});
|
|
this._queue.push({ groupId, filename, cancel, type: "download" });
|
|
const result = await execute();
|
|
this._deleteOp(groupId, "download");
|
|
return result;
|
|
}
|
|
|
|
async uploadFile(groupId, filename) {
|
|
const token = await this.tokenManager.getAccessToken();
|
|
const url = `${hosts.API_HOST}/s3?name=${filename}`;
|
|
const { execute, cancel } = this.fs.uploadFile(filename, {
|
|
url,
|
|
headers: { Authorization: `Bearer ${token}` }
|
|
});
|
|
this._queue.push({ groupId, filename, cancel, type: "upload" });
|
|
const result = await execute();
|
|
this._deleteOp(groupId, "upload");
|
|
return result;
|
|
}
|
|
|
|
async cancel(groupId, type) {
|
|
const [op] = this._deleteOp(groupId, type);
|
|
if (!op) return;
|
|
await op.cancel("Operation canceled.");
|
|
}
|
|
|
|
_deleteOp(groupId, type) {
|
|
const opIndex = this._queue.findIndex(
|
|
(item) => item.groupId === groupId && (!type || item.type === type)
|
|
);
|
|
if (opIndex < 0) return [];
|
|
return this._queue.splice(opIndex, 1);
|
|
}
|
|
|
|
readEncrypted(filename, encryptionKey, cipherData) {
|
|
return this.fs.readEncrypted(filename, encryptionKey, cipherData);
|
|
}
|
|
|
|
writeEncryptedBase64(data, encryptionKey, mimeType) {
|
|
return this.fs.writeEncryptedBase64({
|
|
data,
|
|
key: encryptionKey,
|
|
mimeType
|
|
});
|
|
}
|
|
|
|
async deleteFile(filename, localOnly) {
|
|
if (localOnly) return await this.fs.deleteFile(filename);
|
|
|
|
const token = await this.tokenManager.getAccessToken();
|
|
const url = `${hosts.API_HOST}/s3?name=${filename}`;
|
|
return await this.fs.deleteFile(filename, {
|
|
url,
|
|
headers: { Authorization: `Bearer ${token}` }
|
|
});
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param {string} filename
|
|
* @returns {Promise<boolean>}
|
|
*/
|
|
exists(filename) {
|
|
return this.fs.exists(filename);
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @returns {Promise<void>}
|
|
*/
|
|
clear() {
|
|
return this.fs.clearFileStorage();
|
|
}
|
|
|
|
/**
|
|
* @param {string} data
|
|
* @returns {Promise<{hash: string, type: string}>}
|
|
*/
|
|
hashBase64(data) {
|
|
return this.fs.hashBase64(data);
|
|
}
|
|
}
|