pawnee 0.0.5

data/lib/pawnee/pawnee/templates/newgem/LICENSE.tt

@@ -0,0 +1,22 @@
+Copyright (c) <%=Time.now.year%> <%=config[:author]%>
+
+MIT License
+
+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/lib/pawnee/pawnee/templates/newgem/README.md.tt

@@ -0,0 +1,29 @@
+# <%=config[:constant_name]%>
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem '<%=config[:name]%>'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install <%=config[:name]%>
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/lib/pawnee/pawnee/templates/newgem/Rakefile.tt

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/lib/pawnee/pawnee/templates/newgem/bin/newgem.tt

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+
+require '<%= config[:name] %>'

data/lib/pawnee/pawnee/templates/newgem/gitignore.tt

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/lib/pawnee/pawnee/templates/newgem/lib/pawnee/newgem/base.rb.tt

@@ -0,0 +1,27 @@
+require "<%= config[:name] %>/version"
+require 'pawnee/base'
+
+<%- config[:constant_array].each_with_index do |c,i| -%>
+<%= '  '*i %>module <%= c %>
+<%- end -%>
+<%- j = config[:constant_array].size -%>
+<%= '  '*j %>class Base < Pawnee::Base
+<%= '  '*j %>  global_options do
+<%= '  '*j %>    method_option :servers, :type => :array, :required => true
+<%= '  '*j %>  end
+<%= '  '*j %>  desc "setup", 'setup on the destination server'
+<%= '  '*j %>  def setup
+<%= '  '*j %>    # Add your setup code here
+<%= '  '*j %>  end
+<%= '  '*j %>  
+<%= '  '*j %>  global_options do
+<%= '  '*j %>    method_option :servers, :type => :array, :required => true
+<%= '  '*j %>  end
+<%= '  '*j %>  desc "teardown", 'teardown on the destination server'
+<%= '  '*j %>  def teardown
+<%= '  '*j %>    # Add your teardown code here
+<%= '  '*j %>  end
+<%= '  '*j %>end
+<%- (config[:constant_array].size-1).downto(0) do |i| -%>
+<%= '  '*i %>end
+<%- end -%>

data/lib/pawnee/pawnee/templates/newgem/lib/pawnee/newgem/version.rb.tt

@@ -0,0 +1,7 @@
+<%- config[:constant_array].each_with_index do |c,i| -%>
+<%= '  '*i %>module <%= c %>
+<%- end -%>
+<%= '  '*config[:constant_array].size %>VERSION = "0.0.1"
+<%- (config[:constant_array].size-1).downto(0) do |i| -%>
+<%= '  '*i %>end
+<%- end -%>

data/lib/pawnee/pawnee/templates/newgem/lib/pawnee/newgem.rb.tt

@@ -0,0 +1,10 @@
+require 'pawnee/setup'
+Pawnee::Base.setup('pawnee-<%= config[:name] %>')
+
+<%- config[:constant_array].each_with_index do |c,i| -%>
+<%= '  '*i %>module <%= c %>
+<%- end -%>
+<%= '  '*config[:constant_array].size %># Your code goes here...
+<%- (config[:constant_array].size-1).downto(0) do |i| -%>
+<%= '  '*i %>end
+<%- end -%>

data/lib/pawnee/pawnee/templates/newgem/newgem.gemspec.tt

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/pawnee/<%=config[:name]%>/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = [<%=config[:author].inspect%>]
+  gem.email         = [<%=config[:email].inspect%>]
+  gem.description   = %q{TODO: Write a gem description}
+  gem.summary       = %q{TODO: Write a gem summary}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "pawnee-<%=config[:name]%>"
+  gem.require_paths = ["lib/pawnee"]
+  gem.add_runtime_dependency 'pawnee'
+  gem.version       = <%=config[:constant_name]%>::VERSION
+end

data/lib/pawnee/pawnee/version.rb

@@ -0,0 +1,3 @@
+module Pawnee
+  VERSION = "0.0.5"
+end

data/lib/pawnee/pawnee.rb

@@ -0,0 +1,7 @@
+require 'pawnee/railtie'
+require 'pawnee/base'
+require 'pawnee/cli'
+
+module Pawnee
+  
+end

data/pawnee.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/pawnee/pawnee/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Ryan Stout"]
+  gem.email         = ["ryanstout@gmail.com"]
+  gem.description   = %q{Pawnee attempts to build a better server provisioning system}
+  gem.summary       = %q{Better server provisioing}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "pawnee"
+  gem.require_paths = ["lib/pawnee"]
+  gem.add_runtime_dependency 'bundler'
+  gem.add_runtime_dependency 'thor', '>= 0.14.6'
+  gem.add_runtime_dependency 'net-ssh', '= 2.2.2'
+  gem.add_runtime_dependency 'net-sftp', '= 2.0.5'
+  gem.add_runtime_dependency 'thor-ssh'
+  gem.add_runtime_dependency 'activemodel', "~> 3.2.3"
+  gem.add_runtime_dependency 'activesupport', "~> 3.2.3"
+  gem.add_development_dependency "turn"
+  gem.add_development_dependency 'rspec', '~> 2.10'
+  gem.add_development_dependency 'vagrant', '= 1.0.3'
+  gem.add_development_dependency 'sahara', '>= 0.0.11'
+  gem.add_development_dependency 'rake', '~> 0.9'
+  gem.add_development_dependency 'rdoc', '~> 3.9'
+  gem.add_development_dependency 'simplecov', '~> 0.4'
+  gem.version       = Pawnee::VERSION
+end

data/spec/actions/compile_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+require 'pawnee/base'
+
+
+# NOTE: These are designed to run in order on vagrant
+describe "compile actions" do
+  before do
+    @base = Pawnee::Base.new
+    @base.destination_connection = VagrantManager.connect
+  end
+  
+  it 'should install a package' do
+    @base.install_packages('libpcre3', 'libpcre3-dev', 'libpcrecpp0', 'libssl-dev', 'zlib1g-dev')
+    @base.as_root do
+      @base.exec('rm /usr/local/sbin/nginx')
+    end
+    
+    @base.exec('which nginx').should == ''
+    
+    @base.compile('http://nginx.org/download/nginx-1.2.0.tar.gz', '/home/vagrant/nginx/', {configure: '--sbin-path=/usr/local/sbin', :bin_file => 'nginx'})
+    
+    @base.exec('which nginx').should_not == ''
+  end
+  
+end
+

data/spec/actions/package_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+require 'pawnee/base'
+
+
+# NOTE: These are designed to run in order on vagrant
+describe "package actions" do
+  before do
+    @base = Pawnee::Base.new
+    @base.destination_connection = VagrantManager.connect
+  end
+  
+  it 'should install a package' do
+    @base.install_package('memcached', '1.4.7-0.1ubuntu1')
+    @base.exec('which memcached').strip.should_not == ''
+  end
+
+  it 'should not reinstall a package' do
+    # Should be called once for the package version query
+    @base.destination_server.should_receive(:run).once.and_return('ii memcached 1.4.7-0.1ubuntu1')
+    @base.install_package('memcached', '1.4.7-0.1ubuntu1')
+  end
+  
+  it 'should return the version of a package installed' do
+    @base.installed_package_version('memcached').should == '1.4.7-0.1ubuntu1'
+  end
+  
+  it "should remove a package" do
+    @base.remove_package('memcached')
+    @base.exec('which memcached').should == ''
+  end
+
+  it "should not try to re-remove a package" do
+    @base.destination_server.should_receive(:run).once.and_return('')
+    @base.remove_package('memcached')
+  end
+  
+  it "should not return the package version after the package has been removed" do
+    @base.installed_package_version('memcached').should == nil
+  end
+  
+end

data/spec/actions/user_spec.rb

@@ -0,0 +1,54 @@
+require 'spec_helper'
+require 'pawnee/base'
+
+
+# NOTE: These are designed to run in order on vagrant
+describe "user actions" do
+  before do
+    @base = Pawnee::Base.new
+    @base.destination_connection = VagrantManager.connect
+  end
+  
+  def user_exists?(login)
+    _, _, exit_code, _ = @base.exec("id -u #{login}", true)
+    return exit_code == 0
+  end
+  
+  it "should create a user" do
+    @base.create_user(
+      name: 'Test Blue',
+      login: 'blue'
+    )
+    
+    user_exists?('blue').should == true
+  end
+  
+  it "should shouldn't recreate a user" do
+    @base.should_receive(:say_status).with(:user_exists, 'blue', :blue)
+    @base.create_user(
+      name: 'Test Blue',
+      login: 'blue'
+    )
+  end
+  
+  it "should modify a user" do
+    # @base.should_receive(:exec).any_number_of_times.ordered.with('id -u blue', true).and_return(["1002\n", '', 0, nil])
+    # @base.should_receive(:exec).any_number_of_times.ordered.with('id -u blue', true).and_return(["1002\n", '', 0, nil])
+    # @base.should_receive(:exec).any_number_of_times.ordered.with('id -g blue').and_return("1002\n")
+    # @base.should_receive(:exec).any_number_of_times.ordered.with('groups blue').and_return("blue: blue")
+    # @base.should_receive(:exec).once.ordered.with('useradd -c "comment" blue').and_return('')
+    @base.create_user(
+      name: 'Test Blue',
+      login: 'blue',
+      comment: 'comment'
+    )
+  end
+  
+  it "should delete a user" do
+    user_exists?('blue').should == true
+  
+    @base.delete_user('blue')
+  
+    user_exists?('blue').should == false
+  end
+end

data/spec/base_spec.rb

@@ -0,0 +1,167 @@
+require 'spec_helper'
+
+describe Pawnee::Base do
+  before do
+    @base = Pawnee::Base.new
+  end
+  
+  # after(:all) do
+  #   puts "Clear after base"
+  #   Pawnee::Base.instance_variable_set('@recipes', nil)
+  #   Pawnee::Blue.instance_eval { remove_const :Base }
+  #   Pawnee.instance_eval { remove_const :Blue }
+  #   Pawnee::Red.instance_eval { remove_const :Base }
+  #   Pawnee.instance_eval { remove_const :Red }
+  #   Pawnee.instance_eval { remove_const :CLI }
+  #   require 'pawnee/cli'
+  # end
+  
+  it "should respond to setup" do
+    @base.should respond_to(:setup)
+  end
+  
+  it "should respond to teardown" do
+    @base.should respond_to(:teardown)
+  end
+  
+  it 'should raise an exception when setup is called' do
+    -> { @base.setup() }.should raise_error(RuntimeError)
+  end
+  
+  it 'should raise an exception when teardown is called' do
+    -> { @base.teardown() }.should raise_error(RuntimeError)
+  end
+  
+
+  class Dep
+    attr_accessor :name
+    
+    def initialize(name)
+      @name = name
+    end
+  end
+  
+  it "should have no recipes before they are defined, but should add them after they are defined" do
+    Pawnee::Base.recipes.should == nil
+
+    # These will persist between tests
+    module Pawnee
+      module Red
+        class Base < Pawnee::Base
+          desc "setup", "setup"
+          def setup
+            puts "SETUP RED"
+          end
+        end
+      end
+
+      module Blue
+        class Base < Pawnee::Base
+          desc "setup", "setup"
+          def setup
+            puts "SETUP BLUE"
+          end
+        end
+      end
+    end
+
+    Pawnee::Base.recipes.should == [Pawnee::Red::Base, Pawnee::Blue::Base]
+  end
+  
+  it "should list the recipes in the order declared in the gem" do
+    deps = [Dep.new('blue'), Dep.new('red')]
+    Bundler.load.stub(:dependencies).and_return(deps)
+
+    Pawnee::Base.ordered_recipes.should == [Pawnee::Blue::Base, Pawnee::Red::Base]
+  
+    deps = [Dep.new('red'), Dep.new('blue')]
+    Bundler.load.stub(:dependencies).and_return(deps)
+    Pawnee::Base.instance_variable_set('@ordered_recipes', nil)
+  
+    Pawnee::Base.ordered_recipes.should == [Pawnee::Red::Base, Pawnee::Blue::Base]
+  end
+  
+  
+  it "should invoke roles based on the roles passed in" do
+    red = Pawnee::Red::Base.new([], {})
+    red.should_receive(:setup)
+    Pawnee::Red::Base.should_receive(:new).and_return(red)
+    
+    Pawnee::Blue::Base.should_not_receive(:new)
+    
+    Pawnee::Base.invoke_roles(:setup, ['red'])
+  end
+  
+  it "should invoke roles limiting to server roles" do
+    # Say we have a config file setup like this
+    config_options = {
+      'servers' => [
+        {'domain' => 'test1.com', 'roles' => ['red']}
+      ]
+    }
+
+    # But we want to pass in options with more domains
+    options = {
+      'servers' => [
+        {'domain' => 'test1.com', 'roles' => ['red']},
+        {'domain' => 'test2.com', 'roles' => ['blue', 'red']}
+      ]
+    }
+
+    Pawnee::Base.should_receive(:config_options).at_least(:once).and_return(config_options)
+    
+    red = Pawnee::Red::Base.new([], options)
+    red.should_receive(:setup).twice
+    Pawnee::Red::Base.should_receive(:new).with([], options).and_return(red)
+    
+    blue = Pawnee::Blue::Base.new([], options)
+    blue.should_receive(:setup).once
+    Pawnee::Blue::Base.should_receive(:new).with([], options).and_return(blue)
+    
+    Pawnee::Base.invoke_roles(:setup, ['red', 'blue'], options)
+
+  end
+  
+  it "should invoke for each server when doing an individual recipe's setup task" do
+    options = {
+      'servers' => [
+        {'domain' => 'test1.com', 'roles' => ['red']},
+        {'domain' => 'test2.com', 'roles' => ['red']}
+      ]
+    }
+    
+    Pawnee::Base.should_receive(:config_options).at_least(:once).and_return(options)
+    
+    red = Pawnee::Red::Base.new([], options)
+    red.should_receive(:setup).twice
+    Pawnee::Red::Base.should_receive(:new).and_return(red)
+
+    Pawnee::Blue::Base.should_not_receive(:new)
+    
+    Pawnee::CLI.start(['red', 'setup'])
+    
+  end
+  
+  it "should call all setup methods when called with setup from the command line" do
+    options = {
+      'servers' => [
+        {'domain' => 'test1.com', 'roles' => ['red']},
+        {'domain' => 'test2.com', 'roles' => ['red', 'blue']}
+      ]
+    }
+    
+    Pawnee::Base.should_receive(:config_options).at_least(:once).and_return(options)
+    
+    red = Pawnee::Red::Base.new([], options)
+    red.should_receive(:setup).twice
+    Pawnee::Red::Base.should_receive(:new).and_return(red)
+
+    blue = Pawnee::Blue::Base.new([], options)
+    blue.should_receive(:setup).once
+    Pawnee::Blue::Base.should_receive(:new).with([], options).and_return(blue)
+    
+    Pawnee::CLI.start(['setup'])
+  end
+
+  
+end

data/spec/spec_helper.rb

@@ -0,0 +1,24 @@
+require 'rspec'
+
+libdir = File.expand_path(File.dirname(__FILE__) + "/../lib/pawnee")
+$LOAD_PATH.unshift(libdir) unless $LOAD_PATH.include?(libdir)
+
+
+require 'simplecov'
+SimpleCov.start do
+  add_group 'Libraries', 'lib/pawnee/pawnee'
+  add_group 'Specs', 'spec'
+end
+
+
+require 'pawnee/base'
+require 'vagrant/vagrant'
+
+
+RSpec.configure do |config|
+  config.after(:suite) do
+    # Rollback the server
+    puts "Roll back test server"
+    `cd spec/vagrant/ ; BUNDLE_GEMFILE=../../Gemfile bundle exec vagrant sandbox rollback`
+  end
+end

data/spec/vagrant/.vagrant

@@ -0,0 +1 @@
+{"active":{"default":"21e9da4d-756c-4cf9-8cae-caf4f70820f7"}}

data/spec/vagrant/Vagrantfile

@@ -0,0 +1,99 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+Vagrant::Config.run do |config|
+  # All Vagrant configuration is done here. The most common configuration
+  # options are documented and commented below. For a complete reference,
+  # please see the online documentation at vagrantup.com.
+
+  # Every Vagrant virtual environment requires a box to build off of.
+  config.vm.box = "ubuntu11"
+
+  # The url from where the 'config.vm.box' box will be fetched if it
+  # doesn't already exist on the user's system.
+  # config.vm.box_url = "http://domain.com/path/to/above.box"
+
+  # Boot with a GUI so you can see the screen. (Default is headless)
+  # config.vm.boot_mode = :gui
+
+  # Assign this VM to a host-only network IP, allowing you to access it
+  # via the IP. Host-only networks can talk to the host machine as well as
+  # any other machines on the same network, but cannot be accessed (through this
+  # network interface) by any external networks.
+  # config.vm.network :hostonly, "192.168.33.10"
+
+  # Assign this VM to a bridged network, allowing you to connect directly to a
+  # network using the host's network device. This makes the VM appear as another
+  # physical device on your network.
+  # config.vm.network :bridged
+
+  # Forward a port from the guest to the host, which allows for outside
+  # computers to access the VM, whereas host only networking does not.
+  # config.vm.forward_port 80, 8080
+
+  # Share an additional folder to the guest VM. The first argument is
+  # an identifier, the second is the path on the guest to mount the
+  # folder, and the third is the path on the host to the actual folder.
+  # config.vm.share_folder "v-data", "/vagrant_data", "../data"
+
+  # Enable provisioning with Puppet stand alone.  Puppet manifests
+  # are contained in a directory path relative to this Vagrantfile.
+  # You will need to create the manifests directory and a manifest in
+  # the file ubuntu11.pp in the manifests_path directory.
+  #
+  # An example Puppet manifest to provision the message of the day:
+  #
+  # # group { "puppet":
+  # #   ensure => "present",
+  # # }
+  # #
+  # # File { owner => 0, group => 0, mode => 0644 }
+  # #
+  # # file { '/etc/motd':
+  # #   content => "Welcome to your Vagrant-built virtual machine!
+  # #               Managed by Puppet.\n"
+  # # }
+  #
+  # config.vm.provision :puppet do |puppet|
+  #   puppet.manifests_path = "manifests"
+  #   puppet.manifest_file  = "ubuntu11.pp"
+  # end
+
+  # Enable provisioning with chef solo, specifying a cookbooks path, roles
+  # path, and data_bags path (all relative to this Vagrantfile), and adding 
+  # some recipes and/or roles.
+  #
+  # config.vm.provision :chef_solo do |chef|
+  #   chef.cookbooks_path = "../my-recipes/cookbooks"
+  #   chef.roles_path = "../my-recipes/roles"
+  #   chef.data_bags_path = "../my-recipes/data_bags"
+  #   chef.add_recipe "mysql"
+  #   chef.add_role "web"
+  #
+  #   # You may also specify custom JSON attributes:
+  #   chef.json = { :mysql_password => "foo" }
+  # end
+
+  # Enable provisioning with chef server, specifying the chef server URL,
+  # and the path to the validation key (relative to this Vagrantfile).
+  #
+  # The Opscode Platform uses HTTPS. Substitute your organization for
+  # ORGNAME in the URL and validation key.
+  #
+  # If you have your own Chef Server, use the appropriate URL, which may be
+  # HTTP instead of HTTPS depending on your configuration. Also change the
+  # validation key to validation.pem.
+  #
+  # config.vm.provision :chef_client do |chef|
+  #   chef.chef_server_url = "https://api.opscode.com/organizations/ORGNAME"
+  #   chef.validation_key_path = "ORGNAME-validator.pem"
+  # end
+  #
+  # If you're using the Opscode platform, your validator client is
+  # ORGNAME-validator, replacing ORGNAME with your organization name.
+  #
+  # IF you have your own Chef Server, the default validation client name is
+  # chef-validator, unless you changed the configuration.
+  #
+  #   chef.validation_client_name = "ORGNAME-validator"
+end

data/spec/vagrant/vagrant.rb

@@ -0,0 +1,30 @@
+require 'vagrant'
+require 'net/ssh'
+
+# Return an ssh connection to the vagrant server
+class VagrantManager
+  # Returns a connected Net::SSH connection
+  def self.connect
+    @vm = Vagrant::Environment.new(:cwd => File.dirname(__FILE__)).primary_vm
+
+    ssh_info = @vm.ssh.info
+
+    # Build the options we'll use to initiate the connection via Net::SSH
+    opts = {
+      :port                  => ssh_info[:port],
+      :keys                  => [ssh_info[:private_key_path]],
+      :keys_only             => true,
+      :user_known_hosts_file => [],
+      :paranoid              => false,
+      :config                => false,
+      :forward_agent         => ssh_info[:forward_agent]
+    }
+
+    # Check that the private key permissions are valid
+    @vm.ssh.check_key_permissions(ssh_info[:private_key_path])
+
+    # Connect to SSH, giving it a few tries
+    return Net::SSH.start(ssh_info[:host], ssh_info[:username], opts)
+  end
+end
+