class Object

# File lib/puppet-lint/plugins/non_erb_template_filename.rb, line 2
def check
  resource_indexes.each do |resource|
    if resource[:type].value == 'file'
      resource[:param_tokens].select { |param_token|
        param_token.value == 'content'
      }.each do |content_token|

        value_token = content_token.next_code_token.next_code_token

        if value_token.value == 'template'

          current_token = value_token.next_token

          # iterate over all the code tokens until we hit the closing ')'
          until current_token.type == :RPAREN
            current_token = current_token.next_code_token

            if current_token.type == :SSTRING && !current_token.value.end_with?('.erb')

              warning = 'all template file names should end with ".erb"'

              notify :warning, {
                :message     => warning,
                :line        => value_token.line,
                :column      => value_token.column,
                :param_token => content_token,
                :value_token => value_token,
              }
            end
          end
        end
      end
    end
  end
end