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     ipAddrs = args[0]
   7     yamlfile = args[1]
   8     parser = Puppet::Parser::Parser.new(environment)
   9     parser.watch_file(yamlfile)
  10
  11     ans = {"name" => "unknown"}
  12     yaml = YAML.load_file(yamlfile)
  13
  14     ipAddrs.each do |addr|
  15       yaml.keys.each do |hoster|
  16         next unless yaml[hoster].kind_of?(Hash) and yaml[hoster].has_key?('netrange')
  17         netrange = yaml[hoster]['netrange']
  18
  19         netrange.each do |net|
  20           begin
  21             if IPAddr.new(net).include?(addr)
  22               ans = yaml[hoster]
  23               ans['name'] = hoster
  24             end
  25           rescue => e
  26             raise Puppet::ParseError, "Error while trying to match addr #{addr} for net #{net}: #{e.message}\n#{e.backtrace}"
  27           end
  28         end
  29       end
  30     end
  31     if not ans['longname']
  32       ans['longname'] = ans['name']
  33     end
  34     return ans
  35   end
  36 end
  37 # vim:set ts=2:
  38 # vim:set et:
  39 # vim:set shiftwidth=2: