blob: c16e37725585f0a1f7a31866593557f137c8a302 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# The client class for filebuckets.
class Puppet::Network::Client::Dipper < Puppet::Network::Client
@handler = Puppet::Network::Handler.handler(:filebucket)
@drivername = :Bucket
attr_accessor :name
# Create our bucket client
def initialize(hash = {})
if hash.include?(:Path)
bucket = self.class.handler.new(:Path => hash[:Path])
hash.delete(:Path)
hash[:Bucket] = bucket
end
super(hash)
end
# Back up a file to our bucket
def backup(file)
unless FileTest.exists?(file)
raise(ArgumentError, "File %s does not exist" % file)
end
contents = ::File.read(file)
unless local?
contents = Base64.encode64(contents)
end
return @driver.addfile(contents,file)
end
# Retrieve a file by sum.
def getfile(sum)
if newcontents = @driver.getfile(sum)
unless local?
newcontents = Base64.decode64(newcontents)
end
return newcontents
end
return nil
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
# $Id$
|
