hiera-postgres-backend 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 0cbe610dafd58e2b3222695692a84ea439f4269f
+  data.tar.gz: f7f3946bee8abf69ea12f894c7c15f5cb8ea91cd
+SHA512:
+  metadata.gz: 8d306bed635ec9e2aa8fcd3e878271dc42d87da02cacbda5e2bd10d5d6b5fdbc6ae916911b5ab4e59baadca635773d02dc4de499fcdd73a2ce11dc56caee933c
+  data.tar.gz: 6c3af395d3076858cdacefd7557f4df41e7406e00ea53138673d8ebcb87bd6cd82ba88c25d7092f62d9f32351cdf894a9ae3fd6de22ff5b7f7f69a62edccf69f

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in hiera-postgres-backend.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,22 @@
+PATH
+  remote: .
+  specs:
+    hiera-postgres-backend (0.0.1)
+      hiera
+      pg
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    hiera (1.3.0)
+      json_pure
+    json_pure (1.8.1)
+    pg (0.17.1)
+    rake (10.1.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  hiera-postgres-backend!
+  rake

data/LICENSE

@@ -0,0 +1,20 @@
+The MIT License (MIT)
+
+Copyright (c) 2013 Adrian Lopez
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,68 @@
+## hiera-postgresql-backend
+
+Alternate PostgreSQL backend for Hiera
+
+This is a PostgreSQL backend for Hiera inspired by [hiera-mysql-backend](https://github.com/Telmo/hiera-mysql-backend). 
+
+
+### What goes into the sql files.
+
+The poorly named sql files are really yaml files where the key is the lookup key and the value is the SQL statement (it accepts interpolation)
+
+Lets assume your _datadir_ is `/etc/puppet/hieradata/` and your hierarchy for hiera just have a common. hiera-postgresql-backend would look for /etc/puppet/hieradata/common.sql the common.sql would look like:
+
+```yaml
+---
+applications: SELECT * FROM applications WHERE host='%{fqdn}';
+coats: SELECT cut,name,type FROM coats WHERE color='brown';
+```
+
+running `hiera applications` would run the query against the configured database.
+
+
+### Using
+
+`gem install hiera-postgresql-backend`
+
+
+### Configuring Hiera
+
+Hiera configuration is pretty simple
+
+```yaml
+---
+:backends:
+  - yaml
+  - postgres
+
+:yaml:
+  :datadir: /etc/puppet/hieradata
+
+:postgres:
+  :datadir: /etc/puppet/hieradata
+  :host: hostname
+  :user: username
+  :pass: password
+  :database: database
+
+:hierarchy:
+  - "%{::clientcert}"
+  - "%{::custom_location}"
+  - common
+
+:logger: console
+```
+
+## Known issues
+
+1. It always return an Array of hashes regardless of the number of items returned. (I did this on purpose because it is what I needed but I may be persuaded to do otherwise)
+2. This README is poorly written.
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/hiera-postgres-backend.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |gem|
+  gem.name          = "hiera-postgres-backend"
+  gem.version       = "0.0.1"
+  gem.authors       = ["Adrian"]
+  gem.email         = ["adrianlzt@gmail.com"]
+  gem.description   = %q{Alternative PostgreSQL backend for hiera}
+  gem.summary       = %q{Alternative PostgreSQL backend for hiera}
+  gem.homepage      = "https://github.com/adrianlzt/hiera-postgres-backend"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+  gem.add_dependency('pg')
+  gem.add_dependency('hiera')
+  gem.add_development_dependency('rake')
+end

data/lib/hiera/backend/postgres_backend.rb

@@ -0,0 +1,78 @@
+class Hiera
+  module Backend
+    class Postgres_backend
+
+      def initialize(cache=nil)
+        begin
+          require 'pg'
+        rescue LoadError
+          require 'rubygems'
+          require 'pg'
+        end
+
+        @cache = cache || Filecache.new
+
+        Hiera.debug("Hiera PostgreSQL initialized")
+      end
+
+      def lookup(key, scope, order_override, resolution_type)
+        # default answer is set to nil otherwise the lookup ends up returning
+        # an Array of nils and causing a Puppet::Parser::AST::Resource failed with error ArgumentError
+        # for any other lookup because their default value is overwriten by [nil,nil,nil,nil]
+        # so hiera('myvalue', 'test1') returns [nil,nil,nil,nil]
+      	results = nil
+
+        Hiera.debug("looking up #{key} in PostgreSQL Backend")
+        Hiera.debug("resolution type is #{resolution_type}")
+
+        Backend.datasources(scope, order_override) do |source|
+          Hiera.debug("Looking for data source #{source}")
+          sqlfile = Backend.datafile(:postgres, scope, source, "sql") || next
+
+          next unless File.exist?(sqlfile)
+          data = @cache.read(sqlfile, Hash, {}) do |datafile|
+            YAML.load(datafile)
+          end
+
+          Hiera.debug("data #{data.inspect}")
+          next if data.empty?
+          next unless data.include?(key)
+
+          Hiera.debug("Found #{key} in #{source}")
+
+          new_answer = Backend.parse_answer(data[key], scope)
+          results = query(new_answer)
+
+        end
+          return results
+      end
+
+
+      def query(query)
+        Hiera.debug("Executing SQL Query: #{query}")
+
+        data=nil
+        pg_host = Config[:postgres][:host]
+        pg_user = Config[:postgres][:user]
+        pg_pass = Config[:postgres][:pass]
+        pg_database = Config[:postgres][:database]
+        client = PG::Connection.new(:host     => pg_host, 
+                                    :user     => pg_user, 
+                                    :password => pg_pass, 
+                                    :dbname   => pg_database)
+        begin
+          data = client.exec(query).to_a
+          Hiera.debug("PostgreSQL Query returned #{data.size} rows")
+        rescue => e
+          Hiera.debug e.message
+          data = nil
+        ensure
+          client.close
+        end
+
+        return data
+
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification
+name: hiera-postgres-backend
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Adrian
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hiera
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Alternative PostgreSQL backend for hiera
+email:
+- adrianlzt@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- hiera-postgres-backend.gemspec
+- lib/hiera/backend/postgres_backend.rb
+homepage: https://github.com/adrianlzt/hiera-postgres-backend
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.7
+signing_key: 
+specification_version: 4
+summary: Alternative PostgreSQL backend for hiera
+test_files: []