Renaming the Puppet::PropertyChange class to Puppet::Transaction::Change.

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

1 files changed, 134 insertions, 0 deletions

diff --git a/lib/puppet/transaction/change.rb b/lib/puppet/transaction/change.rb
new file mode 100644
index 000000000..ee92f2249
--- /dev/null
+++ b/lib/puppet/transaction/change.rb
@@ -0,0 +1,134 @@
+require 'puppet/transaction'
+
+# Handle all of the work around performing an actual change,
+# including calling 'sync' on the properties and producing events.
+class Puppet::Transaction::Change
+    attr_accessor :is, :should, :type, :path, :property, :transaction, :changed, :proxy
+    
+    # The log file generated when this object was changed.
+    attr_reader :report
+    
+    # Switch the goals of the property, thus running the change in reverse.
+    def backward
+        @property.should = @is
+        @is = @property.retrieve
+
+        unless defined? @transaction
+            raise Puppet::Error,
+                "PropertyChange '%s' tried to be executed outside of transaction" %
+                self
+        end
+        unless @property.insync?(@is)
+            @property.info "Backing %s" % self
+            return self.go
+        else
+            @property.debug "rollback is already in sync: %s vs. %s" %
+                [@is, @property.should.inspect]
+            return nil
+        end
+    end
+    
+    def changed?
+        self.changed
+    end
+
+    # Create our event object.
+    def event(name)
+        # default to a simple event type
+        unless name.is_a?(Symbol)
+            @property.warning("Property '%s' returned invalid event '%s'; resetting to default" %
+                [@property.class, name])
+
+            event = @property.resource.class.name.id2name + "_changed"
+        end
+        
+        Puppet::Event.new(
+            :event => name,
+            :transaction => @transaction,
+            :source => self.source
+        )
+    end
+
+    def initialize(property, currentvalue)
+        unless property.is_a?(Puppet::Property)
+            raise Puppet::DevError, "Got a %s instead of a property" %
+                property.class
+        end
+        @property = property
+        @path = [property.path,"change"].flatten
+        @is = currentvalue
+
+        @should = property.should
+
+        @changed = false
+    end
+
+    # Perform the actual change.  This method can go either forward or
+    # backward, and produces an event.
+    def go
+        if skip?
+            if self.noop
+                return [event(:noop)]
+            else
+                return nil
+            end
+        end
+
+        # The transaction catches any exceptions here.
+        events = @property.sync
+        if events.nil?
+            return nil
+        end
+
+        if events.is_a?(Array)
+            if events.empty?
+                return nil
+            end
+        else
+            events = [events]
+        end
+        
+        return events.collect { |name|
+            @report = @property.log(@property.change_to_s(@is, @should))
+            event(name)
+        }
+    end
+
+    def forward
+        unless defined? @transaction
+            raise Puppet::Error,
+                "PropertyChange '%s' tried to be executed outside of transaction" %
+                self
+        end
+
+        return self.go
+    end
+    
+    def noop
+        return @property.noop
+    end
+    
+    def skip?
+        if @property.insync?(@is)
+            @property.info "Already in sync"
+            return true
+        end
+
+        if @property.noop
+            @property.log "is %s, should be %s (noop)" %
+                [property.is_to_s(@is), property.should_to_s(@should)]
+            #@property.debug "%s is noop" % @property
+            return true
+        end
+        return false
+    end
+    
+    def source
+        self.proxy || @property.resource
+    end
+
+    def to_s
+        return "change %s.%s(%s)" %
+            [@transaction.object_id, self.object_id, @property.change_to_s(@is, @should)]
+    end
+end