puppet-rest 0.0.3

data/lib/puppet-rest/db/connection.rb

@@ -0,0 +1,26 @@
+module PuppetRestClient::DB
+  class Connection
+    include PuppetRestClient::DB::Connection::Nodes
+    include PuppetRestClient::DB::Connection::FactNames
+    include PuppetRestClient::DB::Connection::Facts
+    include PuppetRestClient::DB::Connection::Resources
+    include PuppetRestClient::DB::Request
+
+    Config::VALID_OPTIONS_KEYS.each do |key|
+      attr_accessor key
+    end
+
+    def initialize(attrs=Mash.new)
+      attrs = PuppetRestClient::DB.options.merge(attrs)
+
+      #unless attrs[:client_key].is_a?(OpenSSL::PKey::RSA)
+      #  attrs[:client_key] = OpenSSL::PKey::RSA.new(attrs[:client_key])
+      #end
+
+      PuppetRestClient::DB::Config::VALID_OPTIONS_KEYS.each do |key|
+        instance_variable_set("@#{key}".to_sym, attrs[key])
+      end
+    end
+
+  end
+end

data/lib/puppet-rest/db/connection/fact-names.rb

@@ -0,0 +1,9 @@
+module PuppetRestClient::DB
+  class Connection
+    module FactNames
+      def fact_names
+        get api_path('fact-names')
+      end
+    end
+  end
+end

data/lib/puppet-rest/db/connection/facts.rb

@@ -0,0 +1,11 @@
+module PuppetRestClient::DB
+  class Connection
+    module Facts
+      def facts(name)
+        get(api_path("facts/#{name}")).map {|fact|
+          PuppetRestClient::DB::Fact.get_or_new fact
+        }
+      end
+    end
+  end
+end

data/lib/puppet-rest/db/connection/nodes.rb

@@ -0,0 +1,32 @@
+module PuppetRestClient::DB
+  class Connection
+    module Nodes
+      def nodes(options=Mash.new)
+        get(api_path('nodes')).map {|node|
+          n = PuppetRestClient::DB::Node.get_or_new(node)
+          n.connection = self
+          n
+        }
+      end
+
+      def node(name)
+        attributes = get(api_path("nodes/#{name}"))
+        n = PuppetRestClient::DB::Node.get_or_new(attributes)
+        n.connection = self
+        n
+      end
+
+      def node_facts(name)
+        get(api_path("nodes/#{name}/facts")).map {|fact|
+          PuppetRestClient::DB::Fact.get_or_new(fact)
+        }
+      end
+
+      def node_resources(name)
+        get(api_path("nodes/#{name}/resources")).map {|resource|
+          PuppetRestClient::DB::Resource.get_or_new(resource)
+        }
+      end
+    end
+  end
+end

data/lib/puppet-rest/db/connection/resources.rb

@@ -0,0 +1,11 @@
+module PuppetRestClient::DB
+  class Connection
+    module Resources
+      def resources(options=Mash.new)
+        get(api_path("resources/#{name}")).map {|resource|
+          PuppetRestClient::DB::Resource.get_or_new resource
+        }
+      end
+    end
+  end
+end

data/lib/puppet-rest/db/entities/base.rb

@@ -0,0 +1,100 @@
+module PuppetRestClient::DB
+  class Base
+    attr_accessor :attrs
+    alias :to_hash :attrs
+
+    @@identity_map = PuppetRestClient::IdentityMap.new
+
+    def self.identity_map
+      @@identity_map
+    end # def self.identity_map
+
+    # Define methods that retrieve the value from an initialized instance variable Hash, using the attribute as a key
+    #
+    # @overload self.attr_reader(attr)
+    #   @param attr [Symbol]
+    # @overload self.attr_reader(attrs)
+    #   @param attrs [Array<Symbol>]
+    def self.attr_reader(*attrs)
+      attrs.each do |attribute|
+        class_eval do
+          define_method attribute do
+            @attrs[attribute.to_s]
+          end
+          define_method "#{attribute}=" do |value|
+            @attrs[attribute.to_s] = value
+          end
+        end
+      end
+    end # def self.attr_reader
+
+    def self.get(attrs=Mash.new)
+      @@identity_map[self] ||= {}
+      if attrs['title']
+        @@identity_map[self][attrs['title']] && @@identity_map[self][attrs['title']].update(attrs)
+      elsif attrs['resource']
+        @@identity_map[self][attrs['resource']] && @@identity_map[self][attrs['resource']].update(attrs)
+      else
+        @@identity_map[self][Marshal.dump(attrs)]
+      end
+    end # def self.get
+
+    # Retrieve an object from the identity map or initialize a new object
+    def self.get_or_new(attrs=Mash.new)
+      self.get(attrs) || self.new(attrs)
+    end # def self.get_or_new
+
+    # Initializes a new object
+    #
+    # @param attrs [Hash]
+    # @return [PuppetRestClient::Base]
+    def initialize(attrs=Mash.new)
+      self.class.attr_reader *attrs.keys
+      attrs.stringify_keys!
+      self.update attrs
+      if attrs['title']
+        @@identity_map[self.class] ||= {}
+        @@identity_map[self.class][attrs['title']] = self
+      elsif attrs['resource']
+        @@identity_map[self.class] ||= {}
+        @@identity_map[self.class][attrs['resource']] = self
+      else
+        @@identity_map[self.class] ||= {}
+        @@identity_map[self.class][Marshal.dump(attrs)] = self
+      end
+    end # def initialize
+
+    # Fetches an attribute of an object using hash notation
+    #
+    # @param method [String, Symbol] Message to send to the object
+    def [](method)
+      self.__send__(method.to_sym)
+    rescue NoMethodError
+      nil
+    end # def []
+
+    # Update the attributes of an object
+    #
+    # @param attrs [Hash]
+    # @return [PuppetRestClient::Base]
+    def update(attrs)
+      @attrs = attrs
+      self
+    end # def update
+
+    # @return [String]
+    def resource
+      @attrs['resource']
+    end # def id
+
+    # @return [String]
+    def title
+      @attrs['title']
+    end # def name
+
+    def keys
+      @attrs.keys
+    end # def keys
+
+  end
+end

data/lib/puppet-rest/db/entities/fact.rb

@@ -0,0 +1,9 @@
+module PuppetRestClient::DB
+  class Fact < PuppetRestClient::DB::Base
+    attr_reader :name, :value, :certname
+
+    def initialize(attrs=Mash.new)
+      super
+    end
+  end
+end

data/lib/puppet-rest/db/entities/node.rb

@@ -0,0 +1,22 @@
+module PuppetRestClient::DB
+  class Node < PuppetRestClient::DB::Base
+    attr_reader :name, :deactivated, :catalog_timestamp, :facts_timestamp, :report_timestamp
+    attr_accessor :connection
+
+    def initialize(attrs = Mash.new)
+      attrs['name'] = attrs['certname'] if attrs['name'].nil? && !attrs['certname'].nil?
+      super attrs
+      @connection = nil
+    end
+
+    def facts
+      return [] if @connection.nil?
+      @connection.node_facts self.name
+    end
+
+    def resources
+      return [] if @connection.nil?
+      @connection.node_resources self.name
+    end
+  end
+end

data/lib/puppet-rest/db/entities/resource.rb

@@ -0,0 +1,12 @@
+module PuppetRestClient::DB
+  class Resource < PuppetRestClient::DB::Base
+    attr_reader :parameters, :sourceline, :sourcefile, :exported, :tags, :title, :type, :resource, :certname
+
+    def initialize(attrs=Mash.new)
+      super
+      @attrs['parameters'] ||= Mash.new
+      @attrs['tags'] ||= Mash.new
+    end
+
+  end
+end

data/lib/puppet-rest/db/request.rb

@@ -0,0 +1,74 @@
+module PuppetRestClient::DB
+  module Request
+
+    def get(path, params=nil, options=Mash.new)
+      request(:get, path, params, options)
+    end
+
+    def api_path(entity)
+      '/%s/%s' % [api_version, entity]
+    end
+
+    private
+
+    def connection
+      # return @connection if defined? @connection
+
+      default_options = {
+        :headers => {
+          :accept => 'application/json',
+          :content_type => 'application/json',
+          :user_agent => user_agent
+        }
+      }
+
+      options = default_options.deep_merge(connection_options)
+
+      # @connection = Faraday.new(PuppetRestClient.server_url, options, &PuppetRestClient.middleware)
+      Faraday.new(server_url, options, &middleware)
+    end
+
+    def request(method, path, params, options)
+      json_params = params ? MultiJson.dump(params) : ''
+      uri = server_url
+      uri = URI(uri) unless uri.respond_to?(:host)
+      full_path = uri.path + URI(path).path
+      full_path_and_params = uri.path + path
+
+      #puts "server_url = #{server_url}"
+      #puts "full_path_and_params = #{full_path_and_params}"
+      #puts "method = #{method}"
+
+      headers = {} #signature_headers(method.to_s.upcase.to_sym, full_path, json_params)
+      # puts headers.inspect
+      # puts client_key.inspect
+      response = connection.run_request(method.to_sym, full_path_and_params, nil, headers) do |request|
+        request.options[:raw] = true if options[:raw]
+
+        #puts request.inspect
+
+        unless params.nil?
+          if request.method == :post || :put
+            request.body = json_params
+          else
+            request.params.update params
+          end
+        end
+        yield request if block_given?
+      end
+
+      if options[:raw]
+        return response
+      end
+
+      if options[:json]
+        return MultiJson.dump(response.body)
+      end
+
+      return response.body
+    #rescue Faraday::Error::ClientError
+    #  raise PuppetRestClient::Error::ClientError
+    end
+
+  end
+end

data/lib/puppet-rest/error.rb

@@ -0,0 +1,42 @@
+module PuppetRestClient
+  class Error < StandardError
+    attr_reader :http_headers, :wrapped_exception
+
+    def initialize(exception=$!, http_headers={})
+      @http_headers = http_headers
+      if exception.respond_to?(:backtrace)
+        super(exception.message)
+        @wrapped_exception = exception
+      else
+        super(exception.to_s)
+      end
+    end
+
+    def backtrace
+      @wrapped_exception ? @wrapped_exception.backtrace : super
+    end
+
+  end
+
+  class Error::BadRequest < PuppetRestClient::Error
+  end
+
+  class Error::Unauthorized < PuppetRestClient::Error
+  end
+
+  class Error::Forbidden < PuppetRestClient::Error
+  end
+
+  class Error::NotFound < PuppetRestClient::Error
+  end
+
+  class Error::NotAcceptable < PuppetRestClient::Error
+  end
+
+  class Error::Conflict < PuppetRestClient::Error
+  end
+
+  class Error::ClientError < PuppetRestClient::Error
+  end
+
+end

data/lib/puppet-rest/identity_map.rb

@@ -0,0 +1,4 @@
+module PuppetRestClient
+  class IdentityMap < Hash
+  end
+end

data/lib/puppet-rest/monkey_patches/array.rb

@@ -0,0 +1,7 @@
+class Array
+
+  def extract_options!
+    last.is_a?(::Hash) ? pop : {}
+  end # def extract_options!
+
+end

data/lib/puppet-rest/monkey_patches/enumerable.rb

@@ -0,0 +1,11 @@
+module Enumerable
+
+  def threaded_map
+    threads = []
+    each do |object|
+      threads << Thread.new{yield object}
+    end
+    threads.map(&:value)
+  end
+
+end

data/lib/puppet-rest/monkey_patches/hash.rb

@@ -0,0 +1,49 @@
+class Hash
+
+  # Return a hash that includes everything but the given keys.
+  #
+  # @param keys [Array, Set]
+  # @return [Hash]
+  def except(*keys)
+    self.dup.except!(*keys)
+  end
+
+  # Replaces the hash without the given keys.
+  #
+  # @param keys [Array, Set]
+  # @return [Hash]
+  def except!(*keys)
+    keys.each{|key| delete(key)}
+    self
+  end
+
+  # Merges self with another hash, recursively
+  #
+  # @param hash [Hash] The hash to merge
+  # @return [Hash]
+  def deep_merge(hash)
+    target = self.dup
+    hash.keys.each do |key|
+      if hash[key].is_a?(Hash) && self[key].is_a?(Hash)
+        target[key] = target[key].deep_merge(hash[key])
+        next
+      end
+      target[key] = hash[key]
+    end
+    target
+  end
+
+  def stringify_keys!
+    keys.each do |key|
+      self[key.to_s] = delete(key)
+    end
+    self
+  end unless defined? stringify_keys
+  
+  def symbolize_keys!
+    keys.each do |key|
+      self[(key.to_sym rescue key) || key] = delete(key)
+    end
+    self
+  end unless defined? symbolize_keys
+end