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

88 lines
2.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';
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();
});
2018-07-05 21:40:49 +02:00
async function decryptStream(request) {
2018-07-12 01:52:46 +02:00
const id = request.url.split('/')[5];
try {
const file = map.get(id);
2018-07-10 02:00:19 +02:00
2018-07-12 01:52:46 +02:00
file.download = downloadStream(id, file.keychain);
2018-07-05 21:40:49 +02:00
2018-07-12 01:52:46 +02:00
const stream = await file.download.result;
2018-07-05 21:40:49 +02:00
2018-07-12 01:52:46 +02:00
// eslint-disable-next-line no-undef
const progStream = new TransformStream({
transform: (chunk, controller) => {
file.progress += chunk.length;
controller.enqueue(chunk);
}
});
2018-07-10 00:39:06 +02:00
2018-07-12 01:52:46 +02:00
const readStream = stream.pipeThrough(progStream);
const decrypted = file.keychain.decryptStream(readStream);
2018-07-10 00:39:06 +02:00
2018-07-12 01:52:46 +02:00
const headers = {
'Content-Disposition': 'attachment; filename=' + file.filename
};
2018-07-10 00:39:06 +02:00
2018-07-12 01:52:46 +02:00
return new Response(decrypted, { headers });
} 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
2018-07-12 01:52:46 +02:00
const redirectRes = await fetch(`/download/${id}`);
return new Response(redirectRes.body, { status: 302 });
}
2018-07-05 21:40:49 +02:00
}
2018-07-07 00:49:50 +02:00
self.onfetch = event => {
2018-07-05 21:40:49 +02:00
const req = event.request.clone();
if (req.url.includes('/api/download')) {
event.respondWith(decryptStream(req));
}
};
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 = {
keychain: new Keychain(event.data.key),
filename: event.data.filename,
progress: 0,
cancelled: false
};
if (event.data.requiresPassword) {
info.keychain.setPassword(event.data.password, event.data.url);
}
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);
if (file.cancelled) {
2018-07-10 02:00:19 +02:00
event.ports[0].postMessage({ error: 'cancelled' });
2018-07-10 00:39:06 +02:00
} else {
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);
file.cancelled = true;
if (file.download) {
file.download.cancel();
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
};