nicer credential management for bin/fog

This commit is contained in:
Wesley Beary 2010-01-23 12:08:37 -08:00
parent 85dceedd13
commit ea5f87aa3f
1 changed files with 22 additions and 16 deletions

38
bin/fog
View File

@ -17,10 +17,12 @@ module AWS
credential = (ARGV.first && :"#{ARGV.first}") || :default credential = (ARGV.first && :"#{ARGV.first}") || :default
if Fog.credentials[credential][:aws_access_key_id] && Fog.credentials[credential][:aws_secret_access_key] if Fog.credentials[credential][:aws_access_key_id] && Fog.credentials[credential][:aws_secret_access_key]
def connections def [](service)
@@connections ||= Hash.new do |hash, key| @@connections ||= Hash.new do |hash, key|
credential = (ARGV.first && :"#{ARGV.first}") || :default credential = (ARGV.first && :"#{ARGV.first}") || :default
credentials = Fog.credentials[credential] credentials = Fog.credentials[credential].reject do |k, v|
![:aws_access_key_id, :aws_secret_access_key].include?(k)
end
hash[key] = case key hash[key] = case key
when :ec2 when :ec2
Fog::AWS::EC2.new(credentials) Fog::AWS::EC2.new(credentials)
@ -28,42 +30,43 @@ module AWS
Fog::AWS::S3.new(credentials) Fog::AWS::S3.new(credentials)
end end
end end
@@connections[service]
end end
def addresses def addresses
connections[:ec2].addresses self[:ec2].addresses
end end
def directories def directories
connections[:s3].directories self[:s3].directories
end end
def flavors def flavors
connections[:ec2].flavors self[:ec2].flavors
end end
def images def images
connections[:ec2].images self[:ec2].images
end end
def servers def servers
connections[:ec2].servers self[:ec2].servers
end end
def key_pairs def key_pairs
connections[:ec2].key_pairs self[:ec2].key_pairs
end end
def security_groups def security_groups
connections[:ec2].security_groups self[:ec2].security_groups
end end
def snapshots def snapshots
connections[:ec2].snapshots self[:ec2].snapshots
end end
def volumes def volumes
connections[:ec2].volumes self[:ec2].volumes
end end
end end
@ -75,10 +78,12 @@ module Rackspace
credential = (ARGV.first && :"#{ARGV.first}") || :default credential = (ARGV.first && :"#{ARGV.first}") || :default
if Fog.credentials[credential][:rackspace_api_key] && Fog.credentials[credential][:rackspace_username] if Fog.credentials[credential][:rackspace_api_key] && Fog.credentials[credential][:rackspace_username]
def connections def [](service)
@@connections ||= Hash.new do |hash, key| @@connections ||= Hash.new do |hash, key|
credential = (ARGV.first && :"#{ARGV.first}") || :default credential = (ARGV.first && :"#{ARGV.first}") || :default
credentials = Fog.credentials[credential] credentials = Fog.credentials[credential].reject do |k,v|
![:rackspace_api_key, :rackspace_username].include?(k)
end
hash[key] = case key hash[key] = case key
when :files when :files
Fog::Rackspace::Files.new(credentials) Fog::Rackspace::Files.new(credentials)
@ -86,18 +91,19 @@ module Rackspace
Fog::Rackspace::Servers.new(credentials) Fog::Rackspace::Servers.new(credentials)
end end
end end
@@connections[service]
end end
def flavors def flavors
connections[:servers].flavors self[:servers].flavors
end end
def images def images
connections[:servers].images self[:servers].images
end end
def servers def servers
connections[:servers].servers self[:servers].servers
end end
end end