tinychef 0.0.2 → 0.0.3

data/bin/tinychef

@@ -4,6 +4,7 @@ $LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
 
 require 'optparse'
 require 'tinychef'
+require 'tinychef/commands'
 
 parser = OptionParser.new do |opts|
   opts.banner = "Usage: tinychef COMMAND [ARGS]"
@@ -23,6 +24,7 @@ parser = OptionParser.new do |opts|
   opts.separator "  bag:create NAME           Creates an empty databag" 
   opts.separator "  bag:encrypt NAME          Encrypts a databag" 
   opts.separator "  bag:decrypt NAME          Decrypts a databag" 
+  opts.separator "  key:generate              Generate a new random key"
   opts.separator "  key:lock                  Lock the key with a password"
   opts.separator "  key:unlock                Unlock the key for a work session" 
   opts.separator "  secure                    Remove all unsecure files from this folder" 
@@ -34,35 +36,24 @@ end
 
 parser.parse!
 
+include Tinychef::Commands
+
 begin
   case ARGV[0]
-  when 'new'
-    Tinychef::NewDirectory.new(ARGV[1]).write
-  when 'boot'
-  when 'bag:create'
-  when 'bag:encrypt'
-  when 'bag:decrypt'
-  when 'key:lock'
-  when 'key:unlock'
-  when 'secure'
-  when 'unsecure'
-  when 'run'
-    options = {}
-
-    options[:node] = ARGV[1]
-
-    if ARGV.size == 3 # assume the destination is not given
-      options[:dest] = nil ; options[:run_list] = ARGV[2]
-    else
-      options[:dest] = ARGV[2] ; options[:run_list] = ARGV[3]
-    end
-
-    Tinychef::NodeRun.new(options).start
+  when 'new' then new_command
+  when 'boot' then boot_command
+  when 'bag:create' then bag_create_command
+  when 'bag:encrypt' then bag_encrypt_command
+  when 'bag:decrypt' then bag_decrypt_command
+  when 'key:generate' then key_generate_command
+  when 'key:lock' then key_lock_command
+  when 'key:unlock' then key_unlock_command
+  when 'secure' then secure_command
+  when 'unsecure' then unsecure_command
+  when 'run' then run_command
   else
     puts parser.help
   end
-rescue Tinychef::NodeRun::OptionsError => e
+rescue Tinychef::OptionsError, RuntimeError => e
   abort e.message
-
-
 end

data/lib/tinychef.rb

@@ -1,5 +1,14 @@
+require 'rubygems'
+require 'chef/encrypted_data_bag_item'
+require 'pp'
+
 require "tinychef/version"
+require 'tinychef/key'
+require 'tinychef/data_bag'
+require 'tinychef/boot_script'
+require 'tinychef/node_run'
 
-module Tinychef; end
+module Tinychef
+  class OptionsError < Exception; end
+end
 
-require 'tinychef/node_run'

data/lib/tinychef/boot_script.rb

@@ -0,0 +1,58 @@
+require 'pathname'
+require 'tinychef/node'
+require 'tinychef/destination'
+
+module Tinychef
+  class BootScript 
+    
+    OPTIONS_ERROR = <<-EOH
+Required arguments:
+
+  DEST:     ssh username@host string to use to connect to host.
+  SCRIPT:   optional path to the boot script (boot.sh by default).
+            You must craft your Chef Solo boot script by yourself, 
+            something like this: https://gist.github.com/3153784
+
+EOH
+    
+    attr_reader :dest, :script
+    
+    def initialize(dest, script)
+      script ||= 'boot.sh'
+
+      begin
+        @dest = Tinychef::Destination.new(dest)
+        @script = Pathname.new(script)
+      rescue => e
+        raise OptionsError.new BootScript::OPTIONS_ERROR
+      end
+      
+      file_check
+    end
+    
+    def run
+      push_to_remote
+      execute_remote
+    end
+    
+    private
+    
+    def file_check
+      unless script.exist? 
+        raise RuntimeError.new "File #{script} not found. Craft your boot script, something like this: https://gist.github.com/3153784"
+      end
+      unless script.extname == '.sh'
+        raise RuntimeError.new "File must be a shell script (.sh extension please)"
+      end
+    end
+    
+    def push_to_remote      
+      system %Q{ scp #{script} #{dest}: }
+    end
+    
+    def execute_remote
+      system %Q{ ssh -t #{dest} "sh #{script.basename}" }
+    end
+    
+  end
+end

data/lib/tinychef/commands.rb

@@ -0,0 +1,61 @@
+module Tinychef
+  module Commands
+    
+    def new_command
+      dir_name = ARGV[1]
+      Tinychef::NewDirectory.new(dir_name).create
+    end
+
+    def boot_command
+      dest = ARGV[1] ; script = ARGV[2]
+      Tinychef::BootScript.new(dest, script).run
+    end
+
+    def bag_create_command
+      recipe = ARGV[1] ; name = ARGV[2]
+      Tinychef::DataBag.new(recipe, name).create
+    end
+
+    def bag_encrypt_command
+      recipe = ARGV[1] ; name = ARGV[2]
+      Tinychef::DataBag.new(recipe, name).encrypt
+    end
+
+    def bag_decrypt_command
+      recipe = ARGV[1] ; name = ARGV[2]
+      Tinychef::DataBag.new(recipe, name).decrypt
+    end
+
+    def key_lock_command
+      Tinychef::Key.new(ARGV[1]).password_protect
+    end
+
+    def key_unlock_command
+      Tinychef::Key.new(ARGV[1]).unlock_and_restore
+    end
+
+    def key_generate_command
+      Tinychef::Key.new(ARGV[1]).generate
+    end
+
+    def secure_command
+    end
+
+    def unsecure_command
+    end
+
+    def run_command
+      options = {}
+
+      options[:node] = ARGV[1]
+
+      if ARGV.size == 3 # assume the destination is not given
+        options[:dest] = nil ; options[:run_list] = ARGV[2]
+      else
+        options[:dest] = ARGV[2] ; options[:run_list] = ARGV[3]
+      end
+
+      Tinychef::NodeRun.new(options).start
+    end
+  end
+end

data/lib/tinychef/data_bag.rb

@@ -0,0 +1,80 @@
+module Tinychef 
+  class DataBag
+    
+    OPTIONS_ERROR = <<-EOH
+Required arguments:
+
+  RECIPE:   Databags must be archived under the name of a recipe.
+  NAME:     Name of this databag. 
+  
+EOH
+    
+    attr_reader :recipe, :name, :ruby_name, :ruby_path, :json_name, :json_path
+    
+    def initialize(recipe, name)
+      
+      if recipe.nil? || name.nil? 
+        raise OptionsError.new OPTIONS_ERROR 
+      end
+      
+      @name      = name
+      @ruby_name = "#{name}.rb"
+      @json_name = "#{name}.json"
+      @recipe    = recipe
+
+      @dir = Pathname.new(File.join('data_bags', @recipe))
+
+      @ruby_path      = @dir.join @ruby_name
+      @json_path      = @dir.join @json_name
+    end
+    
+    def create 
+      prepare_dir
+      create_empty_file
+    end
+
+    def decrypt
+      data = JSON.parse File.read(json_path)
+      secret = Chef::EncryptedDataBagItem.load_secret( key.path )
+      encrypted_data = Chef::EncryptedDataBagItem.new(data, secret)
+      new_hash = {}
+      data.keys.each do |key|
+        new_hash[key] = encrypted_data[key]
+      end
+      file = File.open ruby_path, 'w'
+      file.write new_hash.inspect
+      file.close
+    end
+
+    def encrypt
+      data = eval File.read(ruby_path)
+      secret = Chef::EncryptedDataBagItem.load_secret( key.path )
+      encrypted_data = Chef::EncryptedDataBagItem.encrypt_data_bag_item(data, secret)
+      file = File.open json_path, 'w'
+      file.write encrypted_data.to_json
+      file.close
+    end
+    
+    private 
+    
+    def key
+      Tinychef::Key.new
+    end
+
+    def prepare_dir
+      dir = Pathname.new(File.join 'data_bags', recipe)
+      dir.mkpath unless dir.exist? 
+    end
+      
+    def create_empty_file
+      if ruby_path.exist?
+        RuntimeError.new 'File #{ruby_path.to_s} already exists.'
+      else
+        file = File.new ruby_path, 'a'
+        file.write %Q^{ "id" => "#{name}" }^
+        file.close
+      end
+    end
+      
+  end
+end

data/lib/tinychef/key.rb

@@ -0,0 +1,39 @@
+module Tinychef
+  class Key 
+
+    OPTIONS_ERROR = <<-EOH
+Required arguments:
+
+  NAME:     the name of the secret file. 
+
+EOH
+
+    attr_reader :name, :path
+
+    def initialize(name='secret')
+      raise if name.nil? 
+
+      @name = name
+      @path = Pathname.new "#{name}.key"
+    rescue 
+      raise OptionsError.new OPTIONS_ERROR
+    end
+
+    def password_protect
+      system "openssl aes-256-cbc -salt -in #{path} -out #{path}.aes"
+    end
+
+    def unlock_and_restore
+      system "openssl aes-256-cbc -d -salt -in #{path}.aes -out #{path}"
+    end
+
+    def generate 
+      if path.exist?
+        raise RuntimeError.new "File #{path} exists." 
+      else
+        system "openssl rand -base64 512 | tr -d '\r\n' > #{path}"
+      end
+    end
+    
+  end
+end

data/lib/tinychef/new_directory.rb

@@ -1,6 +1,29 @@
+require 'pathname'
+
 module Tinychef
   class NewDirectory
-    def initialize 
+
+    DIR_MAP = {
+      'cookbooks' => nil,
+      'data_bags' => nil,
+      'imported_cookbooks' => nil,
+      'nodes' => nil,
+      'roles' => nil,
+      'vendor' => nil,
+    }
+
+    attr_reader :path
+
+    def initialize(name)
+      @path = Pathname.new(name)
+    end
+
+    def create
+      path.mkpath unless path.exist?
+
+      DIR_MAP.each do |key, value|
+        path.join(key).mkpath
+      end
     end
   end
 end

data/lib/tinychef/node_run.rb

@@ -1,13 +1,11 @@
 require 'tinychef/node'
 require 'tinychef/destination'
+require 'tinychef/new_directory'
 
 module Tinychef
   class NodeRun
 
-    class OptionsError < Exception
-      def initialize
-        super <<-EOH
-
+    OPTIONS_ERROR = <<-EOH
 Required arguments:
 
   NODE:     node file to execute
@@ -15,11 +13,9 @@ Required arguments:
             If missing, this is guessed by the node file name.
   RUN_LIST: an optional RUN_LIST to be passed to chef-solo executable.
 
-        EOH
-      end
-    end
+EOH
 
-    RECIPES_DIR="recipes"
+    RECIPES_DIR = "recipes"
 
     attr_reader :dest, :node, :run_list, :run_list_command
 
@@ -36,8 +32,7 @@ Required arguments:
       prepare_run_list_command
 
     rescue => e
-      puts e.inspect
-      raise OptionsError
+      raise OptionsError.new NodeRun::OPTIONS_ERROR
     end
 
     def start

data/lib/tinychef/version.rb

@@ -1,3 +1,3 @@
 module Tinychef
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tinychef
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-29 00:00:00.000000000 Z
+date: 2012-12-30 00:00:00.000000000 Z
 dependencies: []
 description: Simple command line tool for Chef Solo recipes
 email:
@@ -26,7 +26,11 @@ files:
 - Rakefile
 - bin/tinychef
 - lib/tinychef.rb
+- lib/tinychef/boot_script.rb
+- lib/tinychef/commands.rb
+- lib/tinychef/data_bag.rb
 - lib/tinychef/destination.rb
+- lib/tinychef/key.rb
 - lib/tinychef/new_directory.rb
 - lib/tinychef/node.rb
 - lib/tinychef/node_run.rb