class Object

# File lib/puppet-lint/plugins/no_erb_template.rb, line 2
def check
  functions = %w[template inline_template]

  function_tokens = tokens.select { |t| (t.type == :NAME or t.type == :FUNCTION_NAME) and functions.include? t.value }

  function_tokens.each do |function_token|
    next unless function_token.next_code_token.type == :LPAREN

    key_token = function_token.next_code_token.next_code_token

    # display the new, epp, version of the function
    original_function = function_token.value
    new_func = function_token.value.sub(/template/, 'epp')

    notify :warning, {
      message: "#{original_function}() function call. Use #{new_func}() instead",
      line:    key_token.line,
      column:  key_token.column,
    }
  end
end