Ordering NAPTR records with Dnsruby

dnsruby-naptr-ordering.rb

require 'rubygems'
require 'dnsruby'

include Dnsruby
$nameserver = Resolver.new

def dig domain, rrtype
        search = $nameserver.query(domain, Types.send(rrtype))
        if rrtype == :NAPTR
                search.answer.sort_by { |a| [ a.order, a.preference ] }.each { |rr|
                        search.answer.delete rr
                        search.add_answer rr
                }
        end
        search
end