simpleworker 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8fe17f8a2b92b4c0ce10fd235a2a2b4ea85e5a70
+  data.tar.gz: 9175ffcdb89eda2144fad564145cbe7c8a009364
+SHA512:
+  metadata.gz: 560792ce1d89592887af9bbd0c069f425e33d4be05438c253d5c1189053870454e4d8db553351de66ea72eba7c9edb617a5d1e4563fead91762dce762b748faf
+  data.tar.gz: 5fc38d0395d77ea5480859edc6c509da306cc2fa62830932a531707f4c4dbce4529ed628f141ca6d71b68c9447ee60465dca21fe77275f0bebe7d0e0d334e887

data/.rspec

@@ -0,0 +1 @@
+--color -fs

data/Gemfile

@@ -0,0 +1,3 @@
+source 'http://rubygems.org'
+
+gemspec

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2014 
+
+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,41 @@
+SimpleWorker
+============
+
+Distribute automation scripts on multiple machines.
+
+Usage
+=====
+
+Create `simpleworker.yml` in the projects working directory.  Ruby must be
+setup on the remote host such that it is available in the login shell.
+
+```yml
+---
+workers:
+  - type: ssh                      # type of worker
+    directory: /tmp/foobar         # directory on remote host
+    user: bill                     # user on remote host
+    host: my.remote.host.com       # remote host name
+```
+
+```ruby
+require 'simpleworker'
+
+# execute with workers configured in $PWD/simpleworker.yml
+SimpleWorker::Runnner.run "my-command"
+
+# execute with a special worker configuration
+SimpleWorker::Runner.load("my-worker-config.yml").run "my-command"
+```
+
+Note on Patches/Pull Requests
+=============================
+
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new
+
+task :default => :spec

data/lib/simpleworker/abstract_worker.rb

@@ -0,0 +1,42 @@
+
+module SimpleWorker::AbstractWorker
+
+  attr_accessor :script, :cmd
+
+  def start
+    if script.kind_of? Array
+      @process = ChildProcess.build *script
+    else
+      @process = ChildProcess.build script
+    end
+
+    @process.io.inherit!
+
+    set_process_env
+
+    @process.start
+  end
+
+  def wait
+    @process.wait
+  end
+
+  def stop
+    @process.stop
+  end
+
+  def env
+    @env
+  end
+
+  private
+
+  def set_process_env
+    env['cmd'] ||= cmd
+
+    env.each do |key, value|
+      @process.environment[key] = value
+    end
+  end
+
+end

data/lib/simpleworker/bash/simple-localworker

@@ -0,0 +1,8 @@
+#!/bin/bash
+
+echo "Ruby Version $(ruby -v)"
+echo "Which ruby: $(which ruby)"
+
+bundle install
+bundle exec ${cmd}
+

data/lib/simpleworker/bash/ssh-remoteworker

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+rsync -ar --delete  -e ssh "${PWD}/" "${user}@${host}:${directory}"
+
+remote_script=$(cat << EOF
+
+cd ${directory};
+echo "Ruby Version: \$(ruby -v)";
+echo "Which ruby: \$(which ruby)";
+
+echo "PWD: \${PWD}";
+
+bundle install;
+bundle exec ${cmd};
+
+EOF
+)
+
+ssh "${user}@${host}" "bash -lc '${remote_script}'"
+
+rsync -ar -e ssh "${user}@${host}:${directory}/" "${PWD}/"

data/lib/simpleworker/local_worker.rb

@@ -0,0 +1,18 @@
+
+module SimpleWorker
+  class LocalWorker
+    include AbstractWorker
+
+    def initialize
+      @script = %W[bash #{File.expand_path(File.dirname(__FILE__))}/bash/simple-localworker]
+      @env = {}
+    end
+
+    def self.create(config = {})
+      worker = new
+      worker.script = config['script'] if config.has_key? 'script'
+      worker.cmd = config['cmd'] if config.has_key? 'cmd'
+      worker
+    end
+  end
+end

data/lib/simpleworker/runner.rb

@@ -0,0 +1,59 @@
+
+module SimpleWorker
+  class Runner
+
+    def initialize
+      @workers = []
+    end
+
+    def self.run(cmd = 'rake')
+      new.load.run(cmd)
+    end
+
+    def self.load(config = 'simpleworker.yml')
+      new.load(config)
+    end
+
+    def load(config = 'simpleworker.yml')
+      data = YAML.load( IO.read(config) )
+      data['workers'].each do |config|
+        case config['type']
+        when 'ssh'
+          @workers << SshWorker.create(config)
+        else
+          @workers << LocalWorker.create(config)
+        end
+      end
+
+      self
+    end
+
+    def run(cmd = 'rake')
+      @workers.each do |worker|
+        worker.cmd = cmd
+      end
+      start
+      process
+      stop
+    rescue Interrupt
+      stop
+    rescue StandardError => e
+      stop
+      raise e
+    end
+
+    private
+
+    def start
+      @workers.each &:start
+    end
+
+    def process
+      @workers.each &:wait
+    end
+
+    def stop
+      @workers.each &:stop
+    end
+  end
+end

data/lib/simpleworker/ssh_worker.rb

@@ -0,0 +1,33 @@
+
+module SimpleWorker
+  class SshWorker
+    include AbstractWorker
+
+    attr_accessor :directory, :user, :host
+
+    def initialize
+      @script = %W[bash #{File.expand_path(File.dirname(__FILE__))}/bash/ssh-remoteworker]
+      @env = {}
+    end
+
+    def self.create(config = {})
+      worker = new
+      worker.script = config['script'] if config.has_key? 'script'
+      worker.cmd = config['cmd'] if config.has_key? 'cmd'
+      worker.user = config['user'] if config.has_key? 'user'
+      worker.host = config['host'] if config.has_key? 'host'
+      worker.directory = config['directory'] if config.has_key? 'directory'
+      worker
+    end
+
+    private
+
+    def set_process_env
+      env['user'] = user || `whoami`
+      env['host'] = host || 'localhost'
+      env['directory'] = directory || Dir.pwd
+
+      super
+    end
+  end
+end

data/lib/simpleworker/version.rb

@@ -0,0 +1,4 @@
+
+module SimpleWorker
+  VERSION = '0.0.1'
+end

data/lib/simpleworker.rb

@@ -0,0 +1,11 @@
+
+require 'childprocess'
+require 'yaml'
+
+module SimpleWorker
+end
+
+require 'simpleworker/runner'
+require 'simpleworker/abstract_worker'
+require 'simpleworker/local_worker'
+require 'simpleworker/ssh_worker'

data/simpleworker.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "simpleworker/version"
+
+Gem::Specification.new do |s|
+  s.name        = "simpleworker"
+  s.version     = SimpleWorker::VERSION
+  s.licenses    = ['MIT']
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Jason Gowan"]
+  s.email       = ["gowanjason@gmail.com"]
+  s.homepage    = "https://github.com/jesg/simpleworker"
+  s.summary     = %q{Distribute ruby scripts on multiple machines}
+  s.description = %q{Distribute ruby scripts on multiple machines for automated testing.}
+
+  s.rubyforge_project = "simpleworker"
+
+  s.add_dependency 'childprocess', '>= 0.5.3'
+
+  s.add_development_dependency "rspec", "~> 2.5"
+  s.add_development_dependency "rake", "~> 0.9.2"
+
+  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"]
+end

data/spec/simpleworker/local_worker_spec.rb

@@ -0,0 +1,33 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+module SimpleWorker
+  describe LocalWorker do
+    let(:mock_env)     { double(Hash) }
+    let(:mock_process) { double(ChildProcess, :start => nil, :stop => nil, :wait => nil, :io => double('io').as_null_object, :environment => mock_env) }
+
+    it 'can start local worker with defaults' do
+      ChildProcess.should_receive(:build).with('bash', end_with('simple-localworker')).and_return(mock_process)
+      mock_process.should_receive(:start)
+      mock_env.should_receive(:[]=).with('cmd', 'rake')
+
+      worker = LocalWorker.new
+      worker.cmd = 'rake'
+      worker.start
+    end
+
+    it 'can start local worker with customization' do
+      ChildProcess.should_receive(:build).with('bash', end_with('simple-localworker')).and_return(mock_process)
+      mock_process.should_receive(:start)
+      mock_env.should_receive(:[]=).with('cmd', 'cuke')
+      mock_process.should_receive(:wait)
+      mock_process.should_receive(:stop)
+
+      worker = LocalWorker.new
+      worker.cmd = 'cuke'
+      worker.start
+      worker.wait
+      worker.stop
+    end
+  end
+end
+

data/spec/simpleworker/runner_spec.rb

@@ -0,0 +1,19 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+module SimpleWorker
+  describe Runner do
+    let(:mock_worker) { double(AbstractWorker, :start => nil, :wait => nil, :stop => nil) }
+
+    it 'can run workers' do
+      mock_worker.should_receive(:start)
+      mock_worker.should_receive(:cmd=).with('cuke')
+      mock_worker.should_receive(:wait)
+      mock_worker.should_receive(:stop)
+      IO.should_receive(:read).with('simpleworker.yml').and_return('hi')
+      YAML.should_receive(:load).with('hi').and_return({'workers' => [{'type' => 'ssh'}]})
+      SshWorker.should_receive(:create).and_return(mock_worker)
+
+      Runner.run 'cuke'
+    end
+  end
+end

data/spec/simpleworker/ssh_worker_spec.rb

@@ -0,0 +1,43 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+module SimpleWorker
+  describe SshWorker do
+    let(:mock_env)     { double(Hash) }
+    let(:mock_process) { double(ChildProcess, :start => nil, :stop => nil, :wait => nil, :io => double('io').as_null_object, :environment => mock_env) }
+
+    it 'can start ssh worker with defaults' do
+      ChildProcess.should_receive(:build).with('bash', end_with('ssh-remoteworker')).and_return(mock_process)
+      mock_process.should_receive(:start)
+      mock_env.should_receive(:[]=).once.with('cmd', 'rake')
+      mock_env.should_receive(:[]=).once.with('user', `whoami`)
+      mock_env.should_receive(:[]=).once.with('host', 'localhost')
+      mock_env.should_receive(:[]=).once.with('directory', Dir.pwd)
+
+      worker = SshWorker.new
+      worker.cmd = 'rake'
+      worker.start
+    end
+
+    it 'can start ssh worker with customization' do
+      ChildProcess.should_receive(:build).with('bash', end_with('ssh-remoteworker')).and_return(mock_process)
+      mock_process.should_receive(:start)
+      mock_env.should_receive(:[]=).once.with('cmd', 'cuke')
+      mock_env.should_receive(:[]=).once.with('user', 'bob')
+      mock_env.should_receive(:[]=).once.with('host', 'foo')
+      mock_env.should_receive(:[]=).once.with('directory', 'bar')
+
+      mock_process.should_receive(:wait)
+      mock_process.should_receive(:stop)
+
+      worker = SshWorker.new
+      worker.cmd = 'cuke'
+      worker.user = 'bob'
+      worker.host = 'foo'
+      worker.directory = 'bar'
+      worker.start
+      worker.wait
+      worker.stop
+    end
+  end
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,3 @@
+
+$LOAD_PATH.unshift File.expand_path("../lib")
+require 'simpleworker'

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: simpleworker
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Jason Gowan
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-08-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: childprocess
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.5.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.5.3
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+description: Distribute ruby scripts on multiple machines for automated testing.
+email:
+- gowanjason@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/simpleworker.rb
+- lib/simpleworker/abstract_worker.rb
+- lib/simpleworker/bash/simple-localworker
+- lib/simpleworker/bash/ssh-remoteworker
+- lib/simpleworker/local_worker.rb
+- lib/simpleworker/runner.rb
+- lib/simpleworker/ssh_worker.rb
+- lib/simpleworker/version.rb
+- simpleworker.gemspec
+- spec/simpleworker/local_worker_spec.rb
+- spec/simpleworker/runner_spec.rb
+- spec/simpleworker/ssh_worker_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/jesg/simpleworker
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: simpleworker
+rubygems_version: 2.2.0
+signing_key: 
+specification_version: 4
+summary: Distribute ruby scripts on multiple machines
+test_files:
+- spec/simpleworker/local_worker_spec.rb
+- spec/simpleworker/runner_spec.rb
+- spec/simpleworker/ssh_worker_spec.rb
+- spec/spec_helper.rb