reports/plugin_formatters/xml/healthmap.rb in arachni-0.3 vs reports/plugin_formatters/xml/healthmap.rb in arachni-0.4

@@ -1,8 +1,8 @@
 =begin
                   Arachni
-  Copyright (c) 2010-2011 Tasos "Zapotek" Laskos <[e-mail]>
+  Copyright (c) 2010-2012 Tasos "Zapotek" Laskos <[e-mail]>
 
   This is free software; you can copy and distribute and modify
   this program under the term of the GPL v2.0 License
   (See LICENSE file for details)
 
@@ -13,70 +13,65 @@
 require Arachni::Options.instance.dir['reports'] + '/xml/buffer.rb'
 
 module Reports
 
 class XML
-    module PluginFormatters
+module PluginFormatters
 
-        #
-        # XML formatter for the results of the HealthMap plugin
-        #
-        # @author: Tasos "Zapotek" Laskos
-        #                                      <[e-mail]>
-        #                                      <[e-mail]>
-        # @version: 0.1
-        #
-        class HealthMap < Arachni::Plugin::Formatter
+    #
+    # XML formatter for the results of the HealthMap plugin
+    #
+    # @author: Tasos "Zapotek" Laskos
+    #                                      <[e-mail]>
+    #                                      <[e-mail]>
+    # @version: 0.1
+    #
+    class HealthMap < Arachni::Plugin::Formatter
 
-            include Buffer
+        include Buffer
 
-            def initialize( plugin_data )
-                @results     = plugin_data[:results]
-                @description = plugin_data[:description]
-            end
+        def run
+            start_tag( 'healthmap' )
+            simple_tag( 'description', @description )
 
-            def run
-                start_tag( 'healthmap' )
-                simple_tag( 'description', @description )
+            start_tag( 'results' )
+            start_tag( 'map' )
+            @results[:map].each {
+                |i|
 
-                start_tag( 'results' )
-                start_tag( 'map' )
-                @results[:map].each {
-                    |i|
+                state = i.keys[0]
+                url   = i.values[0]
 
-                    state = i.keys[0]
-                    url   = i.values[0]
+                if state == :unsafe
+                    add_url( 'unsafe', url )
+                else
+                    add_url( 'safe', url )
+                end
+            }
+            end_tag( 'map' )
 
-                    if state == :unsafe
-                        add_url( 'unsafe', url )
-                    else
-                        add_url( 'safe', url )
-                    end
-                }
-                end_tag( 'map' )
+            start_tag( 'stats' )
 
-                start_tag( 'stats' )
+            simple_tag( 'total', @results[:total].to_s )
+            simple_tag( 'safe', @results[:safe].to_s )
+            simple_tag( 'unsafe', @results[:unsafe].to_s )
+            simple_tag( 'issue_percentage', @results[:issue_percentage].to_s )
 
-                simple_tag( 'total', @results[:total].to_s )
-                simple_tag( 'safe', @results[:safe].to_s )
-                simple_tag( 'unsafe', @results[:unsafe].to_s )
-                simple_tag( 'issue_percentage', @results[:issue_percentage].to_s )
+            end_tag( 'stats' )
+            end_tag( 'results' )
+            end_tag( 'healthmap' )
 
-                end_tag( 'stats' )
-                end_tag( 'results' )
-                end_tag( 'healthmap' )
+            return buffer( )
+        end
 
-                return buffer( )
-            end
-
-            def add_url( type, url )
-                __buffer( "<entry state=\"#{type}\" url=\"#{url}\" />" )
-            end
-
-
+        def add_url( type, url )
+            __buffer( "<entry state=\"#{type}\" url=\"#{url}\" />" )
         end
 
+
     end
+
+end
 end
 
 end
 end