drop.chapril.org-firefoxsend/app/serviceWorker.js

127 lines
3.2 KiB
JavaScript
Raw Normal View History

2018-07-05 21:40:49 +02:00
import Keychain from './keychain';
2018-07-12 01:52:46 +02:00
import { downloadStream } from './api';
2018-07-19 23:46:12 +02:00
import { transformStream } from './streams';
2018-07-26 07:26:11 +02:00
import Zip from './zip';
import contentDisposition from 'content-disposition';
2018-07-12 01:52:46 +02:00
let noSave = false;
const map = new Map();
2018-07-05 21:40:49 +02:00
2018-07-07 00:49:50 +02:00
self.addEventListener('install', event => {
2018-07-05 21:40:49 +02:00
self.skipWaiting();
});
2018-07-12 01:52:46 +02:00
self.addEventListener('activate', event => {
self.clients.claim();
});
async function decryptStream(id) {
const file = map.get(id);
if (!file) {
return new Response(null, { status: 400 });
}
2018-07-12 01:52:46 +02:00
try {
2018-07-26 07:26:11 +02:00
let size = file.size;
let type = file.type;
2018-07-19 22:20:10 +02:00
const keychain = new Keychain(file.key, file.nonce);
if (file.requiresPassword) {
keychain.setPassword(file.password, file.url);
}
2018-07-10 02:00:19 +02:00
2018-07-19 01:39:14 +02:00
file.download = downloadStream(id, keychain);
2018-07-05 21:40:49 +02:00
2018-07-19 01:39:14 +02:00
const body = await file.download.result;
2018-07-05 21:40:49 +02:00
const decrypted = keychain.decryptStream(body);
2018-07-26 07:26:11 +02:00
let zipStream = null;
if (file.type === 'send-archive') {
const zip = new Zip(file.manifest, decrypted);
zipStream = zip.stream;
type = 'application/zip';
size = zip.size;
}
const responseStream = transformStream(
2018-07-26 07:26:11 +02:00
zipStream || decrypted,
{
transform(chunk, controller) {
file.progress += chunk.length;
controller.enqueue(chunk);
}
},
function oncancel() {
// NOTE: cancel doesn't currently fire on chrome
// https://bugs.chromium.org/p/chromium/issues/detail?id=638494
file.download.cancel();
map.delete(id);
2018-07-12 01:52:46 +02:00
}
);
2018-07-10 00:39:06 +02:00
2018-07-12 01:52:46 +02:00
const headers = {
'Content-Disposition': contentDisposition(file.filename),
2018-07-26 07:26:11 +02:00
'Content-Type': type,
'Content-Length': size
2018-07-12 01:52:46 +02:00
};
return new Response(responseStream, { headers });
2018-07-12 01:52:46 +02:00
} catch (e) {
if (noSave) {
return new Response(null, { status: e.message });
2018-07-07 00:49:50 +02:00
}
2018-07-05 21:40:49 +02:00
return new Response(null, {
status: 302,
headers: {
Location: `/download/${id}`
}
});
2018-07-12 01:52:46 +02:00
}
2018-07-05 21:40:49 +02:00
}
2018-07-07 00:49:50 +02:00
self.onfetch = event => {
const req = event.request;
const match = /\/api\/download\/([A-Fa-f0-9]{4,})/.exec(req.url);
if (match) {
event.respondWith(decryptStream(match[1]));
2018-07-05 21:40:49 +02:00
}
};
2018-07-07 00:49:50 +02:00
self.onmessage = event => {
2018-07-12 01:52:46 +02:00
if (event.data.request === 'init') {
noSave = event.data.noSave;
const info = {
2018-07-19 01:39:14 +02:00
key: event.data.key,
2018-07-19 22:20:10 +02:00
nonce: event.data.nonce,
2018-07-12 01:52:46 +02:00
filename: event.data.filename,
requiresPassword: event.data.requiresPassword,
password: event.data.password,
url: event.data.url,
2018-07-13 20:13:09 +02:00
type: event.data.type,
2018-07-26 07:26:11 +02:00
manifest: event.data.manifest,
2018-07-13 00:32:07 +02:00
size: event.data.size,
progress: 0
2018-07-12 01:52:46 +02:00
};
map.set(event.data.id, info);
2018-07-10 02:00:19 +02:00
event.ports[0].postMessage('file info received');
2018-07-12 01:52:46 +02:00
} else if (event.data.request === 'progress') {
const file = map.get(event.data.id);
2018-07-19 01:39:14 +02:00
if (!file) {
2018-07-10 02:00:19 +02:00
event.ports[0].postMessage({ error: 'cancelled' });
2018-07-10 00:39:06 +02:00
} else {
if (file.progress === file.size) {
map.delete(event.data.id);
}
2018-07-12 01:52:46 +02:00
event.ports[0].postMessage({ progress: file.progress });
2018-07-10 00:39:06 +02:00
}
2018-07-12 01:52:46 +02:00
} else if (event.data.request === 'cancel') {
const file = map.get(event.data.id);
2018-07-19 01:39:14 +02:00
if (file) {
if (file.download) {
file.download.cancel();
}
map.delete(event.data.id);
2018-07-10 00:39:06 +02:00
}
2018-07-10 02:00:19 +02:00
event.ports[0].postMessage('download cancelled');
2018-07-10 00:39:06 +02:00
}
2018-07-07 00:49:50 +02:00
};