bumbler 0.1.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in bumbler.gemspec
+gemspec

data/README.md

@@ -0,0 +1,33 @@
+# Bumbler
+Why stare blankly at your terminal window when you can clutter it up with awesome progress bars?
+
+Use Bumbler to track the load progress of your [Bundler](http://gembundler.com/)-based projects!  Maybe you'll find a slow gem or two.
+
+
+## Using Bumbler
+### Step 1:
+
+    gem install bumbler
+
+### Step 1.5:
+Add bumbler to your Gemfile if you want to use `bundle exec`
+
+    gem 'bumbler'
+
+### Step 2:
+Add the following to your .profile, .bash_profile, .zshrc, .wtfrc or whatever shell config you use
+
+    export RUBYOPT=-rbumbler
+
+### Step 3:
+Restart your terminal
+
+
+## Blammo, you're bumbling with bundler and bumbler!
+Run a Bundler-based command, and you should see a spiffy progress bar, such as:
+
+    > rails c
+    [#########                                                                     ]
+    ( 7/59)  492.04ms loaded data_mapper 
+
+And then maybe you'll also want to contribute some patches to make your favorite gems load faster.

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/bumbler.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "bumbler/version"
+
+Gem::Specification.new do |s|
+  s.name        = "bumbler"
+  s.version     = Bumbler::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Ian MacLeod"]
+  s.email       = ["ian@nevir.net"]
+  s.homepage    = "https://github.com/nevir/Bumbler"
+  s.summary     = %q{Track the load progress of your Bundler-based projects}
+  s.description = %q{Why stare blankly at your terminal window when you can clutter it up with awesome progress bars?}
+
+  s.rubyforge_project = "bumbler"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/lib/bumbler.rb

@@ -0,0 +1,21 @@
+# Do nothing unless we're in a bundle
+begin
+  require 'bundler'
+  # This raises if there isn't a gemfile in our root
+  Bundler.default_gemfile
+  
+  module Bumbler
+    autoload :Hooks,    'bumbler/hooks'
+    autoload :Bundler,  'bumbler/bundler'
+    autoload :Progress, 'bumbler/progress'
+    
+    Hooks.hook_require!
+    Hooks.watch_require!
+    
+    Bundler.start!
+    Progress.start!
+  end
+  
+rescue
+  # Welp, if we fail, we fail.
+end

data/lib/bumbler/bundler.rb

@@ -0,0 +1,45 @@
+module Bumbler
+  module Bundler
+    # Returns which gem a require maps to, or nil.
+    def self.gem_for_require(path)
+      self.read_bundler_environment if @require_map.nil?
+      
+      return @require_map[path]
+    end
+    
+    def self.require_finished(path, load_time)
+      self.read_bundler_environment if @gem_state.nil?
+      
+      # Tick it off for the gem.
+      gem_name = self.gem_for_require(path)
+      return unless gem_name
+      
+      @gem_state[gem_name][path] = true
+      
+      if @gem_state[gem_name].values.all?
+        Bumbler::Progress.item_finished(:bundler, gem_name, load_time)
+      end
+    end
+    
+    def self.start!
+      self.read_bundler_environment
+    end
+    
+  private
+    def self.read_bundler_environment
+      @require_map = {}
+      @gem_state = {}
+      
+      ::Bundler.environment.current_dependencies.each do |spec|
+        @gem_state[spec.name] = {}
+        
+        Array(spec.autorequire || spec.name).each do |path|
+          @require_map[path] = spec.name
+          @gem_state[spec.name][path] = false
+        end
+        
+        Bumbler::Progress.register_item(:bundler, spec.name)
+      end
+    end
+  end
+end

data/lib/bumbler/hooks.rb

@@ -0,0 +1,67 @@
+module Bumbler
+  module Hooks
+    # Inject our custom handling of require into the Kernel.
+    def self.hook_require!
+      @hooking_require = true
+      
+      # There are two independent require methods.  Joy!
+      ::Kernel.module_eval do
+        class << self
+          orig_public_require = Kernel.public_method(:require)
+          define_method(:require) do |path, *args|
+            ::Bumbler::Hooks.handle_require(path) do
+              orig_public_require.call(path, *args)
+            end
+          end
+        end
+        
+        orig_instance_require = self.instance_method(:require)
+        define_method(:require) do |path, *args|
+          ::Bumbler::Hooks.handle_require(path) do
+            orig_instance_require.bind(self).call(path, *args)
+          end
+        end
+      end
+      
+      @hooking_require = nil
+    end
+    
+    # Even better: Other gems hook require as well.  The instance method one at least.
+    def self.watch_require!
+      ::Kernel.module_eval do
+        # It isn't previously defined in Kernel.  This could be a bit dangerous, though.
+        def self.method_added(method_name, *args)
+          if method_name == :require && !::Bumbler::Hooks.hooking_require?
+            # Fix those hooks.
+            ::Bumbler::Hooks.hook_require!
+          end
+        end
+      end
+    end
+    
+  private
+    def self.hooking_require?
+      @hooking_require
+    end
+    
+    # Actually do something about a require here.
+    def self.handle_require(path)
+      # break out early if we're already handling this
+      return yield if path == @previous_require
+      @previous_require = path
+      
+      # Shortcut unless we're tracking the gem
+      gem_name = Bumbler::Bundler.gem_for_require(path)
+      return yield unless gem_name
+      
+      # Let's time them
+      start = Time.now.to_f
+      result = yield
+      require_time = (Time.now.to_f - start) * 1000 # ms
+      
+      Bumbler::Bundler.require_finished(path, require_time) if result
+      
+      return result
+    end
+  end
+end

data/lib/bumbler/progress.rb

@@ -0,0 +1,60 @@
+module Bumbler
+  # Singletons are fun, yay!
+  module Progress
+    def self.register_item(type, name)
+      # Build a blank key for the item
+      unless self.registry[type][name]
+        @item_count ||= 0
+        @item_count  += 1
+      end
+      
+      self.registry[type][name] = {}
+    end
+    
+    def self.item_finished(type, name, time)
+      self.registry[type][name] = {:time => time}
+      
+      @loaded_items ||= 0
+      @loaded_items  += 1
+      
+      time_str = ('%.2fms' % time).rjust(9)
+      self.render_progress('%s loaded %s ' % [time_str, name])
+    end
+    
+    def self.start!
+      @loaded_items ||= 0
+      @item_count   ||= 0
+    end
+    
+  private
+    # registry[item_type][item_name] = {:time => 123.45}
+    def self.registry
+      @registry ||= Hash.new { |h,k| h[k] = {} }
+    end
+    
+    def self.tty_width
+      `tput cols`.to_i || 80
+    end
+    
+    def self.bar
+      inner_size = self.tty_width - 2
+      
+      fill_size = ((@loaded_items.to_f / @item_count.to_f) * inner_size).to_i
+      fill  = '#' * fill_size
+      empty = ' ' * (inner_size - fill_size)
+      
+      return "[#{fill}#{empty}]"
+    end
+    
+    def self.render_progress(message)
+      if $stdout.tty?
+        print "\r\e[A\r\e[K\r\e[A" if @outputted_once
+        @outputted_once = true
+        
+        puts self.bar
+      end
+      
+      puts '(%s/%d) %s' % [@loaded_items.to_s.rjust(@item_count.to_s.size), @item_count, message]
+    end
+  end
+end

data/lib/bumbler/version.rb

@@ -0,0 +1,3 @@
+module Bumbler
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification 
+name: bumbler
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Ian MacLeod
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-21 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Why stare blankly at your terminal window when you can clutter it up with awesome progress bars?
+email: 
+- ian@nevir.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bumbler.gemspec
+- lib/bumbler.rb
+- lib/bumbler/bundler.rb
+- lib/bumbler/hooks.rb
+- lib/bumbler/progress.rb
+- lib/bumbler/version.rb
+has_rdoc: true
+homepage: https://github.com/nevir/Bumbler
+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: bumbler
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Track the load progress of your Bundler-based projects
+test_files: []
+