112 lines
3.3 KiB
JavaScript
112 lines
3.3 KiB
JavaScript
/*
|
|
Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import {encryptAttachment} from "../e2ee/attachment.js";
|
|
|
|
export class AttachmentUpload {
|
|
constructor({filename, blob, platform}) {
|
|
this._filename = filename;
|
|
// need to keep around for local preview while uploading
|
|
this._unencryptedBlob = blob;
|
|
this._transferredBlob = this._unencryptedBlob;
|
|
this._platform = platform;
|
|
this._mxcUrl = null;
|
|
this._encryptionInfo = null;
|
|
this._uploadRequest = null;
|
|
this._aborted = false;
|
|
this._error = null;
|
|
this._sentBytes = 0;
|
|
}
|
|
|
|
/** important to call after encrypt() if encryption is needed */
|
|
get size() {
|
|
return this._transferredBlob.size;
|
|
}
|
|
|
|
get sentBytes() {
|
|
return this._sentBytes;
|
|
}
|
|
|
|
/** @public */
|
|
abort() {
|
|
this._uploadRequest?.abort();
|
|
}
|
|
|
|
/** @public */
|
|
get localPreview() {
|
|
return this._unencryptedBlob;
|
|
}
|
|
|
|
/** @package */
|
|
async encrypt() {
|
|
if (this._encryptionInfo) {
|
|
throw new Error("already encrypted");
|
|
}
|
|
const {info, blob} = await encryptAttachment(this._platform, this._transferredBlob);
|
|
this._transferredBlob = blob;
|
|
this._encryptionInfo = info;
|
|
}
|
|
|
|
/** @package */
|
|
async upload(hsApi, progressCallback, log) {
|
|
this._uploadRequest = hsApi.uploadAttachment(this._transferredBlob, this._filename, {
|
|
uploadProgress: sentBytes => {
|
|
this._sentBytes = sentBytes;
|
|
progressCallback();
|
|
},
|
|
log
|
|
});
|
|
const {content_uri} = await this._uploadRequest.response();
|
|
this._mxcUrl = content_uri;
|
|
}
|
|
|
|
/** @package */
|
|
applyToContent(urlPath, content) {
|
|
if (!this._mxcUrl) {
|
|
throw new Error("upload has not finished");
|
|
}
|
|
let prefix = urlPath.substr(0, urlPath.lastIndexOf("url"));
|
|
setPath(`${prefix}info.size`, content, this._transferredBlob.size);
|
|
setPath(`${prefix}info.mimetype`, content, this._unencryptedBlob.mimeType);
|
|
if (this._encryptionInfo) {
|
|
setPath(`${prefix}file`, content, Object.assign(this._encryptionInfo, {
|
|
mimetype: this._unencryptedBlob.mimeType,
|
|
url: this._mxcUrl
|
|
}));
|
|
} else {
|
|
setPath(`${prefix}url`, content, this._mxcUrl);
|
|
}
|
|
}
|
|
|
|
dispose() {
|
|
this._unencryptedBlob.dispose();
|
|
this._transferredBlob.dispose();
|
|
}
|
|
}
|
|
|
|
function setPath(path, content, value) {
|
|
const parts = path.split(".");
|
|
let obj = content;
|
|
for (let i = 0; i < (parts.length - 1); i += 1) {
|
|
const key = parts[i];
|
|
if (!obj[key]) {
|
|
obj[key] = {};
|
|
}
|
|
obj = obj[key];
|
|
}
|
|
const propKey = parts[parts.length - 1];
|
|
obj[propKey] = value;
|
|
}
|