Refactoring how the Facter integration works

I moved all of the extra Fact modifications into the Facts
class, and then moved the calls of those new methods
into the Facter terminus.

Signed-off-by: Luke Kanies <luke@madstop.com>

3 files changed, 49 insertions, 52 deletions

diff --git a/lib/puppet/agent/fact_handler.rb b/lib/puppet/agent/fact_handler.rb
index 4c9280bfc..266ae1815 100644
--- a/lib/puppet/agent/fact_handler.rb
+++ b/lib/puppet/agent/fact_handler.rb
@@ -1,3 +1,5 @@
+require 'puppet/indirector/facts/facter'
+
 # Break out the code related to facts.  This module is
 # just included into the agent, but having it here makes it
 # easier to test.
@@ -24,31 +26,6 @@ module Puppet::Agent::FactHandler
         Puppet::Agent::Downloader.new("fact", Puppet[:factsource], Puppet[:factdest], Puppet[:factsignore]).evaluate
     end
 
-    def load_fact_plugins
-        # LAK:NOTE See http://snurl.com/21zf8  [groups_google_com] 
-        x = Puppet[:factpath].split(":").each do |dir|
-            load_facts_in_dir(dir)
-        end
-    end
-
-    def load_facts_in_dir(dir)
-        return unless FileTest.directory?(dir)
-
-        Dir.chdir(dir) do
-            Dir.glob("*.rb").each do |file|
-                fqfile = ::File.join(dir, file)
-                begin
-                    Puppet.info "Loading facts in %s" % [::File.basename(file.sub(".rb",''))]
-                    Timeout::timeout(Puppet::Agent.timeout) do
-                        load file
-                    end
-                rescue => detail
-                    Puppet.warning "Could not load fact file %s: %s" % [fqfile, detail]
-                end
-            end
-        end
-    end
-
     # Clear out all of the loaded facts and reload them from disk.
     # NOTE: This is clumsy and shouldn't be required for later (1.5.x) versions
     # of Facter.
@@ -66,6 +43,6 @@ module Puppet::Agent::FactHandler
 
         # This loads all existing facts and any new ones.  We have to remove and
         # reload because there's no way to unload specific facts.
-        load_fact_plugins()
+        Puppet::Node::Facts::Facter.load_fact_plugins()
     end
 end
diff --git a/lib/puppet/indirector/facts/facter.rb b/lib/puppet/indirector/facts/facter.rb
index bc4a0e840..6376b71ca 100644
--- a/lib/puppet/indirector/facts/facter.rb
+++ b/lib/puppet/indirector/facts/facter.rb
@@ -6,27 +6,29 @@ class Puppet::Node::Facts::Facter < Puppet::Indirector::Code
         between Puppet and Facter.  It's only `somewhat` abstract because it always
         returns the local host's facts, regardless of what you attempt to find."
 
-    def self.loaddir(dir, type)
-        return unless FileTest.directory?(dir)
 
-        Dir.entries(dir).find_all { |e| e =~ /\.rb$/ }.each do |file|
-            fqfile = ::File.join(dir, file)
-            begin
-                Puppet.info "Loading %s %s" % 
-                    [type, ::File.basename(file.sub(".rb",''))]
-                Timeout::timeout(self.timeout) do
-                    load fqfile
-                end
-            rescue => detail
-                Puppet.warning "Could not load %s %s: %s" % [type, fqfile, detail]
-            end
+    def self.load_fact_plugins
+        # LAK:NOTE See http://snurl.com/21zf8  [groups_google_com] 
+        x = Puppet[:factpath].split(":").each do |dir|
+            load_facts_in_dir(dir)
         end
     end
 
-    def self.loadfacts
-        # LAK:NOTE See http://snurl.com/21zf8  [groups_google_com] 
-        x = Puppet[:factpath].split(":").each do |dir|
-            loaddir(dir, "fact")
+    def self.load_facts_in_dir(dir)
+        return unless FileTest.directory?(dir)
+
+        Dir.chdir(dir) do
+            Dir.glob("*.rb").each do |file|
+                fqfile = ::File.join(dir, file)
+                begin
+                    Puppet.info "Loading facts in %s" % [::File.basename(file.sub(".rb",''))]
+                    Timeout::timeout(self.timeout) do
+                        load file
+                    end
+                rescue => detail
+                    Puppet.warning "Could not load fact file %s: %s" % [fqfile, detail]
+                end
+            end
         end
     end
 
@@ -49,7 +51,7 @@ class Puppet::Node::Facts::Facter < Puppet::Indirector::Code
 
     def initialize(*args)
         super
-        self.class.loadfacts
+        self.class.load_fact_plugins
     end
 
     def destroy(facts)
@@ -59,18 +61,15 @@ class Puppet::Node::Facts::Facter < Puppet::Indirector::Code
     # Look a host's facts up in Facter.
     def find(request)
         result = Puppet::Node::Facts.new(request.key, Facter.to_hash)
-        add_local_facts(result)
+
+        result.add_local_facts
+        result.stringify
+        result.downcase_if_necessary
+
         result
     end
 
     def save(facts)
         raise Puppet::DevError, "You cannot save facts to the code store; it is only used for getting facts from Facter"
     end
-
-    private
-
-    def add_local_facts(facts)
-        facts.values["clientversion"] = Puppet.version.to_s
-        facts.values["environment"] ||= Puppet.settings[:environment]
-    end
 end
diff --git a/lib/puppet/node/facts.rb b/lib/puppet/node/facts.rb
index 8ee90b4ac..dca435c7d 100755
--- a/lib/puppet/node/facts.rb
+++ b/lib/puppet/node/facts.rb
@@ -21,6 +21,11 @@ class Puppet::Node::Facts
 
     attr_accessor :name, :values
 
+    def add_local_facts
+        values["clientversion"] = Puppet.version.to_s
+        values["environment"] ||= Puppet.settings[:environment]
+    end
+
     def initialize(name, values = {})
         @name = name
         @values = values
@@ -28,6 +33,22 @@ class Puppet::Node::Facts
         add_internal
     end
 
+    def downcase_if_necessary
+        return unless Puppet.settings[:downcasefacts]
+
+        Puppet.warning "DEPRECATION NOTICE: Fact downcasing is deprecated; please disable (20080122)"
+        values.each do |fact, value|
+            values[fact] = value.downcase if value.is_a?(String)
+        end
+    end
+
+    # Convert all fact values into strings.
+    def stringify
+        values.each do |fact, value|
+            values[fact] = value.to_s
+        end
+    end
+
     private
 
     # Add internal data to the facts for storage.