minionizer 0.0.1 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd7dc40fdbe374f013b0e9e807c3bc974215a951
-  data.tar.gz: ddcb859dd3f2d985341553cf923cc4de4422b19d
+  metadata.gz: 82bef054c00cc362c8c6bde7b0557b1ec7ea8966
+  data.tar.gz: 496df270bcb0c2c4fe65d3bb291ef5d9d9673e69
 SHA512:
-  metadata.gz: 2556cc919d6e3dc13fafd41406726bdd8bc640c5b27d75ca67614a4b14a06c9d465f08558e3943caa6daf0752e1a49d4044672996cfc6206db660336a922ed6f
-  data.tar.gz: b73ba6895d71e3b8d3aa9bd37dc3c01368bd160f71383cb09feb49cb7112738346620d283b5448dbcb4eb024184eee9bce9322c9c14ec0afa3f022f9edfb200d
+  metadata.gz: b437e19e86a6bc97c546ba32193a77812c0710c50b375927ddd0c49b129fab0e96b4b91ab9e6456a409b10061e7ca04398220bb125c4a85a7c56382d06d67621
+  data.tar.gz: ef1cc521975e07f8fbee54cd3e3f5415d0610f2dfb6bc6cb5a0f5b79725d05942f4c9f1baf8e915729ecb6a9d32a9bdf111c936f233f7c76dd9abf1797668588

data/.coveralls.yml

@@ -0,0 +1 @@
+service_name: 'travis-ci'

data/.gitignore

@@ -1,3 +1,4 @@
 *.swp
 test/.vagrant
+coverage
 

data/.travis.yml

@@ -0,0 +1,6 @@
+language: ruby
+rvm:
+  - "2.0.0"
+  - "2.1.1"
+# uncomment this line if your project needs to run something other than `rake`:
+# # script: bundle exec rspec spec

data/Gemfile

@@ -1,2 +1,4 @@
 source 'http://rubygems.org'
 gemspec
+
+gem 'coveralls', require: false

data/Gemfile.lock

@@ -2,35 +2,57 @@ PATH
   remote: .
   specs:
     minionizer (0.0.1)
-      activesupport
-      net-ssh
+      activesupport (~> 4.1)
+      net-ssh (~> 2.9)
 
 GEM
   remote: http://rubygems.org/
   specs:
-    activesupport (4.0.4)
+    activesupport (4.1.1)
       i18n (~> 0.6, >= 0.6.9)
-      minitest (~> 4.2)
-      multi_json (~> 1.3)
+      json (~> 1.7, >= 1.7.7)
+      minitest (~> 5.1)
       thread_safe (~> 0.1)
-      tzinfo (~> 0.3.37)
-    atomic (1.1.16)
+      tzinfo (~> 1.1)
+    coveralls (0.7.0)
+      multi_json (~> 1.3)
+      rest-client
+      simplecov (>= 0.7)
+      term-ansicolor
+      thor
+    docile (1.1.3)
     fakefs (0.5.2)
     i18n (0.6.9)
+    json (1.8.1)
     metaclass (0.0.4)
-    minitest (4.7.5)
+    mime-types (2.2)
+    minitest (5.3.3)
     mocha (1.0.0)
       metaclass (~> 0.0.1)
-    multi_json (1.9.2)
-    net-ssh (2.8.0)
-    thread_safe (0.3.1)
-      atomic (>= 1.1.7, < 2)
-    tzinfo (0.3.39)
+    multi_json (1.10.0)
+    net-ssh (2.9.0)
+    rake (10.3.1)
+    rest-client (1.6.7)
+      mime-types (>= 1.16)
+    simplecov (0.8.2)
+      docile (~> 1.1.0)
+      multi_json
+      simplecov-html (~> 0.8.0)
+    simplecov-html (0.8.0)
+    term-ansicolor (1.3.0)
+      tins (~> 1.0)
+    thor (0.19.1)
+    thread_safe (0.3.3)
+    tins (1.1.0)
+    tzinfo (1.1.0)
+      thread_safe (~> 0.1)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  fakefs
+  coveralls
+  fakefs (~> 0.5)
   minionizer!
-  mocha
+  mocha (~> 1.0)
+  rake (~> 10.3)

data/README.md

@@ -1,4 +1,8 @@
 [![Code Climate](https://codeclimate.com/github/jsgarvin/minionizer.png)](https://codeclimate.com/github/jsgarvin/minionizer)
+[![Build Status](https://travis-ci.org/jsgarvin/minionizer.svg?branch=master)](https://travis-ci.org/jsgarvin/minionizer)
+[![Coverage Status](https://coveralls.io/repos/jsgarvin/minionizer/badge.png)](https://coveralls.io/r/jsgarvin/minionizer)
+[![Gem Version](https://badge.fury.io/rb/minionizer.svg)](http://badge.fury.io/rb/minionizer)
+[![Dependency Status](https://gemnasium.com/jsgarvin/minionizer.svg)](https://gemnasium.com/jsgarvin/minionizer)
 
 # Minionizer
 

data/Rakefile

@@ -18,6 +18,9 @@ namespace :test do
       relay_output(vagrant_command(:up))
       unless snapshot_plugin_installed?
         relay_output(vagrant_command('plugin install vagrant-vbox-snapshot'))
+      end
+      unless test_snapshot_exists?
+        sleep 5
         relay_output(vagrant_command('snapshot take blank-test-slate'))
       end
     end
@@ -36,6 +39,10 @@ def snapshot_plugin_installed?
     Gem::Version.new(vagrant_plugins['vagrant-vbox-snapshot']) >= Gem::Version.new('0.0.4')
 end
 
+def test_snapshot_exists?
+  vagrant_snapshots.include?('blank-test-slate')
+end
+
 def vagrant_plugins
   Hash.new.tap do |hash|
     `cd #{vagrant_path}; vagrant plugin list`.split("\n").each do |plugin_string|
@@ -46,6 +53,16 @@ def vagrant_plugins
   end
 end
 
+def vagrant_snapshots
+  Array.new.tap do |snapshots|
+    `cd #{vagrant_path}; vagrant snapshot list`.split("\n").each do |snapshot_string|
+       if snapshot_string.match(/Name\: ([^\(]+)/)
+         snapshots << $1 
+       end
+    end
+  end
+end
+
 def vagrant_path
   File.expand_path('../test', __FILE__)
 end

data/lib/core/file_injection.rb

@@ -1,19 +1,22 @@
 module Minionizer
-  class FileInjection
-    attr_reader :session
+  class FileInjection < TaskTemplate
 
-    def initialize(session)
-      @session = session
-    end
-
-    def inject(source, target)
-      session.exec("echo '#{contents_from(source)}' > #{target}")
+    def call
+      session.exec("mkdir --parents #{target_directory}")
+      session.exec("echo '#{contents_from(source_path)}' > #{target_path}")
+      session.exec("chmod #{mode} #{target_path}") if respond_to?(:mode)
+      session.exec("chown #{owner} #{target_path}") if respond_to?(:owner)
+      session.exec("chgrp #{group} #{target_path}") if respond_to?(:group)
     end
 
     #######
     private
     #######
 
+    def target_directory
+      File.dirname(target_path)
+    end
+
     def contents_from(source)
       File.open(source).read.strip
     end

data/lib/core/folder_creation.rb

@@ -0,0 +1,12 @@
+module Minionizer
+  class FolderCreation < TaskTemplate
+
+    def call
+      session.exec("mkdir --parents #{path}")
+      session.exec("chmod #{mode} #{path}") if respond_to?(:mode)
+      session.exec("chown #{owner} #{path}") if respond_to?(:owner)
+      session.exec("chgrp #{group} #{path}") if respond_to?(:group)
+    end
+
+  end
+end

data/lib/core/public_ssh_key_injection.rb

@@ -0,0 +1,40 @@
+module Minionizer
+  class PublicSshKeyInjection < TaskTemplate
+
+    def call
+      file_injection.call
+    ensure
+      temp_file.unlink
+    end
+
+    #######
+    private
+    #######
+
+    def file_injection
+      @file_injection ||= file_injection_creator.new(session, file_injection_options)
+    end
+
+    def file_injection_creator
+      options[:file_injection_creator] ||= FileInjection
+    end
+
+    def file_injection_options
+      {
+        source_path: temp_file.path,
+        target_path: "~#{target_username}/.ssh/authorized_keys",
+        owner: target_username,
+        group: target_username
+      }
+    end
+
+    def temp_file
+      @temp_file ||= Tempfile.new('MinionizerPublicKeys').tap do |temp_file|
+        Dir.glob("data/public_keys/*.pubkey") do |key_file|
+          temp_file.puts File.open(key_file).read
+        end
+      end
+    end
+
+  end
+end

data/lib/core/task_template.rb

@@ -0,0 +1,25 @@
+module Minionizer
+  class TaskTemplate
+    attr_reader :session, :options
+
+    def initialize(session, options = {})
+      @session = session
+      @options = options.with_indifferent_access
+    end
+
+    def method_missing(method_name, *arguments, &block)
+      if options.key?(method_name)
+        options[method_name]
+      else
+        super
+      end
+    end
+
+    def respond_to?(method_name, include_private = false)
+      options.key?(method_name) || super
+    end
+
+  end
+end
+
+

data/lib/core/user_creation.rb

@@ -0,0 +1,18 @@
+module Minionizer
+  class UserCreation < TaskTemplate
+
+    def call
+      unless user_exists?
+        session.exec("adduser --disabled-password --gecos '#{name}' #{username}")
+      end
+    end
+
+    private
+
+    def user_exists?
+      session.exec("id #{username}")
+    rescue CommandExecution::CommandError
+      return false
+    end
+  end
+end

data/lib/minionizer.rb

@@ -1,6 +1,9 @@
+require 'active_support/core_ext/hash/indifferent_access'
 require 'active_support/inflector'
+require 'erb'
 require 'net/ssh'
 require 'singleton'
 require 'yaml'
 
+require_relative 'core/task_template'
 Dir[File.dirname(__FILE__) + '/**/*.rb'].each { |file| require file }

data/lib/minionizer/command_execution.rb

@@ -0,0 +1,60 @@
+module Minionizer
+
+  class CommandExecution
+    class CommandError < StandardError; end
+    class InvocationError < StandardError; end
+
+    attr_reader :connection, :command
+
+    def initialize(connection, command)
+      @connection = connection
+      @command = command
+    end
+
+    def call
+      execute_command
+      check_exit_code
+      return results
+    end
+
+    private
+
+    def execute_command
+      connection.open_channel do |channel|
+        execute_command_inside_channel(channel)
+      end
+      connection.loop
+    end
+
+    def execute_command_inside_channel(channel)
+      channel.exec(command) do |_, success|
+        if success
+          compile_results(channel)
+        else
+          raise InvocationError.new("Failed to invoke command: #{command} ")
+        end
+      end
+    end
+
+    def check_exit_code
+      if exit_failure?
+        raise CommandError.new("\"#{command}\" returned exit code #{results[:exit_code]}/#{results[:exit_signal]}/#{results[:stderr]}")
+      end
+    end
+
+    def exit_failure?
+      results[:exit_code].to_i != 0
+    end
+
+    def results
+      @results ||= {stdout: '', stderr: ''}
+    end
+
+    def compile_results(channel)
+      channel.on_data { |_, data| results[:stdout] += data.strip }
+      channel.on_extended_data { |_, data| results[:stderr] += data.to_s }
+      channel.on_request('exit-status') { |_,data| results[:exit_code] = data.read_long }
+      channel.on_request('exit-signal') { |_,data| results[:exit_signal] = data.read_string }
+    end
+  end
+end

data/lib/minionizer/session.rb

@@ -1,39 +1,54 @@
 module Minionizer
   class Session
-    attr_reader :fqdn, :username, :password, :connector
+    attr_reader :fqdn, :username, :password, :connector, :command_executor
 
-    def initialize(fqdn, credentials, connector = Net::SSH)
+    def initialize(fqdn, credentials, connector = Net::SSH, command_executor = CommandExecution)
       @fqdn = fqdn
       @username = credentials['username']
       @password = credentials['password']
       @connector = connector
+      @command_executor = command_executor
     end
 
-    def exec(arg)
-      if arg.is_a?(Array)
-        arg.map { |command| exec_single_command(command) }
+    def sudo(*commands)
+      @with_sudo = true
+      if commands.any?
+        return exec(*commands)
       else
-        exec_single_command(arg)
+        yield self
       end
+    ensure
+      @with_sudo = false
+    end
+
+    def exec(*commands)
+      results = commands.map { |command| execution(command).call }
+      results.length == 1 ? results.first : results
     end
 
     #######
     private
     #######
 
-    def exec_single_command(command)
-      connection.exec(command) do |channel, stream, output|
-        if stream == :stdout
-          return output.strip
-        else
-          raise StandardError.new(output)
-        end
+    def execution(command)
+      if with_sudo?
+        command_executor.new(connection, prefix_sudo(command))
+      else
+        command_executor.new(connection, command)
       end
-      connection.loop
     end
 
     def connection
       @connection ||= connector.start(fqdn, username, password: password)
     end
+
+    def prefix_sudo(command)
+      %Q{sudo bash -c "#{command}"}
+    end
+
+    def with_sudo?
+      @with_sudo
+    end
+
   end
 end

data/lib/minionizer/version.rb

@@ -1,3 +1,3 @@
 module Minionizer
-  VERSION = '0.0.1'
+  VERSION = '0.1.0'
 end

data/minionizer.gemspec

@@ -10,11 +10,12 @@ Gem::Specification.new do |s|
   s.summary = %q{Simple server provisioning and management.}
   s.description = %q{Minionizer aims to be a light weight server provisioning tool without bloat or steep learning curves.}
 
-  s.add_dependency('activesupport')
-  s.add_dependency('net-ssh')
+  s.add_dependency('activesupport', '~> 4.1')
+  s.add_dependency('net-ssh', '~> 2.9')
 
-  s.add_development_dependency('fakefs')
-  s.add_development_dependency('mocha')
+  s.add_development_dependency('fakefs', '~> 0.5')
+  s.add_development_dependency('mocha', '~> 1.0')
+  s.add_development_dependency('rake', '~> 10.3')
 
   s.files = `git ls-files`.split("\n")
   s.test_files = `git ls-files -- test/*`.split("\n")