files/etc/puppet/lib/puppet/parser/functions/nodeinfo.rb

   1 module Puppet::Parser::Functions
   2   newfunction(:nodeinfo, :type => :rvalue) do |args|
   3
   4     host = args[0]
   5     yamlfile = args[1]
   6     parser.watch_file(yamlfile)
   7
   8     require 'ldap'
   9     require 'yaml'
  10
  11     $KCODE = 'utf-8'
  12
  13     yaml = YAML.load_file(yamlfile)
  14     results = {}
  15
  16     ['nameinfo', 'footer'].each do |detail|
  17       if yaml.has_key?(detail)
  18         if yaml[detail].has_key?(host)
  19           results[detail] = yaml[detail][host]
  20         end
  21       end
  22     end
  23
  24 #    results['bugsmaster']     = host == 'rietz.debian.org'
  25 #    results['qamaster']       = host == 'merkel.debian.org'
  26 #    results['mailrelay']      = host == 'spohr.debian.org'
  27 #    results['rtmaster']       = host == 'spohr.debian.org'
  28 #    results['packagesmaster'] = host == 'powell.debian.org'
  29 #
  30 #    nodeinfo['reservedaddrs'] = case host
  31 #      when "ball.debian.org": "0.0.0.0/8 : 127.0.0.0/8 : 169.254.0.0/16 : \
  32 #            172.16.0.0/12 : 192.0.0.0/17 : 192.168.0.0/16 : \
  33 #            224.0.0.0/4 : 240.0.0.0/5 : 248.0.0.0/5"
  34 #      else "0.0.0.0/8 : 127.0.0.0/8 : 10.0.0.0/8 : 169.254.0.0/16 : \
  35 #            172.16.0.0/12 : 192.0.0.0/17 : 192.168.0.0/16 : \
  36 #            224.0.0.0/4 : 240.0.0.0/5 : 248.0.0.0/5"
  37 #    end
  38
  39     ldap = LDAP::Conn.new('db.debian.org')
  40
  41     results['ldap'] = []
  42     filter = '(hostname=' + host +')'
  43     begin
  44       ldap.search2('ou=hosts,dc=debian,dc=org', LDAP::LDAP_SCOPE_SUBTREE, filter) do |x|
  45         results['ldap'] << x
  46       end
  47     rescue LDAP::ResultError
  48     rescue RuntimeError
  49     ensure
  50       ldap.unbind
  51     end
  52     return(results)
  53   end
  54 end