db_debug 0.0.1

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2012 YOURNAME
+
+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,3 @@
+= DbDebug
+
+This project rocks and uses MIT-LICENSE.

data/Rakefile

@@ -0,0 +1,27 @@
+#!/usr/bin/env rake
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+begin
+  require 'rdoc/task'
+rescue LoadError
+  require 'rdoc/rdoc'
+  require 'rake/rdoctask'
+  RDoc::Task = Rake::RDocTask
+end
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'DbDebug'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.rdoc')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+
+
+
+Bundler::GemHelper.install_tasks
+

data/lib/db_debug/action_controller.rb

@@ -0,0 +1,41 @@
+module  DbDebug
+  module ActionController
+    module Base
+      class << self
+        def included(klass)
+            klass.class_eval do
+            around_filter :hack_to_pass_shit_to_rack_middleware
+
+            protected
+
+            def hack_to_pass_shit_to_rack_middleware
+              klasses = [ActiveRecord::Base, ActiveRecord::Base.class]
+              methods = ["session", "cookies", "params", "request"]
+
+              methods.each do |shenanigan|
+                oops = instance_variable_get(:"@_#{shenanigan}") 
+
+                klasses.each do |klass|
+                  klass.send(:define_method, shenanigan, proc { oops })
+                end
+              end
+
+              yield
+
+              methods.each do |shenanigan|      
+                klasses.each do |klass|
+                  klass.send :remove_method, shenanigan
+                end
+              end
+            end 
+          end
+        end
+      end
+    end
+  end
+  
+end
+
+
+
+ActionController::Base.send(:include, DbDebug::ActionController::Base)

data/lib/db_debug/active_record.rb

@@ -0,0 +1,36 @@
+module DbDebug
+  module ActiveRecord
+    module Base
+      class << self
+        def included(klass)
+          klass.extend ClassMethods
+          klass.class_eval do
+            class << self
+              alias_method_chain :find_by_sql, :record_counter
+            end
+          end
+          include InstanceMethods
+        end
+      end
+
+      module ClassMethods
+        def find_by_sql_with_record_counter(*args)
+          resp = nil
+          request.env['db.count'] = request.env['db.count'] + 1
+          time = Benchmark.realtime { resp = find_by_sql_without_record_counter(*args) } * 1000
+          if request.params[:db_debug]
+            Logger.log "DB Query took: #{time} ms"
+            Logger.log caller.delete_if{ |str| str.include?("/gems/") || str.include?("script/rails") }.join("\n")
+          end
+          request.env['db.time'] = request.env['db.time'] + time
+          resp
+        end
+      end
+
+      module InstanceMethods
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.send(:include, DbDebug::ActiveRecord::Base)

data/lib/db_debug/engine.rb

@@ -0,0 +1,5 @@
+module DbDebug
+  class Engine < ::Rails::Engine
+    config.app_middleware.use DbDebug::Middleware
+  end
+end

data/lib/db_debug/logger.rb

@@ -0,0 +1,19 @@
+module DbDebug
+  module Logger
+    COLOR_MAP = {
+      :black => 30,
+      :red  => 31,
+      :green  => 32,
+      :yellow  => 33,
+      :blue  => 34,
+      :magenta  => 35,
+      :cyan  => 36,
+      :white => 37
+    }
+    
+    def self.log color, str
+      code = COLOR_MAP[color]
+      Rails.logger.debug "\033[0;#{code}m#{str}\033[0;#37m"      
+    end
+  end
+end

data/lib/db_debug/middleware.rb

@@ -0,0 +1,26 @@
+module DbDebug
+  class Middleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      env['db.count'] = 0
+      env['db.time'] = 0.0
+      resp = @app.call(env)
+      color = determine_color env['db.count'], env['db.time']
+      Logger.log color, "You hit the database #{env['db.count']} times and it took %5.3f ms" % env['db.time']
+      resp
+    end
+    
+    def determine_color count, time
+      if count <= 10 || time < 50.0
+        :green
+      elsif count <= 20 || time < 100.0
+        :yellow
+      else
+        :red
+      end
+    end
+  end
+end

data/lib/db_debug/version.rb

@@ -0,0 +1,3 @@
+module DbDebug
+  VERSION = "0.0.1"
+end

data/lib/db_debug.rb

@@ -0,0 +1,9 @@
+require "db_debug/middleware"
+require "db_debug/active_record"
+require "db_debug/action_controller"
+require "db_debug/version"
+require "db_debug/logger"
+require "db_debug/engine"
+
+module DbDebug
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification
+name: db_debug
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Phil Monroe
+- Identified
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+description: Description of DbDebug.
+email:
+- phil@identified.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/db_debug/action_controller.rb
+- lib/db_debug/active_record.rb
+- lib/db_debug/engine.rb
+- lib/db_debug/logger.rb
+- lib/db_debug/middleware.rb
+- lib/db_debug/version.rb
+- lib/db_debug.rb
+- MIT-LICENSE
+- Rakefile
+- README.rdoc
+homepage: https://github.com/Identified/db_debug
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Summary of DbDebug.
+test_files: []