344 lines
12 KiB
JavaScript
344 lines
12 KiB
JavaScript
define([
|
|
'/common/cryptget.js',
|
|
'/file/file-crypto.js',
|
|
'/common/common-hash.js',
|
|
'/common/common-util.js',
|
|
'/customize/messages.js',
|
|
'/bower_components/nthen/index.js',
|
|
'/bower_components/saferphore/index.js',
|
|
'/bower_components/jszip/dist/jszip.min.js',
|
|
], function (Crypt, FileCrypto, Hash, Util, Messages, nThen, Saferphore, JsZip) {
|
|
var saveAs = window.saveAs;
|
|
|
|
var sanitize = function (str) {
|
|
return str.replace(/[\\/?%*:|"<>]/gi, '_')/*.toLowerCase()*/;
|
|
};
|
|
|
|
var getUnique = function (name, ext, existing) {
|
|
var n = name + ext;
|
|
var i = 1;
|
|
while (existing.indexOf(n.toLowerCase()) !== -1) {
|
|
n = name + ' ('+ i++ + ')' + ext;
|
|
}
|
|
return n;
|
|
};
|
|
|
|
var transform = function (ctx, type, sjson, cb) {
|
|
var result = {
|
|
data: sjson,
|
|
ext: '.json',
|
|
};
|
|
var json;
|
|
try {
|
|
json = JSON.parse(sjson);
|
|
} catch (e) {
|
|
return void cb(result);
|
|
}
|
|
var path = '/' + type + '/export.js';
|
|
require([path], function (Exporter) {
|
|
Exporter.main(json, function (data) {
|
|
result.ext = Exporter.ext || '';
|
|
result.data = data;
|
|
cb(result);
|
|
});
|
|
}, function () {
|
|
cb(result);
|
|
});
|
|
};
|
|
|
|
|
|
var _downloadFile = function (ctx, fData, cb, updateProgress) {
|
|
var cancelled = false;
|
|
var cancel = function () {
|
|
cancelled = true;
|
|
};
|
|
var href = (fData.href && fData.href.indexOf('#') !== -1) ? fData.href : fData.roHref;
|
|
var parsed = Hash.parsePadUrl(href);
|
|
var hash = parsed.hash;
|
|
var name = fData.filename || fData.title;
|
|
var secret = Hash.getSecrets('file', hash, fData.password);
|
|
var src = (ctx.fileHost || '') + Hash.getBlobPathFromHex(secret.channel);
|
|
var key = secret.keys && secret.keys.cryptKey;
|
|
Util.fetch(src, function (err, u8) {
|
|
if (cancelled) { return; }
|
|
if (err) { return void cb('E404'); }
|
|
FileCrypto.decrypt(u8, key, function (err, res) {
|
|
if (cancelled) { return; }
|
|
if (err) { return void cb(err); }
|
|
if (!res.content) { return void cb('EEMPTY'); }
|
|
var dl = function () {
|
|
saveAs(res.content, name || res.metadata.name);
|
|
};
|
|
cb(null, {
|
|
metadata: res.metadata,
|
|
content: res.content,
|
|
download: dl
|
|
});
|
|
}, updateProgress && updateProgress.progress2);
|
|
}, updateProgress && updateProgress.progress);
|
|
return {
|
|
cancel: cancel
|
|
};
|
|
|
|
};
|
|
|
|
|
|
var _downloadPad = function (ctx, pData, cb, updateProgress) {
|
|
var cancelled = false;
|
|
var cancel = function () {
|
|
cancelled = true;
|
|
};
|
|
|
|
var href = (pData.href && pData.href.indexOf('#') !== -1) ? pData.href : pData.roHref;
|
|
var parsed = Hash.parsePadUrl(href);
|
|
var name = pData.filename || pData.title;
|
|
var opts = {
|
|
password: pData.password
|
|
};
|
|
var handler = ctx.sframeChan.on("EV_CRYPTGET_PROGRESS", function (data) {
|
|
if (data.hash !== parsed.hash) { return; }
|
|
updateProgress.progress(data.progress);
|
|
if (data.progress === 1) {
|
|
handler.stop();
|
|
updateProgress.progress2(1);
|
|
}
|
|
});
|
|
ctx.get({
|
|
hash: parsed.hash,
|
|
opts: opts
|
|
}, function (err, val) {
|
|
if (cancelled) { return; }
|
|
if (err) { return; }
|
|
if (!val) { return; }
|
|
transform(ctx, parsed.type, val, function (res) {
|
|
if (cancelled) { return; }
|
|
if (!res.data) { return; }
|
|
var dl = function () {
|
|
saveAs(res.data, Util.fixFileName(name));
|
|
};
|
|
cb(null, {
|
|
metadata: res.metadata,
|
|
content: res.data,
|
|
download: dl
|
|
});
|
|
});
|
|
});
|
|
return {
|
|
cancel: cancel
|
|
};
|
|
|
|
};
|
|
|
|
// Add a file to the zip. We have to cryptget&transform it if it's a pad
|
|
// or fetch&decrypt it if it's a file.
|
|
var addFile = function (ctx, zip, fData, existingNames) {
|
|
if (!fData.href && !fData.roHref) {
|
|
return void ctx.errors.push({
|
|
error: 'EINVAL',
|
|
data: fData
|
|
});
|
|
}
|
|
|
|
var href = (fData.href && fData.href.indexOf('#') !== -1) ? fData.href : fData.roHref;
|
|
var parsed = Hash.parsePadUrl(href);
|
|
if (['pad', 'file'].indexOf(parsed.hashData.type) === -1) { return; }
|
|
|
|
// waitFor is used to make sure all the pads and files are process before downloading the zip.
|
|
var w = ctx.waitFor();
|
|
|
|
ctx.max++;
|
|
// Work with only 10 pad/files at a time
|
|
ctx.sem.take(function (give) {
|
|
var g = give();
|
|
if (ctx.stop) { return; }
|
|
|
|
var to;
|
|
|
|
var done = function () {
|
|
if (ctx.stop) { return; }
|
|
if (to) { clearTimeout(to); }
|
|
//setTimeout(g, 2000);
|
|
g();
|
|
w();
|
|
ctx.done++;
|
|
ctx.updateProgress('download', {max: ctx.max, current: ctx.done});
|
|
};
|
|
|
|
var error = function (err) {
|
|
if (ctx.stop) { return; }
|
|
done();
|
|
return void ctx.errors.push({
|
|
error: err,
|
|
data: fData
|
|
});
|
|
};
|
|
|
|
to = setTimeout(function () {
|
|
error('TIMEOUT');
|
|
}, 60000);
|
|
|
|
setTimeout(function () {
|
|
if (ctx.stop) { return; }
|
|
var opts = {
|
|
password: fData.password
|
|
};
|
|
var rawName = fData.filename || fData.title || 'File';
|
|
console.log(rawName);
|
|
|
|
// Pads (pad,code,slide,kanban,poll,...)
|
|
var todoPad = function () {
|
|
ctx.get({
|
|
hash: parsed.hash,
|
|
opts: opts
|
|
}, function (err, val) {
|
|
if (ctx.stop) { return; }
|
|
if (err) { return void error(err); }
|
|
if (!val) { return void error('EEMPTY'); }
|
|
|
|
var opts = {
|
|
binary: true,
|
|
};
|
|
transform(ctx, parsed.type, val, function (res) {
|
|
if (ctx.stop) { return; }
|
|
if (!res.data) { return void error('EEMPTY'); }
|
|
var fileName = getUnique(sanitize(rawName), res.ext, existingNames);
|
|
existingNames.push(fileName.toLowerCase());
|
|
zip.file(fileName, res.data, opts);
|
|
console.log('DONE ---- ' + fileName);
|
|
setTimeout(done, 500);
|
|
});
|
|
});
|
|
};
|
|
|
|
// Files (mediatags...)
|
|
var todoFile = function () {
|
|
var it;
|
|
var dl = _downloadFile(ctx, fData, function (err, res) {
|
|
if (it) { clearInterval(it); }
|
|
if (err) { return void error(err); }
|
|
var opts = {
|
|
binary: true,
|
|
};
|
|
var extIdx = rawName.lastIndexOf('.');
|
|
var name = extIdx !== -1 ? rawName.slice(0,extIdx) : rawName;
|
|
var ext = extIdx !== -1 ? rawName.slice(extIdx) : "";
|
|
var fileName = getUnique(sanitize(name), ext, existingNames);
|
|
existingNames.push(fileName.toLowerCase());
|
|
zip.file(fileName, res.content, opts);
|
|
console.log('DONE ---- ' + fileName);
|
|
setTimeout(done, 1000);
|
|
});
|
|
it = setInterval(function () {
|
|
if (ctx.stop) {
|
|
clearInterval(it);
|
|
dl.cancel();
|
|
}
|
|
}, 50);
|
|
};
|
|
if (parsed.hashData.type === 'file') {
|
|
return void todoFile();
|
|
}
|
|
todoPad();
|
|
});
|
|
});
|
|
// cb(err, blob);
|
|
};
|
|
|
|
// Add folders and their content recursively in the zip
|
|
var makeFolder = function (ctx, root, zip, fd) {
|
|
if (typeof (root) !== "object") { return; }
|
|
var existingNames = [];
|
|
Object.keys(root).forEach(function (k) {
|
|
var el = root[k];
|
|
if (typeof el === "object" && el.metadata !== true) { // if folder
|
|
var fName = getUnique(sanitize(k), '', existingNames);
|
|
existingNames.push(fName.toLowerCase());
|
|
return void makeFolder(ctx, el, zip.folder(fName), fd);
|
|
}
|
|
if (ctx.data.sharedFolders[el]) { // if shared folder
|
|
var sfData = ctx.sf[el].metadata;
|
|
var sfName = getUnique(sanitize(sfData.title || 'Folder'), '', existingNames);
|
|
existingNames.push(sfName.toLowerCase());
|
|
return void makeFolder(ctx, ctx.sf[el].root, zip.folder(sfName), ctx.sf[el].filesData);
|
|
}
|
|
var fData = fd[el];
|
|
if (fData) {
|
|
addFile(ctx, zip, fData, existingNames);
|
|
return;
|
|
}
|
|
});
|
|
};
|
|
|
|
// Main function. Create the empty zip and fill it starting from drive.root
|
|
var create = function (data, getPad, fileHost, cb, progress) {
|
|
if (!data || !data.uo || !data.uo.drive) { return void cb('EEMPTY'); }
|
|
var sem = Saferphore.create(5);
|
|
var ctx = {
|
|
fileHost: fileHost,
|
|
get: getPad,
|
|
data: data.uo.drive,
|
|
folder: data.folder,
|
|
sf: data.sf,
|
|
zip: new JsZip(),
|
|
errors: [],
|
|
sem: sem,
|
|
updateProgress: progress,
|
|
max: 0,
|
|
done: 0
|
|
};
|
|
var filesData = data.sharedFolderId && ctx.sf[data.sharedFolderId] ? ctx.sf[data.sharedFolderId].filesData : ctx.data.filesData;
|
|
progress('reading', -1);
|
|
nThen(function (waitFor) {
|
|
ctx.waitFor = waitFor;
|
|
var zipRoot = ctx.zip.folder(data.name || Messages.fm_rootName);
|
|
makeFolder(ctx, ctx.folder || ctx.data.root, zipRoot, filesData);
|
|
progress('download', {});
|
|
}).nThen(function () {
|
|
console.log(ctx.zip);
|
|
console.log(ctx.errors);
|
|
progress('compressing', -1);
|
|
ctx.zip.generateAsync({type: 'blob'}).then(function (content) {
|
|
progress('done', -1);
|
|
cb(content, ctx.errors);
|
|
});
|
|
});
|
|
|
|
var stop = function () {
|
|
ctx.stop = true;
|
|
delete ctx.zip;
|
|
};
|
|
return {
|
|
stop: stop
|
|
};
|
|
};
|
|
|
|
|
|
var _downloadFolder = function (ctx, data, cb, updateProgress) {
|
|
create(data, ctx.get, ctx.fileHost, function (blob, errors) {
|
|
if (errors && errors.length) { console.error(errors); } // TODO show user errors
|
|
var dl = function () {
|
|
saveAs(blob, data.folderName);
|
|
};
|
|
cb(null, {download: dl});
|
|
}, function (state, progress) {
|
|
if (state === "reading") {
|
|
updateProgress.folderProgress(0);
|
|
}
|
|
if (state === "download") {
|
|
if (typeof progress.current !== "number") { return; }
|
|
updateProgress.folderProgress(progress.current / progress.max);
|
|
}
|
|
else if (state === "done") {
|
|
updateProgress.folderProgress(1);
|
|
}
|
|
});
|
|
};
|
|
|
|
|
|
return {
|
|
create: create,
|
|
downloadFile: _downloadFile,
|
|
downloadPad: _downloadPad,
|
|
downloadFolder: _downloadFolder,
|
|
};
|
|
});
|