114 lines
3.0 KiB
JavaScript
114 lines
3.0 KiB
JavaScript
'use strict';
|
|
const fsp = require('fs-promise');
|
|
const path = require('path');
|
|
const Transform = require('stream').Transform;
|
|
const debug = require('debug')('psitransfer:store');
|
|
const httpErrors = require('http-errors');
|
|
|
|
class StreamLen extends Transform {
|
|
constructor(options) {
|
|
super(options);
|
|
this.bytes = 0;
|
|
}
|
|
|
|
_transform(chunk, encoding, cb) {
|
|
this.bytes += chunk.length;
|
|
this.push(chunk);
|
|
cb();
|
|
}
|
|
}
|
|
|
|
// TODO ???: make tus-store complaint: https://github.com/blockai/abstract-tus-store
|
|
class Store {
|
|
|
|
constructor(targetDir) {
|
|
this.dir = path.normalize(targetDir);
|
|
}
|
|
|
|
|
|
getFilename(fid) {
|
|
let p = path.resolve(this.dir, fid.replace('++', '/'));
|
|
if(!p.startsWith(this.dir)) {
|
|
throw new Error('file name not in jail path. aborting');
|
|
}
|
|
return p;
|
|
}
|
|
|
|
|
|
async create(fid, opts = {}) {
|
|
debug(`New File ${this.getFilename(fid)}`);
|
|
await fsp.ensureDir(path.dirname(this.getFilename(fid)));
|
|
await fsp.writeJson(this.getFilename(fid) + '.json', Object.assign(opts, {
|
|
isPartial: true
|
|
}));
|
|
return {uploadId: fid};
|
|
}
|
|
|
|
|
|
async info(fid) {
|
|
try {
|
|
const info = await fsp.readJson(this.getFilename(fid) + '.json');
|
|
const stat = await fsp.stat(this.getFilename(fid));
|
|
info.size = stat.size;
|
|
info.offset = stat.size;
|
|
debug(`Fetched Fileinfo ${this.getFilename(fid)}`);
|
|
return info;
|
|
} catch(e) {
|
|
if(e.code === 'ENOENT') {
|
|
throw httpErrors.NotFound();
|
|
}
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
|
|
async append(fid, readStream, offset) {
|
|
debug(`Append Data to ${this.getFilename(fid)}`);
|
|
const uploadSize = new StreamLen();
|
|
const ws = fsp.createWriteStream(this.getFilename(fid), {flags: 'a', start: offset});
|
|
|
|
const ret = new Promise((resolve, reject) => {
|
|
ws.on('finish', async() => {
|
|
const info = await this.info(fid);
|
|
if(info.size >= info.uploadLength) delete info.isPartial;
|
|
await fsp.writeJson(this.getFilename(fid) + '.json', info);
|
|
debug(`Finished appending Data to ${this.getFilename(fid)}`);
|
|
return resolve({ offset: info.offset, upload: info });
|
|
});
|
|
ws.on('error', reject);
|
|
// End writeStream on connection abort and wait for drain
|
|
readStream.on('close', () => {
|
|
ws.on('drain', () => {
|
|
ws.end();
|
|
});
|
|
});
|
|
});
|
|
|
|
readStream.pipe(uploadSize).pipe(ws);
|
|
return ret;
|
|
}
|
|
|
|
|
|
createReadStream(fid, start, end, cb) {
|
|
debug(`Create ReadStream for ${this.getFilename(fid)}`);
|
|
this.info(fid).then(info => {
|
|
let contentLength = info.size;
|
|
if(start > 0) {
|
|
if(!end) end = info.size - 1;
|
|
contentLength = end - start + 1
|
|
}
|
|
if(cb) cb({ contentLength, metadata: info.metadata, info });
|
|
});
|
|
return fsp.createReadStream(this.getFilename(fid), {start, end});
|
|
}
|
|
|
|
|
|
async del(fid) {
|
|
debug(`Delete ${this.getFilename(fid)}`);
|
|
await fsp.unlink(this.getFilename(fid) + '.json');
|
|
await fsp.unlink(this.getFilename(fid));
|
|
}
|
|
}
|
|
|
|
module.exports = Store;
|