starttime 0.0.2

data/bin/starttime

@@ -0,0 +1,167 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'erb'
+
+options = {
+  :output_format => 'flat'
+}
+files = []
+
+opts = OptionParser.new do |opts|
+  opts.banner = "Usage: starttime [options] file(s)"
+  
+  opts.on "-r", "--rails", "Profile environment time for a Rails project. Implies config/environment as a file to require" do
+    files << 'config/environment'
+  end
+  
+  opts.on "-o", "--output=FILENAME", "Output to FILENAME. Default is STDOUT." do |filename|
+    options[:output_file] = filename
+  end
+  
+  opts.on "-f", "--format=FORMAT", "Choose output format. Available are html and flat." do |format|
+    options[:output_format] = format
+  end
+  
+  opts.on "-e", "--execute=COMMAND", "Execute a Ruby command" do |command|
+    options[:command] = command
+  end
+end
+
+opts.parse!(ARGV)
+
+unless options[:command]
+  files += ARGV unless ARGV.empty?
+
+  if files.empty?
+    puts opts.help
+    exit
+  end
+end
+
+class SourceFile
+  attr_accessor :name, :duration
+  
+  def initialize(name)
+    @name = name
+  end
+  
+  def children
+    @children ||= []
+  end
+  
+  def children_sorted_by_duration
+    children.sort_by { |c| c.duration }.reverse
+  end
+  
+  def own_time
+    duration - children.collect { |c| c.duration }.sum rescue 0
+  end
+
+  class BasePrinter
+    def initialize(file, output)
+      @file = file
+      @output = output
+    end
+  end
+  
+  class FlatPrinter < BasePrinter
+    def print
+      print_with_indent(0, @file)
+    end
+  
+    def print_with_indent(num, file)
+      @output.puts((" " * num) + "#{file.name} (#{file.duration})")
+      file.children_sorted_by_duration.each do |child|
+        print_with_indent(num + 2, child)
+      end
+    end
+  end
+  
+  class HTMLPrinter < BasePrinter
+    TEMPLATE_PATH = File.dirname(__FILE__) + '/../templates'
+    
+    def print
+      @output.puts render("#{TEMPLATE_PATH}/template.html.erb")
+    end
+    
+    def render(filename, vars = {})
+      b = binding
+      vars.each do |name, var|
+        eval("#{name}=vars[:#{name}]", b)
+      end
+      ERB.new(File.read(filename)).result(b)
+    end
+  end
+end
+
+$current_source_file = SourceFile.new(__FILE__)
+
+module Kernel
+  def require_with_benchmark(*args)
+    old_source_file = $current_source_file
+  
+    file_name = args.first
+  
+    $current_source_file = current_source_file = SourceFile.new(file_name)
+  
+    before = Time.now.to_f
+    result = require_without_benchmark(*args)
+  ensure
+    after = Time.now.to_f
+  
+    current_source_file.duration = after - before
+    old_source_file.children << current_source_file
+    
+    $current_source_file = old_source_file
+  
+    result
+  end
+
+  alias require_without_benchmark require
+  alias require require_with_benchmark
+  
+  def load_with_benchmark(*args)
+    old_source_file = $current_source_file
+  
+    file_name = args.first
+  
+    $current_source_file = current_source_file = SourceFile.new(file_name)
+  
+    before = Time.now.to_f
+    result = load_without_benchmark(*args)
+  ensure
+    after = Time.now.to_f
+  
+    current_source_file.duration = after - before
+    old_source_file.children << current_source_file
+    
+    $current_source_file = old_source_file
+    result
+  end
+  
+  alias load_without_benchmark load
+  alias load load_with_benchmark
+end
+
+if options[:command]
+  eval(options[:command])
+else
+  files.each do |file|
+    require file
+  end
+end
+
+output_stream = if options[:output_file]
+  File.open(options[:output_file], 'w')
+else
+  $stdout
+end
+
+case options[:output_format]
+when 'html'
+  SourceFile::HTMLPrinter.new($current_source_file, output_stream).print
+when 'flat'
+  SourceFile::FlatPrinter.new($current_source_file, output_stream).print
+end
+
+output_stream.close

data/templates/file.html.erb

@@ -0,0 +1,15 @@
+<% if file.children.empty? %>
+  <%= file.name.gsub(Dir.pwd, '$PWD') %> (<%= (file.duration * 1000).to_i %> ms)
+<% else %>
+  <a href='#'><%= file.name.gsub(Dir.pwd, '$PWD') %></a>
+  <% if file.duration %>
+    (total <%= (file.duration * 1000).to_i %> ms, own <%= (file.own_time * 1000).to_i %> ms)
+  <% end %>
+  <ul style='display:none'>
+  <% file.children_sorted_by_duration.each do |child| %>
+    <li>
+      <%= render "#{TEMPLATE_PATH}/file.html.erb", :file => child %>
+    </li>
+  <% end %>
+  </ul>
+<% end %>

data/templates/template.html.erb

@@ -0,0 +1,27 @@
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
+        "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
+<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="de" lang="de">
+  <head>
+    <style>
+      .report ul li { list-style: none; padding: 2px 0 2px 20px; text-indent: -19px; background-position: 9px 0; }
+      .report ul li > * { text-indent: 0px; }
+    </style>
+    <script type="text/javascript" src="http://ajax.googleapis.com/ajax/libs/prototype/1.6.0.3/prototype.js"></script>
+    <script type="text/javascript" src="http://ajax.googleapis.com/ajax/libs/scriptaculous/1.8.3/scriptaculous.js"></script>
+    <script>
+      $(document).observe('dom:loaded', function() {
+        $('report').getElementsBySelector('a').each(function(link) {
+          link.observe('click', function(event) {
+            Effect.toggle(link.up().down('ul'), 'blind', { duration: 0.3 });
+            event.stop();
+          });
+        });
+      });
+    </script>
+  </head>
+  <body>
+    <div id="report" class="report">
+      <%= render "#{TEMPLATE_PATH}/file.html.erb", :file => @file %>
+    </div>
+  </body>
+</html>

metadata

@@ -0,0 +1,55 @@
+--- !ruby/object:Gem::Specification 
+name: starttime
+version: !ruby/object:Gem::Version 
+  version: 0.0.2
+platform: ruby
+authors: 
+- Thomas Kadauke
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-07 00:00:00 +01:00
+default_executable: 
+dependencies: []
+
+description: 
+email: tkadauke@imedo.de
+executables: 
+- starttime
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/starttime
+- templates/file.html.erb
+- templates/template.html.erb
+has_rdoc: false
+homepage: http://www.imedo.de/
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.1
+signing_key: 
+specification_version: 2
+summary: Instrument startup time of Ruby applications
+test_files: []
+