Use ldap's purpose field (ganeti/kvm host) to decide which hosts get the puppet ganet...
[mirror/dsa-puppet.git] / modules / puppetmaster / lib / puppet / parser / functions / nodeinfo.rb
1 module Puppet::Parser::Functions
2   newfunction(:nodeinfo, :type => :rvalue) do |args|
3     require 'ipaddr'
4
5     host = args[0]
6     yamlfile = args[1]
7     begin
8       nodeinfo         = function_yamlinfo([host, yamlfile])
9       nodeinfo['ldap'] = function_ldapinfo([host, '*'])
10       unless nodeinfo['ldap']['ipHostNumber']
11         raise Puppet::ParseError, "Host #{host} does not have ipHostNumber values in ldap"
12       end
13       nodeinfo['hoster'] = function_whohosts([nodeinfo['ldap']['ipHostNumber']])
14       nodeinfo['buildd'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('buildd'))
15       nodeinfo['timeserver'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('timeserver'))
16       nodeinfo['porterbox'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('porterbox'))
17       nodeinfo['ganeti'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('ganeti/kvm host'))
18
19       if lookupvar('::mta') == 'exim4'
20         unless nodeinfo['heavy_exim']
21           nodeinfo['smarthost'] = 'mailout.debian.org'
22         end
23       elsif lookupvar('::mta') == 'postfix'
24         unless nodeinfo['heavy_postfix']
25           nodeinfo['smarthost'] = 'mailout.debian.org'
26         end
27       end
28
29       nodeinfo['misc'] = {}
30       fqdn = lookupvar('::fqdn')
31       if fqdn and fqdn == host
32         v4ips = lookupvar('::v4ips')
33         if v4ips and v4ips.to_s != "" and v4ips.to_s != 'undefined'
34           nodeinfo['misc']['v4addrs'] = v4ips.split(',').uniq()
35
36           # find out if we are behind nat
37           intersection = nodeinfo['misc']['v4addrs'] & nodeinfo['ldap']['ipHostNumber']
38           nodeinfo['misc']['natted'] = intersection.empty?
39         else
40           nodeinfo['misc']['v4addrs'] = []
41         end
42
43         v6ips = lookupvar('::v6ips')
44         if v6ips and v6ips.to_s != "" and v6ips.to_s != 'undefined'
45           nodeinfo['misc']['v6addrs'] = v6ips.split(',').uniq()
46         else
47           nodeinfo['misc']['v6addrs'] = []
48         end
49
50         # find out if we have an ipv4 and/or an ipv6 address for our host in ldap.
51         nodeinfo['misc']['has_v4_ldap'] = nodeinfo['ldap']['ipHostNumber'].any? { |x| IPAddr.new(x).ipv4? }
52         nodeinfo['misc']['has_v6_ldap'] = nodeinfo['ldap']['ipHostNumber'].any? { |x| IPAddr.new(x).ipv6? }
53         nodeinfo['misc']['v4_ldap'] = nodeinfo['ldap']['ipHostNumber'].select { |x| IPAddr.new(x).ipv4? }
54         nodeinfo['misc']['v6_ldap'] = nodeinfo['ldap']['ipHostNumber'].select { |x| IPAddr.new(x).ipv6? }
55       end
56
57       ns = call_function('hiera',['nameservers'])
58       allow_dns_q = call_function('hiera',['allow_dns_query'])
59       if ns.empty?
60         # no nameservers known for this hoster
61         nodeinfo['misc']['resolver-recursive'] = true
62
63         if not allow_dns_q.empty?
64           raise Puppet::ParseError, "No nameservers listed for #{nodeinfo['hoster']['name']} yet we should answer somebody's queries?  That makes no sense.  allow_dns_q: #{allow_dns_q}."
65         end
66       elsif (nodeinfo['misc']['v4addrs'] and (ns & nodeinfo['misc']['v4addrs']).size > 0) or
67             (nodeinfo['misc']['v6addrs'] and (ns & nodeinfo['misc']['v6addrs']).size > 0)
68         # this host is listed as a nameserver at this location
69         nodeinfo['misc']['resolver-recursive'] = true
70
71         if allow_dns_q.empty?
72           raise Puppet::ParseError, "Host #{host} is listed as a nameserver for #{nodeinfo['hoster']['name']} but no allow_dns_query networks are defined for this location"
73         end
74       else
75         nodeinfo['misc']['resolver-recursive'] = false
76       end
77
78       return(nodeinfo)
79     rescue => e
80       raise Puppet::ParseError, "Error in nodeinfo for node #{host}, yamlfile #{yamlfile}: #{e.message}\n#{e.backtrace}"
81     end
82   end
83 end
84
85 # vim: set fdm=marker ts=2 sw=2 et: