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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
require 'digest'
require 'erb'
require 'fileutils'
require_relative 'resource'
module RCM
# Backup the file on change
module FileBackup
def backup!(file_path, checksum)
backup_dir = "#{::File.dirname(file_path)}/.rcm"
Dir.mkdir(backup_dir) unless ::File.directory?(backup_dir)
backup_path = "#{backup_dir}/#{::File.basename(file_path)}.#{checksum}"
return if ::File.exist?(backup_path)
info("Backing up #{file_path} -> #{backup_path}")
::File.rename(file_path, backup_path)
end
end
# Managing files
class File < Resource
include FileBackup
class UnsupportedOperation < StandardError; end
def initialize(file_path)
super(file_path)
@file_path = file_path
@is = :installed
end
def content(text = nil)
return @content if text.nil?
@content = text.instance_of?(Array) ? text.join("\n") : text
end
def create_parent_directory = @create_parent = true
def from_sourcefile = @from_sourcefile = true
def from_template = @from_template = true
def line(line) = @ensure_line = line
def path(file_path = nil)
return @file_path if file_path.nil?
@file_path = file_path
end
def is(what)
@is = what.to_sym
raise UnsupportedOperation, "Unsupported operation #{@is}" unless %i[present absent].include?(@is)
end
def evaluate!
return unless super
return evaluate_ensure_line! unless @ensure_line.nil?
write!(real_content)
end
private
def evaluate_ensure_line!
return evaluate_ensure_line_absent! if @is == :absent
return write!(@ensure_line) unless ::File.file?(@file_path)
return if ::File.readlines(@file_path, chomp: true).include?(@ensure_line)
::File.open(@file_path, 'a') do |fd|
fd.puts(@ensure_line)
end
end
def evaluate_ensure_line_absent!
return unless ::File.file?(@file_path)
write!(::File.readlines(@file_path, chomp: true).reject do |line|
line == @ensure_line
end.join("\n"))
end
def write!(text)
info "Managing file #{@file_path}"
create_parent_directory!
debug text if option :debug
tmp_path = "#{@file_path}.rcmtmp"
::File.write(tmp_path, text)
if ::File.file?(@file_path)
checksum = Digest::SHA256.file(@file_path).hexdigest
tmp_checksum = Digest::SHA256.file(tmp_path).hexdigest
if tmp_checksum == checksum
::File.delete(tmp_path) # File has not changed, not doing anything
return
end
backup!(@file_path, checksum) # File changed, backup!
end
::File.rename(tmp_path, @file_path)
end
def create_parent_directory!
dirname = ::File.dirname(@file_path)
return unless !::File.directory?(dirname) && @create_parent
info "Creating parent directory #{dirname}"
FileUtils.mkdir_p(dirname)
end
def real_content
text = @from_sourcefile ? ::File.read(@content) : @content
@from_template ? ERB.new(text).result : text
end
end
# Add file keyword to the DSL
class DSL
def file(file_path, &block)
return unless @conds_met
f = File.new(file_path)
f.content(f.instance_eval(&block))
self << f
f
end
end
end
|
