ronin-wrapper 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dbba7faf4124e6faf33232719132c7d3ee2f0aa9
-  data.tar.gz: 7ff5f5f4fd2095a528624d8e3ec555a85f658a36
+  metadata.gz: 2c3c00f54efdb0caece795e59b3647d555f25ea3
+  data.tar.gz: 8cb9cae2f49b2bc0453e32fefdc5f649f34dac24
 SHA512:
-  metadata.gz: de8637c2dd77b9b9f3e20a13360ac4fbbc709a164945de83eab5d81167a29d0530e8eedf5b578ae01948371269260c21a68d79e27f40fa890ad605d343e5f768
-  data.tar.gz: 01933a1e37887cffa3ccd6d34d0f6d0c2781136118d5508d901ecf2813a158e4b2fdaa3edf9e9e8e40110076ea95001e77d0e59b85493d3c06cebda2f5641a15
+  metadata.gz: 2f1fb190ffc1853a1341fcc19741b563898e82691c493b3c222d5713e9e3bece065c7d04204fc9212d6719ae43a512f6eff06ae5c8cdc19ed8d31f32221f5cc1
+  data.tar.gz: 7e4bcdf75d4b40262809e7958a2f7a0fc7e10bdd37d44cd3b24ba47b8bd57d1c4ae700d43278d89d3db293c43192d47789633762f7aa3532e2c2967cae15bb03

data/bin/ronin

@@ -16,4 +16,9 @@
 # limitations under the License.
 require 'ronin'
 
-Ronin.run
+unless Process.uid == 0
+  puts 'You need to be root to perform this command.'
+  exit 1
+end
+
+Ronin.run

data/lib/ronin.rb

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 require 'ronin/config'
-require 'ronin/module_runner'
+require 'ronin/artifact_runner'
 require 'ronin/run_list'
 require 'ronin/version'
 require 'ronin/puppet'

data/lib/ronin/{module_runner.rb → artifact_runner.rb}

@@ -19,7 +19,7 @@ require 'ronin/log'
 require 'fileutils'
 
 module Ronin
-  class ModuleRunner
+  class ArtifactRunner
     def initialize
       @changes    = false
       @run_list   = Ronin::RunList.new
@@ -29,7 +29,7 @@ module Ronin
       @run_list.items.each do |item|
          @actual_branch = Ronin::Git.branch(item[:name])
 
-        if File.exist?("#{Ronin::Config[:module_path]}/#{item[:name]}")
+        if File.exist?("#{Ronin::Config[:artifact_path]}/#{item[:name]}")
           if item[:branch] != 'master'
             Ronin::Log.info("Module #{item[:name]} is being pulled from the #{item[:branch]} branch and not master.")
           end
@@ -43,12 +43,12 @@ module Ronin
             end
           else
             Ronin::Log.info("Module #{item[:name]} already cached, but is the wrong branch. Deleting cached copy of branch #{@actual_branch}")
-            FileUtils.rm_rf("#{Ronin::Config[:module_path]}/#{item[:name]}/")
+            FileUtils.rm_rf("#{Ronin::Config[:artifact_path]}/#{item[:name]}/")
             Ronin::Git.clone(item[:repo], item[:branch])
             @changes = true if Ronin::Config[:update_on_change]
           end
         else
-          Ronin::Log.info("Module #{item[:name]} not cached, cloning branch #{item[:branch]} of #{item[:repo]} to #{Ronin::Config[:module_path]}.")
+          Ronin::Log.info("Module #{item[:name]} not cached, cloning branch #{item[:branch]} of #{item[:repo]} to #{Ronin::Config[:artifact_path]}.")
           Ronin::Git.clone(item)
           @changes = true if Ronin::Config[:update_on_change]
         end
@@ -57,13 +57,13 @@ module Ronin
     end
 
     def purge_unused
-      @local_modules = Dir.entries(Ronin::Config[:module_path]).select { |dir| File.directory?("#{Ronin::Config[:module_path]}/#{dir}") and !(dir =='.' || dir == '..') }
-      @modules = @run_list.modules
+      @local_artifacts = Dir.entries(Ronin::Config[:artifact_path]).select { |dir| File.directory?("#{Ronin::Config[:artifact_path]}/#{dir}") and !(dir =='.' || dir == '..') }
+      @artifacts = @run_list.artifacts
 
-      @local_modules.each do |mod|
-        unless @modules.include?(mod)
-          Ronin::Log.info("No module named #{mod} in run list, but it exists in #{Ronin::Config[:module_path]}. Purging it.")
-          FileUtils.rm_rf("#{Ronin::Config[:module_path]}/#{mod}/")
+      @local_artifacts.each do |a|
+        unless @artifacts.include?(a)
+          Ronin::Log.info("No module named #{a} in run list, but it exists in #{Ronin::Config[:artifact_path]}. Purging it.")
+          FileUtils.rm_rf("#{Ronin::Config[:artifact_path]}/#{a}/")
         end
       end
     end

data/lib/ronin/chef.rb

@@ -16,9 +16,59 @@
 require 'ronin/run_list'
 require 'ronin/config'
 require 'ronin/log'
+require 'json'
 
 module Ronin
   module Chef
-    # tbd
+    @run_list  = "#{Ronin::Config[:artifact_path]}/ronin.json"
+    @solo_conf = "#{Ronin::Config[:artifact_path]}/ronin-chef-solo.rb"
+    @recipes   = Ronin::RunList.new.artifacts
+
+    def create_run_list
+      Ronin::Log.info("Building Chef run list at #{@run_list}.")
+
+      @rl = []
+      @rl_obj = {}
+
+      @recipes.each do |r|
+        @rl << "recipe[#{r}]"
+        Ronin::Log.info("Adding recipe '#{r}' to run list.")
+      end
+
+      @rl_obj['run_list'] = @rl
+
+      File.open(@run_list, "w") do |f|
+        f.write(@rl_obj.to_json)
+      end
+    end
+    module_function :create_run_list
+
+    def run
+      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.run_command
+      self.clean_up
+    end
+    module_function :run
+
+    def create_solo_conf
+      @solo_config = "file_cache_path '/var/tmp/ronin/chef-solo'\ncookbook_path '#{Ronin::Config[:artifact_path]}'\n"
+
+      File.open(@solo_conf, "w") do |f|
+        f.write(@solo_config)
+      end
+    end
+    module_function :create_solo_conf
+
+    def clean_up
+      Ronin::Log.info("Cleaning up Chef run list at #{@run_list}.")
+      File.delete(@run_list)
+
+      Ronin::Log.info("Cleaning up Chef-Solo config at #{@solo_conf}.")
+      File.delete(@solo_conf)
+    end
+    module_function :clean_up
   end
 end

data/lib/ronin/config.rb

@@ -33,8 +33,8 @@ module Ronin
     default :update_on_change, true
     default :run_list_type,    :yaml
     default :interpreter,      :puppet
-    default :run_list_file,    '/etc/ronin/modules.yaml'
-    default :module_path,      '/var/lib/modules'
+    default :run_list_file,    '/etc/ronin/artifacts.yaml'
+    default :artifact_path,    '/var/lib/ronin/artifacts'
   end
 end
 

data/lib/ronin/git.rb

@@ -18,24 +18,24 @@ require 'ronin/config'
 
 module Ronin
   module Git
-    def branch(mod)
-      @cmd = Mixlib::ShellOut.new("git --git-dir=#{Ronin::Config[:module_path]}/#{mod}/.git --work-tree=#{Ronin::Config[:module_path]}/#{mod}/ branch")
+    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.run_command
       @branch = @cmd.stdout.chomp.split(' ')[1]
       @branch
     end
     module_function :branch
 
-    def pull_and_report_updated(mod)
-      @cmd = Mixlib::ShellOut.new("git --git-dir=#{Ronin::Config[:module_path]}/#{mod}/.git --work-tree=#{Ronin::Config[:module_path]}/#{mod}/ pull")
+    def pull_and_report_updated(artifact)
+      @cmd = Mixlib::ShellOut.new("git --git-dir=#{Ronin::Config[:artifact_path]}/#{artifact}/.git --work-tree=#{Ronin::Config[:artifact_path]}/#{artifact}/ pull")
       @cmd.run_command
       @updated = @cmd.stdout.include?("Updating")
       @updated ? true : false
     end
     module_function :pull_and_report_updated
 
-    def clone(mod_data)
-      @cmd = Mixlib::ShellOut.new("git clone #{mod_data[:repo]} #{Ronin::Config[:module_path]}/#{mod_data[:name]}/ -b #{mod_data[:branch]}")
+    def clone(artifact_data)
+      @cmd = Mixlib::ShellOut.new("git clone #{artifact_data[:repo]} #{Ronin::Config[:artifact_path]}/#{artifact_data[:name]}/ -b #{artifact_data[:branch]}")
       @cmd.run_command
       @cmd.stdout
     end

data/lib/ronin/puppet.rb

@@ -20,8 +20,8 @@ require 'ronin/log'
 module Ronin
   module Puppet
 
-    @run_list = "#{Ronin::Config[:module_path]}/ronin.pp"
-    @modules = Ronin::RunList.new.modules
+    @run_list = "#{Ronin::Config[:artifact_path]}/ronin.pp"
+    @modules = Ronin::RunList.new.artifacts
 
     def create_run_list
       Ronin::Log.info("Building Puppet run list at #{@run_list}.")
@@ -36,8 +36,8 @@ module Ronin
 
     def run
       self.create_run_list
-      Ronin::Log.info("Running Puppet, logging puppet output 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[:module_path]} #{@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.run_command
       self.clean_up
     end

data/lib/ronin/ronin.rb

@@ -14,28 +14,25 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 $LOAD_PATH << "."
-require 'ronin/module_runner'
+require 'ronin/artifact_runner'
 require 'ronin/run_list'
 require 'ronin/config'
 
 module Ronin
   def run
 
-    unless Process.uid == 0
-      puts 'You need to be root to perform this command.'
-      exit 1
-    end
-
     Ronin::Log.level = Ronin::Config[:log_level]
 
-    @r = Ronin::ModuleRunner.new
+    puts Ronin::Config[:interpreter].inspect
+
+    @r = Ronin::ArtifactRunner.new
     @changes = @r.download_and_report_changes
     @r.purge_unused
 
     if @changes
-      if Ronin::Config[:interpreter] = :puppet
+      if Ronin::Config[:interpreter] == :puppet
         Ronin::Puppet.run
-      elsif Ronin::Config[:interpreter] = :chef
+      elsif Ronin::Config[:interpreter] == :chef
         Ronin::Chef.run
       end
     end

data/lib/ronin/run_list.rb

@@ -23,7 +23,7 @@ module Ronin
     def initialize
       @run_list = {}
       if Ronin::Config['run_list_type'] = :yaml
-         @modules_raw = YAML.load_file(Ronin::Config['run_list_file'])['modules']
+         @modules_raw = YAML.load_file(Ronin::Config['run_list_file'])['artifacts']
       end
 
       @modules_raw.each do |m|
@@ -43,7 +43,7 @@ module Ronin
       @run_list
     end
 
-    def modules
+    def artifacts
       @mods = []
       @run_list.each { |k,v| @mods << k }
       @mods

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.8'
+  VERSION = '0.0.9'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ronin-wrapper
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - Nathan Milford
@@ -96,7 +96,7 @@ files:
 - lib/ronin/chef.rb
 - lib/ronin/git.rb
 - lib/ronin/log.rb
-- lib/ronin/module_runner.rb
+- lib/ronin/artifact_runner.rb
 - lib/ronin/run_list.rb
 - lib/ronin/version.rb
 - README.md