helpers.rb

# Cookbook:: vagrant
# Library:: helpers

# Author:: Joshua Timberman <opensource@housepub.org>
# Copyright:: Copyright (c) 2014, Joshua Timberman
# 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 'uri'
require 'open-uri'

module Vagrant
  module Cookbook
    module Helpers
      def vagrant_package_uri
        "#{vagrant_base_uri}#{package_version}/#{package_name}"
      end

      def vagrant_sha256sum
        sha256sums = fetch_platform_checksums_for_version
        extract_checksum(sha256sums)
      end

      private

      def vagrant_base_uri
        'https://releases.hashicorp.com/vagrant/'
      end

      def package_name
        if @appimage
          "vagrant_#{package_version}#{package_extension}"
        elsif Gem::Version.new(package_version) >= Gem::Version.new('2.3.0')
          separator = platform_family?(%w(rhel suse fedora amazon)) ? '-' : '_'
          "vagrant#{separator}#{package_version}#{package_extension}"
        else
          "vagrant_#{package_version}#{package_extension}"
        end
      end

      def package_version
        @vagrant_version
      end

      def package_extension
        extension =
          case node['platform_family']
          when 'mac_os_x'
            mac_os_x_extension
          when 'windows'
            windows_extension
          when 'debian'
            deb_extension
          when 'rhel', 'suse', 'fedora', 'amazon'
            rpm_extension
          end
        extension = '_linux_amd64.zip' if @appimage
        raise "HashiCorp doesn't provide a Vagrant package for the #{node['platform']} platform." if extension.nil?

        extension
      end

      def fetch_platform_checksums_for_version
        checksums_url = "#{vagrant_base_uri}#{package_version}/vagrant_#{package_version}_SHA256SUMS?direct"
        URI.open(checksums_url).readlines
      end

      def extract_checksum(sha256sums)
        raise "SHA 256 sum not found for the Vagrant package #{package_name}" unless sha256sums.grep(/#{package_name}/).first.respond_to?(:split)

        sha256sums.grep(/#{package_name}/).first.split.first
      end

      def deb_extension
        Gem::Version.new(package_version) < Gem::Version.new('2.3.0') ? '_x86_64.deb' : '-1_amd64.deb'
      end

      def rpm_extension
        Gem::Version.new(package_version) < Gem::Version.new('2.3.0') ? '_x86_64.rpm' : '-1.x86_64.rpm'
      end

      def mac_os_x_extension
        if Gem::Version.new(package_version) < Gem::Version.new('2.3.0')
          last_using_dmg = Gem::Version.new('1.9.2')
          Gem::Version.new(package_version) > last_using_dmg ? '_x86_64.dmg' : '.dmg'
        else
          '_darwin_amd64.dmg'
        end
      end

      def windows_extension
        last_using_msi = Gem::Version.new('1.9.5')
        Gem::Version.new(package_version) > last_using_msi ? "#{windows_machine}.msi" : '.msi'
      end

      def windows_machine
        if Gem::Version.new(package_version) < Gem::Version.new('2.3.0')
          node['kernel']['machine'] == 'x86_64' ? '_x86_64' : '_i686'
        else
          node['kernel']['machine'] == 'x86_64' ? '_windows_amd64' : '_windows_i686'
        end
      end
    end
  end
end
Chef::DSL::Recipe.include ::Vagrant::Cookbook::Helpers
Chef::Resource.include ::Vagrant::Cookbook::Helpers