bootstrap_template.rb

# Author:: Matt Ray <matt@chef.io>
# Cookbook Name:: pxe_dust
# Recipe:: bootstrap_template
#
# Copyright 2012-2013, Chef Software, Inc
#
# 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 'mixlib/shellout'

include_recipe 'pxe_dust::common'
include_recipe 'pxe_dust::installers'

# write out a new pxedust.erb template
remote_file "#{node['pxe_dust']['dir']}/chef-full.erb" do
  source "https://raw.githubusercontent.com/chef/chef/master/lib/chef/knife/bootstrap/templates/chef-full.erb"
end

# change URL from
# url="http://www.chef.io/chef/download?v=${version}&p=${platform}&pv=${platform_version}&m=${machine}"
# to
# url="http://hypnotoad/${platform}-${platform_version}-${machine}/${filename}"
ruby_block "template url" do
  block do
    sed = "sed 's/https:\\/\\/www.chef.io\\/chef\\//"
    sed += "http:\\/\\/#{node['ipaddress']}\\/#{node['hostname']}-/'"
    sed += " #{node['pxe_dust']['dir']}/chef-full.erb"
    Chef::Log.debug sed
    cmd = Mixlib::ShellOut.new(sed)
    output = cmd.run_command
    Chef::Log.debug output.stdout
    nodetemplate = File.new("#{node['pxe_dust']['dir']}/#{node['hostname']}.erb", "w+")
    nodetemplate.puts output.stdout
    nodetemplate.chmod(0644)
  end
  action :nothing
  subscribes :create, resources("remote_file[#{node['pxe_dust']['dir']}/chef-full.erb]")
end

# write out a new install.sh
remote_file "#{node['pxe_dust']['dir']}/original-install.sh" do
  source "https://chef.io/chef/install.sh"
end

# change URL from
# url="http://www.chef.io/chef/download?v=${version}&p=${platform}&pv=${platform_version}&m=${machine}"
# to
# url="http://hypnotoad/chef-full-stack/${platform}-${platform_version}-${machine}/${filename}"
ruby_block "install.sh url" do
  block do
    sed = "sed 's/https:\\/\\/chef.io\\/chef\\/download?v=${version}&prerelease=${prerelease}&p=${platform}&pv=${platform_version}&m=${machine}/"
    sed += "http:\\/\\/#{node['ipaddress']}\\/chef-full-stack\\/${platform}-${platform_version}-${machine}\\/${filename}/'"
    sed += " #{node['pxe_dust']['dir']}/original-install.sh"
    Chef::Log.debug sed
    cmd = Mixlib::ShellOut.new(sed)
    output = cmd.run_command
    Chef::Log.debug output.stdout
    nodeinstall = File.new("#{node['pxe_dust']['dir']}/#{node['hostname']}-install.sh", "w+")
    nodeinstall.puts output.stdout
    nodeinstall.chmod(0644)
  end
  action :nothing
  subscribes :create, resources("remote_file[#{node['pxe_dust']['dir']}/original-install.sh]")
end

# straighten up symlinks for downloading from install.sh
ruby_block "create symlinks that match up with the urls" do
  block do
    Dir.glob("#{node['pxe_dust']['dir']}/chef-full-stack/*").each do |distro|
      if File.directory?(distro) && !File.symlink?(distro)
        packages = Dir.glob("#{distro}/chef_*")
        versions = []
        packages.each {|x| versions << x.split('/').last.split('-')[0].split('_')[1]}
        versions.uniq.each do |version|
          vpackages = Dir.glob("#{distro}/chef_#{version}-*").sort
          vpackages.each do |filename|
            unless File.symlink?(filename)
              lname = "#{distro}/chef_#{version}_#{filename.split('_').last}"
              ln = "ln -sf #{filename} #{lname}"
              Chef::Log.debug ln
              cmd = Mixlib::ShellOut.new(ln)
              output = cmd.run_command
              Chef::Log.debug output.stdout
            end
          end
        end
      end
    end
  end
end