solokit 0.0.1

data/.rvmrc

@@ -0,0 +1 @@
+rvm ree-1.8.7-2010.02@solokit --create

data/Gemfile

@@ -0,0 +1,2 @@
+source "http://rubygems.org"
+gemspec

data/LICENCE

@@ -0,0 +1,19 @@
+Copyright (c) 2011 Joakim Kolsjö
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README

@@ -0,0 +1 @@
+Extracting out some code I use to provision servers with chef-solo. Only supports ubuntu.

data/lib/chef.rb

@@ -0,0 +1,57 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'ssh.rb'))
+
+class Chef
+  def initialize(ip, name, env, debug_ssh, user = 'root')
+    @ip, @name, @env = ip, name, env
+    @ssh = SSH.new(ip, user, debug_ssh)
+  end
+
+  def install
+    return true if installed?
+    puts "#{@name} (#{@env}): Installing chef..."
+    @ssh.run('export DEBIAN_FRONTEND=noninteractive; apt-get update && apt-get upgrade -y && apt-get install ruby ruby1.8-dev libopenssl-ruby wget rsync build-essential -y && wget http://production.cf.rubygems.org/rubygems/rubygems-1.3.7.tgz && tar xfz rubygems-1.3.7.tgz && cd rubygems-1.3.7 && ruby setup.rb && cd .. && rm -rf rubygems-1.3.7* && ln -s /usr/bin/gem1.8 /usr/bin/gem && gem install chef ohai --no-ri --no-rdoc')
+  end
+
+  def upload(root = "/")
+    @ssh.run("rm -rf #{root}var/chef-solo", false) &&
+    @ssh.run("rm -rf #{root}etc/chef", false) &&
+    upload_files("cookbooks/upstream/", "#{root}var/chef-solo/upstream-cookbooks") &&
+    upload_files("cookbooks/site/", "#{root}var/chef-solo/site-cookbooks") &&
+    (File.exists?("envs/#{@env}/cookbooks") ? 
+     upload_files("envs/#{@env}/cookbooks/*", "#{root}var/chef-solo/site-cookbooks") :
+     true) &&
+    upload_files("chef/*", "#{root}etc/chef") &&
+    upload_files("envs/#{@env}/chef/", "#{root}etc/chef") 
+  end
+
+  def run(debug = false, root = "/")
+    puts "\n#{@name} (#{@env}): Running chef..."
+    if debug
+      @ssh.run("#{custom_ruby_path(root)} chef-solo -c #{root}etc/chef/solo.rb -j #{root}etc/chef/#{@name}.json -l debug", false)
+    else
+      @ssh.run("#{custom_ruby_path(root)} chef-solo -c #{root}etc/chef/solo.rb -j #{root}etc/chef/#{@name}.json", false)
+    end
+  end
+
+  def self.provision!(env, host, config)
+    chef = Chef.new(host, "server", env, config.debug_ssh)
+    chef.install || raise("Chef install failed on #{env}")
+    chef.upload || raise("Chef upload failed on #{env}")
+    chef.run(config.debug_chef) || raise("Chef failed on #{env}")
+  end
+
+  private
+  
+  def custom_ruby_path(root)
+    "PATH=\"#{root}ruby/bin:/sbin:$PATH\""
+  end
+
+  def upload_files(from, to)
+    @ssh.run("mkdir -p #{to}") &&
+    @ssh.rsync(from, to, true) 
+  end
+
+  def installed?
+    @ssh.run('gem list chef | grep chef')
+  end
+end

data/lib/configuration.rb

@@ -0,0 +1,19 @@
+require 'ostruct'
+
+class Configuration < OpenStruct
+  def initialize(*envs)
+    shared_config = YAML.load_file("config.yml")
+    
+    config = shared_config
+    envs.each do |env|
+      env_config_file = "envs/#{env}/config.yml"
+
+      config = File.exists?(env_config_file) ?
+               YAML.load_file(env_config_file).merge(config) :
+               config
+    end
+
+    super(config)
+  end
+end
+

data/lib/solokit/version.rb

@@ -0,0 +1,3 @@
+module Solokit
+  VERSION = "0.0.1"
+end

data/lib/solokit.rb

@@ -0,0 +1,4 @@
+require File.join(File.dirname(__FILE__), "chef.rb")
+require File.join(File.dirname(__FILE__), "user_json_generator.rb")
+require File.join(File.dirname(__FILE__), "configuration.rb")
+

data/lib/ssh.rb

@@ -0,0 +1,33 @@
+class SSH
+  def initialize(ip, user, debug)
+    @ip, @user, @debug = ip, user, debug
+  end
+
+  def run(command, quiet = true)
+    run_command("ssh #{ssh_opts} #{@user}@#{@ip} '#{command}' #{supress_output(quiet)}")
+  end
+
+  def rsync(source, target, quiet = false)
+    run_command("rsync -e 'ssh #{ssh_opts}' -az #{source} #{@user}@#{@ip}:#{target} #{supress_output(quiet)}")
+  end
+
+  private
+
+  def supress_output(hide_stdout)
+    if @debug
+      ""
+    else
+      "#{hide_stdout ? '2> /dev/null 1> /dev/null' : '2> /dev/null'}"
+    end
+  end
+
+  def run_command(cmd)
+    puts "SSH | #{cmd}" if @debug
+    system(cmd)
+  end
+
+  def ssh_opts
+    "-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no"
+  end
+end
+

data/lib/user_json_generator.rb

@@ -0,0 +1,100 @@
+require 'yaml'
+require 'active_support'
+require 'active_support/json/encoding'
+
+class UserJsonGenerator
+
+  def initialize(key_path, env, opts = {})
+    @key_path = key_path
+
+    data = YAML.load_file("users.yml")
+
+    @default_hash = data["default_hash"]
+    @users = data["users"]
+    @groups = data["groups"]
+    @env = data["envs"][env]
+    @opts = { :home => "/home" }.merge(opts)
+  end
+
+  def generate(target)
+    if ENV['DEBUG_USER_JSON']
+      require 'pp'
+      ::PP.pp generate_json
+      exit 0
+    end
+    File.open(target, 'w') { |f| f.write(generate_json) }
+  end
+
+  def self.generate!(env, opts = {})
+    UserJsonGenerator.new("public_keys", env, opts).generate("envs/#{env}/chef/roles/users.json")
+  end
+
+  private
+
+  def generate_json
+    {
+      :name => "Users",
+      :chef_type => "role",
+      :json_class => "Chef::Role",
+      :override_attributes => {
+        :users => env_users.inject({}) { |h, user|
+          h[user] = {
+              :password => (@users[user] || {})["hash"] || @default_hash,
+              :home => (user == 'root' ? '/root' : [ @opts[:home], user ].join('/')),
+              :hidden_home => !!(@users[user] || {})["hidden_home"]
+          }; h
+        },
+        :ssh_keys => env_users.inject({}) { |h, user|
+          h[user] = get_keys(user); h
+        },
+        :authorization => {
+          :sudo => { :users => env_sudo_users }
+        }
+      },
+      :run_list => [ "recipe[users]", "recipe[sudo]" ]
+    }.send(ENV['DEBUG_USER_JSON'] ? 'to_hash' : 'to_json')
+    #.gsub(/\{/, "{\n").gsub(/\}/, "}\n")
+  end
+
+  def env_users
+    resolve_users(@env["users"].split)
+  end
+
+  def env_sudo_users
+    @env["sudo"] && resolve_users(@env["sudo"].split)
+  end
+
+  def resolve_users(list)
+    list.map { |item|
+      if item.include?("group")
+        fetch_group_users(item)
+      else
+        item
+      end
+    }.flatten
+  end
+
+  def get_keys(user)
+    (@users[user] || {})["keys"].to_s.split.map do |key_name|
+      if key_name.include?("group")
+        keys_from_group(key_name)
+      else
+        key_file = "#{@key_path}/#{key_name}.pub"
+        if File.exists?(key_file)
+          File.read(key_file).chomp
+        else
+          raise "Could not find key file: #{key_file}."
+        end
+      end
+    end.join("\n")
+  end
+
+  def fetch_group_users(name)
+    @groups[name.split('/').last].split
+  end
+
+  def keys_from_group(name)
+    @groups[name.split("/").last].split.map { |user| get_keys(user) }
+  end
+end
+

data/solokit.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "solokit/version"
+
+Gem::Specification.new do |s|
+  s.name        = "solokit"
+  s.version     = Solokit::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Joakim Kolsjö"]
+  s.email       = ["joakim.kolsjo@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{}
+  s.description = %q{}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {spec}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map { |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency "activesupport", ">= 3.0.0"
+  s.add_development_dependency "bundler"
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification 
+name: solokit
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- "Joakim Kolsj\xC3\xB6"
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-25 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        - 0
+        version: 3.0.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: ""
+email: 
+- joakim.kolsjo@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .rvmrc
+- Gemfile
+- LICENCE
+- README
+- lib/chef.rb
+- lib/configuration.rb
+- lib/solokit.rb
+- lib/solokit/version.rb
+- lib/ssh.rb
+- lib/user_json_generator.rb
+- solokit.gemspec
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: ""
+test_files: []
+