lib/specinfra/ec2_metadata.rb in specinfra-2.18.3 vs lib/specinfra/ec2_metadata.rb in specinfra-2.19.0

@@ -1,23 +1,76 @@
 # -*- coding: utf-8 -*-
 module Specinfra
   class Ec2Metadata
     def initialize
       @base_uri = 'http://169.254.169.254/latest/meta-data/'
+      @metadata = {}
     end
 
-    def get(path='')
+    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(path + key)
+          metadata[key[0..-2]] = get_metadata(path + key)
         else
           if key =~ %r{=}
             key = key.split('=')[0] + '/'
-            metadata[key[0..-2]] = get(path + key)
+            metadata[key[0..-2]] = get_metadata(path + key)
           else
             ret = get_endpoint(path)
             metadata[key] = get_endpoint(path + key) if ret
           end
         end