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     if yaml.has_key?('services')
  25       ['bugsmaster', 'qamaster', 'mailrelay', 'rtmaster', 'packagesmaster'].each do |service|
  26         if yaml['services'].has_key?(service)
  27           results[service] = host == yaml['services'][service]
  28         end
  29       end
  30     end
  31
  32     results['mail_port']      = ''
  33     results['smarthost']      = ''
  34     results['heavy_exim']     = ""
  35     results['smarthost_port'] = 587
  36     results['reservedaddrs']  = '0.0.0.0/8 : 127.0.0.0/8 : 10.0.0.0/8 : 169.254.0.0/16 : 172.16.0.0/12 : 192.0.0.0/17 : 192.168.0.0/16 : 224.0.0.0/4 : 240.0.0.0/5 : 248.0.0.0/5'
  37
  38     if yaml.has_key?('mail_port') and yaml['mail_port'].has_key?(host)
  39       results['mail_port'] = yaml['mail_port'][host]
  40     end
  41
  42     if yaml.has_key?('need_smarthost') and yaml['need_smarthost'].include?(host)
  43       results['smarthost']     = "mailout.debian.org"
  44     end
  45
  46     if yaml.has_key?('reservedaddrs') and yaml['reservedaddrs'].has_key?(host)
  47       results['reservedaddrs'] = yaml['reservedaddrs'][host]
  48     end
  49
  50     if yaml.has_key?('heavy_exim') and yaml['heavy_exim'].include?(host)
  51       results['heavy_exim']    = "true"
  52     end
  53
  54     ldap = LDAP::Conn.new('db.debian.org')
  55
  56     results['ldap'] = []
  57     filter = '(hostname=' + host +')'
  58     begin
  59       ldap.search2('ou=hosts,dc=debian,dc=org', LDAP::LDAP_SCOPE_SUBTREE, filter) do |x|
  60         results['ldap'] << x
  61       end
  62     rescue LDAP::ResultError
  63     rescue RuntimeError
  64     ensure
  65       ldap.unbind
  66     end
  67     return(results)
  68   end
  69 end