Initial Chef repository

cookbooks/omnibus_updater/libraries/omnibus_checker.rb

@@ -0,0 +1,30 @@
+#
+# Cookbook Name:: omnibus_updater
+# Library:: omnibuschecker
+#
+# Copyright 2014, Heavy Water Ops, LLC
+#
+# 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.
+#
+
+module OmnibusChecker
+  def is_omnibus?
+    Gem.bindir =~ %r{/opt/(opscode|chef)/}
+  end
+end
+
+OmnibusChecker.send(:extend, OmnibusChecker)
+
+unless(Chef::Recipe.instance_methods.include?(:is_omnibus?))
+  Chef::Recipe.send(:include, OmnibusChecker)
+end

cookbooks/omnibus_updater/libraries/omnitrucker.rb

@@ -0,0 +1,92 @@
+#
+# Cookbook Name:: omnibus_updater
+# Library:: omnitrucker
+#
+# Copyright 2014, Heavy Water Ops, LLC
+#
+# 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.
+#
+
+module OmnibusTrucker
+  class << self
+    URL_MAP = {
+      :p => :platform, :pv => :platform_version, :m => :machine,
+      :v => :version, :prerelease => :prerelease,
+      :nightlies => :nightlies
+    }
+
+    def build_url(*opts)
+      args = node = nil
+      opts.each do |o|
+        if(o.kind_of?(Hash))
+          args = o
+        elsif(o.kind_of?(Chef::Node))
+          node = o
+        else
+          raise ArgumentError.new "Provided argument is not allowed: #{o.class}"
+        end
+      end
+      args ||= {}
+      if(node)
+        args = collect_attributes(node).merge(args)
+      end
+      url = args[:url] || "http://www.opscode.com/chef/download#{'-server' if args[:server]}"
+      u_args = URL_MAP.map do |u_k, a_k|
+        "#{u_k}=#{args[a_k]}" unless args[a_k].nil?
+      end.compact
+      "#{url}?#{u_args.join('&')}"
+    end
+
+    def collect_attributes(node, args={})
+      set = Mash[
+        [:platform_family, :platform, :platform_version].map do |k|
+          [k, args[k] || node[k]]
+        end
+      ]
+      unless(@attrs)
+        if(set[:platform] == 'amazon')
+          @attrs = {:platform => 'el', :platform_version => 6}
+        elsif(set[:platform_family] == 'fedora')
+          @attrs = {:platform => 'el', :platform_version => 6}
+        elsif(set[:platform_family] == 'rhel')
+          @attrs = {:platform => 'el', :platform_version => set[:platform_version].to_i}
+        elsif(set[:platform] == 'debian')
+          @attrs = {:platform => set[:platform], :platform_version => set[:platform_version].to_i}
+        elsif(set[:platform_family] == 'mac_os_x')
+          @attrs = {:platform => set[:platform_family], :platform_version => [set[:platform_version].to_f, 10.7].min}
+        else
+          @attrs = {:platform => set[:platform], :platform_version => set[:platform_version]}
+        end
+        @attrs[:machine] = args[:machine] || node[:kernel][:machine]
+        @attrs[:machine] = "i386" if(set[:platform_family] == 'solaris2' && @attrs[:machine] == "i86pc")
+      end
+      @attrs
+    end
+
+    def url(url_or_node, node = nil)
+      if(url_or_node.is_a?(Chef::Node))
+        url = build_url(url_or_node)
+        node = url_or_node
+      else
+        url = url_or_node
+        raise "Node instance is required for Omnitruck.url!" unless node
+      end
+      request = Chef::REST::RESTRequest.new(:head, URI.parse(url), nil)
+      result = request.call
+      if(result.kind_of?(Net::HTTPRedirection))
+        result['location']
+      end
+    end
+
+  end
+end