53 lines
1.5 KiB
JavaScript
53 lines
1.5 KiB
JavaScript
import { dirname, join } from 'path';
|
|
import StatsD from 'statsy';
|
|
import Client from './ssh2-sftp-client';
|
|
import debugname from 'debug';
|
|
const debug = debugname('hostr:sftp');
|
|
|
|
const statsdOpts = { prefix: 'hostr-api', host: process.env.STATSD_HOST };
|
|
const statsd = new StatsD(statsdOpts);
|
|
|
|
export function get(remotePath) {
|
|
debug('fetching', join('hostr', 'uploads', remotePath));
|
|
const sftp = new Client();
|
|
return sftp.connect({
|
|
host: process.env.SFTP_HOST,
|
|
port: process.env.SFTP_PORT,
|
|
username: process.env.SFTP_USERNAME,
|
|
password: process.env.SFTP_PASSWORD,
|
|
})
|
|
.then(() => sftp.get(join('hostr', 'uploads', remotePath), { encoding: null }));
|
|
}
|
|
|
|
function sendFile(localPath, remotePath) {
|
|
const sftp = new Client();
|
|
return sftp.connect({
|
|
host: process.env.SFTP_HOST,
|
|
port: process.env.SFTP_PORT,
|
|
username: process.env.SFTP_USERNAME,
|
|
password: process.env.SFTP_PASSWORD,
|
|
})
|
|
.then(() => sftp.put(localPath, remotePath, true))
|
|
.catch((err) => {
|
|
if (err.message === 'No such file') {
|
|
debug('Creating directory');
|
|
return sftp.mkdir(dirname(remotePath), true)
|
|
.then(() => sftp.put(localPath, remotePath, true));
|
|
}
|
|
throw err;
|
|
});
|
|
}
|
|
|
|
export function *upload(localPath, remotePath) {
|
|
let done = false;
|
|
for (let retries = 0; retries < 5; retries++) {
|
|
try {
|
|
done = yield sendFile(localPath, remotePath);
|
|
break;
|
|
} catch (err) {
|
|
statsd.incr('file.upload.retry', 1);
|
|
debug('retry');
|
|
}
|
|
}
|
|
return done;
|
|
}
|