hostr/lib/sftp.js

import { dirname, join } from 'path';
import Client from './ssh2-sftp-client';
import debugname from 'debug';
const debug = debugname('hostr:sftp');

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) {
      debug('retry');
    }
  }
  return done;
}