lib/capistrano-db-tasks/asset.rb in capistrano-db-tasks-0.2.1 vs lib/capistrano-db-tasks/asset.rb in capistrano-db-tasks-0.3

@@ -1,23 +1,27 @@
 module Asset
   extend self
 
   def remote_to_local(cap)
-    servers = cap.find_servers :roles => :app
-    port = cap.port rescue 22
-    [cap.assets_dir].flatten.each do |dir|
-      system("rsync -a --del -L -K -vv --progress --rsh='ssh -p #{port}' #{cap.user}@#{servers.first}:#{cap.current_path}/#{dir} #{cap.local_assets_dir}")
+    servers = Capistrano::Configuration.env.send(:servers)
+    server = servers.detect { |s| s.roles.include?(:app) }
+    port = server.netssh_options[:port] || 22
+    user = server.netssh_options[:user]
+    [cap.fetch(:assets_dir)].flatten.each do |dir|
+      system("rsync -a --del -L -K -vv --progress --rsh='ssh -p #{port}' #{user}@#{server}:#{cap.current_path}/#{dir} #{cap.fetch(:local_assets_dir)}")
     end
   end
 
   def local_to_remote(cap)
-    servers = cap.find_servers :roles => :app
-    port = cap.port rescue 22
-    [cap.assets_dir].flatten.each do |dir|
-      system("rsync -a --del -L -K -vv --progress --rsh='ssh -p #{port}' ./#{dir} #{cap.user}@#{servers.first}:#{cap.current_path}/#{cap.local_assets_dir}")
+    servers = Capistrano::Configuration.env.send(:servers)
+    server = servers.detect { |s| s.roles.include?(:app) }
+    port = server.netssh_options[:port] || 22
+    user = server.netssh_options[:user]
+    [cap.fetch(:assets_dir)].flatten.each do |dir|
+      system("rsync -a --del -L -K -vv --progress --rsh='ssh -p #{port}' ./#{dir} #{user}@#{server}:#{cap.current_path}/#{cap.fetch(:local_assets_dir)}")
     end
   end
 
   def to_string(cap)
-    [cap.assets_dir].flatten.join(" ")
+    [cap.fetch(:assets_dir)].flatten.join(" ")
   end
 end