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
|
#
# Author:: Adam Jacob (<adam@opscode.com>)
# Copyright:: Copyright (c) 2008 Opscode, Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
require 'chef/provider/file'
require 'chef/mixin/template'
require 'chef/mixin/checksum'
require 'chef/rest'
require 'chef/file_cache'
require 'uri'
require 'tempfile'
class Chef
class Provider
class Template < Chef::Provider::File
include Chef::Mixin::Checksum
include Chef::Mixin::Template
def action_create
raw_template_file = nil
cache_file_name = "cookbooks/#{@new_resource.cookbook_name}/templates/default/#{@new_resource.source}"
if Chef::Config[:solo]
filename = ::File.join(Chef::Config[:cookbook_path], "#{@new_resource.cookbook_name}/templates/default/#{@new_resource.source}")
Chef::Log.debug("using local file for template:#{filename}")
raw_template_file = ::File.open(filename)
else
r = Chef::REST.new(Chef::Config[:template_url])
current_checksum = nil
if Chef::FileCache.has_key?(cache_file_name)
current_checksum = self.checksum(Chef::FileCache.load(cache_file_name, false))
else
Chef::Log.debug("Template #{@new_resource} is not in the template cache")
end
template_url = generate_url(
@new_resource.source,
"templates",
{
:checksum => current_checksum
}
)
template_updated = true
begin
raw_template_file = r.get_rest(template_url, true)
rescue Net::HTTPRetriableError => e
if e.response.kind_of?(Net::HTTPNotModified)
template_updated = false
Chef::Log.debug("Cached template for #{@new_resource} is unchanged")
else
raise e
end
end
end
if template_updated
Chef::Log.debug("Updating template for #{@new_resource} in the cache")
Chef::FileCache.move_to(raw_template_file.path, cache_file_name)
end
context = @new_resource.variables
context[:node] = @node
template_file = render_template(Chef::FileCache.load(cache_file_name), context)
update = false
if ::File.exists?(@new_resource.path)
@new_resource.checksum(self.checksum(template_file.path))
if @new_resource.checksum != @current_resource.checksum
Chef::Log.debug("#{@new_resource} changed from #{@current_resource.checksum} to #{@new_resource.checksum}")
Chef::Log.info("Updating #{@new_resource} at #{@new_resource.path}")
update = true
end
else
Chef::Log.info("Creating #{@new_resource} at #{@new_resource.path}")
update = true
end
if update
backup
FileUtils.cp(template_file.path, @new_resource.path)
@new_resource.updated = true
else
Chef::Log.debug("#{@new_resource} is unchanged")
end
set_owner if @new_resource.owner != nil
set_group if @new_resource.group != nil
set_mode if @new_resource.mode != nil
end
def action_create_if_missing
if ::File.exists?(@new_resource.path)
Chef::Log.debug("Template #{@new_resource} exists, taking no action.")
else
action_create
end
end
end
end
end
|
