puppetsdb 0.0.1

data/bin/puppetsdb

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'puppetsdb/cli'
+
+PuppetSDB::CLI.run

data/lib/puppetsdb/cli.rb

@@ -0,0 +1,126 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'subcommand'
+require 'puppetsdb'
+require 'pp'
+
+include Subcommands
+
+module PuppetSDB
+  class CLI
+    class << self
+      COL_WIDTH = 40
+      def ask_confirm(question)
+        print "Are you sure you want to delete this node? (yes/[no]) "
+        answer = $stdin.gets.chomp
+        accepted_confirms = ['y','yes']
+        accepted_confirms.include?(answer.downcase)
+      end
+
+      def run
+        options = {}
+        global_cmd = global_options do |opts|
+          opts.banner = "Usage: ", $PROGRAM_NAME, " [subcommand [options]]"
+          opts.description = "CLI for Puppet ENC on Amazon's SimpleDB"
+          opts.separator ""
+          opts.separator "Global options are:"
+          opts.on("-v", "--[no-]verbose", "Run verbosely") do |v|
+            options[:verbose] = v
+          end
+        end
+
+        list_cmd = command :list do |opts|
+          opts.banner = "Usage: list"
+          opts.description = "List all nodes in the ENC"
+        end
+        listdomains_cmd = command :listdomains do |opts|
+          opts.banner = "Usage: listdomains"
+          opts.description = "List all SimpleDB domains in the AWS account"
+        end
+        createdomain_cmd = command :createdomain do |opts|
+          opts.banner = "Usage: createdomain domain_name"
+          opts.description = "Create a SimpleDB domain"
+        end
+        deletedomain_cmd = command :deletedomain do |opts|
+          opts.banner = "Usage: deletedomain [options] domain_name"
+          opts.description = "Delete a SimpleDB domain"
+          opts.on("-f", "--force", "Do not prompt for confirmation") do |v|
+            options[:deletedomain_force] = v
+          end
+          opts.on("-i", "--items", "Delete the domain and all associated items") do |v|
+            options[:deletedomain_items] = v
+          end
+        end
+        get_cmd = command :get do |opts|
+          opts.banner = "Usage: get [options] nodename"
+          opts.description = "Get YAML data for a node"
+        end
+        set_cmd = command :set do |opts|
+          opts.banner = "Usage: set [options] nodename [yaml_file]"
+          opts.description = "Set YAML data for a node. If yaml_file is not specified, the yaml data is read from STDIN."
+        end
+        delete_cmd = command :delete do |opts|
+          opts.banner = "Usage: delete [options] nodename [...]"
+          opts.description = "Delete the ENC record(s)"
+          opts.on("-f", "--force", "Do not prompt for confirmation") do |v|
+            options[:delete_force] = v
+          end
+        end
+
+        cmd = opt_parse
+        puppet_enc = PuppetSDB.new
+        case cmd
+          when "list"
+            strfmt = "%-#{COL_WIDTH.to_s}s %s"
+            puts strfmt.%(['Node','Time Modified'])
+            puts strfmt.%(['-'*COL_WIDTH,'-'*COL_WIDTH])
+            puppet_enc.list_items_detailed.each do |node|
+              puts strfmt.%([node[0],node[1]['Time Modified']])
+            end
+          when "listdomains"
+            strfmt = "%s"
+            puts strfmt.%(['Domain'])
+            puts strfmt.%(['-'*COL_WIDTH])
+            puppet_enc.list_domains.each { |domain| puts strfmt.%([domain]) }
+          when "createdomain"
+            domain_name = ARGV.shift
+            puppet_enc.create_domain(domain_name)
+            puts "Domain create successfully"
+          when "deletedomain"
+            domain_name = ARGV.shift
+            unless options[:deletedomain_force]
+              (puts "Cancelled..."; exit 0) unless ask_confirm("Are you sure you want to delete this domain? (yes/[no]) ")
+            end
+            if options[:deletedomain_items]
+              puppet_enc.delete_domain(domain_name, with_items=true)
+            else
+              puppet_enc.delete_domain(domain_name, with_items=false)
+            end
+            puts "Domain deleted successfully"
+          when "get"
+            (puts get_cmd.call.help; exit 1) if ARGV.empty?
+            yaml = puppet_enc.get_node_yaml(ARGV[0])
+            puts yaml unless yaml.empty?
+          when "set"
+            (puts set_cmd.call.help; exit 1) if ARGV.empty?
+            nodename = ARGV.shift
+            if ARGV.empty?
+              yaml_data = $stdin.read
+            else
+              yaml_file = ARGV.shift
+              yaml_data = File.new(yaml_file).read
+            end
+            puppet_enc.write_node_data(nodename, yaml_data)
+          when "delete"
+            (puts delete_cmd.call.help; exit 1) if ARGV.empty?
+            unless options[:delete_force]
+              (puts "Cancelled..."; exit 0) unless ask_confirm("Are you sure you want to delete this node? (yes/[no]) ")
+            end
+            ARGV.each { |node| puppet_enc.delete_item(node) }
+        end
+        (add_subcommand_help; puts global_cmd.help; exit 1) unless cmd
+      end
+    end
+  end
+end

data/lib/puppetsdb/version.rb

@@ -0,0 +1,3 @@
+module PuppetSDB
+  VERSION = "0.0.1"
+end

data/lib/puppetsdb.rb

@@ -0,0 +1,119 @@
+#!/usr/bin/env ruby
+
+require 'pathname'
+require 'base64'
+require 'rubygems'
+require 'aws'
+require 'yaml'
+
+module PuppetSDB
+  class PuppetSDB
+    attr_accessor :simpledb, :domain
+     
+    def initialize(domain=nil)
+      cfg_file_paths = ['/etc/puppetsdb/config.yml']
+      if ENV['HOME']
+        cfg_file_paths << '~/.puppetsdb/config.yml'
+      else
+        cfg_file_paths << '~puppet/.puppetsdb/config.yml'
+      end
+      conf = {}
+      cfg_file_paths.each do |file|
+        file = File.expand_path(file)
+        next unless File.exists?(file)
+        conf.merge!(YAML.load(File.read(file)))
+      end
+
+      AWS.config(conf['aws'])
+      domain ||= conf['puppetsdb']['domain']
+      @simpledb = AWS::SimpleDB.new
+      @domain = domain
+    end
+
+    def list_domains
+      @simpledb.domains.collect(&:name)
+    end
+
+    def create_domain(domain_name)
+      @simpledb.domains.create(domain_name)
+    end
+
+    def delete_domain(domain_name, with_items=false)
+      if with_items
+        @simpledb.domains[domain_name].delete!
+      else
+        @simpledb.domains[domain_name].delete
+      end
+    end
+
+    def list_items
+      @simpledb.domains[@domain].items.collect(&:name)
+    end
+
+    def list_items_detailed
+      attrs = ['Time Modified']
+      results = []
+      @simpledb.domains[@domain].items.select(*attrs).each do |item_data|
+        results << [item_data.name, item_data.attributes]
+      end
+      results
+    end
+
+    def list_item_attrs(item)
+      @simpledb.domains[@domain].items[item].attributes.collect(&:name)
+    end
+
+    def delete_item(item)
+      @simpledb.domains[@domain].items[item].delete
+    end
+
+    def get_item_attrs(item)
+      @simpledb.domains[@domain].items[item].data.attributes
+    end
+
+    def get_node_yaml(item)
+      attrs = get_item_attrs(item)
+      if attrs.empty?
+        ""
+      else
+        decode_base64(attrs['Node Data'][0])
+      end
+    end
+
+    def update_item(item, attrs)
+      @simpledb.domains[@domain].items[item].attributes.put(
+        :replace => attrs)
+    end
+
+    def encode_yaml(yaml)
+      Base64.encode64(yaml)
+    end
+
+    def decode_base64(base64)
+      Base64.decode64(base64)
+    end
+
+    def get_timestamp_utc
+      Time.now.utc
+    end
+
+    def construct_attrs(yaml)
+      attrs = {'Time Modified' => get_timestamp_utc,
+               'Node Data' => encode_yaml(yaml)}
+    end
+
+    def validate_yaml(yaml)
+      begin
+        YAML.load(yaml)
+      rescue
+        puts "Validation of YAML failed"
+        raise
+      end
+    end
+
+    def write_node_data(nodename, yaml)
+      validate_yaml(yaml)
+      update_item(nodename, construct_attrs(yaml))
+    end
+  end
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification 
+name: puppetsdb
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Brian Wong
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-10-26 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: subcommand
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: aws-sdk
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: A simple interface to Puppet's ENC stored on Amazon's SimpleDB
+email: bwong114@gmail.com
+executables: 
+- puppetsdb
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/puppetsdb.rb
+- lib/puppetsdb/cli.rb
+- lib/puppetsdb/version.rb
+- bin/puppetsdb
+has_rdoc: true
+homepage: http://rubygems.org/gems/puppetsdb
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Puppet ENC script stored against Amazon's SimpleDB
+test_files: []
+