babushka_vagrant.rb

class BabushkaProvisioner < Vagrant::Provisioners::Base
  class Config < Vagrant::Config::Base
    attr_accessor :args
    attr_accessor :deps

    def initialize
      @deps = []
    end

    def dep(dep_spec, args = {})
      deps << [dep_spec, args]
    end
  end

  def self.config_class
    BabushkaProvisioner::Config
  end

  def provision!
    # Only install babushka if it's not already instealled on the remote.
    bootstrap_babushka! unless remote_host.test('babushka --version')

    # Run each named dep in order, failing fast.
    config.deps.all? do |(dep_name, dep_args)|
      remote_command(
        'babushka',
        '--update --defaults --colour', [config.args].flatten(1).join(' '),
        "'#{dep_name}'",
        *arg_pairs(dep_args)
      ).tap {
        # Disconnect after running each dep, so a new session is started
        # each time, for things like setting the locale.
        disconnect!
      }
    end
  end

  private

  def bootstrap_babushka!
    env[:ui].info("Installing babushka on #{host_name}.")
    tmpfile = "/tmp/babushka_me_up"
    File.open(tmpfile, 'w') {|f| f.write `curl https://babushka.me/up` }
    remote_host.upload(tmpfile, tmpfile)
    remote_command("sh #{tmpfile}")
  end

  def remote_command *cmd
    # Truncate long arguments (like ssh pubkeys, etc) so as not to spam the terminal.
    logged_cmd = cmd.map {|i| i.sub(/^(.{50})(.{3}).*/m, '\1...').sub(/\n/, ' ') }.join(' ') # the command, with long args truncated
    env[:ui].info "#{host_name}$ #{logged_cmd}"
    remote_host.sudo(cmd.join(' ')) do |type, data|
      if [:stderr, :stdout].include?(type)
        env[:ui].info(data.chomp, :prefix => false)
      end
    end
  end

  def arg_pairs arg_hash
    arg_hash.keys.map {|k| "#{k}='#{arg_hash[k]}'" }
  end

  def disconnect!
    remote_host.instance_variable_get('@connection').close
  end

  def remote_host
    env[:vm].channel
  end

  def host_name
    env[:vm].config.vm.host_name
  end
end