docker.rb in dockly-2.1.0

- old
+ new
@@ -23,33 +23,50 @@
 
   def push_cache(version)
     ensure_present! :output_dir
     if cache = pull_from_s3(version)
       debug "inserting to #{output_directory}"
-      path = File.expand_path(cache.path)
-      path_parent = File.dirname(path)
-      tar_flags = keep_old_files ? '-xkf' : 'xf'
-      container = ::Docker::Container.create(
-        'Image' => image.id,
-        'Cmd' => ['/bin/bash', '-c', [
-            "mkdir -p #{File.dirname(output_directory)}",
-            '&&',
-            "tar #{tar_flags} #{File.join('/', 'host', path)} -C #{File.dirname(output_directory)}"
-          ].join(' ')
-        ],
-        'Volumes' => {
-          File.join('/', 'host', path_parent) => { path_parent => 'rw' }
-        }
-      )
-      container.start('Binds' => ["#{path_parent}:#{File.join('/', 'host', path_parent)}"])
-      result = container.wait['StatusCode']
-      raise "Got bad status code when copying build cache: #{result}" unless result.zero?
-      self.image = container.commit
+      if safe_push_cache
+        push_cache_safe(cache)
+      else
+        push_cache_with_volumes(cache)
+      end
       debug "inserted cache into #{output_directory}"
       cache.close
     else
       info "could not find #{s3_object(version)}"
     end
+  end
+
+  def push_cache_safe(cache)
+    container = image.run("mkdir -p #{File.dirname(output_directory)}")
+    image_with_dir = container.tap(&:wait).commit
+    self.image = image_with_dir.insert_local(
+      'localPath' => cache.path,
+      'outputPath' => File.dirname(output_directory)
+    )
+  end
+
+  def push_cache_with_volumes(cache)
+    path = File.expand_path(cache.path)
+    path_parent = File.dirname(path)
+    tar_flags = keep_old_files ? '-xkf' : 'xf'
+    container = ::Docker::Container.create(
+      'Image' => image.id,
+      'Cmd' => ['/bin/bash', '-c', [
+          "mkdir -p #{File.dirname(output_directory)}",
+          '&&',
+          "tar #{tar_flags} #{File.join('/', 'host', path)} -C #{File.dirname(output_directory)}"
+        ].join(' ')
+      ],
+      'Volumes' => {
+        File.join('/', 'host', path_parent) => { path_parent => 'rw' }
+      }
+    )
+    container.start('Binds' => ["#{path_parent}:#{File.join('/', 'host', path_parent)}"])
+    result = container.wait['StatusCode']
+    raise "Got bad status code when copying build cache: #{result}" unless result.zero?
+    self.image = container.commit
   end
 
   def copy_output_dir(container)
     ensure_present! :output_dir
     file_path = File.join(tmp_dir,s3_object(hash_output))