perfectline-rack-bug 0.1.1

data/lib/rack/bug/panels/cache_panel/panel_app.rb

@@ -0,0 +1,50 @@
+require "rack/bug/panel_app"
+
+module Rack
+  module Bug
+    class CachePanel
+      
+      class PanelApp < ::Rack::Bug::PanelApp
+        
+        def dispatch
+          case request.path_info
+          when "/__rack_bug__/view_cache"         then view_cache
+          when "/__rack_bug__/delete_cache"       then delete_cache
+          when "/__rack_bug__/delete_cache_list"  then delete_cache_list
+          else not_found
+          end
+        end
+        
+        def ok
+          Rack::Response.new(["OK"]).to_a
+        end
+        
+        def view_cache
+          validate_params
+          render_template "panels/view_cache", :key => params["key"], :value => Rails.cache.read(params["key"])
+        end
+        
+        def delete_cache
+          validate_params
+          raise "Rails not found... can't delete key" unless defined?(Rails)
+          Rails.cache.delete(params["key"])
+          ok
+        end
+        
+        def delete_cache_list
+          validate_params
+          raise "Rails not found... can't delete key" unless defined?(Rails)
+          
+          params.each do |key, value|
+            next unless key =~ /^keys_/
+            Rails.cache.delete(value)
+          end
+          
+          ok
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/rack/bug/panels/cache_panel/stats.rb

@@ -0,0 +1,97 @@
+module Rack
+  module Bug
+    class CachePanel
+      
+      class Stats
+        class Query
+          attr_reader :method, :time, :hit, :keys
+          
+          def initialize(method, time, hit, keys)
+            @method = method
+            @time = time
+            @hit = hit
+            @keys = keys
+          end
+          
+          def display_time
+            "%.2fms" % time
+          end
+          
+          def display_keys
+            if keys.size == 1
+              keys.first
+            else
+              keys.join(", ")
+            end
+          end
+        end
+        
+        attr_reader :calls
+        attr_reader :keys
+        attr_reader :queries
+        
+        def initialize
+          @queries = []
+          @misses = 
+          @calls = 0
+          @time = 0.0
+          @keys = []
+        end
+        
+        def record_call(method, time, hit, *keys)
+          @queries << Query.new(method, time, hit, keys)
+          @calls += 1
+          @time += time
+          @keys += keys
+        end
+        
+        def display_time
+          "%.2fms" % time
+        end
+        
+        def time
+          @queries.inject(0) do |memo, query|
+            memo + query.time
+          end
+        end
+        
+        def gets
+          count_queries(:get)
+        end
+        
+        def sets
+          count_queries(:set)
+        end
+        
+        def deletes
+          count_queries(:delete)
+        end
+        
+        def get_multis
+          count_queries(:get_multi)
+        end
+        
+        def hits
+          @queries.select { |q| [:get, :get_multi].include?(q.method) && q.hit }.size
+        end
+        
+        def misses
+          @queries.select { |q| [:get, :get_multi].include?(q.method) && !q.hit }.size
+        end
+        
+        def count_queries(method)
+          @queries.select { |q| q.method == method }.size
+        end
+        
+        def queries_to_param
+          params = {}
+          @queries.each_with_index do |query, index|
+            params["keys_#{index}"] = query.keys.first
+          end
+          params
+        end
+      end
+      
+    end
+  end
+end

data/lib/rack/bug/panels/env_panel.rb

@@ -0,0 +1,25 @@
+module Rack
+  module Bug
+    
+    class EnvPanel < Panel
+      
+      def name
+        "env"
+      end
+      
+      def before(env)
+        @env = env
+      end
+      
+      def heading
+        "Rack Env"
+      end
+      
+      def content
+        render_template "panels/env", :env => @env
+      end
+
+    end
+    
+  end
+end

data/lib/rack/bug/panels/log_panel.rb

@@ -0,0 +1,39 @@
+require "rack/bug/panels/log_panel/rails_extension"
+
+module Rack
+  module Bug
+    
+    class LogPanel < Panel
+      
+      def self.record(message)
+        return unless Rack::Bug.enabled?
+        return unless message
+        logs << message.to_s
+      end
+      
+      def self.reset
+        Thread.current["rack.bug.logs"] = []
+      end
+      
+      def self.logs
+        Thread.current["rack.bug.logs"] ||= []
+      end
+      
+      def name
+        "log"
+      end
+      
+      def heading
+        "Log"
+      end
+
+      def content
+        result = render_template "panels/log", :logs => self.class.logs
+        self.class.reset
+        return result
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/panels/log_panel/rails_extension.rb

@@ -0,0 +1,11 @@
+if defined?(Rails) && Rails.logger
+  module LoggingExtensions
+    def add(*args, &block)
+      logged_message = super
+      Rack::Bug::LogPanel.record(logged_message)
+      return logged_message
+    end
+  end
+
+  Rails.logger.extend LoggingExtensions
+end

data/lib/rack/bug/panels/memory_panel.rb

@@ -0,0 +1,27 @@
+# 
+module Rack
+  module Bug
+    
+    class MemoryPanel < Panel
+      
+      def before(env)
+        @original_memory = `ps -o rss= -p #{$$}`.to_i
+      end
+      
+      def after(env, status, headers, body)
+        @total_memory = `ps -o rss= -p #{$$}`.to_i
+        @memory_increase = @total_memory - @original_memory
+      end
+      
+      def heading
+        "#{@memory_increase} KB &#916;, #{@total_memory} KB total"
+      end
+
+      def has_content?
+        false
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/panels/rails_info_panel.rb

@@ -0,0 +1,23 @@
+module Rack
+  module Bug
+    
+    class RailsInfoPanel < Panel
+      
+      def name
+        "rails_info"
+      end
+      
+      def heading
+        return unless defined?(Rails)
+        "Rails #{Rails.version}"
+      end
+
+      def content
+        return unless defined?(Rails)
+        render_template "panels/rails_info"
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/panels/request_variables_panel.rb

@@ -0,0 +1,25 @@
+module Rack
+  module Bug
+    
+    class RequestVariablesPanel < Panel
+      
+      def name
+        "request_variables"
+      end
+      
+      def before(env)
+        @env = env
+      end
+      
+      def heading
+        "Request Vars"
+      end
+      
+      def content
+        render_template "panels/request_variables", :request => @request
+      end
+
+    end
+    
+  end
+end

data/lib/rack/bug/panels/sql_panel.rb

@@ -0,0 +1,55 @@
+require "digest"
+
+module Rack
+  module Bug
+    
+    class SQLPanel < Panel
+      
+      require "rack/bug/panels/sql_panel/sql_extension"
+      require "rack/bug/panels/sql_panel/query"
+      require "rack/bug/panels/sql_panel/panel_app"
+      
+      def panel_app
+        PanelApp.new
+      end
+      
+      def self.record(sql, backtrace = [], &block)
+        return block.call unless Rack::Bug.enabled?
+        
+        start_time = Time.now
+        result = block.call
+        queries << Query.new(sql, Time.now - start_time, backtrace)
+        
+        return result
+      end
+      
+      def self.reset
+        Thread.current["rack.test.queries"] = []
+      end
+      
+      def self.queries
+        Thread.current["rack.test.queries"] ||= []
+      end
+      
+      def self.total_time
+        (queries.inject(0) { |memo, query| memo + query.time}) * 1_000
+      end
+      
+      def name
+        "sql"
+      end
+      
+      def heading
+        "#{self.class.queries.size} Queries (%.2fms)" % self.class.total_time
+      end
+
+      def content
+        result = render_template "panels/sql", :queries => self.class.queries
+        self.class.reset
+        return result
+      end
+      
+    end
+    
+  end
+end

data/lib/rack/bug/panels/sql_panel/panel_app.rb

@@ -0,0 +1,39 @@
+require "rack/bug/panel_app"
+
+module Rack
+  module Bug
+    class SQLPanel
+      
+      class PanelApp < ::Rack::Bug::PanelApp
+  
+        def dispatch
+          case request.path_info
+          when "/__rack_bug__/explain_sql" then explain_sql
+          when "/__rack_bug__/profile_sql" then profile_sql
+          when "/__rack_bug__/execute_sql" then execute_sql
+          else not_found
+          end
+        end
+        
+        def explain_sql
+          validate_params
+          query = Query.new(params["query"], params["time"].to_f)
+          render_template "panels/explain_sql", :result => query.explain, :query => query.sql, :time => query.time
+        end
+  
+        def profile_sql
+          validate_params
+          query = Query.new(params["query"], params["time"].to_f)
+          render_template "panels/profile_sql", :result => query.profile, :query => query.sql, :time => query.time
+        end
+  
+        def execute_sql
+          validate_params
+          query = Query.new(params["query"], params["time"].to_f)
+          render_template "panels/execute_sql", :result => query.execute, :query => query.sql, :time => query.time
+        end
+        
+      end
+    end
+  end
+end

data/lib/rack/bug/panels/sql_panel/query.rb

@@ -0,0 +1,73 @@
+module Rack
+  module Bug
+    class SQLPanel
+      
+      class Query
+        attr_reader :sql
+        attr_reader :time
+        attr_reader :backtrace
+        
+        def initialize(sql, time, backtrace = [])
+          @sql = sql
+          @time = time
+          @backtrace = backtrace
+        end
+        
+        def human_time
+          "%.2fms" % (@time * 1_000)
+        end
+
+        def inspectable?
+          sql.strip =~ /^SELECT /i
+        end
+        
+        def with_profiling
+          self.class.execute("SET PROFILING=1")
+          result = yield
+          self.class.execute("SET PROFILING=0")
+          return result
+        end
+        
+        def explain
+          self.class.execute "EXPLAIN #{@sql}"
+        end
+        
+        def profile
+          with_profiling do
+            execute
+            self.class.execute <<-SQL
+              SELECT *
+                FROM information_schema.profiling
+               WHERE query_id = (SELECT query_id FROM information_schema.profiling ORDER BY query_id DESC LIMIT 1)
+            SQL
+          end
+        end
+        
+        def execute
+          self.class.execute(@sql)
+        end
+        
+        def valid_hash?(secret_key, possible_hash)
+          hash = Digest::SHA1.hexdigest [secret_key, @sql].join(":")
+          possible_hash == hash
+        end
+        
+        def self.execute(sql)
+          ActiveRecord::Base.connection.execute(sql)
+        end
+        
+        def has_backtrace?
+          filtered_backtrace.any?
+        end
+        
+        def filtered_backtrace
+          @filtered_backtrace ||= @backtrace.map { |l| l.to_s.strip }.select do |line|
+            line.starts_with?(Rails.root) &&
+            !line.starts_with?(Rails.root.join("vendor"))
+          end
+        end
+      end
+      
+    end
+  end
+end