fox-send/app/fileReceiver.js

157 lines
3.9 KiB
JavaScript
Raw Normal View History

import Nanobus from 'nanobus';
2018-01-24 18:23:13 +00:00
import Keychain from './keychain';
2018-07-09 22:39:06 +00:00
import { delay, bytes } from './utils';
2018-07-11 23:52:46 +00:00
import { metadata } from './api';
2017-06-02 19:38:05 +00:00
export default class FileReceiver extends Nanobus {
2018-01-24 18:23:13 +00:00
constructor(fileInfo) {
super('FileReceiver');
2018-01-24 18:23:13 +00:00
this.keychain = new Keychain(fileInfo.secretKey, fileInfo.nonce);
if (fileInfo.requiresPassword) {
this.keychain.setPassword(fileInfo.password, fileInfo.url);
}
2018-01-24 18:23:13 +00:00
this.fileInfo = fileInfo;
2018-02-05 02:30:33 +00:00
this.reset();
}
get progressRatio() {
return this.progress[0] / this.progress[1];
}
2018-02-21 21:59:06 +00:00
get progressIndefinite() {
return this.state !== 'downloading';
}
get sizes() {
return {
partialSize: bytes(this.progress[0]),
totalSize: bytes(this.progress[1])
};
}
cancel() {
if (this.downloadRequest) {
this.downloadRequest.cancel();
2018-01-24 18:23:13 +00:00
}
}
2018-02-05 02:30:33 +00:00
reset() {
this.msg = 'fileSizeProgress';
this.state = 'initialized';
this.progress = [0, 1];
}
2018-01-24 18:23:13 +00:00
async getMetadata() {
const meta = await metadata(this.fileInfo.id, this.keychain);
this.keychain.setIV(meta.iv);
this.fileInfo.name = meta.name;
this.fileInfo.type = meta.type;
this.fileInfo.iv = meta.iv;
this.fileInfo.size = meta.size;
this.state = 'ready';
}
2018-07-05 19:40:49 +00:00
async streamToArrayBuffer(stream, streamSize, onprogress) {
2018-07-06 22:49:50 +00:00
const result = new Uint8Array(streamSize);
let offset = 0;
const reader = stream.getReader();
let state = await reader.read();
while (!state.done) {
result.set(state.value, offset);
offset += state.value.length;
state = await reader.read();
onprogress([offset, streamSize]);
2018-06-29 16:36:08 +00:00
}
2018-07-06 22:49:50 +00:00
onprogress([streamSize, streamSize]);
return result.slice(0, offset).buffer;
2018-06-21 00:05:33 +00:00
}
2018-07-09 22:39:06 +00:00
sendMessageToSw(msg) {
2018-07-10 00:00:19 +00:00
return new Promise((resolve, reject) => {
2018-07-09 22:39:06 +00:00
const channel = new MessageChannel();
channel.port1.onmessage = function(event) {
2018-07-11 23:52:46 +00:00
if (event.data === undefined) {
reject('bad response from serviceWorker');
} else if (event.data.error !== undefined) {
2018-07-09 22:39:06 +00:00
reject(event.data.error);
} else {
resolve(event.data);
}
2018-07-10 00:00:19 +00:00
};
2018-07-11 23:52:46 +00:00
2018-07-10 00:00:19 +00:00
navigator.serviceWorker.controller.postMessage(msg, [channel.port2]);
2018-07-09 22:39:06 +00:00
});
}
2018-02-21 04:31:27 +00:00
async download(noSave = false) {
2018-07-05 19:40:49 +00:00
const onprogress = p => {
this.progress = p;
this.emit('progress');
2018-07-06 22:49:50 +00:00
};
2018-06-21 00:05:33 +00:00
2018-07-09 22:39:06 +00:00
this.downloadRequest = {
cancel: () => {
2018-07-11 23:52:46 +00:00
this.sendMessageToSw({ request: 'cancel', id: this.fileInfo.id });
2018-07-09 22:39:06 +00:00
}
2018-07-10 00:00:19 +00:00
};
2018-07-09 22:39:06 +00:00
try {
2018-07-05 19:40:49 +00:00
this.state = 'downloading';
2018-07-06 22:49:50 +00:00
const info = {
2018-07-11 23:52:46 +00:00
request: 'init',
id: this.fileInfo.id,
2018-07-06 22:49:50 +00:00
filename: this.fileInfo.name,
2018-07-13 18:13:09 +00:00
type: this.fileInfo.type,
2018-07-11 23:52:46 +00:00
key: this.fileInfo.secretKey,
requiresPassword: this.fileInfo.requiresPassword,
password: this.fileInfo.password,
url: this.fileInfo.url,
2018-07-12 22:32:07 +00:00
size: this.fileInfo.size,
2018-07-11 23:52:46 +00:00
noSave
2018-07-06 22:49:50 +00:00
};
2018-07-09 22:39:06 +00:00
await this.sendMessageToSw(info);
2018-06-29 16:36:08 +00:00
2018-07-11 23:52:46 +00:00
onprogress([0, this.fileInfo.size]);
2018-06-21 00:05:33 +00:00
2018-07-11 23:52:46 +00:00
if (noSave) {
const res = await fetch(`/api/download/${this.fileInfo.id}`);
if (res.status !== 200) {
throw new Error(res.status);
}
} else {
2018-07-06 22:49:50 +00:00
const downloadUrl = `${location.protocol}//${
location.host
}/api/download/${this.fileInfo.id}`;
const a = document.createElement('a');
a.href = downloadUrl;
document.body.appendChild(a);
a.click();
2018-07-12 18:22:49 +00:00
}
2018-07-09 22:39:06 +00:00
2018-07-12 18:22:49 +00:00
let prog = 0;
while (prog < this.fileInfo.size) {
const msg = await this.sendMessageToSw({
request: 'progress',
id: this.fileInfo.id
});
prog = msg.progress;
onprogress([prog, this.fileInfo.size]);
2018-07-13 18:13:09 +00:00
await delay(1000);
2018-02-21 04:31:27 +00:00
}
2018-06-29 16:36:08 +00:00
2018-07-09 22:39:06 +00:00
this.downloadRequest = null;
this.msg = 'downloadFinish';
this.state = 'complete';
} catch (e) {
this.downloadRequest = null;
2018-07-12 18:22:49 +00:00
if (e === 'cancelled') {
throw new Error(0);
}
throw e;
}
2017-06-02 19:38:05 +00:00
}
}