fog--fog/bin/fog

#!/usr/bin/env ruby
require File.join(File.dirname(__FILE__), '..', 'lib', 'fog')
require 'irb'
require 'yaml'

module Fog
  module Credentials
    credential = (ARGV.first && :"#{ARGV.first}") || :default
    unless Fog.credentials[credential]
      exit
    end
  end
end

module AWS
  class << self
    credential = (ARGV.first && :"#{ARGV.first}") || :default
    if Fog.credentials[credential][:aws_access_key_id] && Fog.credentials[credential][:aws_secret_access_key]

      def connections
        @@connections ||= Hash.new do |hash, key|
          credential = (ARGV.first && :"#{ARGV.first}") || :default
          credentials = Fog.credentials[credential]
          hash[key] = case key
          when :ec2
            Fog::AWS::EC2.new(credentials)
          when :s3
            Fog::AWS::S3.new(credentials)
          end
        end
      end

      def addresses
        connections[:ec2].addresses
      end

      def directories
        connections[:s3].directories
      end

      def flavors
        connections[:ec2].flavors
      end

      def images
        connections[:ec2].images
      end

      def servers
        connections[:ec2].servers
      end

      def key_pairs
        connections[:ec2].key_pairs
      end

      def security_groups
        connections[:ec2].security_groups
      end

      def snapshots
        connections[:ec2].snapshots
      end

      def volumes
        connections[:ec2].volumes
      end

    end
  end
end

module Rackspace
  class << self
    credential = (ARGV.first && :"#{ARGV.first}") || :default
    if Fog.credentials[credential][:rackspace_api_key] && Fog.credentials[credential][:rackspace_username]

      def connections
        @@connections ||= Hash.new do |hash, key|
          credential = (ARGV.first && :"#{ARGV.first}") || :default
          credentials = Fog.credentials[credential]
          hash[key] = case key
          when :files
            Fog::Rackspace::Files.new(credentials)
          when :servers
            Fog::Rackspace::Servers.new(credentials)
          end
        end
      end

      def flavors
        connections[:servers].flavors
      end

      def images
        connections[:servers].images
      end

      def servers
        connections[:servers].servers
      end
      
    end
  end
end

ARGV.clear # Avoid passing args to IRB
IRB.setup(nil)
@irb = IRB::Irb.new(nil)
IRB.conf[:MAIN_CONTEXT] = @irb.context
@irb.context.workspace = IRB::WorkSpace.new(binding)
catch(:IRB_EXIT) { @irb.eval_input }