reports/plugin_formatters/html/cookie_collector.rb in arachni-0.3 vs reports/plugin_formatters/html/cookie_collector.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)
 
@@ -11,56 +11,48 @@
 module Arachni
 
 module Reports
 
 class HTML
-    module PluginFormatters
+module PluginFormatters
 
-        #
-        # HTML formatter for the results of the CookieCollector plugin
-        #
-        # @author: Tasos "Zapotek" Laskos
-        #                                      <[e-mail]>
-        #                                      <[e-mail]>
-        # @version: 0.1
-        #
-        class CookieCollector < Arachni::Plugin::Formatter
+    #
+    # HTML formatter for the results of the CookieCollector plugin
+    #
+    # @author: Tasos "Zapotek" Laskos
+    #                                      <[e-mail]>
+    #                                      <[e-mail]>
+    # @version: 0.1
+    #
+    class CookieCollector < Arachni::Plugin::Formatter
 
-            def initialize( plugin_data )
-                @results     = plugin_data[:results]
-                @description = plugin_data[:description]
-            end
+        def run
+            return ERB.new( tpl ).result( binding )
+        end
 
-            def run
-                return ERB.new( tpl ).result( binding )
-            end
-
-            def tpl
-                %q{
-                    <h3>Cookie Collector</h3>
-                    <blockquote><%=@description%></blockquote>
-
-                    <h4>Cookies</h4>
-                    <ul>
-                    <% @results.each do |entry| %>
-                        <li>
-                            On <%=entry[:time].to_s%> @ <a href="<%=CGI.escapeHTML(entry[:res]['effective_url'])%>"><%=CGI.escapeHTML(entry[:res]['effective_url'])%></a>
-                            <br/>
-                            Cookies were forced to:
-                            <ul>
-                                <% entry[:cookies].each_pair do |name, val| %>
-                                    <li><%=CGI.escapeHTML(name)%> = <%=CGI.escapeHTML(val)%></li>
-                                <%end%>
-                            </ul>
-                        </li>
-                    <%end%>
-                    </ul>
-                }
-            end
-
+        def tpl
+            %q{
+                <h3>Cookies</h3>
+                <ul>
+                <% @results.each do |entry| %>
+                    <li>
+                        On <%=entry[:time].to_s%> @ <a href="<%=CGI.escapeHTML(entry[:res]['effective_url'])%>"><%=CGI.escapeHTML(entry[:res]['effective_url'])%></a>
+                        <br/>
+                        Cookies were forced to:
+                        <ul>
+                            <% entry[:cookies].each_pair do |name, val| %>
+                                <li><%=CGI.escapeHTML(name)%> = <%=CGI.escapeHTML(val)%></li>
+                            <%end%>
+                        </ul>
+                    </li>
+                <%end%>
+                </ul>
+            }
         end
 
     end
+
+end
 end
 
 end
 end