init

lib/store.js

@@ -0,0 +1,113 @@
+'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
+      }
+      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;