mirror of
https://github.com/fog/fog.git
synced 2022-11-09 13:51:43 -05:00
95 lines
3 KiB
Ruby
95 lines
3 KiB
Ruby
module Fog
|
|
module SCP
|
|
|
|
def self.new(address, username, options = {})
|
|
if Fog.mocking?
|
|
Fog::SCP::Mock.new(address, username, options)
|
|
else
|
|
Fog::SCP::Real.new(address, username, options)
|
|
end
|
|
end
|
|
|
|
class Mock
|
|
|
|
def self.data
|
|
@data ||= Hash.new do |hash, key|
|
|
hash[key] = []
|
|
end
|
|
end
|
|
|
|
def initialize(address, username, options)
|
|
@address = address
|
|
@username = username
|
|
@options = options
|
|
end
|
|
|
|
def upload(local_path, remote_path, upload_options = {})
|
|
self.class.data[@address] << { :username => @username,
|
|
:options => @options,
|
|
:local_path => local_path,
|
|
:remote_path => remote_path,
|
|
:upload_options => upload_options }
|
|
end
|
|
|
|
def download(remote_path, local_path, download_options = {})
|
|
self.class.data[@address] << { :username => @username,
|
|
:options => @options,
|
|
:remote_path => remote_path,
|
|
:local_path => local_path,
|
|
:download_options => download_options }
|
|
end
|
|
|
|
end
|
|
|
|
class Real
|
|
|
|
def initialize(address, username, options)
|
|
require 'net/scp'
|
|
|
|
key_manager = Net::SSH::Authentication::KeyManager.new(nil, options)
|
|
|
|
unless options[:key_data] || options[:keys] || options[:password] || key_manager.agent
|
|
raise ArgumentError.new(':key_data, :keys, :password or a loaded ssh-agent is required to initialize SSH')
|
|
end
|
|
|
|
if options[:key_data] || options[:keys]
|
|
options[:keys_only] = true
|
|
#Explicitly set these so net-ssh doesn't add the default keys
|
|
#as seen at https://github.com/net-ssh/net-ssh/blob/master/lib/net/ssh/authentication/session.rb#L131-146
|
|
options[:keys] = [] unless options[:keys]
|
|
options[:key_data] = [] unless options[:key_data]
|
|
end
|
|
|
|
@address = address
|
|
@username = username
|
|
@options = { :paranoid => false }.merge(options)
|
|
end
|
|
|
|
def upload(local_path, remote_path, upload_options = {})
|
|
begin
|
|
Net::SCP.start(@address, @username, @options) do |scp|
|
|
scp.upload!(local_path, remote_path, upload_options) do |ch, name, sent, total|
|
|
# TODO: handle progress display?
|
|
end
|
|
end
|
|
rescue Exception => error
|
|
raise error
|
|
end
|
|
end
|
|
|
|
def download(remote_path, local_path, download_options = {})
|
|
begin
|
|
Net::SCP.start(@address, @username, @options) do |scp|
|
|
scp.download!(remote_path, local_path, download_options) do |ch, name, sent, total|
|
|
# TODO: handle progress display?
|
|
end
|
|
end
|
|
rescue Exception => error
|
|
raise error
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|