hiera-puppet 0.1.0

data/bin/extlookup2hiera

@@ -0,0 +1,56 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'csv'
+
+options = {:in => nil, :out => nil, :format => :yaml}
+
+OptionParser.new do |opts|
+    opts.banner = "Converter for extlookup CSV files into Hiera JSON and YAML files"
+
+    opts.on("--in FILE", "-i", "Input CSV file") do |v|
+        options[:in] = v
+    end
+
+    opts.on("--out FILE", "-o", "Output Hiera file") do |v|
+        options[:out] = v
+    end
+
+    opts.on("--json", "-j", "Create JSON format file") do |v|
+        options[:format] = :json
+    end
+end.parse!
+
+if options[:in].nil? || options[:out].nil?
+    STDERR.puts "Please specify an input and output file with --in and --out"
+    exit 1
+end
+
+unless File.exist?(options[:in])
+    STDERR.puts "Cannot find input file #{options[:in]}"
+    exit 1
+end
+
+csvdata = CSV.read(options[:in])
+hieradata = {}
+
+csvdata.each do |d|
+    d = d.map{|item| item.to_s}
+
+    if d.size > 2
+        hieradata[d[0]] = d[1, d.size].flatten
+    else
+        hieradata[d[0]] = d[1]
+    end
+end
+
+case options[:format]
+when :yaml
+    require 'yaml'
+    File.open(options[:out], "w") {|f| f.write hieradata.to_yaml}
+
+when :json
+    require 'rubygems'
+    require 'json'
+    File.open(options[:out], "w") {|f| f.write JSON.pretty_generate hieradata}
+end

data/lib/hiera/backend/puppet_backend.rb

@@ -0,0 +1,66 @@
+class Hiera
+    module Backend
+        class Puppet_backend
+            def initialize
+                Hiera.debug("Hiera Puppet backend starting")
+            end
+
+            def hierarchy(scope, override)
+                begin
+                    data_class = Config[:puppet][:datasource] || "data"
+                rescue
+                    data_class = "data"
+                end
+
+                calling_class = scope.real.resource.name.to_s.downcase
+                calling_module = calling_class.split("::").first
+
+                hierarchy = Config[:hierarchy] || [calling_class, calling_module]
+
+                hierarchy = [hierarchy].flatten.map do |klass|
+                    klass = Backend.parse_string(klass, scope, {"calling_class" => calling_class, "calling_module" => calling_module})
+
+                    next if klass == ""
+
+                    [data_class, klass].join("::")
+                end.compact
+
+                hierarchy << [calling_class, data_class].join("::")
+                hierarchy << [calling_module, data_class].join("::") unless calling_module == calling_class
+
+                hierarchy.insert(0, [data_class, override].join("::")) if override
+
+                hierarchy
+            end
+
+            def lookup(key, scope, order_override, resolution_type)
+                answer = nil
+
+                Hiera.debug("Looking up #{key} in Puppet backend")
+
+                include_class = Puppet::Parser::Functions.function(:include)
+                loaded_classes = scope.real.catalog.classes
+
+                hierarchy(scope, order_override).each do |klass|
+                    unless answer
+                        Hiera.debug("Looking for data in #{klass}")
+
+                        varname = [klass, key].join("::")
+                        unless loaded_classes.include?(klass)
+                            begin
+                                scope.real.function_include(klass)
+                                answer = scope[varname]
+                                Hiera.debug("Found data in class #{klass}")
+                            rescue
+                            end
+                        else
+                            answer = scope[varname]
+                        end
+                    end
+                end
+
+                answer
+            end
+        end
+    end
+end

data/lib/hiera/puppet_logger.rb

@@ -0,0 +1,8 @@
+class Hiera
+    module Puppet_logger
+        class << self
+            def warn(msg); Puppet.notice("hiera(): #{msg}"); end
+            def debug(msg); Puppet.debug("hiera(): #{msg}"); end
+        end
+    end
+end

data/lib/hiera/scope.rb

@@ -0,0 +1,21 @@
+class Hiera
+    class Scope
+        attr_reader :real
+
+        def initialize(real)
+            @real = real
+        end
+
+        def [](key)
+            ans = @real.lookupvar(key)
+
+            # damn you puppet visual basic style variables.
+            return nil if ans == ""
+            return ans
+        end
+
+        def include?(key)
+            @real.lookupvar(key) != ""
+        end
+    end
+end

data/lib/puppet/parser/functions/hiera.rb

@@ -0,0 +1,40 @@
+module Puppet::Parser::Functions
+    newfunction(:hiera, :type => :rvalue) do |*args|
+        # Functions called from puppet manifests that look like this:
+        #   lookup("foo", "bar")
+        # internally in puppet are invoked:  func(["foo", "bar"])
+        #
+        # where as calling from templates should work like this:
+        #   scope.function_lookup("foo", "bar")
+        #
+        #  Therefore, declare this function with args '*args' to accept any number
+        #  of arguments and deal with puppet's special calling mechanism now:
+        if args[0].is_a?(Array)
+            args = args[0]
+        end
+
+        key = args[0] || nil
+        default = args[1] || nil
+        override = args[2] || nil
+
+        configfile = File.join([File.dirname(Puppet.settings[:config]), "hiera.yaml"])
+
+        raise(Puppet::ParseError, "Hiera config file #{configfile} not readable") unless File.exist?(configfile)
+        raise(Puppet::ParseError, "You need rubygems to use Hiera") unless Puppet.features.rubygems?
+
+        require 'hiera'
+        require 'hiera/scope'
+
+        config = YAML.load_file(configfile)
+        config[:logger] = "puppet"
+
+        hiera = Hiera.new(:config => config)
+        hiera_scope = Hiera::Scope.new(self)
+
+        answer = hiera.lookup(key, default, hiera_scope, override, :priority)
+
+        raise(Puppet::ParseError, "Could not find data item #{key} in any Hiera data file and no default supplied") unless answer
+
+        return answer
+    end
+end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification 
+name: hiera-puppet
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- R.I.Pienaar
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-06 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hiera
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Store and query Hiera data from Puppet
+email: rip@devco.net
+executables: 
+- extlookup2hiera
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/extlookup2hiera
+- lib/puppet/parser/functions/hiera.rb
+- lib/hiera/puppet_logger.rb
+- lib/hiera/scope.rb
+- lib/hiera/backend/puppet_backend.rb
+has_rdoc: true
+homepage: https://github.com/ripienaar/hiera-puppet
+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 query interface and backend for Hiera
+test_files: []
+