modules/puppetmaster/lib/puppet/parser/functions/whohosts.rb

   1 module Puppet::Parser::Functions
   2   newfunction(:whohosts, :type => :rvalue) do |args|
   3     require 'ipaddr'
   4     require 'yaml'
   5
   6     nodeinfo = args[0]
   7     yamlfile = args[1]
   8     parser.watch_file(yamlfile)
   9
  10     $KCODE = 'utf-8'
  11
  12     ans = "unknown"
  13     yaml = YAML.load_file(yamlfile)
  14
  15     if (nodeinfo['ldap'].has_key?('ipHostNumber'))
  16       nodeinfo['ldap']['ipHostNumber'].each do |addr|
  17         yaml.keys.each do |hoster|
  18           if yaml[hoster].kind_of?(Hash) and yaml[hoster].has_key?('netrange')
  19             netrange = yaml[hoster]['netrange']
  20           else
  21             next
  22           end
  23
  24           netrange.each do |net|
  25             begin
  26               if IPAddr.new(net).include?(addr)
  27                 ans = hoster
  28               end
  29             rescue Exception => e
  30               raise "Error while trying to match addr #{addr} for net #{net}: #{e.message}\n#{e.backtrace}"
  31             end
  32           end
  33         end
  34       end
  35     end
  36     return ans
  37   end
  38 end
  39 # vim:set ts=2:
  40 # vim:set et:
  41 # vim:set shiftwidth=2: