environment.rb

Facter.add(:environment) do
  setcode do
    if defined? Puppet and Puppet.application_name == :agent
      Puppet[:environment]
    else
      Facter::Util::Resolution.exec('puppet agent --configprint environment')
    end
  end
end