ronin-wrapper 0.0.9 → 0.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2c3c00f54efdb0caece795e59b3647d555f25ea3
-  data.tar.gz: 8cb9cae2f49b2bc0453e32fefdc5f649f34dac24
+  metadata.gz: 11143e25ba860d375511daa655d3bce4518e78b4
+  data.tar.gz: 3bd2050c5d6cd7250518d2de9ddb542fc08a26fa
 SHA512:
-  metadata.gz: 2f1fb190ffc1853a1341fcc19741b563898e82691c493b3c222d5713e9e3bece065c7d04204fc9212d6719ae43a512f6eff06ae5c8cdc19ed8d31f32221f5cc1
-  data.tar.gz: 7e4bcdf75d4b40262809e7958a2f7a0fc7e10bdd37d44cd3b24ba47b8bd57d1c4ae700d43278d89d3db293c43192d47789633762f7aa3532e2c2967cae15bb03
+  metadata.gz: b280ea075de35614afea3a9cfd4b3b516794bc43660abb34fa86312cf251c6db736e8ce08c90c3ab84f6ca4c8c2bb6d2629b1f22012a44be37a41799d275e71d
+  data.tar.gz: 01a7f99f884aabff11e0c181bd244b9cc080b6dce87a3bbed926f17e0fe92dc6af448dfc4878071cff758475d359105607f73740ab8ce17c2686eb7b2464e684

data/bin/ronin

@@ -14,11 +14,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-require 'ronin'
 
 unless Process.uid == 0
   puts 'You need to be root to perform this command.'
   exit 1
 end
 
+require 'ronin'
+
 Ronin.run

data/lib/ronin/chef.rb

@@ -47,7 +47,7 @@ module Ronin
       self.create_run_list
       self.create_solo_conf
       Ronin::Log.info("Running Chef, logging to #{Ronin::Config[:log_path]}/ronin-chef.log.")
-      @cmd = Mixlib::ShellOut.new("chef-solo --logfile #{Ronin::Config[:log_path]}/ronin-chef.log --config #{@solo_conf} --json-attributes #{@run_list}")
+      @cmd = Mixlib::ShellOut.new("#{$CHEFSOLO_BIN}  --logfile #{Ronin::Config[:log_path]}/ronin-chef.log --config #{@solo_conf} --json-attributes #{@run_list}")
       @cmd.run_command
       self.clean_up
     end

data/lib/ronin/config.rb

@@ -28,13 +28,17 @@ module Ronin
     end
 
     config_strict_mode         true
+    default :lock_file,        '/var/tmp/ronin.lock'
     default :log_path,         '/var/log/ronin'
     default :log_level,        :info
     default :update_on_change, true
-    default :run_list_type,    :yaml
     default :interpreter,      :puppet
-    default :run_list_file,    '/etc/ronin/artifacts.yaml'
     default :artifact_path,    '/var/lib/ronin/artifacts'
+    default :run_list_type,    :yaml
+    default :run_list_file,    '/etc/ronin/artifacts.yaml'
+    default :etcd_host,        '127.0.0.1'
+    default :etcd_port,        4001
+
   end
 end
 

data/lib/ronin/etcd.rb

@@ -0,0 +1,44 @@
+# Author:: Nathan Milford (<nathan@milford.io>)
+# Copyright:: Copyright (c) 2013 Nathan Milford
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+require 'mixlib/config'
+require 'net/https'
+require 'net/http'
+require 'socket'
+require 'json'
+
+# Add test data thus.
+#curl -L http://127.0.0.1:4001/v2/keys/ronin/run_lists/localhost.localdomain -X PUT -d value='{
+# "run_list": [
+#   "https://github.com/opscode-cookbooks/motd-tail"
+#  ]
+#}'
+
+module Ronin
+  module Etcd
+    def get_run_list
+      # Will add error handling... one day.
+      @hostname = Socket.gethostname
+      @path = "/v2/keys/ronin/run_lists/#{@hostname}"
+      @http = Net::HTTP.new(Ronin::Config['etcd_host'], Ronin::Config['etcd_port'])
+      @http.use_ssl = false
+      @request = Net::HTTP::Get.new(@path)
+      @result = @http.request(@request)
+      @raw = JSON.parse(@result.body)['node']['value']
+      return JSON.parse(@raw)['run_list']
+    end
+    module_function :get_run_list
+  end
+end

data/lib/ronin/git.rb

@@ -19,7 +19,7 @@ require 'ronin/config'
 module Ronin
   module Git
     def branch(artifact)
-      @cmd = Mixlib::ShellOut.new("git --git-dir=#{Ronin::Config[:artifact_path]}/#{artifact}/.git --work-tree=#{Ronin::Config[:artifact_path]}/#{artifact}/ branch")
+      @cmd = Mixlib::ShellOut.new("#{$GIT_BIN} --git-dir=#{Ronin::Config[:artifact_path]}/#{artifact}/.git --work-tree=#{Ronin::Config[:artifact_path]}/#{artifact}/ branch")
       @cmd.run_command
       @branch = @cmd.stdout.chomp.split(' ')[1]
       @branch

data/lib/ronin/puppet.rb

@@ -37,7 +37,7 @@ module Ronin
     def run
       self.create_run_list
       Ronin::Log.info("Running Puppet, logging to #{Ronin::Config[:log_path]}/ronin-puppet.log.")
-      @cmd = Mixlib::ShellOut.new("puppet apply --verbose --ordering manifest --logdest #{Ronin::Config[:log_path]}/ronin-puppet.log --modulepath #{Ronin::Config[:artifact_path]} #{@run_list}")
+      @cmd = Mixlib::ShellOut.new("#{$PUPPET_BIN} apply --verbose --ordering manifest --logdest #{Ronin::Config[:log_path]}/ronin-puppet.log --modulepath #{Ronin::Config[:artifact_path]} #{@run_list}")
       @cmd.run_command
       self.clean_up
     end

data/lib/ronin/ronin.rb

@@ -17,24 +17,51 @@ $LOAD_PATH << "."
 require 'ronin/artifact_runner'
 require 'ronin/run_list'
 require 'ronin/config'
+require 'ronin/util'
+require 'ronin/log'
 
 module Ronin
   def run
 
     Ronin::Log.level = Ronin::Config[:log_level]
 
-    puts Ronin::Config[:interpreter].inspect
+    if Ronin::Util.find_cmd("git").nil?
+      abort("You need to have git installed to perform this command.")
+    else
+      $GIT_BIN = Ronin::Util.find_cmd("git")
+    end
+
+    if Ronin::Util.find_cmd("puppet").nil? and Ronin::Config[:interpreter] == :puppet
+      abort("You need to have Puppet installed to perform this command with Puppet set as the interpreter.")
+    else
+      $PUPPET_BIN = Ronin::Util.find_cmd("puppet")
+    end
+
+    if Ronin::Util.find_cmd("chef-solo").nil? and Ronin::Config[:interpreter] == :puppet
+      abort("You need to have Chef-Solo installed to perform this command with Chef set as the interpreter.")
+    else
+      $CHEFSOLO_BIN = Ronin::Util.find_cmd("puppet")
+    end
+
+    unless File.exists?(Ronin::Config[:lock_file])
+      Ronin::Log.info("Dropping lock file. (#{Ronin::Config[:lock_file]})")
+      File.new(Ronin::Config[:lock_file], "w")
 
-    @r = Ronin::ArtifactRunner.new
-    @changes = @r.download_and_report_changes
-    @r.purge_unused
+      @r = Ronin::ArtifactRunner.new
+      @changes = @r.download_and_report_changes
+      @r.purge_unused
 
-    if @changes
-      if Ronin::Config[:interpreter] == :puppet
-        Ronin::Puppet.run
-      elsif Ronin::Config[:interpreter] == :chef
-        Ronin::Chef.run
+      if @changes
+        if Ronin::Config[:interpreter] == :puppet
+          Ronin::Puppet.run
+        elsif Ronin::Config[:interpreter] == :chef
+          Ronin::Chef.run
+        end
       end
+      Ronin::Log.info("Deleting lock file and exiting. (#{Ronin::Config[:lock_file]})")
+      File.delete(Ronin::Config[:lock_file])
+    else
+      abort("Lock file (#{Ronin::Config[:lock_file]}) exists! Check to see if this Ronin is already running. Exiting.")
     end
   end
   module_function :run

data/lib/ronin/run_list.rb

@@ -14,6 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 require 'mixlib/config'
+require 'ronin/etcd'
 require 'yaml'
 
 module Ronin
@@ -22,16 +23,18 @@ module Ronin
 
     def initialize
       @run_list = {}
-      if Ronin::Config['run_list_type'] = :yaml
-         @modules_raw = YAML.load_file(Ronin::Config['run_list_file'])['artifacts']
+      if Ronin::Config[:run_list_type] == :yaml
+         @artifacts_raw = YAML.load_file(Ronin::Config['run_list_file'])['artifacts']
+      elsif Ronin::Config[:run_list_type] == :etcd
+         @artifacts_raw = Ronin::Etcd.get_run_list
       end
 
-      @modules_raw.each do |m|
-        if m.include?(";")
-          @repo   = m.split(";")[0].sub(/(\/)+$/,'')
-          @branch = m.split(";")[1]
+      @artifacts_raw.each do |a|
+        if a.include?(";")
+          @repo   = a.split(";")[0].sub(/(\/)+$/,'')
+          @branch = a.split(";")[1]
         else
-          @repo   = m
+          @repo   = a
           @branch = 'master'
         end
 
@@ -44,9 +47,9 @@ module Ronin
     end
 
     def artifacts
-      @mods = []
-      @run_list.each { |k,v| @mods << k }
-      @mods
+      @arts = []
+      @run_list.each { |k,v| @arts << k }
+      @arts
     end
 
     def items

data/lib/ronin/util.rb

@@ -0,0 +1,30 @@
+# Author:: Nathan Milford (<nathan@milford.io>)
+# Copyright:: Copyright (c) 2013 Nathan Milford
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Ronin
+  module Util
+
+    def find_cmd(cmd)
+      ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
+        cmd_path = File.join(path, cmd)
+        return cmd_path if File.executable? cmd_path
+      end
+      return nil
+    end
+    module_function :find_cmd
+
+  end
+end

data/lib/ronin/version.rb

@@ -14,5 +14,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 module Ronin
-  VERSION = '0.0.9'
+  VERSION = '0.0.11'
 end

data/lib/ronin.rb

@@ -13,12 +13,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+require 'ronin/ronin'
 require 'ronin/config'
 require 'ronin/artifact_runner'
 require 'ronin/run_list'
 require 'ronin/version'
 require 'ronin/puppet'
-require 'ronin/ronin'
 require 'ronin/chef'
 require 'ronin/git'
 require 'ronin/log'
+require 'ronin/etcd'
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ronin-wrapper
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.11
 platform: ruby
 authors:
 - Nathan Milford
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-12-27 00:00:00.000000000 Z
+date: 2013-12-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yajl-ruby
@@ -91,11 +91,13 @@ extra_rdoc_files:
 files:
 - lib/ronin.rb
 - lib/ronin/ronin.rb
+- lib/ronin/util.rb
 - lib/ronin/config.rb
 - lib/ronin/puppet.rb
 - lib/ronin/chef.rb
 - lib/ronin/git.rb
 - lib/ronin/log.rb
+- lib/ronin/etcd.rb
 - lib/ronin/artifact_runner.rb
 - lib/ronin/run_list.rb
 - lib/ronin/version.rb