dylan.derr/Send
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixes #543 added FILE_DIR environment variable

Browse source
This commit is contained in:
Danny Coates 2017-09-06 13:25:27 -07:00
parent bbaeb44b26
commit 7a8e9b5de1
No known key found for this signature in database
GPG key ID: 4C442633C62E00CB
4 changed files with 19 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
server/config.js
View file

@ -1,4 +1,7 @@
const convict = require('convict');
const { tmpdir } = require('os');
const path = require('path');
const { randomBytes } = require('crypto');
const conf = convict({
s3_bucket: {
@ -56,6 +59,11 @@ const conf = convict({
format: 'url',
default: 'https://send.firefox.com',
env: 'BASE_URL'
},
file_dir: {
format: 'String',
default: `${tmpdir()}${path.sep}send-${randomBytes(4).toString('hex')}`,
env: 'FILE_DIR'
}
});

18
server/storage.js
View file

@ -1,8 +1,8 @@
const AWS = require('aws-sdk');
const s3 = new AWS.S3();
const mkdirp = require('mkdirp');
const config = require('./config');
const { tmpdir } = require('os');
const fs = require('fs');
const path = require('path');
@ -25,7 +25,7 @@ redis_client.on('error', err => {
log.error('Redis:', err);
});
let tempDir = null;
const fileDir = config.file_dir;
if (config.s3_bucket) {
module.exports = {
@ -44,8 +44,8 @@ if (config.s3_bucket) {
metadata
};
} else {
tempDir = fs.mkdtempSync(`${tmpdir()}${path.sep}send-`);
log.info('tempDir', tempDir);
mkdirp.sync(config.file_dir);
log.info('fileDir', fileDir);
module.exports = {
filename: filename,
exists: exists,
@ -123,7 +123,7 @@ function setField(id, key, value) {
function localLength(id) {
return new Promise((resolve, reject) => {
try {
resolve(fs.statSync(path.join(tempDir, id)).size);
resolve(fs.statSync(path.join(fileDir, id)).size);
} catch (err) {
reject();
}
@ -131,12 +131,12 @@ function localLength(id) {
}
function localGet(id) {
return fs.createReadStream(path.join(tempDir, id));
return fs.createReadStream(path.join(fileDir, id));
}
function localSet(newId, file, filename, meta) {
return new Promise((resolve, reject) => {
const filepath = path.join(tempDir, newId);
const filepath = path.join(fileDir, newId);
const fstream = fs.createWriteStream(filepath);
file.pipe(fstream);
file.on('limit', () => {
@ -166,7 +166,7 @@ function localDelete(id, delete_token) {
} else {
redis_client.del(id);
log.info('Deleted:', id);
resolve(fs.unlinkSync(path.join(tempDir, id)));
resolve(fs.unlinkSync(path.join(fileDir, id)));
}
});
});
@ -175,7 +175,7 @@ function localDelete(id, delete_token) {
function localForceDelete(id) {
return new Promise((resolve, reject) => {
redis_client.del(id);
resolve(fs.unlinkSync(path.join(tempDir, id)));
resolve(fs.unlinkSync(path.join(fileDir, id)));
});
}