# -*- coding: utf-8 -*-
module Specinfra
  class Ec2Metadata
    def initialize
      @base_uri = 'http://169.254.169.254/latest/meta-data/'
    end

    def get(path='')
      metadata = {}

      keys = Specinfra::Runner.run_command("curl #{@base_uri}#{path}").stdout.split("\n")

      keys.each do |key|
        if key =~ %r{/$}
          metadata[key[0..-2]] = get(path + key)
        else
          if key =~ %r{=}
            key = key.split('=')[0] + '/'
            metadata[key[0..-2]] = get(path + key)
          else
            ret = get_endpoint(path)
            metadata[key] = get_endpoint(path + key) if ret
          end
        end
      end

      metadata
    end

    def get_endpoint(path)
      ret = Specinfra::Runner.run_command("curl #{@base_uri}#{path}")
      if ret.success?
        ret.stdout
      else
        nil
      end
    end

  end
end

# -*- coding: utf-8 -*-
module Specinfra
  class Ec2Metadata
    def initialize
      @base_uri = 'http://169.254.169.254/latest/meta-data/'
      @metadata = {}
    end

    def get
      @metadata = get_metadata
      self
    end

    def [](key)
      if @metadata[key].nil?
        begin
          require "specinfra/ec2_metadata/#{key}"
          inventory_class = Specinfra::Ec2Metadata.const_get(key.to_camel_case)
          @metadata[key] = inventory_class.get
        rescue LoadError
          @metadata[key] = nil
        rescue
          @metadata[key] = nil
        end
      end

      @metadata[key]
    end

    def empty?
      @metadata.empty?
    end

    def each
      keys.each do |k|
        yield k, @metadata[k]
      end
    end

    def each_key
      keys.each do |k|
        yield k
      end
    end

    def each_value
      keys.each do |k|
        yield @metadata[k]
      end
    end

    def keys
      @metadata.keys
    end

    def inspect
      @metadata
    end

    private
    def get_metadata(path='')
      metadata = {}

      keys = Specinfra::Runner.run_command("curl #{@base_uri}#{path}").stdout.split("\n")

      keys.each do |key|
        if key =~ %r{/$}
          metadata[key[0..-2]] = get_metadata(path + key)
        else
          if key =~ %r{=}
            key = key.split('=')[0] + '/'
            metadata[key[0..-2]] = get_metadata(path + key)
          else
            ret = get_endpoint(path)
            metadata[key] = get_endpoint(path + key) if ret
          end
        end
      end

      metadata
    end

    def get_endpoint(path)
      ret = Specinfra::Runner.run_command("curl #{@base_uri}#{path}")
      if ret.success?
        ret.stdout
      else
        nil
      end
    end

  end
end

Fix to extend Ec2Metadata easily by external gem
Fix to extend Ec2Metadata easily by external gem
