ssp 0.1

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2010 Secret Sauce Partners, Inc.
+
+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.md

@@ -0,0 +1 @@
+The SSP gem provides various command line tools that we use internally. It might be useful to others as well, but we're not trying to make them general.

data/bin/ssp

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+$:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
+require 'ssp/app'
+
+SSP::App.start

data/lib/ssp/app.rb

@@ -0,0 +1,106 @@
+require 'thor'
+require 'thor/group'
+
+module SSP
+  class App < Thor
+    namespace :app
+    map "-T" => :list
+
+    # Override Thor#help so it can give information about any class and any method.
+    #
+    def help(meth=nil)
+      if meth && !self.respond_to?(meth)
+        initialize_commands
+        klass, task = Thor::Util.find_class_and_task_by_namespace(meth)
+        if klass
+          klass.start(["-h", task].compact, :shell => self.shell)
+        else
+          super
+        end
+      else
+        super
+      end
+    end
+
+    # If a task is not found on SSP::App, method missing is invoked and
+    # SSP::App is then responsable for finding the task in all classes.
+    #
+    def method_missing(meth, *args)
+      meth = meth.to_s
+      initialize_commands
+      klass, task = Thor::Util.find_class_and_task_by_namespace(meth)
+      args.unshift(task) if task
+      klass.start(args, :shell => self.shell)
+    end
+
+    desc "version", "Show SSP version"
+    def version
+      require 'ssp/version'
+      say "SSP #{SSP::VERSION}"
+    end
+
+    desc "list [SEARCH]", "List the available ssp tasks (--substring means .*SEARCH)"
+    method_options :substring => :boolean, :group => :string, :all => :boolean
+    def list(search="")
+      initialize_commands
+
+      search = ".*#{search}" if options["substring"]
+      search = /^#{search}.*/i
+      group  = options[:group] || "standard"
+
+      klasses = Thor::Base.subclasses.select do |k|
+        (options[:all] || k.group == group) && k.namespace =~ search
+      end
+
+      display_klasses(klasses)
+    end
+
+
+    private
+      def initialize_commands
+        unless @_commands_initialized
+          Dir[File.join(File.dirname(__FILE__), "application", "*.rb")].each do |command|
+            require(command)
+          end
+          @_commands_initialized = true
+        end
+      end
+
+      # Display information about the given klasses.
+      #
+      def display_klasses(klasses=Thor::Base.subclasses)
+        klasses -= [Thor, SSP::App, Thor::Group]
+
+        raise Error, "No Thor tasks available" if klasses.empty?
+
+        # Remove subclasses
+        klasses.dup.each do |klass|
+          klasses -= Thor::Util.thor_classes_in(klass)
+        end
+
+        list = Hash.new { |h,k| h[k] = [] }
+        groups = klasses.select { |k| k.ancestors.include?(Thor::Group) }
+
+        # Get classes which inherit from Thor
+        (klasses - groups).each { |k| list[k.namespace] += k.printable_tasks(false) }
+
+        # Get classes which inherit from Thor::Base
+        groups.map! { |k| k.printable_tasks(false).first }
+        list["root"] = groups
+
+        # Order namespaces with default coming first
+        list = list.sort{ |a,b| a[0].sub(/^default/, '') <=> b[0].sub(/^default/, '') }
+        list.each { |n, tasks| display_tasks(n, tasks) unless tasks.empty? }
+      end
+
+      def display_tasks(namespace, list) #:nodoc:
+        list.sort!{ |a,b| a[0] <=> b[0] }
+
+        say shell.set_color(namespace, :blue, true)
+        say "-" * namespace.size
+
+        print_table(list, :truncate => true)
+        say
+      end
+  end
+end

data/lib/ssp/application/bags.rb

@@ -0,0 +1,67 @@
+require 'json'
+require 'pathname'
+
+class SSP::App::Bags < Thor
+  namespace :bags
+
+  class_option :knife_config, :aliases => "-c",
+               :desc => "Location of the knife configuration file",
+               :default => File.join(ENV['HOME'], '.chef', 'knife.rb')
+
+  class_option :path, :aliases => "-p",
+               :desc => "Root path for the data bag files",
+               :default => "./databags"
+
+
+  desc "download", "Downloads data bags from the server"
+  def download
+    bags = JSON.parse(knife_cmd("list"))
+    bags.each do |bag|
+      say "Downloading #{shell.set_color(bag, :bold)}: "
+      bag_path = data_bag_root.join(bag)
+      bag_path.mkpath
+      items = JSON.parse(knife_cmd("show #{bag}"))
+      items.each_with_index do |item, i|
+        say "#{", " unless i == 0}#{item}", nil, false
+        knife_cmd "show #{bag} #{item} > #{bag_path.join("#{item}.json")}"
+      end
+      puts
+    end
+  end
+
+  desc "upload [BAG [ITEM]]", "Uploads data bags to the server"
+  def upload(selected_bag = nil, selected_item = nil)
+    bags = if selected_bag
+      [data_bag_root.join(selected_bag)]
+    else
+      Pathname.glob(data_bag_root.join("*")).select {|p| p.directory?}
+    end
+    bags.each do |bag_path|
+      bag = bag_path.basename
+      say "Uploading #{shell.set_color(bag, :bold)}: "
+      knife_cmd "create #{bag} > /dev/null"
+      current_items = JSON.parse(knife_cmd("show #{bag}"))
+      items = if selected_bag and selected_item
+        [data_bag_root.join(selected_bag, "#{selected_item}.json")]
+      else
+        Pathname.glob(bag_path.join("*.json"))
+      end
+      items.each_with_index do |item, i|
+        key = item.basename.to_s.split(".").first
+        say "#{", " unless i == 0}#{key}", nil, false
+        command = current_items.include?(key) ? "edit" : "create"
+        knife_cmd "#{command} -e 'cat #{item} >' #{bag} #{key}"
+      end
+      puts
+    end
+  end
+
+  private
+    def knife_cmd(cmd)
+      %x{knife data bag -c #{options[:knife_config]} #{cmd}}
+    end
+
+    def data_bag_root
+      Pathname.new(options[:path])
+    end
+end

data/lib/ssp/application/node.rb

@@ -0,0 +1,128 @@
+require 'fog'
+require 'chef/config'
+require 'chef/knife/ssh'
+require 'ssp/config'
+
+class SSP::App::Node < Thor
+  namespace :node
+
+  class_option :api_key,
+               :desc => "Rackspace API key",
+               :default => SSP::Config[:rackspace][:api_key]
+
+  class_option :api_username,
+               :desc => "Rackspace API username",
+               :default => SSP::Config[:rackspace][:username]
+
+  class_option :chef_config, :aliases => "-c",
+               :desc => "Location of the chef configuration file",
+               :default => File.join(ENV['HOME'], '.chef', 'knife.rb')
+
+
+  desc "create FQDN [ROLES...]", "Creates a new chef client on Rackspace"
+  method_option :flavor, :aliases => "-f", :type => :numeric,
+                :desc => "Sets the flavor of the server (1: 256MB, 2: 512MB, 3: 1024MB, ...)",
+                :default => 2
+  method_option :image, :aliases => "-i", :type => :numeric,
+                :desc => "Sets the image to be used (49: Ubuntu 10.04, 4151016: chef-node)",
+                :default => 4151016
+  def create(fqdn, *roles)
+    server = connection.servers.new
+
+    # Convert roles array to proper run list
+    run_list = roles.map { |r| r =~ /recipe\[(.*)\]/ ? $1 : "role[#{r}]" }
+
+    # Script for creating an /etc/hosts file that will result in correct
+    # fqdn and will point to the temporary chef server if we're testing
+    setup_node_script = <<-EOH
+#!/bin/bash
+
+IP=$(ifconfig eth0 | sed -ne 's/.*inet addr:\\([0-9\\.]*\\).*/\\1/p')
+
+echo "127.0.0.1	localhost localhost.localdomain" > /etc/hosts
+echo "$IP	#{fqdn} #{fqdn.split(".").first}" >> /etc/hosts
+    EOH
+    if chef_config[:chef_server_url] == "https://chef-server.rackspace"
+      chef_server_ip = %x{dscl . read Hosts/chef-server.rackspace 2>/dev/null}[/[\d\.]+/]
+      setup_node_script << %{echo "#{chef_server_ip}	chef.secretsaucepartners.com" >> /etc/hosts\n}
+      chef_config[:chef_server_url] = "https://chef.secretsaucepartners.com"
+    end
+    setup_node_script << "exit 0\n"
+
+    # Always include the chef-client role in the run list
+    unless run_list.include? "role[chef-client]"
+      run_list.unshift "role[chef-client]"
+    end
+
+    server.flavor_id = options[:flavor]
+    server.image_id = options[:image]
+    server.name = fqdn
+    server.personality = [
+      {
+        'path' => '/etc/setup-node',
+        'contents' => setup_node_script
+      },
+      {
+        'path' => "/etc/chef/validation.pem",
+        'contents' => IO.read(chef_config[:validation_key])
+      },
+      {
+        'path' => "/etc/chef/client.rb",
+        'contents' => <<-EOH
+log_level        :info
+log_location     STDOUT
+chef_server_url  "#{chef_config[:chef_server_url]}"
+validation_client_name "#{chef_config[:validation_client_name]}"
+        EOH
+      },
+      {
+        'path' => "/etc/chef/first-boot.json",
+        'contents' => { "run_list" => run_list }.to_json
+      },
+    ]
+
+    server.save
+
+    say_status "Name: ", server.name, :cyan
+    say_status "Flavor: ", server.flavor_id, :cyan
+    say_status "Image: ", server.image_id, :cyan
+    say_status "Public IP: ", server.addresses["public"], :cyan
+    say_status "Private IP: ", server.addresses["private"], :cyan
+    say_status "Password: ", server.password, :cyan
+
+    say "\nRequesting server", :magenta
+    saved_password = server.password
+
+    # wait for it to be ready to do stuff
+    server.wait_for { print "."; ready? }
+
+    say "\nServer ready, waiting 15 seconds to bootstrap."
+    sleep 15
+
+    say "\nBootstrapping #{shell.set_color(server.name, :bold)}..."
+
+    ssh = Chef::Knife::Ssh.new
+    ssh.name_args = [ server.addresses["public"][0], "/bin/bash /etc/setup-node && /usr/local/bin/chef-client -j /etc/chef/first-boot.json" ]
+    ssh.config[:ssh_user] = "root"
+    ssh.config[:manual] = true
+    ssh.config[:password] = saved_password
+    ssh.password = saved_password
+    ssh.run
+  end
+
+  protected
+    def connection
+      @connection ||= Fog::Rackspace::Servers.new(
+        :rackspace_api_key => options[:api_key],
+        :rackspace_username => options[:api_username]
+      )
+    end
+
+    def chef_config
+      unless defined?(@_chef_config_loaded)
+        Chef::Config.from_file(options[:chef_config])
+        @_chef_config_loaded = true
+      end
+      Chef::Config
+    end
+end

data/lib/ssp/application/pair.rb

@@ -0,0 +1,107 @@
+require 'yaml'
+require 'open-uri'
+
+class SSP::App::Pair < Thor
+  namespace :pair
+
+  default_task :set_author
+  desc "set_author [PAIR]", "Sets the commit author for the repository to a pair or just yourself"
+  method_option :devsfile,
+                :desc => "Location of the SSP developers file",
+                :default => File.join(ENV['HOME'], '.sspdevs')
+  method_option :pair_email,
+                :desc => "Base email to use for constructing pair emails",
+                :default => "dev@secretsaucepartners.com"
+  def set_author(pair=nil)
+    chek_dot_git!
+
+    me = %x{git config --global user.name}.chomp
+    my_email = sspdevs[me]
+    devs = sspdevs.map {|name,_| name}.sort
+    devs.delete(me)
+
+    if pair.nil?
+      say "Pairing?", :bold
+      devs.each_with_index do |name, index|
+        say_status "#{index + 1}:", name, :cyan
+      end
+      print "Enter the index of your pair or leave blank if not pairing: "
+
+      while (choice = File.new("/dev/tty").readline.chomp) !~ /\A\d*\Z/
+        say "Bad input `#{choice}'"
+      end
+    elsif pair =~ /^(|x|me|not)$/i
+      choice = ""
+    else
+      pair = pair.downcase
+      pair = devs.detect do |dev|
+        initial = dev.split(" ").map {|x| x[0..0]}.join.downcase
+        initial == pair or dev.downcase.split(" ").include?(pair)
+      end
+      raise Thor::Error, "No such pair" unless pair
+      choice = devs.index(pair) + 1
+    end
+
+    commit_name, commit_email = if choice == ''
+      [me, my_email]
+    else
+      pair = devs[choice.to_i - 1]
+      ["#{me} & #{pair}", options[:pair_email].sub('@', initials(me, pair)+'@')]
+    end
+
+    say "Setting #{shell.set_color("commit author", :bold)} to #{shell.set_color(commit_name, :green)}"
+    %x{git config user.name '#{commit_name}'}
+    say "Setting #{shell.set_color("commit email", :bold)} to #{shell.set_color(commit_email, :green)}"
+    %x{git config user.email '#{commit_email}'}
+  end
+
+  desc "status", "Displays current commit author and email and asks for confirmation (used by the install hook)"
+  def status
+    chek_dot_git!
+
+    commit_name = %x{git config user.name}.chomp
+    commit_email = %x{git config user.email}.chomp
+    say "Current #{shell.set_color("commit author", :bold)} is #{shell.set_color(commit_name, :yellow, true)}"
+    say "Current #{shell.set_color("commit email", :bold)} is #{shell.set_color(commit_email, :yellow, true)}"
+
+    say "Is this ok? [Y/n] "
+    unless File.new("/dev/tty").readline.chomp =~ /^(y(es)?)?$/i
+      exit 1
+    end
+  end
+
+  desc "install_hook", "Installs a git pre-commit hook which checks commit author and email"
+  def install_hook
+    chek_dot_git!
+
+    hook_file = ".git/hooks/pre-commit"
+    raise Thor::Error, "a pre-commit hook file already exists" if File.exists?(hook_file)
+
+    File.open(hook_file, "w") { |f| f.write "#!/bin/sh\nssp pair:status\n" }
+    File.chmod(0755, hook_file)
+
+    say "pre-commit hook is installed in #{hook_file}"
+  end
+
+  private
+    def chek_dot_git!
+      unless File.exists?(".git")
+        raise Thor::Error, "This doesn't look like a git repository."
+      end
+    end
+
+    def sspdevs
+      @sspdevs ||= begin
+        unless File.exists?(options[:devsfile])
+          File.open(options[:devsfile], "w") do |f|
+            f.write(open("https://gist.github.com/bba2eeabf72da8d61254.txt").read)
+          end
+        end
+        YAML.load_file(options[:devsfile])
+      end
+    end
+
+    def initials(*names)
+      "+" + names.map {|n| n.split(" ").map {|x| x[0..0]}.join}.join("+").downcase
+    end
+end

data/lib/ssp/config.rb

@@ -0,0 +1,7 @@
+require 'yaml'
+
+module SSP
+  CONFIG_FILE = File.expand_path("~/.ssp")
+  Config = Hash.new({})
+  Config.merge!(YAML.load_file(CONFIG_FILE)) if File.exists?(CONFIG_FILE)
+end

data/lib/ssp/version.rb

@@ -0,0 +1,3 @@
+module SSP
+  VERSION = "0.1"
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification 
+name: ssp
+version: !ruby/object:Gem::Version 
+  hash: 9
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  version: "0.1"
+platform: ruby
+authors: 
+- !binary |
+  TMOhc3psw7MgQsOhY3Np
+
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-06-30 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 17
+        segments: 
+        - 0
+        - 13
+        version: "0.13"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: chef
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 25
+        segments: 
+        - 0
+        - 9
+        version: "0.9"
+  type: :runtime
+  version_requirements: *id002
+description: "  The SSP gem provides various command line tools that\n  Secret Sauce Partners, Inc. uses internally.\n"
+email: dev@secretsaucepartners.com
+executables: 
+- ssp
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README.md
+- LICENSE
+- bin/ssp
+- lib/ssp/app.rb
+- lib/ssp/application/bags.rb
+- lib/ssp/application/node.rb
+- lib/ssp/application/pair.rb
+- lib/ssp/config.rb
+- lib/ssp/version.rb
+has_rdoc: true
+homepage: http://github.com/sspinc/ssp
+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.3.7
+signing_key: 
+specification_version: 3
+summary: Various command line tools for Secret Sauce Partners, Inc.
+test_files: []
+