Line 0
Link Here
|
|
|
1 |
--- /dev/null 2011-11-21 16:48:43.000000000 +0900 |
2 |
+++ lib/chef/provider/package/pkgupgrade.rb 2011-11-21 16:04:09.951953315 +0900 |
3 |
@@ -0,0 +1,399 @@ |
4 |
+# |
5 |
+# Author:: Mitsuru Y (<mitsuruy@reallyenglish.com>) |
6 |
+# Copyright:: Copyright (c) 2011 Real English Broadband |
7 |
+# License:: Apache License, Version 2.0 |
8 |
+# |
9 |
+# Licensed under the Apache License, Version 2.0 (the "License"); |
10 |
+# you may not use this file except in compliance with the License. |
11 |
+# You may obtain a copy of the License at |
12 |
+# |
13 |
+# http://www.apache.org/licenses/LICENSE-2.0 |
14 |
+# |
15 |
+# Unless required by applicable law or agreed to in writing, software |
16 |
+# distributed under the License is distributed on an "AS IS" BASIS, |
17 |
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
18 |
+# See the License for the specific language governing permissions and |
19 |
+# limitations under the License. |
20 |
+# |
21 |
+ |
22 |
+require 'chef/provider/package' |
23 |
+require 'chef/mixin/command' |
24 |
+require 'chef/resource/package' |
25 |
+require 'singleton' |
26 |
+require 'chef/mixin/get_source_from_package' |
27 |
+ |
28 |
+class Hash |
29 |
+ def safe_invert |
30 |
+ result = Hash.new{|h,key| h[key] = [] } |
31 |
+ self.each{|key, value| |
32 |
+ result[value] << key |
33 |
+ } |
34 |
+ result |
35 |
+ end |
36 |
+end |
37 |
+ |
38 |
+class Chef |
39 |
+ class Provider |
40 |
+ class Package |
41 |
+ class PkgUpgrade < Chef::Provider::Package |
42 |
+ |
43 |
+ # Cache for our installed and available packages |
44 |
+ |
45 |
+ class PkgCache |
46 |
+ include Chef::Mixin::Command |
47 |
+ include Singleton |
48 |
+ |
49 |
+ def initialize |
50 |
+ @pkg_index = Hash.new |
51 |
+ @pkg_index[:name] = Hash.new |
52 |
+ @pkg_index[:version] = Hash.new |
53 |
+ @pkg_index[:origin] = Hash.new |
54 |
+ @pkg_pkgdb = Hash.new |
55 |
+ @pkg_pkgdb[:name] = Hash.new |
56 |
+ @pkg_pkgdb[:version] = Hash.new |
57 |
+ |
58 |
+# Next time @pkgdb is accessed: |
59 |
+# :all - Trigger updates pkg's cache - this data is slow to gather. |
60 |
+# :installed - Trigger updates only the local pkg db. |
61 |
+# Used between client runs for a quick refresh. |
62 |
+# :none - Do nothing, a call to one of the reload methods is required. |
63 |
+ @next_refresh = :all |
64 |
+ |
65 |
+ |
66 |
+ # these are for subsequent runs if we are on an interval |
67 |
+ Chef::Client.when_run_starts do |
68 |
+ PkgCache.instance.reload |
69 |
+ end |
70 |
+ end |
71 |
+ |
72 |
+ # Cache management |
73 |
+ # |
74 |
+ |
75 |
+ def refresh |
76 |
+ case @next_refresh |
77 |
+ when :none |
78 |
+ return nil |
79 |
+ when :installed |
80 |
+ reset_installed |
81 |
+ # fast |
82 |
+ refresh_pkgdb |
83 |
+ when :all |
84 |
+ reset |
85 |
+ # medium |
86 |
+ refresh_pkgdb |
87 |
+ refresh_index |
88 |
+ else |
89 |
+ raise ArgumentError, "Unexpected value in next_refresh: #{@next_refresh}" |
90 |
+ end |
91 |
+ |
92 |
+ # A reload method must be called before the cache is altered |
93 |
+ @next_refresh = :none |
94 |
+ end |
95 |
+ |
96 |
+ def refresh_index |
97 |
+ Chef::Log.debug("#{@new_resource} refresh index") |
98 |
+ run_command_with_systems_locale( |
99 |
+ :command => "uma fetch ftpindex" |
100 |
+ ) |
101 |
+ indexfile = "" |
102 |
+ status = popen4("uma env") do |pid, stdin, stdout, stderr| |
103 |
+ stdout.each do |env_var| |
104 |
+ env_var.chomp! |
105 |
+ if env_var =~ /^PKG_INDEX='(.*)'/ |
106 |
+ indexfile = $1 |
107 |
+ end |
108 |
+ end |
109 |
+ end |
110 |
+ IO.foreach(indexfile) do |line| |
111 |
+ indexline = line.split("|") |
112 |
+ dir = indexline[0] |
113 |
+ origin = indexline[1].sub(/^\/usr\/ports\//,'') |
114 |
+ pkg_name = pkgversion(dir) |
115 |
+ version = pkg_name[1] |
116 |
+ if version |
117 |
+ @pkg_index[:name][origin] = pkg_name[0] |
118 |
+ @pkg_index[:version][origin] = pkg_name[1] |
119 |
+ end |
120 |
+ end |
121 |
+ @pkg_index[:origin] = @pkg_index[:name].safe_invert |
122 |
+ end |
123 |
+ |
124 |
+ def refresh_pkgdb |
125 |
+ Chef::Log.debug("#{@new_resource} refresh pkgdb") |
126 |
+ status = popen4("pkg_info -Ea") do |pid, stdin, stdout, stderr| |
127 |
+ stdout.each do |pkgname| |
128 |
+ pkgname.chomp! |
129 |
+ pkg_name = pkgversion(pkgname) |
130 |
+ version = pkg_name[1] |
131 |
+ status = popen4("pkg_info -qo #{pkgname}") do |pid, stdin, stdout, stderr| |
132 |
+ stdout.each do |line| |
133 |
+ origin = line.chomp |
134 |
+ @pkg_pkgdb[:name][origin] = pkg_name[0] |
135 |
+ @pkg_pkgdb[:version][origin] = pkg_name[1] |
136 |
+ end |
137 |
+ end |
138 |
+ end |
139 |
+ end |
140 |
+ end |
141 |
+ |
142 |
+ def pkgversion(pkgname) |
143 |
+ if /\s/ =~ pkgname |
144 |
+ return nil |
145 |
+ # raise ArgumentError, "Must not contain whitespace." |
146 |
+ end |
147 |
+ |
148 |
+ if /^(.+)-([^-]+)$/ !~ pkgname |
149 |
+ return nil |
150 |
+ # raise ArgumentError, "Not in due form: <name>-<version>" |
151 |
+ end |
152 |
+ name = $1 |
153 |
+ version = splitversion($2) |
154 |
+ return [name, version] |
155 |
+ end |
156 |
+ |
157 |
+ def splitversion(pkgversion) |
158 |
+ if /[\s-]/ =~ pkgversion |
159 |
+ return nil |
160 |
+ # raise ArgumentError, "#{pkgversion}: Must not contain a '-' or whitespace." |
161 |
+ end |
162 |
+ |
163 |
+ if /^([^_,]+)(?:_(\d+))?(?:,(\d+))?$/ !~ pkgversion |
164 |
+ return nil |
165 |
+ # raise ArgumentError, "#{pkgversion}: Not in due form: '<version>[_<revision>][,<epoch>]'." |
166 |
+ end |
167 |
+ |
168 |
+ version = $1 |
169 |
+ revision = $2 ? $2.to_i : 0 |
170 |
+ epoch = $3 ? $3.to_i : 0 |
171 |
+ |
172 |
+ return pkgversion |
173 |
+# return [version, revision, epoch] |
174 |
+ end |
175 |
+ |
176 |
+ def reload |
177 |
+ @next_refresh = :all |
178 |
+ end |
179 |
+ |
180 |
+ def reload_installed |
181 |
+ @next_refresh = :installed |
182 |
+ end |
183 |
+ |
184 |
+ def reset |
185 |
+ @pkg_index[:name].clear |
186 |
+ @pkg_index[:version].clear |
187 |
+ @pkg_index[:origin].clear |
188 |
+ @pkg_pkgdb[:name].clear |
189 |
+ @pkg_pkgdb[:version].clear |
190 |
+ end |
191 |
+ |
192 |
+ def reset_installed |
193 |
+ @pkg_pkgdb[:name].clear |
194 |
+ @pkg_pkgdb[:version].clear |
195 |
+ end |
196 |
+ |
197 |
+ # Querying the cache |
198 |
+ # |
199 |
+ |
200 |
+ def available_version(package_name) |
201 |
+ refresh |
202 |
+ origin = nil |
203 |
+ if package_name =~ /\// |
204 |
+ origin = package_name |
205 |
+ else |
206 |
+ origin = @pkg_pkgdb[:name].index(package_name) |
207 |
+ unless origin # not installed |
208 |
+ origins = @pkg_index[:origin][package_name] |
209 |
+ case origins.size |
210 |
+ when 1 |
211 |
+ origin = origins.first |
212 |
+ when 0 |
213 |
+ raise Chef::Exceptions::Package, "Package #{@new_resource} not found." |
214 |
+ else |
215 |
+ origins.each do |o| |
216 |
+ if o =~ /^[^\/]*\/#{package_name}$/ |
217 |
+ Chef::Log.warn("#{@new_resource} more than one match and use guessed origin. consider specifying origin. ") |
218 |
+ origin = o |
219 |
+ break |
220 |
+ end |
221 |
+ end |
222 |
+ unless origin |
223 |
+ raise Chef::Exceptions::Package, "Package #{@new_resource} matches multiple packages which origins are #{origins.join(' ')}." |
224 |
+ end |
225 |
+ end |
226 |
+ end |
227 |
+ end |
228 |
+ if origin |
229 |
+ @pkg_index[:version][origin] |
230 |
+ else |
231 |
+ nil |
232 |
+ end |
233 |
+ end |
234 |
+ alias :candidate_version :available_version |
235 |
+ |
236 |
+ def installed_version(package_name) |
237 |
+ refresh |
238 |
+ if package_name =~ /\// |
239 |
+ origin = package_name |
240 |
+ else |
241 |
+ origin = @pkg_pkgdb[:name].index(package_name) |
242 |
+ end |
243 |
+ if origin |
244 |
+ @pkg_pkgdb[:version][origin] |
245 |
+ else |
246 |
+ nil |
247 |
+ end |
248 |
+ end |
249 |
+ |
250 |
+ end # PkgCache |
251 |
+ |
252 |
+ include Chef::Mixin::GetSourceFromPackage |
253 |
+ |
254 |
+ def initialize(new_resource, run_context) |
255 |
+ super |
256 |
+ |
257 |
+ @pkg = PkgCache.instance |
258 |
+ end |
259 |
+ |
260 |
+ # Extra attributes |
261 |
+ # |
262 |
+ |
263 |
+ def flush_cache |
264 |
+ if @new_resource.respond_to?("flush_cache") |
265 |
+ @new_resource.flush_cache |
266 |
+ else |
267 |
+ { :before => false, :after => false } |
268 |
+ end |
269 |
+ end |
270 |
+ |
271 |
+ def agree_license |
272 |
+ if @new_resource.respond_to?("agree_license") |
273 |
+ if @new_resource.agree_license |
274 |
+ "yes | " |
275 |
+ else |
276 |
+ "" |
277 |
+ end |
278 |
+ else |
279 |
+ "" |
280 |
+ end |
281 |
+ end |
282 |
+ |
283 |
+ # Standard Provider methods for Parent |
284 |
+ # |
285 |
+ |
286 |
+ def load_current_resource |
287 |
+ if flush_cache[:before] |
288 |
+ @pkg.reload |
289 |
+ end |
290 |
+ |
291 |
+ @current_resource = Chef::Resource::Package.new(@new_resource.name) |
292 |
+ @current_resource.package_name(@new_resource.package_name) |
293 |
+ |
294 |
+ if @new_resource.source |
295 |
+ case @new_resource.source |
296 |
+ when /^\// |
297 |
+ unless ::File.exists?(@new_resource.source) |
298 |
+ raise Chef::Exceptions::Package, "Package #{@new_resource.name} not found: #{@new_resource.source}" |
299 |
+ end |
300 |
+ end |
301 |
+ |
302 |
+# Chef::Log.debug("#{@new_resource} checking rpm status") |
303 |
+# status = popen4("rpm -qp --queryformat '%{NAME} %{VERSION}-%{RELEASE}\n' #{@new_resource.source}") do |pid, stdin, stdout, stderr| |
304 |
+# stdout.each do |line| |
305 |
+# case line |
306 |
+# when /([\w\d_.-]+)\s([\w\d_.-]+)/ |
307 |
+# @current_resource.package_name($1) |
308 |
+# @new_resource.version($2) |
309 |
+# end |
310 |
+# end |
311 |
+# end |
312 |
+ end |
313 |
+ |
314 |
+ if @new_resource.version |
315 |
+ new_resource = "#{@new_resource.package_name}-#{@new_resource.version}" |
316 |
+ else |
317 |
+ new_resource = "#{@new_resource.package_name}" |
318 |
+ end |
319 |
+ |
320 |
+ Chef::Log.debug("#{@new_resource} checking package info for #{new_resource}") |
321 |
+ |
322 |
+ installed_version = @pkg.installed_version(@new_resource.package_name) |
323 |
+ @current_resource.version(installed_version) |
324 |
+ |
325 |
+ @candidate_version = @pkg.candidate_version(@new_resource.package_name) |
326 |
+ |
327 |
+ Chef::Log.debug("#{@new_resource} installed version: #{installed_version || "(none)"} candidate version: " + |
328 |
+ "#{@candidate_version || "(none)"}") |
329 |
+ |
330 |
+ @current_resource |
331 |
+ end |
332 |
+ |
333 |
+ def install_package(name, version) |
334 |
+ if @new_resource.source && name =~ /^\// |
335 |
+ run_command_with_systems_locale( |
336 |
+ :command => "#{agree_license} pkg_add #{expand_options(@new_resource.options)} #{@new_resource.source}" |
337 |
+ ) |
338 |
+ else |
339 |
+ if name =~ /\// |
340 |
+ run_command_with_systems_locale( |
341 |
+ :command => "#{agree_license} pkg_upgrade --clean #{expand_options(@new_resource.options)} #{name}" |
342 |
+ ) |
343 |
+ else |
344 |
+ run_command_with_systems_locale( |
345 |
+ :command => "#{agree_license} pkg_upgrade --clean #{expand_options(@new_resource.options)} #{name}-#{version}" |
346 |
+ ) |
347 |
+ end |
348 |
+ end |
349 |
+ if flush_cache[:after] |
350 |
+ @pkg.reload |
351 |
+ else |
352 |
+ @pkg.reload_installed |
353 |
+ end |
354 |
+ end |
355 |
+ |
356 |
+ # Keep upgrades from trying to install an older candidate version. Can happen when a new |
357 |
+ # version is installed then removed from a repository, now the older available version |
358 |
+ # shows up as a viable install candidate. |
359 |
+ # |
360 |
+ # Can be done in upgrade_package but an upgraded from->to log message slips out |
361 |
+ # |
362 |
+ # Hacky - better overall solution? Custom compare in Package provider? |
363 |
+# def action_upgrade |
364 |
+ # Ensure the candidate is newer |
365 |
+# if RPMVersion.parse(candidate_version) > RPMVersion.parse(@current_resource.version) |
366 |
+# super |
367 |
+ # Candidate is older |
368 |
+# else |
369 |
+# Chef::Log.debug("#{@new_resource} is at the latest version - nothing to do") |
370 |
+# end |
371 |
+# end |
372 |
+ |
373 |
+ def upgrade_package(name, version) |
374 |
+ install_package(name, version) |
375 |
+ end |
376 |
+ |
377 |
+ def remove_package(name, version) |
378 |
+ if version |
379 |
+ run_command_with_systems_locale( |
380 |
+ :command => "pkg_delete #{expand_options(@new_resource.options)} #{name}-#{version}" |
381 |
+ ) |
382 |
+ else |
383 |
+ run_command_with_systems_locale( |
384 |
+ :command => "pkg_delete #{expand_options(@new_resource.options)} #{name}-#{@current_resource.version}" |
385 |
+ ) |
386 |
+ end |
387 |
+ if flush_cache[:after] |
388 |
+ @pkg.reload |
389 |
+ else |
390 |
+ @pkg.reload_installed |
391 |
+ end |
392 |
+ end |
393 |
+ |
394 |
+ def purge_package(name, version) |
395 |
+ remove_package(name, version) |
396 |
+ end |
397 |
+ |
398 |
+ |
399 |
+ end |
400 |
+ end |
401 |
+ end |
402 |
+end |