1 files changed, 97 insertions, 0 deletions
diff --git a/lib/puppet/file_bucket/dipper.rb b/lib/puppet/file_bucket/dipper.rb
new file mode 100644
index 000000000..c73d76345
--- /dev/null
+++ b/lib/puppet/file_bucket/dipper.rb
@@ -0,0 +1,97 @@
+require 'puppet/file_bucket'
+require 'puppet/file_bucket/file'
+require 'puppet/indirector/request'
+
+class Puppet::FileBucket::Dipper
+    # This is a transitional implementation that uses REST
+    # to access remote filebucket files.
+
+    attr_accessor :name
+
+    # Create our bucket client
+    def initialize(hash = {})
+        # Emulate the XMLRPC client
+        server      = hash[:Server]
+        port        = hash[:Port] || Puppet[:masterport]
+        environment = Puppet[:environment]
+
+        if hash.include?(:Path)
+            @local_path = hash[:Path]
+            @rest_path  = nil
+        else
+            @local_path = nil
+            @rest_path = "https://#{server}:#{port}/#{environment}/file_bucket_file/"
+        end
+    end
+
+    def local?
+        !! @local_path
+    end
+
+    # Back up a file to our bucket
+    def backup(file)
+        unless ::File.exist?(file)
+            raise(ArgumentError, "File %s does not exist" % file)
+        end
+        contents = ::File.read(file)
+        begin
+            file_bucket_file = Puppet::FileBucket::File.new(contents, :bucket_path => @local_path, :path => file)
+            dest_path = "#{@rest_path}#{file_bucket_file.name}"
+
+            request = Puppet::Indirector::Request.new(:file_bucket_file, :save, dest_path)
+
+            file_bucket_file.save(request)
+            return file_bucket_file.checksum_data
+        rescue => detail
+            puts detail.backtrace if Puppet[:trace]
+            raise Puppet::Error, "Could not back up %s: %s" % [file, detail]
+        end
+    end
+
+    # Retrieve a file by sum.
+    def getfile(sum)
+        source_path = "#{@rest_path}md5/#{sum}"
+        file_bucket_file = Puppet::FileBucket::File.find(source_path)
+
+        return file_bucket_file.to_s
+    end
+
+    # Restore the file
+    def restore(file,sum)
+        restore = true
+        if FileTest.exists?(file)
+            cursum = Digest::MD5.hexdigest(::File.read(file))
+
+            # if the checksum has changed...
+            # this might be extra effort
+            if cursum == sum
+                restore = false
+            end
+        end
+
+        if restore
+            if newcontents = getfile(sum)
+                tmp = ""
+                newsum = Digest::MD5.hexdigest(newcontents)
+                changed = nil
+                if FileTest.exists?(file) and ! FileTest.writable?(file)
+                    changed = ::File.stat(file).mode
+                    ::File.chmod(changed | 0200, file)
+                end
+                ::File.open(file, ::File::WRONLY|::File::TRUNC|::File::CREAT) { |of|
+                    of.print(newcontents)
+                }
+                if changed
+                    ::File.chmod(changed, file)
+                end
+            else
+                Puppet.err "Could not find file with checksum %s" % sum
+                return nil
+            end
+            return newsum
+        else
+            return nil
+        end
+    end
+end
+