wested-rack-bug 0.2.2.1

data/.gitignore

@@ -0,0 +1,3 @@
+coverage
+pkg
+TODO

data/MIT-LICENSE.txt

@@ -0,0 +1,19 @@
+Copyright (c) 2009 Bryan Helmkamp
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,29 @@
+Usage:
+
+ script/plugin install git://github.com/brynary/rack-bug.git 
+
+ # config/environments/development.rb
+ config.middleware.use "Rack::Bug"
+
+ # add bookmarklet to browser
+ open http://RAILS_APP/__rack_bug__/bookmarklet.html
+
+Notes:
+
+ To use rack-bug with hoptoad, you need to stub hoptoad from trying to serialize the environment variables
+ that rack-bug uses like so:
+ 
+ HoptoadNotifier.configure do |config|
+   config.environment_filters << 'rack-bug.*'
+ end
+ 
+ 
+Thanks to:
+
+  Django debug toolbar
+  Rails footnotes
+  Rack's ShowException middleware
+  Oink
+  Rack::Cache
+  
+  

data/Rakefile

@@ -0,0 +1,20 @@
+require "rubygems"
+require "spec/rake/spectask"
+
+Spec::Rake::SpecTask.new do |t|
+  t.spec_opts = ['--options', "\"#{File.dirname(__FILE__)}/spec/spec.opts\""]
+end
+
+desc "Run all specs in spec directory with RCov"
+Spec::Rake::SpecTask.new(:rcov) do |t|
+  t.spec_opts = ['--options', "\"#{File.dirname(__FILE__)}/spec/spec.opts\""]
+  t.rcov = true
+  t.rcov_opts = lambda do
+    IO.readlines(File.dirname(__FILE__) + "/spec/rcov.opts").map {|l| l.chomp.split " "}.flatten
+  end
+end
+
+task :spec => :check_dependencies
+
+desc "Run the specs"
+task :default => :spec

data/Thorfile

@@ -0,0 +1,109 @@
+module GemHelpers
+
+  def generate_gemspec
+    $LOAD_PATH.unshift(File.expand_path(File.join(File.dirname(__FILE__), "lib")))
+    require "rack/bug"
+
+    Gem::Specification.new do |s|
+      s.name      = "rack-bug"
+      s.version   = Rack::Bug::VERSION
+      s.author    = "Bryan Helmkamp"
+      s.email     = "bryan@brynary.com"
+      s.homepage  = "http://github.com/brynary/rack-bug"
+      s.summary   = "Debugging toolbar for Rack applications implemented as middleware"
+      # s.description  = "TODO"
+      s.rubyforge_project = "rack-bug"
+
+      require "git"
+      repo = Git.open(".")
+
+      s.files      = normalize_files(repo.ls_files.keys - repo.lib.ignored_files)
+      s.test_files = normalize_files(Dir['spec/**/*.rb'] - repo.lib.ignored_files)
+
+      s.has_rdoc = true
+      s.extra_rdoc_files = %w[README.rdoc MIT-LICENSE.txt]
+
+      s.add_dependency "rack", ">= 1.0"
+    end
+  end
+
+  def normalize_files(array)
+    # only keep files, no directories, and sort
+    array.select do |path|
+      File.file?(path)
+    end.sort
+  end
+
+  # Adds extra space when outputting an array. This helps create better version
+  # control diffs, because otherwise it is all on the same line.
+  def prettyify_array(gemspec_ruby, array_name)
+    gemspec_ruby.gsub(/s\.#{array_name.to_s} = \[.+?\]/) do |match|
+      leadin, files = match[0..-2].split("[")
+      leadin + "[\n    #{files.split(",").join(",\n   ")}\n  ]"
+    end
+  end
+
+  def read_gemspec
+    @read_gemspec ||= eval(File.read("rack-bug.gemspec"))
+  end
+
+  def sh(command)
+    puts command
+    system command
+  end
+end
+
+class Default < Thor
+  include GemHelpers
+
+  desc "gemspec", "Regenerate rack-bug.gemspec"
+  def gemspec
+    File.open("rack-bug.gemspec", "w") do |file|
+      gemspec_ruby = generate_gemspec.to_ruby
+      gemspec_ruby = prettyify_array(gemspec_ruby, :files)
+      gemspec_ruby = prettyify_array(gemspec_ruby, :test_files)
+      gemspec_ruby = prettyify_array(gemspec_ruby, :extra_rdoc_files)
+
+      file.write gemspec_ruby
+    end
+
+    puts "Wrote gemspec to rack-bug.gemspec"
+    read_gemspec.validate
+  end
+
+  desc "build", "Build a rack-bug gem"
+  def build
+    sh "gem build rack-bug.gemspec"
+    FileUtils.mkdir_p "pkg"
+    FileUtils.mv read_gemspec.file_name, "pkg"
+  end
+
+  desc "install", "Install the latest built gem"
+  def install
+    sh "gem install --local pkg/#{read_gemspec.file_name}"
+  end
+
+  desc "release", "Release the current branch to GitHub and Gemcutter"
+  def release
+    gemspec
+    build
+    Release.new.tag
+    Release.new.gem
+  end
+end
+
+class Release < Thor
+  include GemHelpers
+
+  desc "tag", "Tag the gem on the origin server"
+  def tag
+    release_tag = "v#{read_gemspec.version}"
+    sh "git tag -a #{release_tag} -m 'Tagging #{release_tag}'"
+    sh "git push origin #{release_tag}"
+  end
+
+  desc "gem", "Push the gem to Gemcutter"
+  def gem
+    sh "gem push pkg/#{read_gemspec.file_name}"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.2.1

data/lib/rack/bug.rb

@@ -0,0 +1,43 @@
+require "rack"
+
+module Rack::Bug
+  autoload :Options,                "rack/bug/options"
+  autoload :Panel,                  "rack/bug/panel"
+  autoload :PanelApp,               "rack/bug/panel_app"
+  autoload :ParamsSignature,        "rack/bug/params_signature"
+  autoload :Render,                 "rack/bug/render"
+  autoload :Toolbar,                "rack/bug/toolbar"
+
+  # Panels
+  autoload :ActiveRecordPanel,      "rack/bug/panels/active_record_panel"
+  autoload :CachePanel,             "rack/bug/panels/cache_panel"
+  autoload :LogPanel,               "rack/bug/panels/log_panel"
+  autoload :MemoryPanel,            "rack/bug/panels/memory_panel"
+  autoload :RailsInfoPanel,         "rack/bug/panels/rails_info_panel"
+  autoload :RedisPanel,             "rack/bug/panels/redis_panel"
+  autoload :RequestVariablesPanel,  "rack/bug/panels/request_variables_panel"
+  autoload :SQLPanel,               "rack/bug/panels/sql_panel"
+  autoload :TemplatesPanel,         "rack/bug/panels/templates_panel"
+  autoload :TimerPanel,             "rack/bug/panels/timer_panel"
+
+  VERSION = "0.2.2.pre"
+
+  class SecurityError < StandardError
+  end
+
+  def self.enable
+    Thread.current["rack-bug.enabled"] = true
+  end
+
+  def self.disable
+    Thread.current["rack-bug.enabled"] = false
+  end
+
+  def self.enabled?
+    Thread.current["rack-bug.enabled"] == true
+  end
+
+  def self.new(*args, &block)
+    Toolbar.new(*args, &block)
+  end
+end

data/lib/rack/bug/options.rb

@@ -0,0 +1,89 @@
+module 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
+  module 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
+  module 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
+  module 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