ghazel-rack-bug 0.3.0.1

data/lib/rack/bug/options.rb

@@ -0,0 +1,89 @@
+class Rack::Bug
+
+  module Options
+    class << self
+      private
+      def option_accessor(key)
+        define_method(key) { || read_option(key) }
+        define_method("#{key}=") { |value| write_option(key, value) }
+        define_method("#{key}?") { || !! read_option(key) }
+      end
+    end
+
+    option_accessor :secret_key
+    option_accessor :ip_masks
+    option_accessor :password
+    option_accessor :panel_classes
+    option_accessor :intercept_redirects
+
+    # The underlying options Hash. During initialization (or outside of a
+    # request), this is a default values Hash. During a request, this is the
+    # Rack environment Hash. The default values Hash is merged in underneath
+    # the Rack environment before each request is processed.
+    def options
+      @env || @default_options
+    end
+
+    # Set multiple options.
+    def options=(hash={})
+      hash.each { |key,value| write_option(key, value) }
+    end
+
+    # Set an option. When +option+ is a Symbol, it is set in the Rack
+    # Environment as "rack-cache.option". When +option+ is a String, it
+    # exactly as specified. The +option+ argument may also be a Hash in
+    # which case each key/value pair is merged into the environment as if
+    # the #set method were called on each.
+    def set(option, value=self, &block)
+      if block_given?
+        write_option option, block
+      elsif value == self
+        self.options = option.to_hash
+      else
+        write_option option, value
+      end
+    end
+
+  private
+
+    def read_option(key)
+      options[option_name(key)]
+    end
+
+    def write_option(key, value)
+      options[option_name(key)] = value
+    end
+
+    def option_name(key)
+      case key
+      when Symbol ; "rack-bug.#{key}"
+      when String ; key
+      else raise ArgumentError
+      end
+    end
+
+    def initialize_options(options={})
+      @default_options = {
+        'rack-bug.ip_masks' => [IPAddr.new("127.0.0.1")],
+        'rack-bug.password' => nil,
+        'rack-bug.verbose'  => nil,
+        'rack-bug.secret_key' => nil,
+        'rack-bug.intercept_redirects' => false,
+        'rack-bug.panels' => [],
+        'rack-bug.panel_classes' => [
+          RailsInfoPanel,
+          TimerPanel,
+          RequestVariablesPanel,
+          SQLPanel,
+          ActiveRecordPanel,
+          CachePanel,
+          TemplatesPanel,
+          LogPanel,
+          MemoryPanel
+        ]
+      }
+      self.options = options
+    end
+
+  end
+end

data/lib/rack/bug/panel.rb

@@ -0,0 +1,50 @@
+require "erb"
+
+module Rack
+  class Bug
+
+    # Panels are also Rack middleware
+    class Panel
+      include Render
+      include ERB::Util
+
+      attr_reader :request
+
+      def initialize(app)
+        if panel_app
+          @app = Rack::Cascade.new([panel_app, app])
+        else
+          @app = app
+        end
+      end
+
+      def call(env)
+        before(env)
+        status, headers, body = @app.call(env)
+        @request = Request.new(env)
+        after(env, status, headers, body)
+        env["rack-bug.panels"] << self
+        return [status, headers, body]
+      end
+
+      def panel_app
+        nil
+      end
+
+      def has_content?
+        true
+      end
+
+      def before(env)
+      end
+
+      def after(env, status, headers, body)
+      end
+
+      def render(template)
+      end
+
+    end
+
+  end
+end

data/lib/rack/bug/panel_app.rb

@@ -0,0 +1,33 @@
+module Rack
+  class Bug
+
+    class PanelApp
+      include Rack::Bug::Render
+
+      attr_reader :request
+
+      def call(env)
+        @request = Rack::Request.new(env)
+        dispatch
+      end
+
+      def render_template(*args)
+        Rack::Response.new([super]).to_a
+      end
+
+      def params
+        @request.GET
+      end
+
+      def not_found
+        [404, {}, []]
+      end
+
+      def validate_params
+        ParamsSignature.new(request).validate!
+      end
+
+    end
+
+  end
+end

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

@@ -0,0 +1,45 @@
+require "rack/bug/panels/active_record_panel/activerecord_extensions"
+
+module Rack
+  class Bug
+
+    class ActiveRecordPanel < Panel
+
+      def self.record(class_name)
+        return unless Rack::Bug.enabled?
+        records[class_name] += 1
+      end
+
+      def self.reset
+        Thread.current["rack.bug.active_records"] = Hash.new { 0 }
+      end
+
+      def self.records
+        Thread.current["rack.bug.active_records"] ||= Hash.new { 0 }
+      end
+
+      def self.total
+        records.inject(0) do |memo, (key, value)|
+          memo + value
+        end
+      end
+
+      def name
+        "active_record"
+      end
+
+      def heading
+        "#{self.class.total} AR Objects"
+      end
+
+      def content
+        records = self.class.records.to_a.sort_by { |key, value| value }.reverse
+        result = render_template "panels/active_record", :records => records
+        self.class.reset
+        result
+      end
+
+    end
+
+  end
+end

data/lib/rack/bug/panels/active_record_panel/activerecord_extensions.rb

@@ -0,0 +1,18 @@
+if defined?(ActiveRecord)
+  ActiveRecord::Base.class_eval do
+
+    if instance_methods.include?("after_initialize")
+      def after_initialize_with_rack_bug
+        Rack::Bug::ActiveRecordPanel.record(self.class.base_class.name)
+        after_initialize_without_rack_bug
+      end
+
+      alias_method_chain :after_initialize, :rack_bug
+    else
+      def after_initialize
+        Rack::Bug::ActiveRecordPanel.record(self.class.base_class.name)
+      end
+    end
+
+  end
+end

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

@@ -0,0 +1,51 @@
+require "rack/bug/panels/cache_panel/memcache_extension"
+require "rack/bug/panels/cache_panel/dalli_extension"
+
+module Rack
+  class Bug
+
+    class CachePanel < Panel
+      autoload :PanelApp, "rack/bug/panels/cache_panel/panel_app"
+      autoload :Stats,    "rack/bug/panels/cache_panel/stats"
+
+      def self.record(method, *keys, &block)
+        return block.call unless Rack::Bug.enabled?
+
+        start_time = Time.now
+        result = block.call
+        total_time = Time.now - start_time
+        hit = result.nil? ? false : true
+        stats.record_call(method, total_time * 1_000, hit, *keys)
+        return result
+      end
+
+      def self.reset
+        Thread.current["rack.bug.cache"] = Stats.new
+      end
+
+      def self.stats
+        Thread.current["rack.bug.cache"] ||= Stats.new
+      end
+
+      def panel_app
+        PanelApp.new
+      end
+
+      def name
+        "cache"
+      end
+
+      def heading
+        "Cache: %.2fms (#{self.class.stats.queries.size} calls)" % self.class.stats.time
+      end
+
+      def content
+        result = render_template "panels/cache", :stats => self.class.stats
+        self.class.reset
+        return result
+      end
+
+    end
+
+  end
+end

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

@@ -0,0 +1,16 @@
+begin
+
+  require 'dalli'
+
+  Dalli::Client.class_eval do
+    def perform_with_rack_bug(op, *args)
+      Rack::Bug::CachePanel.record(op, args.first) do
+        perform_without_rack_bug(op, *args)
+      end
+    end
+    
+    alias_method_chain :perform, :rack_bug
+  end
+    
+rescue NameError, LoadError
+end

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

@@ -0,0 +1,129 @@
+if defined?(Memcached)
+  Memcached.class_eval do
+
+    def set_with_rack_bug(key, value, timeout=0, marshal=true)
+      Rack::Bug::CachePanel.record(:set, key) do
+        set_without_rack_bug(key, value, timeout, marshal)
+      end
+    end
+
+    def add_with_rack_bug(key, value, timeout=0, marshal=true)
+      Rack::Bug::CachePanel.record(:add, key) do
+        add_without_rack_bug(key, value, timeout, marshal)
+      end
+    end
+
+    def increment_with_rack_bug(key, offset=1)
+      Rack::Bug::CachePanel.record(:incr, key) do
+        increment_without_rack_bug(key, offset)
+      end
+    end
+
+    def decrement_with_rack_bug(key, offset=1)
+      Rack::Bug::CachePanel.record(:decr, key) do
+        decrement_without_rack_bug(key, offset)
+      end
+    end
+
+    def replace_with_rack_bug(key, value, timeout=0, marshal=true)
+      Rack::Bug::CachePanel.record(:replace, key) do
+        replace_without_rack_bug(key, value, timeout, marshal)
+      end
+    end
+
+    def append_with_rack_bug(key, value)
+      Rack::Bug::CachePanel.record(:append, key) do
+        append_without_rack_bug(key, value)
+      end
+    end
+
+    def prepend_with_rack_bug(key, value)
+      Rack::Bug::CachePanel.record(:prepend, key) do
+        prepend_without_rack_bug(key, value)
+      end
+    end
+
+    def delete_with_rack_bug(key)
+      Rack::Bug::CachePanel.record(:delete, key) do
+        delete_without_rack_bug(key)
+      end
+    end
+
+    def get_with_rack_bug(keys, marshal=true)
+      if keys.is_a? Array
+        Rack::Bug::CachePanel.record(:get_multi, *keys) do
+          get_without_rack_bug(keys, marshal)
+        end
+      else
+        Rack::Bug::CachePanel.record(:get, keys) do
+          get_without_rack_bug(keys, marshal)
+        end
+      end
+    end
+
+    alias_method_chain :decrement,  :rack_bug
+    alias_method_chain :get,        :rack_bug
+    alias_method_chain :increment,  :rack_bug
+    alias_method_chain :set,        :rack_bug
+    alias_method_chain :add,        :rack_bug
+    alias_method_chain :replace,    :rack_bug
+    alias_method_chain :delete,     :rack_bug
+    alias_method_chain :prepend,    :rack_bug
+    alias_method_chain :append,     :rack_bug
+  end
+end
+
+if defined?(MemCache)
+  MemCache.class_eval do
+
+    def decr_with_rack_bug(key, amount = 1)
+      Rack::Bug::CachePanel.record(:decr, key) do
+        decr_without_rack_bug(key, amount)
+      end
+    end
+
+    def get_with_rack_bug(key, raw = false)
+      Rack::Bug::CachePanel.record(:get, key) do
+        get_without_rack_bug(key, raw)
+      end
+    end
+
+    def get_multi_with_rack_bug(*keys)
+      Rack::Bug::CachePanel.record(:get_multi, *keys) do
+        get_multi_without_rack_bug(*keys)
+      end
+    end
+
+    def incr_with_rack_bug(key, amount = 1)
+      Rack::Bug::CachePanel.record(:incr, key) do
+        incr_without_rack_bug(key, amount)
+      end
+    end
+
+    def set_with_rack_bug(key, value, expiry = 0, raw = false)
+      Rack::Bug::CachePanel.record(:set, key) do
+        set_without_rack_bug(key, value, expiry, raw)
+      end
+    end
+
+    def add_with_rack_bug(key, value, expiry = 0, raw = false)
+      Rack::Bug::CachePanel.record(:add, key) do
+        add_without_rack_bug(key, value, expiry, raw)
+      end
+    end
+
+    def delete_with_rack_bug(key, expiry = 0)
+      Rack::Bug::CachePanel.record(:delete, key) do
+        delete_without_rack_bug(key, expiry)
+      end
+    end
+
+    alias_method_chain :decr,       :rack_bug
+    alias_method_chain :get,        :rack_bug
+    alias_method_chain :get_multi,  :rack_bug
+    alias_method_chain :incr,       :rack_bug
+    alias_method_chain :set,        :rack_bug
+    alias_method_chain :add,        :rack_bug
+    alias_method_chain :delete,     :rack_bug
+  end
+end

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

@@ -0,0 +1,48 @@
+module Rack
+  class 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