lib/commands/dns.rb in yawast-0.6.0 vs lib/commands/dns.rb in yawast-0.7.0.beta1

@@ -4,13 +4,18 @@
       def self.process(args, options)
         uri = Yawast::Commands::Utils.extract_uri(args)
 
         Yawast.header
 
+        if options.output != nil
+          Yawast::Shared::Output.setup uri, options
+        end
+
         puts "Scanning: #{uri}"
         puts
 
         Yawast::Scanner::Plugins::DNS::Generic.dns_info uri, options
+        Yawast::Shared::Output.write_file
       end
     end
   end
 end