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

   1 module Puppet::Parser::Functions
   2   newfunction(:gen_tlsa_entry, :type => :rvalue) do |args|
   3     certfile = args.shift()
   4     hostname = args.shift()
   5     port = args.shift()
   6
   7     if port.kind_of?(Array)
   8       ports = port
   9     else
  10       ports = [ port ]
  11     end
  12
  13     res = []
  14     ports.each{ |port|
  15       if File.exist?(certfile)
  16         cmd = ['swede', 'create', '--usage=3', '--selector=1', '--mtype=1', '--certificate', certfile, '--port', port.to_s, hostname]
  17         IO.popen(cmd, "r") {|i| res << i.read }
  18       else
  19         res << "; certfile #{certfile} did not exist to create TLSA record for #{hostname}:#{port}."
  20       end
  21
  22       if certfile.gsub!(/\.crt$/, '-new.crt') and File.exist?(certfile)
  23         cmd = ['swede', 'create', '--usage=3', '--selector=1', '--mtype=1', '--certificate', certfile, '--port', port.to_s, hostname]
  24         new_entry = ''
  25         IO.popen(cmd, "r") {|i| new_entry = i.read }
  26         if not res.include?(new_entry)
  27           res << new_entry
  28         end
  29       end
  30     }
  31
  32     return res.join("\n")
  33   end
  34 end