perfectline-rack-bug 0.1.1

data/lib/rack/bug/render.rb

@@ -0,0 +1,67 @@
+require "erb"
+
+module Rack
+  module Bug
+    
+    module Render
+      include ERB::Util
+      
+      def signed_params(hash)
+        # require "rubygems"; require "ruby-debug"; Debugger.start; debugger
+        ParamsSignature.sign(request, hash)
+      end
+      
+      module CompiledTemplates
+      end
+      include CompiledTemplates
+      
+      def render_template(filename, local_assigns = {})
+        compile(filename, local_assigns)
+        render_symbol = method_name(filename, local_assigns)
+        send(render_symbol, local_assigns)
+      end
+      
+      def compile(filename, local_assigns)
+        render_symbol = method_name(filename, local_assigns)
+        
+        if !CompiledTemplates.instance_methods.include?(render_symbol.to_s)
+          compile!(filename, local_assigns)
+        end
+      end
+      
+      def compile!(filename, local_assigns)
+        render_symbol = method_name(filename, local_assigns) 
+        locals_code = local_assigns.keys.map { |key| "#{key} = local_assigns[:#{key}];" }.join
+        
+        source = <<-end_src
+          def #{render_symbol}(local_assigns)
+            #{locals_code}
+            #{compiled_source(filename)}
+          end
+        end_src
+        
+        CompiledTemplates.module_eval(source, filename, 0)
+      end
+      
+      def compiled_source(filename)
+        ::ERB.new(::File.read(::File.dirname(__FILE__) + "/../bug/views/#{filename}.html.erb"), nil, "-").src
+      end
+      
+      def method_name(filename, local_assigns)
+        if local_assigns && local_assigns.any?
+          method_name = method_name_without_locals(filename).dup
+          method_name << "_locals_#{local_assigns.keys.map { |k| k.to_s }.sort.join('_')}"
+        else
+          method_name = method_name_without_locals(filename)
+        end
+        method_name.to_sym
+      end
+      
+      def method_name_without_locals(filename)
+        filename.split("/").join("_")
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/toolbar.rb

@@ -0,0 +1,145 @@
+require "ipaddr"
+require "digest"
+
+require "rack/bug/options"
+require "rack/bug/render"
+
+Dir[File.dirname(__FILE__) + "/panels/*.rb"].each do |panel_name|
+  require "rack/bug/panels/" + File.basename(panel_name)
+end
+
+module Rack
+  module Bug
+    
+    class RackStaticBugAvoider
+      def initialize(app, static_app)
+        @app = app
+        @static_app = static_app
+      end
+      
+      def call(env)
+        if env["PATH_INFO"]
+          @static_app.call(env)
+        else
+          @app.call(env)
+        end
+      end
+    end
+    
+    class Toolbar
+      include Options
+      include Render
+      
+      MIME_TYPES = ["text/html", "application/xhtml+xml"]
+      
+      def initialize(app, options = {})
+        @app = asset_server(app)
+        initialize_options options
+        instance_eval(&block) if block_given?
+      end
+      
+      def asset_server(app)
+        RackStaticBugAvoider.new(app, Rack::Static.new(app, :urls => ["/__rack_bug__"], :root => public_path))
+      end
+      
+      def public_path
+        ::File.expand_path(::File.dirname(__FILE__) + "/../bug/public")
+      end
+      
+      def call(env)
+        env.replace @default_options.merge(env)
+        @env = env
+        @original_request = Request.new(@env)
+
+        if toolbar_requested? && ip_authorized? && password_authorized?
+          dispatch
+        else
+          pass
+        end
+      end
+      
+      def pass
+        @app.call(@env)
+      end
+      
+      def dispatch
+        @env["rack-bug.panels"] = []
+        
+        Rack::Bug.enable
+        status, headers, body = builder.call(@env)
+        Rack::Bug.disable
+        
+        @response = Rack::Response.new(body, status, headers)
+        
+        if @response.redirect? && options["rack-bug.intercept_redirects"]
+          intercept_redirect
+        elsif modify?
+          inject_toolbar
+        end
+        
+        return @response.to_a
+      end
+      
+      def intercept_redirect
+        redirect_to = @response.location
+        new_body = render_template("redirect", :redirect_to => @response.location)
+        new_response = Rack::Response.new(new_body, 200, { "Content-Type" => "text/html" })
+        new_response["Content-Length"] = new_body.size.to_s
+        @response = new_response
+      end
+      
+      def toolbar_requested?
+        @original_request.cookies["rack_bug_enabled"]
+      end
+      
+      def ip_authorized?
+        return true unless options["rack-bug.ip_masks"]
+        
+        options["rack-bug.ip_masks"].any? do |ip_mask|
+          ip_mask.include?(IPAddr.new(@original_request.ip))
+        end
+      end
+      
+      def password_authorized?
+        return true unless options["rack-bug.password"]
+        
+        expected_sha = Digest::SHA1.hexdigest ["rack_bug", options["rack-bug.password"]].join(":")
+        actual_sha = @original_request.cookies["rack_bug_password"]
+        
+        actual_sha == expected_sha
+      end
+      
+      def modify?
+        @response.ok? &&
+        @env["X-Requested-With"] != "XMLHttpRequest" &&
+        MIME_TYPES.include?(@response.content_type.split(";").first)
+      end
+      
+      def builder
+        builder = Rack::Builder.new
+        
+        options["rack-bug.panel_classes"].each do |panel_class|
+          builder.use panel_class
+        end
+        
+        builder.run @app
+        
+        return builder
+      end
+      
+      def inject_toolbar
+        full_body = @response.body.join
+        full_body.sub! /<\/body>/, render + "</body>"
+        
+        @response["Content-Length"] = full_body.size.to_s
+        @response.body = [full_body]
+      end
+      
+      def render
+        render_template("toolbar", :panels => @env["rack-bug.panels"].reverse)
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/views/error.html.erb

@@ -0,0 +1,16 @@
+<script type="text/javascript" charset="utf-8">
+  if (typeof jQuery == 'undefined') {
+    var jquery_url = '/__rack_bug__/jquery-1.3.2.js';
+    document.write(unescape('%3Cscript src="' + jquery_url + '" type="text/javascript"%3E%3C/script%3E'));
+  }
+</script>
+<script type="text/javascript" src="/__rack_bug__/bug.js"></script>
+<style type="text/css" media="screen">
+  @import url(/__rack_bug__/bug.css);
+</style>
+
+<div id="rack_bug" class="rack_bug_error">
+  <div id="rack_bug_toolbar">
+    <p>There was an error within Rack::Bug!</p>
+  </div>
+</div>

data/lib/rack/bug/views/panels/active_record.html.erb

@@ -0,0 +1,17 @@
+<h3>ActiveRecord Objects</h3>
+<table>
+  <thead>
+    <tr>
+      <th>Count</th>
+      <th>Class</th>
+    </tr>
+  </thead>
+  <tbody>
+    <% records.each do |class_name, count| %>
+      <tr>
+        <td><%= count %></td>
+        <td><%= class_name %></td>
+      </tr>
+    <% end %>
+  </tbody>
+</table>

data/lib/rack/bug/views/panels/cache.html.erb

@@ -0,0 +1,93 @@
+<h3>Cache Usage</h3>
+<table id="cache_usage">
+  <colgroup>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+    <col width="12%"/>
+  </colgroup>
+  <tr>
+    <th>Total Calls</th>
+    <td><%= stats.calls %></td>
+    
+    <th>Total Time</th>
+    <td><%= stats.display_time %></td>
+    
+    <th>Hits</th>
+    <td><%= stats.hits %></td>
+    
+    <th>Misses</th>
+    <td><%= stats.misses %></td>
+  </tr>
+  <tr>
+    <th>gets</th>
+    <td><%= stats.gets %></td>
+    
+    <th>sets</th>
+    <td><%= stats.sets %></td>
+    
+    <th>deletes</th>
+    <td><%= stats.deletes %></td>
+    
+    <th>get_multis</th>
+    <td><%= stats.get_multis %></td>
+  </tr>
+</table>
+
+<% if stats.queries.any? %>
+  <h3>Breakdown</h3>
+  <table id="cache_breakdown">
+    <thead>
+      <tr>
+        <th>Time&nbsp;(ms)</th>
+        <th>Type</th>
+        <th>Parameters</th>
+        <th>Function</th>
+        <th>
+          <a href="/__rack_bug__/delete_cache_list?<%= signed_params(stats.queries_to_param) %>" class="rb_delete_cache">
+            Delete All
+          </a>
+        </th>
+      </tr>
+    </thead>
+    <tbody>
+      <% i = 1 %>
+      <% stats.queries.each do |query| %>
+        <tr class="<%= i % 2 == 0 ? "even" : "odd" %>">
+          <td><%= query.display_time %></td>
+          <td><%= query.method %></td>
+          <td><%= query.display_keys %></td>
+          <td></td>
+          <td>
+            <a href="/__rack_bug__/view_cache?<%= signed_params("key" => query.keys.first) %>" class="remote_call">View</a> |
+            <a href="/__rack_bug__/delete_cache?<%= signed_params("key" => query.keys.first) %>" class="rb_delete_cache">Delete</a>
+          </td>
+        </tr>
+        <% i += 1 %>
+      <% end %>
+    </tbody>
+  </table>
+<% end %>
+
+<script type="text/javascript" charset="utf-8">
+  jQuery(function () {
+    jQuery("#rack_bug .rb_delete_cache").click(function (evt) {
+      jQuery.ajax({
+        url: this.href,
+        beforeSend: function() {
+          jQuery(evt.target).parent("td, th").addClass("rack_bug_spinner");
+        },
+        success: function () {
+          jQuery(evt.target).parent("td, th").removeClass("rack_bug_spinner");
+          jQuery(evt.target).replaceWith("Deleted");
+        }
+      });
+
+      return false;
+    });
+  });
+</script>

data/lib/rack/bug/views/panels/env.html.erb

@@ -0,0 +1,19 @@
+<h3>Rack ENV</h3>
+<table>
+  <thead>
+    <tr>
+      <th>Variable</th>
+      <th>Value</th>
+    </tr>
+  </thead>
+  <tbody>
+    <% i = 1 %>
+    <% env.sort_by { |k, v| k.to_s }.each do |key, val| %>
+      <tr class="<%= i % 2 == 0 ? "even" : "odd" %>">
+        <td><%=h key %></td>
+        <td class="code"><div><%=h val %></div></td>
+      </tr>
+      <% i += 1 %>
+    <% end %>
+  </tbody>
+</table>

data/lib/rack/bug/views/panels/execute_sql.html.erb

@@ -0,0 +1,32 @@
+<a class="back" href="">&laquo;&nbsp;Back</a>
+
+<h3>SQL Results</h3>
+
+<dl>
+  <dt>Executed SQL</dt>
+  <dd><pre><%=h query %></pre></dd>
+  
+  <dt>Time</dt>
+  <dd><%=h "%.2f" % (time * 1_000) %>ms</dd>
+</dl>
+
+<table>
+  <thead>
+    <tr>
+      <% result.fetch_fields.each do |field| %>
+        <th><%= field.name.upcase %></th>
+      <% end %>
+    </tr>
+  </thead>
+  <tbody>
+    <% i = 1 %>
+    <% result.each do |row| %>
+      <tr class="<%= i % 2 == 0 ? "even" : "odd" %>">
+        <% row.each do |value| %>
+          <td><%= value %></td>
+        <% end %>
+      </tr>
+      <% i += 1 %>
+    <% end %>
+  </tbody>
+</table>

data/lib/rack/bug/views/panels/explain_sql.html.erb

@@ -0,0 +1,32 @@
+<a class="back" href="">&laquo;&nbsp;Back</a>
+
+<h3>SQL Explained</h3>
+
+<dl>
+  <dt>Executed SQL</dt>
+  <dd><pre><%=h query %></pre></dd>
+  
+  <dt>Time</dt>
+  <dd><%=h "%.2f" % (time * 1_000) %>ms</dd>
+</dl>
+
+<table>
+  <thead>
+    <tr>
+      <% result.fetch_fields.each do |field| %>
+        <th><%= field.name.upcase %></th>
+      <% end %>
+    </tr>
+  </thead>
+  <tbody>
+    <% i = 1 %>
+    <% result.each do |row| %>
+      <tr class="<%= i % 2 == 0 ? "even" : "odd" %>">
+        <% row.each do |value| %>
+          <td><%= value %></td>
+        <% end %>
+      </tr>
+      <% i += 1 %>
+    <% end %>
+  </tbody>
+</table>

data/lib/rack/bug/views/panels/log.html.erb

@@ -0,0 +1,23 @@
+<h3>Log Messages</h3>
+<table>
+  <thead>
+    <tr>
+      <th>Level</th>
+      <th>Time</th>
+      <th>Message</th>
+      <th>Location</th>
+    </tr>
+  </thead>
+  <tbody>
+    <% i = 1 %>
+    <% logs.each do |log| %>
+      <tr class="<%= i % 2 == 0 ? "even" : "odd" %>">
+        <td></td>
+        <td></td>
+        <td><%= log.to_s.gsub(/\e\[[;\d]+m/, "") %></td>
+        <td></td>
+      </tr>
+      <% i += 1 %>
+    <% end %>
+  </tbody>
+</table>