class Object

# File lib/puppet-lint/plugins/check_file_ensure.rb, line 2
def check
  resource_indexes.each do |resource|
    if resource[:type].value == 'file'
      attr = resource[:tokens].select { |t| t.type == :NAME && \
                                            t.value == 'ensure' && \
                                            t.next_code_token.type == :FARROW }
      next if attr.empty?
      val_token = attr[0].next_code_token.next_code_token
      next unless val_token.value == 'present'
      next unless [:NAME, :STRING, :SSTRING].include? val_token.type
      notify :warning, {
        :message  => 'ensure set to present on file resource',
        :line     => val_token.line,
        :column   => val_token.column,
        :token    => val_token,
        :resource => resource,
      }
    end
  end
end

# File lib/puppet-lint/plugins/check_file_ensure.rb, line 23
def fix(problem)
  target_attr = problem[:resource][:tokens].select { |t| t.type == :NAME && \
                                                         t.value == 'target' && \
                                                         t.next_code_token.type == :FARROW }
  if target_attr.empty?
    problem[:token].value = 'file'
  else
    problem[:token].value = 'link'
  end
end