from-scratch 0.1.0

data/cookbooks/rvm/providers/shell.rb

@@ -0,0 +1,121 @@
+#
+# Cookbook Name:: rvm
+# Provider:: shell
+#
+# Author:: Fletcher Nichol <fnichol@nichol.ca>
+#
+# Copyright 2011, Fletcher Nichol
+#
+# 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.
+#
+
+include Chef::RVM::StringHelpers
+include Chef::RVM::EnvironmentHelpers
+include Chef::RVM::ShellHelpers
+
+def load_current_resource
+  @user_rvm     = user_installed_rvm? ? new_resource.user : nil
+  @rubie        = normalize_ruby_string(select_ruby(new_resource.ruby_string), @user_rvm)
+  @gemset       = select_gemset(new_resource.ruby_string)
+  @ruby_string  = @gemset.nil? ? @rubie : "#{@rubie}@#{@gemset}"
+  @rvm_env      = ::RVM::ChefUserEnvironment.new(@user_rvm)
+end
+
+action :run do
+  user_rvm = @user_rvm
+
+  # ensure ruby is installed and gemset exists
+  unless env_exists?(@ruby_string)
+    e = rvm_environment @ruby_string do
+      user    user_rvm
+      action :nothing
+    end
+    e.run_action(:create)
+  end
+
+  script_wrapper :run
+  new_resource.updated_by_last_action(true)
+end
+
+private
+
+##
+# Wraps the script resource for RVM-dependent code.
+#
+# @param [Symbol] action to be performed with gem_package provider
+def script_wrapper(exec_action)
+  profile = find_profile_to_source
+
+  script_code = <<-CODE
+    if [ -s "${HOME}/.rvm/scripts/rvm" ]; then
+      source "${HOME}/.rvm/scripts/rvm"
+    elif [ -s "#{profile}" ]; then
+      source "#{profile}"
+    fi
+
+    rvm use #{@ruby_string}
+
+    #{new_resource.code}
+  CODE
+
+  if new_resource.user
+    user_rvm  = user_installed_rvm?
+    user_home = user_dir
+  end
+
+  s = script new_resource.name do
+    interpreter   "bash"
+
+    if new_resource.user
+      user        new_resource.user
+      if user_rvm && new_resource.environment
+        environment({ 'USER' => new_resource.user, 'HOME' => user_home }.merge(
+          new_resource.environment))
+      elsif user_rvm
+        environment({ 'USER' => new_resource.user, 'HOME' => user_home })
+      end
+    end
+
+    code          script_code
+    creates       new_resource.creates      if new_resource.creates
+    cwd           new_resource.cwd          if new_resource.cwd
+    group         new_resource.group        if new_resource.group
+    path          new_resource.path         if new_resource.path
+    returns       new_resource.returns      if new_resource.returns
+    timeout       new_resource.timeout      if new_resource.timeout
+    umask         new_resource.umask        if new_resource.umask
+    action        :nothing
+  end
+  s.run_action(exec_action)
+  new_resource.updated_by_last_action(true) if s.updated_by_last_action?
+end
+
+##
+# Whether or not the user has an isolated RVM installation
+#
+# @return [true,false] does the user have RVM installed for themselves?
+def user_installed_rvm?
+  return false unless new_resource.user
+
+  ::File.exists?("#{user_dir}/.rvm/VERSION")
+end
+
+##
+# Determines the user's home directory
+#
+# @return [String] the path to the user's home directory
+def user_dir
+  return nil unless new_resource.user
+
+  Etc.getpwnam(new_resource.user).dir
+end

data/cookbooks/rvm/providers/wrapper.rb

@@ -0,0 +1,73 @@
+#
+# Cookbook Name:: rvm
+# Provider:: wrapper
+#
+# Author:: Fletcher Nichol <fnichol@nichol.ca>
+#
+# Copyright 2011, Fletcher Nichol
+#
+# 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.
+#
+
+include Chef::RVM::StringHelpers
+include Chef::RVM::EnvironmentHelpers
+
+def load_current_resource
+  @rubie        = normalize_ruby_string(select_ruby(new_resource.ruby_string))
+  @gemset       = select_gemset(new_resource.ruby_string)
+  @ruby_string  = @gemset.nil? ? @rubie : "#{@rubie}@#{@gemset}"
+  @rvm_env      = ::RVM::ChefUserEnvironment.new(new_resource.user)
+
+  if new_resource.binary.nil?
+    @binaries = new_resource.binaries || []
+  else
+    @binaries = [ new_resource.binary ] || []
+  end
+end
+
+action :create do
+  # ensure ruby is installed and gemset exists
+  unless env_exists?(@ruby_string)
+    e = rvm_environment @ruby_string do
+      user    new_resource.user
+      action :nothing
+    end
+    e.run_action(:create)
+  end
+
+  @rvm_env.use @ruby_string
+
+  @binaries.each { |b| create_wrapper(b) }
+  new_resource.updated_by_last_action(true)
+end
+
+private
+
+def create_wrapper(bin)
+  full_bin = "#{new_resource.prefix}_#{bin}"
+  resource_name = "rvm_wrapper[#{full_bin}::#{@ruby_string}]"
+  script = ::File.join(@rvm_env.config["rvm_path"], "bin", full_bin)
+
+  if ::File.exists?(script)
+    Chef::Log.debug("#{resource_name} already exists, so updating")
+  else
+    Chef::Log.info("Creating #{resource_name}")
+  end
+
+  if @rvm_env.wrapper @ruby_string, new_resource.prefix, bin
+    Chef::Log.debug("Creation/Update of #{resource_name} was successful.")
+    new_resource.updated_by_last_action(true)
+  else
+    Chef::Log.warn("Failed to create/update #{resource_name}.")
+  end
+end

data/cookbooks/rvm/recipes/default.rb

@@ -0,0 +1,40 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: default
+#
+# Copyright 2010, 2011, Fletcher Nichol
+#
+# 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.
+#
+
+# install rvm api gem during chef compile phase
+chef_gem 'rvm' do
+  action :install
+  version '>= 1.11.3.6'
+end
+require 'rvm'
+
+create_rvm_shell_chef_wrapper
+create_rvm_chef_user_environment
+
+class Chef::Resource
+  # mix in #rvm_cmd_wrap helper into resources
+  include Chef::RVM::ShellHelpers
+end
+
+class Chef::Recipe
+  # mix in recipe helpers
+  include Chef::RVM::ShellHelpers
+  include Chef::RVM::RecipeHelpers
+  include Chef::RVM::StringHelpers
+end

data/cookbooks/rvm/recipes/gem_package.rb

@@ -0,0 +1,33 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: gem_package
+#
+# Copyright 2011, Fletcher Nichol
+#
+# 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.
+#
+
+node_val = node['rvm']['gem_package']['rvm_string']
+case node_val
+when String
+  rvm_descriptor = node_val + " RVM Ruby"
+when Array
+  last = node_val.pop
+  rvm_descriptor = [ node_val.join(', '), last ].join(' & ') + " RVM Rubies"
+end
+
+patch_gem_package
+
+::Chef::Log.info "gem_package resource has been patched to use provider " <<
+  "Chef::Provider::Package::RVMRubygems and will install gems to " <<
+  "the #{rvm_descriptor}."

data/cookbooks/rvm/recipes/system.rb

@@ -0,0 +1,37 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: system
+#
+# Copyright 2010, 2011 Fletcher Nichol
+#
+# 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.
+#
+
+include_recipe "rvm::system_install"
+
+perform_install_rubies  = node['rvm']['install_rubies'] == true ||
+                  node['rvm']['install_rubies'] == "true"
+
+if perform_install_rubies
+  install_rubies  :rubies => node['rvm']['rubies'],
+                  :default_ruby => node['rvm']['default_ruby'],
+                  :global_gems => node['rvm']['global_gems'],
+                  :gems => node['rvm']['gems']
+end
+
+# add users to rvm group
+group 'rvm' do
+  members node['rvm']['group_users']
+
+  only_if { node['rvm']['group_users'].any? }
+end

data/cookbooks/rvm/recipes/system_install.rb

@@ -0,0 +1,42 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: system_install
+#
+# Copyright 2010, 2011 Fletcher Nichol
+#
+# 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.
+#
+
+include_recipe 'rvm'
+
+# Build the rvm group ahead of time, if it is set. This allows avoiding
+# collision with later processes which may set a guid explicitly
+if node['rvm']['group_id'] != 'default'
+  g = group 'rvm' do
+    group_name 'rvm'
+    gid        node['rvm']['group_id']
+    action     :nothing
+  end
+  g.run_action(:create)
+end
+
+key_server = node['rvm']['gpg']['keyserver'] || "hkp://keys.gnupg.net"
+home_dir = "#{node['rvm']['gpg']['homedir'] || '~'}/.gnupg"
+
+execute 'Adding gpg key' do
+  command "`which gpg2 || which gpg` --keyserver #{key_server} --homedir #{home_dir} --recv-keys #{node['rvm']['gpg_key']}"
+  only_if 'which gpg2 || which gpg'
+  not_if { node['rvm']['gpg_key'].empty? }
+end
+
+rvm_installation("root")

data/cookbooks/rvm/recipes/user.rb

@@ -0,0 +1,43 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: user
+#
+# Copyright 2011 Fletcher Nichol
+#
+# 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.
+#
+
+include_recipe 'rvm::user_install'
+
+Array(node['rvm']['user_installs']).each do |rvm_user|
+  perform_install_rubies  = rvm_user['install_rubies'] == true ||
+                            rvm_user['install_rubies'] == "true" ||
+                            node['rvm']['user_install_rubies'] == true ||
+                            node['rvm']['user_install_rubies'] == "true"
+  rubies                  = rvm_user['rubies'] ||
+                            node['rvm']['user_rubies']
+  default_ruby            = rvm_user['default_ruby'] ||
+                            node['rvm']['user_default_ruby']
+  global_gems             = rvm_user['global_gems'] ||
+                            node['rvm']['user_global_gems']
+  gems                    = rvm_user['gems'] ||
+                            node['rvm']['user_gems']
+
+  if perform_install_rubies
+    install_rubies  :rubies => rubies,
+                    :default_ruby => default_ruby,
+                    :global_gems => global_gems,
+                    :gems => gems,
+                    :user => rvm_user['user']
+  end
+end

data/cookbooks/rvm/recipes/user_install.rb

@@ -0,0 +1,58 @@
+#
+# Cookbook Name:: rvm
+# Recipe:: user_install
+#
+# Copyright 2011, 2012, 2013 Fletcher Nichol
+#
+# 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.
+#
+
+include_recipe "rvm"
+
+node["rvm"]["installs"].each do |user, opts|
+  # if user hash is falsy (nil, false) then we're not installing
+  next unless opts
+
+  # if user hash is not a hash (i.e. set to true), init an empty Hash
+  opts = Hash.new if opts == true
+
+  ruby_block "Conditionally add RVM gpg key" do # ~FC022 this will be fixed in the LWRP rewrite
+    block do
+      cmd = Mixlib::ShellOut.new('which gpg2 || which gpg')
+      cmd.run_command
+
+      if cmd.exitstatus == 0
+        gpg_command = cmd.stdout.chomp
+
+        exec = Chef::Resource::Execute.new 'Add RVM gpg key', run_context
+        exec.command "#{gpg_command} --keyserver hkp://keys.gnupg.net --recv-keys #{node['rvm']['gpg_key']}"
+        exec.user user['user']
+        exec.environment 'HOME' => user['home']
+        exec.guard_interpreter :bash
+        exec.not_if "#{gpg_command} -k #{node['rvm']['gpg_key']} > /dev/null", user: user['user'], environment: { 'HOME' => user['home'] }
+        exec.run_action :run
+      else
+        Chef::Log.info 'Skipping adding RVM key because gpg/gpg2 not installed'
+      end
+    end
+  end
+
+  rvm_installation(user.to_s) do
+    %w(installer_url installer_flags install_pkgs rvmrc_template_source
+      rvmrc_template_cookbook rvmrc_env action
+    ).each do |attr|
+      # if user hash attr is set, then set the resource attr
+      send(attr, opts[attr]) if opts.fetch(attr, false)
+    end
+  end
+end