Call site display for deprecation warnings

1 files changed, 6 insertions, 2 deletions

diff --git a/lib/puppet/functions/deprecation.rb b/lib/puppet/functions/deprecation.rb
index a860aa2..7082068 100644
--- a/lib/puppet/functions/deprecation.rb
+++ b/lib/puppet/functions/deprecation.rb
@@ -8,15 +8,19 @@ Puppet::Functions.create_function(:deprecation) do
   end
 
   def deprecation(key, message)
+    stacktrace = Puppet::Pops::PuppetStack.stacktrace()
+    file = stacktrace[0]
+    line = stacktrace[1]
+    output_message = "#{message} at #{file}:#{line}"
     # depending on configuration setting of strict
     case Puppet.settings[:strict]
     when :off
       # do nothing
     when :error
-      fail("deprecation. #{key}. #{message}")
+      fail("deprecation. #{key}. #{output_message}")
     else
       unless ENV['STDLIB_LOG_DEPRECATIONS'] == 'false'
-        Puppet.deprecation_warning(message, key)
+        Puppet.deprecation_warning(output_message, key)
       end
     end
   end