train-core 3.4.9 → 3.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6272024b67e2dac1df86499e4687dc44d5df58b328146d52cb864433d7c6e6a7
-  data.tar.gz: a4133877056cc7feed58ac9dab964ecfc435d28cc30a09d6dd104577a8a30374
+  metadata.gz: 4debe973595fd78f3fc61e3ebb8d62061d5d6a72f72077ae972422a24c8fa403
+  data.tar.gz: 9696bd7bc5ad5de2cbaa1fef3b8cf38deb34ebb4a78b61fe85acf648559d50c4
 SHA512:
-  metadata.gz: 53bcc25af9bade96ee63aa1f0940927cb4e4183220737269bb4882c800df94c1bf6cb924abf5cb6b354b18ae000b6fba3d7a35787a3764d9402cd63f7b90bdd2
-  data.tar.gz: 775d8a24be41a87f8427f3850b829b1acbc78d2a9ef8e299e1e4c9f9dbd352838d7bd9ef906723832171d0f822a10dac39827ab8a1b92c7ebec220bf37e337dd
+  metadata.gz: a4edf4fadbfe0f9e9ee8febd26bfa806d953129ed06e1c892acde68ffa3fcb0820bef51f5eeacf31f61ba83e3ed0084cec5ed2cc34911974338a20e1b3c0c580
+  data.tar.gz: e37d56303596018a0185b569c85fe72a38a00d0d99ee597f9cd42012142eb33eb0483c74e7d17fd1141c77982e7346507ad4f1d0c7fd31614fc84b66803ad99a

data/lib/train/file/local.rb

@@ -15,6 +15,12 @@ module Train
         nil
       end
 
+      def content=(new_content)
+        ::File.open(@path, "w", encoding: "UTF-8") { |fp| fp.write(new_content) }
+
+        @content = new_content
+      end
+
       def link_path
         return nil unless symlink?
 

data/lib/train/file/remote/unix.rb

@@ -1,3 +1,4 @@
+require "base64" unless defined?(Base64)
 require "shellwords" unless defined?(Shellwords)
 
 module Train
@@ -19,6 +20,21 @@ module Train
             end
         end
 
+        def content=(new_content)
+          execute_result = @backend.run_command("base64 --help")
+          if execute_result.exit_status != 0
+            raise TransportError, "#{self.class} found no base64 binary for file writes"
+          end
+
+          unix_cmd = format("echo '%<base64>s' | base64 --decode > %<file>s",
+                            base64: Base64.strict_encode64(new_content),
+                            file: @spath)
+
+          @backend.run_command(unix_cmd)
+
+          @content = new_content
+        end
+
         def exist?
           @exist ||= begin
             f = @follow_symlink ? "" : " || test -L #{@spath}"

data/lib/train/file/remote/windows.rb

@@ -26,6 +26,16 @@ module Train
           @content
         end
 
+        def content=(new_content)
+          win_cmd = format('[IO.File]::WriteAllBytes("%<file>s", [Convert]::FromBase64String("%<base64>s"))',
+                           base64: Base64.strict_encode64(new_content),
+                           file: @spath)
+
+          @backend.run_command(win_cmd)
+
+          @content = new_content
+        end
+
         def exist?
           return @exist if defined?(@exist)
 

data/lib/train/plugins/base_connection.rb

@@ -1,6 +1,7 @@
 require_relative "../errors"
 require_relative "../extras"
 require_relative "../file"
+require "fileutils" unless defined?(FileUtils)
 require "logger"
 
 class Train::Plugins::Transport
@@ -161,6 +162,48 @@ class Train::Plugins::Transport
       @cache[:file][path] ||= file_via_connection(path, *args)
     end
 
+    # Uploads local files or directories to remote host.
+    #
+    # @param locals [Array<String>] paths to local files or directories
+    # @param remote [String] path to remote destination
+    # @raise [TransportFailed] if the files could not all be uploaded
+    #   successfully, which may vary by implementation
+    def upload(locals, remote)
+      unless file(remote).directory?
+        raise TransportError, "#{self.class} expects remote directory as second upload parameter"
+      end
+
+      Array(locals).each do |local|
+        new_content = File.read(local)
+        remote_file = File.join(remote, File.basename(local))
+
+        logger.debug("Attempting to upload '#{local}' as file #{remote_file}")
+
+        file(remote_file).content = new_content
+      end
+    end
+
+    # Download remote files or directories to local host.
+    #
+    # @param remotes [Array<String>] paths to remote files or directories
+    # @param local [String] path to local destination. If `local` is an
+    #   existing directory, `remote` will be downloaded into the directory
+    #   using its original name
+    # @raise [TransportFailed] if the files could not all be downloaded
+    #   successfully, which may vary by implementation
+    def download(remotes, local)
+      FileUtils.mkdir_p(File.dirname(local))
+
+      Array(remotes).each do |remote|
+        new_content = file(remote).content
+        local_file = File.join(local, File.basename(remote))
+
+        logger.debug("Attempting to download '#{remote}' as file #{local_file}")
+
+        File.open(local_file, "w") { |fp| fp.write(new_content) }
+      end
+    end
+
     # Builds a LoginCommand which can be used to open an interactive
     # session on the remote host.
     #

data/lib/train/transports/cisco_ios_connection.rb

@@ -29,6 +29,14 @@ class Train::Transports::SSH
       result.stdout.split(" ")[-1]
     end
 
+    def upload(locals, remote)
+      raise NotImplementedError, "#{self.class} does not implement #upload()"
+    end
+
+    def download(remotes, local)
+      raise NotImplementedError, "#{self.class} does not implement #download()"
+    end
+
     private
 
     def establish_connection

data/lib/train/transports/local.rb

@@ -39,6 +39,18 @@ module Train::Transports
         "local://"
       end
 
+      def upload(locals, remote)
+        FileUtils.mkdir_p(remote)
+
+        Array(locals).each do |local|
+          FileUtils.cp_r(local, remote)
+        end
+      end
+
+      def download(remotes, local)
+        upload(remotes, local)
+      end
+
       private
 
       def select_runner(options)
@@ -187,7 +199,7 @@ module Train::Transports
           @server_pid = start_pipe_server(pipe_name)
 
           # Ensure process is killed when the Train process exits
-          at_exit { close }
+          at_exit { close rescue Errno::EIO }
 
           pipe = nil
 
@@ -254,4 +266,4 @@ module Train::Transports
       end
     end
   end
-end
+end

data/lib/train/version.rb

@@ -2,5 +2,5 @@
 # Author:: Dominik Richter (<dominik.richter@gmail.com>)
 
 module Train
-  VERSION = "3.4.9".freeze
+  VERSION = "3.5.2".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: train-core
 version: !ruby/object:Gem::Version
-  version: 3.4.9
+  version: 3.5.2
 platform: ruby
 authors:
 - Chef InSpec Team
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-01-27 00:00:00.000000000 Z
+date: 2021-02-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: addressable