Changing some of the internals of autoloading so that there is a class-level method to query whether a given file has been loaded.

git-svn-id: https://reductivelabs.com/svn/puppet/trunk@2668 980ebf18-57e1-0310-9a29-db15c13687c0

1 files changed, 44 insertions, 15 deletions

diff --git a/lib/puppet/util/autoload.rb b/lib/puppet/util/autoload.rb
index ac4939a0b..42aa56c32 100644
--- a/lib/puppet/util/autoload.rb
+++ b/lib/puppet/util/autoload.rb
@@ -3,20 +3,42 @@ class Puppet::Util::Autoload
     include Puppet::Util
 
     @autoloaders = {}
-
-    attr_accessor :object, :path, :objwarn, :wrap
-
+    @loaded = {}
 
     class << self
         attr_reader :autoloaders
         private :autoloaders
     end
-    Puppet::Util.classproxy self, :autoloaders, "[]", "[]=", :clear
 
-    attr_reader :loaded
-    private :loaded
+    # Send [], []=, and :clear to the @autloaders hash
+    Puppet::Util.classproxy self, :autoloaders, "[]", "[]="
+
+    # Clear the list of autoloaders and loaded files.
+    def self.clear
+        @autoloaders.clear
+        @loaded.clear
+    end
+
+    # List all loaded files.
+    def self.list_loaded
+        @loaded.sort { |a,b| a[0] <=> b[0] }.collect do |path, hash|
+            "%s: %s" % [path, hash[:file]]
+        end
+    end
+
+    # Has a given path been loaded?  This is used for testing whether a
+    # changed file should be loaded or just ignored.
+    def self.loaded?(path)
+        path = path.to_s.sub(/\.rb$/, '')
+        @loaded[path]
+    end
+
+    # Save the fact that a given path has been loaded
+    def self.loaded(path, file, loader)
+        @loaded[path] = {:file => file, :autoloader => loader}
+    end
 
-    Puppet::Util.proxy self, :loaded, :clear
+    attr_accessor :object, :path, :objwarn, :wrap
 
     def initialize(obj, path, options = {})
         @path = path.to_s
@@ -39,11 +61,10 @@ class Puppet::Util::Autoload
         unless defined? @wrap
             @wrap = true
         end
-
-        @loaded = {}
     end
 
-    # Load a single plugin by name.
+    # Load a single plugin by name.  We use 'load' here so we can reload a
+    # given plugin.
     def load(name)
         path = name.to_s + ".rb"
 
@@ -53,7 +74,7 @@ class Puppet::Util::Autoload
             begin
                 Kernel.load file, @wrap
                 name = symbolize(name)
-                @loaded[name] = true
+                loaded name, file
                 return true
             rescue LoadError => detail
                 # I have no idea what's going on here, but different versions
@@ -70,11 +91,19 @@ class Puppet::Util::Autoload
         return false
     end
 
+    # Mark the named object as loaded.  Note that this supports unqualified
+    # queries, while we store the result as a qualified query in the class.
+    def loaded(name, file)
+        self.class.loaded(File.join(@path, name.to_s), file, object)
+    end
+
     # Indicate whether the specfied plugin has been loaded.
     def loaded?(name)
-        @loaded[symbolize(name)]
+        self.class.loaded?(File.join(@path, name.to_s))
     end
 
+    # Load all instances that we can.  This uses require, rather than load,
+    # so that already-loaded files don't get reloaded unnecessarily.
     def loadall
         # Load every instance of everything we can find.
         eachdir do |dir|
@@ -84,12 +113,12 @@ class Puppet::Util::Autoload
                 # believe, but it works as long as real classes
                 # aren't used.
                 name = File.basename(file).sub(".rb", '').intern
-                next if @loaded.include? name
+                next if loaded?(name)
                 next if $".include?(File.join(@path, name.to_s + ".rb"))
                 filepath = File.join(@path, name.to_s + ".rb")
                 begin
-                    Kernel.require filepath
-                    @loaded[name] = true
+                    Kernel.require file
+                    loaded(name, file)
                 rescue => detail
                     if Puppet[:trace]
                         puts detail.backtrace