feature: download all command

Author: vslinko

lib/RemoteSync.coffee

@@ -1,3 +1,4 @@
+DownloadAllCommand = require "./commands/DownloadAllCommand"
 SettingsLocator = require "./SettingsLocator"
 UploadListener = require "./UploadListener"
 ScpTransport = require "./transports/ScpTransport"
@@ -15,3 +16,7 @@ class RemoteSync
       scp: new ScpTransport logger
 
     new UploadListener logger, settingsLocator, transports
+
+    downloadAll = new DownloadAllCommand logger, settingsLocator, transports
+
+    atom.workspaceView.command "remote-sync:download-all", downloadAll.run.bind(downloadAll)

lib/commands/DownloadAllCommand.coffee

@@ -0,0 +1,33 @@
+minimatch = require "minimatch"
+async = require "async"
+
+
+module.exports =
+class DownloadAllCommand
+  constructor: (@logger, @settingsLocator, @transports) ->
+
+  run: ->
+    buffer = atom.workspace.getActiveEditor().getBuffer()
+    return unless buffer.file
+    filePath = buffer.file.path
+    @settingsLocator.locate filePath, (err, result) =>
+      return if err
+
+      settings = result.settings
+
+      @transports[settings.transport].fetchFileTree settings, (err, files) =>
+        return if err
+
+        if settings.ignore
+          patterns = settings.ignore
+          files = files.filter (file) ->
+            for pattern in patterns
+              if minimatch file, pattern
+                return false
+            return true
+
+        async.mapSeries files, (file, callback) =>
+          @transports[settings.transport].download result.rootDirectory, file, settings, callback
+        , (err) =>
+          return @logger.error if err
+          @logger.log "Downloaded all files"

lib/transports/ScpTransport.coffee

@@ -1,5 +1,6 @@
 {MessagePanelView, PlainMessageView} = require "atom-message-panel"
 SSHConnection = require "ssh2"
+mkdirp = require "mkdirp"
 path = require "path"
 
 
@@ -34,6 +35,50 @@ class ScpTransport
 
             sftp.end()
             callback()
+    @connections = {}
+
+  download: (rootDirectory, relativeFilePath, settings, callback) ->
+    localFilePath = path.join(rootDirectory, relativeFilePath)
+    targetFilePath = path.join(settings.target, relativeFilePath)
+
+    errorHandler = (err) =>
+      @logger.error err
+      callback()
+
+    @_getConnection settings.hostname, settings.username, settings.password, (err, c) =>
+      return errorHandler err if err
+
+      @logger.log "Downloading: #{relativeFilePath}"
+
+      c.sftp (err, sftp) =>
+        return errorHandler err if err
+
+        mkdirp path.dirname(localFilePath), (err) =>
+          return errorHandler err if err
+
+          sftp.fastGet targetFilePath, localFilePath, (err) =>
+            return errorHandler err if err
+
+            @logger.log "Downloaded: #{relativeFilePath}"
+
+            sftp.end()
+            callback()
+
+  fetchFileTree: (settings, callback) ->
+    @_getConnection settings.hostname, settings.username, settings.password, (err, c) =>
+      return callback err if err
+
+      c.exec "find \"#{settings.target}\" -type f", (err, result) ->
+        return callback err if err
+
+        buf = ""
+        result.on "data", (data) -> buf += data.toString()
+        result.on "end", ->
+          targetRegexp = new RegExp "^#{settings.target}/"
+          files = buf.split("\n")
+            .filter((f) -> targetRegexp.test(f))
+            .map((f) -> f.replace(targetRegexp, ""))
+          callback null, files
 
   _getConnection: (hostname, username, password, callback) ->
     key = "#{username}@#{hostname}"

package.json

@@ -13,6 +13,7 @@
     "ssh2": "^0.2",
     "minimatch": "^0.2",
     "async": "^0.2",
+    "mkdirp": "^0.3",
     "atom-message-panel": "git+https://github.com/vslinko/atom-message-panel.git#patch-1"
   }
 }