pupcap 0.0.1

data/.gitignore

@@ -0,0 +1,2 @@
+*.swp
+pkg/

data/Gemfile

@@ -0,0 +1,12 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in capistrano.gemspec
+gemspec
+
+#
+# Development Dependencies from the Gemfile
+# are merged here.
+#
+group :development do
+  gem 'rake', '0.8.7'
+end

data/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+task :default => :test

data/bin/pupcap

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.expand_path("../../lib", __FILE__)
+
+require 'pupcap/cli'
+
+Pupcap::CLI.start

data/lib/pupcap.rb

@@ -0,0 +1,2 @@
+module Pupcap
+end

data/lib/pupcap/action.rb

@@ -0,0 +1,53 @@
+require 'pupcap'
+require 'capistrano'
+require 'capistrano/cli'
+
+module Pupcap::Action
+  class Base
+
+    def initialize(file, options = {})
+      @file = file
+      @options = options
+      set_default_vars
+      ENV["ROLES"] = @options[:role]
+    end
+
+
+    class << self
+      def ensure_local_dir(dir, mode = "0755")
+        run_local("mkdir -p #{dir} && chmod #{mode} #{dir}")
+      end
+
+      def run_local(cmd)
+        system cmd
+        fail "#{cmd} fail" if $?.to_i != 0
+      end
+    end
+
+    def load_recipes(config) #:nodoc:
+      # load the standard recipe definition
+      cap.load "standard"
+    end
+
+    private
+      def set_default_vars
+        app = ENV['app'] || File.basename(File.dirname(@file))
+        cap.set :application,       app
+        cap.set :local_root,         File.dirname(@file)
+        cap.set :provision_key,     "#{cap.local_root}/.keys/provision"
+        cap.set :provision_key_pub, "#{cap.local_root}/.keys/provision.pub"
+        cap.ssh_options[:keys]          = cap.provision_key
+        cap.ssh_options[:forward_agent] = true
+        cap.default_run_options[:pty]   = true
+        cap.load @file
+      end
+
+      def cap
+        unless @cap
+          @cap = Capistrano::Configuration.new
+          @cap.logger.level = Capistrano::Logger::DEBUG
+        end
+        @cap
+      end
+  end
+end

data/lib/pupcap/action/cook.rb

@@ -0,0 +1,14 @@
+require 'pupcap/action'
+
+class Pupcap::Action::Cook < Pupcap::Action::Base
+  def start
+    cap.load "#{root}/cook/Capfile"
+    cap.trigger(:load)
+    cap.find_and_execute_task("cook", :before => :start, :after => :finish)
+    cap.trigger(:exit)
+  end
+
+  def root
+    File.dirname(File.expand_path __FILE__)
+  end
+end

data/lib/pupcap/action/cook/Capfile

@@ -0,0 +1,26 @@
+namespace :cook do
+  task :default do
+    remote_path = "/tmp/puppet"
+    find_servers.each do |server|
+      ssh_cmd = "-e \"ssh -p #{cook_ssh_port(server)} -i #{provision_key}\""
+      rsync_options = "-p --chmod=o+r,g+r -az --exclude=\".keys/\""
+      rsync_path = "--rsync-path=\"sudo rsync\""
+      cmd = "rsync #{rsync_options} #{ssh_cmd} #{local_root}/ #{rsync_path} #{cook_rsync_host(server)}:#{remote_path}"
+      logger.important(cmd)
+      system(cmd)
+    end
+    modules = "--modulepath=#{remote_path}/modules:#{remote_path}/manifests"
+    nook = ENV["try"] ? "--noop" : ""
+    debug = ENV["debug"] ? "--debug" : ""
+    sudo("/usr/local/bin/puppet apply --detailed-exitcodes #{nook} #{debug} #{modules} #{remote_path}/manifests/#{ENV["ROLES"]}.pp || true")
+  end
+
+  def cook_ssh_port(server)
+    server.port || ssh_options[:port] || 22
+  end
+
+  def cook_rsync_host(server)
+    u = server.user || fetch(:user)
+    u ? "#{u}@#{server.host}" : server.host
+  end
+end

data/lib/pupcap/action/prepare.rb

@@ -0,0 +1,28 @@
+require 'pupcap/action'
+
+class Pupcap::Action::Prepare < Pupcap::Action::Base
+  def start
+    cap.load "#{root}/prepare/Capfile"
+    cap.trigger(:load)
+    cap.find_and_execute_task("prepare", :before => :start, :after => :finish)
+    prepare_dist
+    cap.trigger(:exit)
+  end
+
+  def root
+    File.dirname(File.expand_path __FILE__)
+  end
+
+  def prepare_dist
+    rel = cap.capture("lsb_release -d")
+    codename = rel.match(/^.*:\s(.*)$/)
+    if codename && codename[1]
+      codename = codename[1].to_s.strip[0...12]
+      case codename
+      when 'Ubuntu 12.04'
+        cap.load "#{root}/prepare/ubuntu/precise.rb"
+        cap.find_and_execute_task("prepare:ubuntu:precise")
+      end
+    end
+  end
+end

data/lib/pupcap/action/prepare/Capfile

@@ -0,0 +1,24 @@
+namespace :prepare do
+  task :default do
+    generate_keys
+    ssh_copy_id
+  end
+
+  task :ssh_copy_id do
+    remote_key_path = "/tmp/#{application}_id.pub.#{Time.now.to_i}"
+    upload(provision_key_pub, remote_key_path)
+    run("mkdir -p ~/.ssh && chmod 0700 ~/.ssh")
+    run("cat #{remote_key_path} >> ~/.ssh/authorized_keys && chmod 0600 ~/.ssh/authorized_keys")
+    run("rm -f #{remote_key_path}")
+  end
+
+  task :generate_keys do
+    unless File.exists?(provision_key)
+      logger.important("Generate keys")
+      Pupcap::Action.ensure_local_dir("#{local_root}/.keys", "0700")
+      Pupcap::Action.run_local("ssh-keygen -q -b 1024 -t rsa -C \"#{application} provision key\" -f #{provision_key}")
+      Pupcap::Action.run_local("chmod 0600 #{provision_key}")
+      Pupcap::Action.run_local("chmod 0600 #{provision_key_pub}")
+    end
+  end
+end

data/lib/pupcap/action/prepare/ubuntu/precise.rb

@@ -0,0 +1,48 @@
+namespace :prepare do
+  namespace :ubuntu do
+    task :precise do
+      run("#{sudo} [ ! -f /root/.prepare_locale_ok ] && #{sudo} locale-gen && #{sudo} update-locale LANG=en_US.UTF8 && #{sudo} touch /root/.prepare_locale_ok || true")
+
+      tmp_list = "/tmp/sources.list.#{Time.now.to_i}"
+      sources_list = "/etc/apt/sources.list"
+
+      dist_upgraded = false
+      current_list = capture("md5sum #{sources_list}").split(" ")[0].to_s.strip
+      if current_list != "1f722c129d80782e4fa323a6a70dee59"
+        sudo("cp #{sources_list} #{sources_list}.back#{Time.now.to_i}")
+        put(precise_sources_list, tmp_list)
+        sudo("cp #{tmp_list} #{sources_list}")
+        sudo("chown root:root #{sources_list}")
+        sudo("chmod 0644 #{sources_list}")
+        run("rm -f #{tmp_list}")
+        sudo("apt-get -qy update > /dev/null")
+        if ENV['upgrade']
+          run("echo \"grub-pc hold\" | #{sudo} dpkg --set-selections")
+          sudo("apt-get -qy upgrade")
+          run("echo \"grub-pc install\" | #{sudo} dpkg --set-selections")
+          dist_upgraded = true
+        end
+      end
+
+      sudo("apt-get install -qy rsync wget rubygems vim git-core build-essential > /dev/null")
+      sudo("/usr/bin/gem install -q --no-ri --no-rdoc --version '~> 2.7.1' puppet")
+      sudo("reboot") if dist_upgraded
+    end
+  end
+end
+
+UBUNTU_PRECISE_SOURCES_LIST = <<-CODE
+#############################################################
+################### OFFICIAL UBUNTU REPOS ###################
+#############################################################
+
+###### Ubuntu Main Repos
+deb mirror://mirrors.ubuntu.com/mirrors.txt precise main
+deb-src mirror://mirrors.ubuntu.com/mirrors.txt precise main
+
+###### Ubuntu Update Repos
+deb mirror://mirrors.ubuntu.com/mirrors.txt precise-security main
+deb mirror://mirrors.ubuntu.com/mirrors.txt precise-updates main
+deb-src mirror://mirrors.ubuntu.com/mirrors.txt precise-security main
+deb-src mirror://mirrors.ubuntu.com/mirrors.txt precise-updates main
+CODE

data/lib/pupcap/cli.rb

@@ -0,0 +1,63 @@
+require 'optparse'
+require 'pupcap'
+
+class Pupcap::CLI
+  class << self
+    def start
+      get_action
+      require "pupcap/action/#{options[:action]}"
+      klass = Pupcap::Action.const_get(options[:action].capitalize)
+      inst = klass.new(options[:file], options)
+      inst.start
+    end
+
+    private
+      def get_action
+        options_parser
+
+        action = ARGV[1]
+        role   = ARGV[0]
+        if !action || !valid_actions.include?(action)
+          $stderr.puts "Please specify a action to run: #{valid_actions.join("|")}"
+          exit 1
+        end
+
+        if !role
+          $stderr.puts "Please specify a role to run"
+          exit 1
+        end
+
+        unless File.exists?(options[:file])
+          $stderr.puts "A file #{options[:file]} does not exists"
+          exit 1
+        end
+
+        options.merge!(:action => action, :role => role)
+      end
+
+      def valid_actions
+        @valid_actions ||= %w{ prepare cook }
+      end
+
+      def options
+        @options ||= {
+          :file => File.expand_path("Puppetfile")
+        }
+      end
+
+      def options_parser
+        @options_parser ||= OptionParser.new do |opts|
+          opts.banner = "Usage: #{File.basename($0)} [options] role #{valid_actions.join("|")}"
+
+          opts.on("-h", "--help", "Displays this help info") do
+            puts opts
+            exit 0
+          end
+
+          opts.on("-f", "--file FILE", "A recipe file to load") do |file|
+            options[:file] = File.expand_path(file)
+          end
+        end.parse!
+      end
+  end
+end

data/lib/pupcap/version.rb

@@ -0,0 +1,16 @@
+require 'scanf'
+module Pupcap
+
+  class Version
+
+    MAJOR = 0
+    MINOR = 0
+    PATCH = 1
+
+    def self.to_s
+      "#{MAJOR}.#{MINOR}.#{PATCH}"
+    end
+
+  end
+
+end

data/pupcap.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "pupcap/version"
+
+Gem::Specification.new do |s|
+
+  s.name        = "pupcap"
+  s.version     = Pupcap::Version.to_s
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Dmitry Galinsky"]
+  s.email       = ["dima.exe@gmail.com"]
+  s.homepage    = "http://github.com/dima-exe/pupcap"
+  s.summary     = %q{under development}
+  s.description = %q{under development description}
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<capistrano>, [">= 2.12.0"])
+    else
+      s.add_dependency(%q<capistrano>, [">= 2.12.0"])
+    end
+  else
+    s.add_dependency(%q<capistrano>, [">= 2.12.0"])
+  end
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: pupcap
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Dmitry Galinsky
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: capistrano
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.12.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.12.0
+description: under development description
+email:
+- dima.exe@gmail.com
+executables:
+- pupcap
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/pupcap
+- lib/pupcap.rb
+- lib/pupcap/action.rb
+- lib/pupcap/action/.prepare.rb.swp
+- lib/pupcap/action/cook.rb
+- lib/pupcap/action/cook/Capfile
+- lib/pupcap/action/prepare.rb
+- lib/pupcap/action/prepare/Capfile
+- lib/pupcap/action/prepare/ubuntu/precise.rb
+- lib/pupcap/cli.rb
+- lib/pupcap/version.rb
+- pupcap.gemspec
+homepage: http://github.com/dima-exe/pupcap
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: under development
+test_files: []
+has_rdoc: