support puppetdb token authentication

[h0tw1r3]

No description provided.

[ekohl]

I'm not familiar with PE and their tokens, but this is a general Ruby thing.

[ekohl]

I think you should use a block to cache. The diff may be hard to read, but it's essentially

@@client ||= begin
  # ...
end

Suggested change

	options = {
	server: ENV['PUPPETDB_URL'] || @@puppetdb,
	}
	if ENV['PE_TOKEN']
	options[:token] = ENV['PE_TOKEN']
	options[:cacert] = ENV['PUPPETDB_CACERT_FILE'] || Puppet[:localcacert]
	else
	options[:pem] = {
	'key' => ENV['PUPPETDB_KEY_FILE'] || Puppet[:hostprivkey],
	'cert' => ENV['PUPPETDB_CERT_FILE'] || Puppet[:hostcert],
	'ca_file' => ENV['PUPPETDB_CACERT_FILE'] || Puppet[:localcacert],
	}
	end
	@@client ||= ::PuppetDB::Client.new(options, 4)
	@@client ||= begin
	options = {
	server: ENV['PUPPETDB_URL'] || @@puppetdb,
	}
	if ENV['PE_TOKEN']
	options[:token] = ENV['PE_TOKEN']
	options[:cacert] = ENV['PUPPETDB_CACERT_FILE'] || Puppet[:localcacert]
	else
	options[:pem] = {
	'key' => ENV['PUPPETDB_KEY_FILE'] || Puppet[:hostprivkey],
	'cert' => ENV['PUPPETDB_CERT_FILE'] || Puppet[:hostcert],
	'ca_file' => ENV['PUPPETDB_CACERT_FILE'] || Puppet[:localcacert],
	}
	end
	::PuppetDB::Client.new(options, 4)
	end

[h0tw1r3]

Cheers. Fixed.