from-scratch 0.1.0

data/cookbooks/rvm/libraries/chef_rvm_shell_helpers.rb

@@ -0,0 +1,57 @@
+#
+# Cookbook Name:: rvm
+# Library:: Chef::RVM::ShellHelpers
+#
+# 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.
+#
+
+class Chef
+  module RVM
+    module ShellHelpers
+      ##
+      # Finds the correct shell profile to source to init an RVM-aware
+      # shell environment
+      #
+      # @param [String] a user's home directory path if this is for a user or
+      #                 nil if it is in a system context
+      # @return [String] full path the shell profile
+      def find_profile_to_source(user_dir = nil)
+        if user_dir
+          "#{user_dir}/.rvm/scripts/rvm"
+        else
+          if ::File.directory?("/etc/profile.d")
+            "/etc/profile.d/rvm.sh"
+          else
+            "/etc/profile"
+          end
+        end
+      end
+
+      ##
+      # Returns a shell command that is RVM-aware
+      #
+      # @param [String, #to_s] the shell command to be wrapped
+      # @param [String] A user's home directory path if this is for a user or
+      #                 nil if it is in a system context
+      # @return [String] the command wrapped in RVM-initialized bash command
+      def rvm_wrap_cmd(cmd, user_dir = nil)
+        profile = find_profile_to_source(user_dir)
+        %{bash -c "source #{profile} && #{cmd.gsub(/"/, '\"')}"}
+      end
+    end
+  end
+end

data/cookbooks/rvm/libraries/chef_rvm_string_cache.rb

@@ -0,0 +1,104 @@
+#
+# Cookbook Name:: rvm
+# Library:: Chef::RVM::StringCache
+#
+# 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.
+#
+
+require 'chef/mixin/command'
+
+class Chef
+  module RVM
+    module ShellHelpers
+      # stub to satisfy StringCache (library load order not guarenteed)
+    end
+
+    class StringCache
+      class << self
+        include Chef::Mixin::Command
+        include Chef::RVM::ShellHelpers
+      end
+
+      ##
+      # Returns a fully qualified RVM Ruby string for the given input string
+      #
+      # @param [String] a string that can interpreted by RVM
+      # @param [String] the username if this is for a user install or nil if
+      #                 it is a system install
+      # @return [String] a fully qualified RVM Ruby string
+      def self.fetch(str, user = nil)
+        @@strings ||= Hash.new
+        rvm_install = user || "system"
+        @@strings[rvm_install] ||= Hash.new
+
+        return @@strings[rvm_install][str] if @@strings[rvm_install].has_key?(str)
+
+        result = canonical_ruby_string(str, user)
+        # cache everything except default environment
+        if str == 'default'
+          result
+        else
+          @@strings[rvm_install][str] = result
+        end
+      end
+
+      protected
+
+      def self.canonical_ruby_string(str, user)
+        Chef::Log.debug("Fetching canonical RVM string for: #{str} " +
+                        "(#{user || 'system'})")
+        if user
+          user_dir = Etc.getpwnam(user).dir
+        else
+          user_dir = nil
+        end
+
+        cmd = ["source #{find_profile_to_source(user_dir)}",
+          "rvm_ruby_string='#{str}'", "__rvm_ruby_string",
+          "echo $rvm_ruby_string"].join(" && ")
+        pid, stdin, stdout, stderr = popen4('bash', shell_params(user, user_dir))
+        stdin.puts(cmd)
+        stdin.close
+
+        result = stdout.read.split('\n').first.chomp
+        if result =~ /^-/   # if the result has a leading dash, value is bogus
+          Chef::Log.warn("Could not determine canonical RVM string for: #{str} " +
+                         "(#{user || 'system'})")
+          nil
+        else
+          Chef::Log.debug("Canonical RVM string is: #{str} => #{result} " +
+                          "(#{user || 'system'})")
+          result
+        end
+      end
+
+      def self.shell_params(user, user_dir)
+        if user
+          {
+            :user => user,
+            :environment => {
+              'USER' => user,
+              'HOME' => user_dir
+            }
+          }
+        else
+          Hash.new
+        end
+      end
+    end
+  end
+end

data/cookbooks/rvm/libraries/chef_rvm_string_helpers.rb

@@ -0,0 +1,61 @@
+#
+# Cookbook Name:: rvm
+# Library:: Chef::RVM::StringHelpers
+#
+# 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.
+#
+
+class Chef
+  module RVM
+    module StringHelpers
+      ##
+      # Filters out any gemset declarations in an RVM Ruby string
+      #
+      # @param [String, #to_s] the RVM Ruby string
+      # @return [String] the Ruby string, minus gemset
+      def select_ruby(ruby_string)
+        ruby_string.split('@').first
+      end
+
+      ##
+      # Filters out any Ruby declaration in an RVM Ruby string
+      #
+      # @param [String, #to_s] the RVM Ruby string
+      # @return [String] the gemset string, minus Ruby or nil if no gemset given
+      def select_gemset(ruby_string)
+        if ruby_string.include?('@')
+          ruby_string.split('@').last
+        else
+          nil
+        end
+      end
+
+      ##
+      # Sanitizes a Ruby string so that it's more normalized.
+      #
+      # @param [String, #to_s] an RVM Ruby string
+      # @param [String] a specific user RVM or nil for system-wide
+      # @return [String] a fully qualified RVM Ruby string
+      def normalize_ruby_string(ruby_string, user = new_resource.user, patch = new_resource.patch)
+        return "system" if ruby_string == "system"
+        fetched_ruby_string = StringCache.fetch(ruby_string, user)
+        return "#{fetched_ruby_string} --patch #{patch}" if patch
+        fetched_ruby_string
+      end
+    end
+  end
+end

data/cookbooks/rvm/libraries/chef_rvm_version_helpers.rb

@@ -0,0 +1,45 @@
+class Chef
+  module RVM
+    module ShellHelpers
+      # stub to satisfy VersionCache (library load order not guarenteed)
+    end
+
+    module VersionHelpers
+      def rvm_version(user = nil)
+        VersionCache.fetch_version(user)
+      end
+    end
+
+    class VersionCache
+      class << self
+        include Chef::Mixin::ShellOut
+        include Chef::RVM::ShellHelpers
+      end
+
+      def self.fetch_version(user = nil)
+        @@versions ||= Hash.new
+        rvm_install = user || "system"
+        @@versions[rvm_install] ||= rvm_version(user)
+      end
+
+      def self.rvm_version(user = nil)
+        cmd = "rvm version | cut -d ' ' -f 2"
+
+        if user
+          user_dir    = Etc.getpwnam(user).dir
+          environment = { 'USER' => user, 'HOME' => user_dir }
+        else
+          user_dir    = nil
+          environment = nil
+        end
+
+        version = shell_out!(
+          rvm_wrap_cmd(cmd, user_dir), :env => environment).stdout.strip
+
+        Chef::Log.debug "RVM version = #{version} (#{user || 'system'})"
+
+        version
+      end
+    end
+  end
+end

data/cookbooks/rvm/libraries/gem_package_monkeypatch.rb

@@ -0,0 +1,34 @@
+#
+# Cookbook Name:: rvm
+# Library: gem_package resource monkey patch
+#
+# 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.
+#
+
+##
+# Patch Chef::Resource::GemPackage resource to use the RVMRubygems provider.
+# This has potentially dangerous side effects and should be considered
+# experimental. You have been warned.
+def patch_gem_package
+  ::Chef::Resource::GemPackage.class_eval do
+    def initialize(name, run_context=nil)
+      super
+      @resource_name = :gem_package
+      @provider = Chef::Provider::Package::RVMRubygems
+    end
+  end
+end

data/cookbooks/rvm/libraries/provider_rvm_installation.rb

@@ -0,0 +1,185 @@
+#
+# Cookbook Name:: rvm
+# Provider:: rvm_installation
+#
+# Author:: Fletcher Nichol <fnichol@nichol.ca>
+#
+# Copyright 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.
+#
+
+require "chef/provider"
+require "chef/resource/remote_file"
+require "chef/resource/package"
+require "chef/resource/template"
+require "chef/mixin/shell_out"
+
+class Chef
+
+  class Provider
+
+    class RvmInstallation < Chef::Provider
+
+      include Chef::Mixin::ShellOut
+
+      def load_current_resource
+        @current_resource = Chef::Resource::RvmInstallation.new(new_resource.name)
+        if @current_resource.installed(installed?)
+          @current_resource.version(version)
+        end
+        @current_resource
+      end
+
+      def action_install
+        converge_rvmrc
+        if current_resource.installed
+          Chef::Log.info("#{new_resource} #{current_resource.version} " +
+            "already installed - nothing to do")
+        else
+          converge_install
+        end
+      end
+
+      def action_force
+        converge_rvmrc
+      end
+
+      def converge_rvmrc
+        converge_by("manage rvmrc for #{new_resource}") { write_rvmrc }
+      end
+
+      def converge_install
+        converge_by("install RVM for #{new_resource}") { install_rvm }
+        Chef::Log.info("#{new_resource} #{new_resource.version} installed")
+      end
+
+      def install_rvm
+        install_packages
+        download_installer
+        run_install_cmd
+
+        new_resource.version(version)
+      end
+
+      def install_packages
+        Array(new_resource.install_pkgs).map do |pkg|
+          r = Chef::Resource::Package.new(pkg, run_context)
+          r.run_action(:install)
+          r
+        end
+      end
+
+      def write_rvmrc
+        r = Chef::Resource::Template.new(rvmrc_path, run_context)
+        r.owner(new_resource.user)
+        r.group(etc_user.gid)
+        r.mode("0644")
+        r.source(new_resource.rvmrc_template_source)
+        r.cookbook(new_resource.rvmrc_template_cookbook)
+        r.variables(
+          :user => new_resource.user,
+          :rvm_path => rvm_path,
+          :rvmrc_env => new_resource.rvmrc_env
+        )
+        r.run_action(:create)
+        r
+      end
+
+      def download_installer
+        r = Chef::Resource::RemoteFile.new(rvm_installer_path, run_context)
+        r.source(new_resource.installer_url)
+        r.run_action(:create)
+        r
+      end
+
+      def run_install_cmd
+        rvm_shell_out!(%{bash #{rvm_installer_path} #{new_resource.installer_flags}})
+      end
+
+      def rvm_installer_path
+        ::File.join(
+          Chef::Config[:file_cache_path], "rvm-installer-#{new_resource.user}"
+        )
+      end
+
+      def installed?
+        cmd = rvm_shell_out(
+          %{bash -c "source #{rvm_path}/scripts/rvm && type rvm"}
+        )
+        (cmd.exitstatus == 0 && cmd.stdout.lines.first == "rvm is a function\n")
+      end
+
+      def version
+        cmd = rvm("version")
+        matches = /^rvm ([\w.]+)/.match(cmd.stdout)
+
+        if cmd.exitstatus != 0
+          raise "Could not determine version for #{new_resource}, " +
+            "exited (#{cmd.exitstatus})"
+        end
+
+        if matches && matches[1]
+          return matches[1]
+        else
+          raise "Could not determine version for #{new_resource} " +
+            "from version string [#{cmd.stdout}]"
+        end
+      end
+
+      def rvm_shell_out(cmd)
+        user = new_resource.user
+        home_dir = etc_user.dir
+        opts = {
+          :user => user,
+          :group => etc_user.gid,
+          :cwd => home_dir,
+          :env => { "HOME" => home_dir, "USER" => user, "TERM" => "dumb" }
+        }
+
+        Chef::Log.debug("Running [#{cmd}] with #{opts}")
+        shell_out(cmd, opts)
+      end
+
+      def rvm_shell_out!(*args)
+        cmd = rvm_shell_out(*args)
+        cmd.error!
+        cmd
+      end
+
+      def rvm(subcommand)
+        rvm_shell_out(%{#{rvm_path}/bin/rvm #{subcommand}})
+      end
+
+      def rvm_path
+        if new_resource.user == "root"
+          "/usr/local/rvm"
+        else
+          ::File.join(etc_user.dir, ".rvm")
+        end
+      end
+
+      def rvmrc_path
+        if new_resource.user == "root"
+          "/etc/rvmrc"
+        else
+          ::File.join(etc_user.dir, ".rvmrc")
+        end
+      end
+
+      def etc_user
+        Etc.getpwnam(new_resource.user)
+      end
+    end
+  end
+end