ghazel-fiveruns_tuneup 0.8.22

data/lib/fiveruns/tuneup/asset_tags.rb

@@ -0,0 +1,54 @@
+module Fiveruns
+  module Tuneup
+    
+    module AssetTags
+      
+      def add_asset_tags_to(response)
+        return unless show_for?(response)
+        before, after = response.body.split(/<\/head>/i, 2)
+        if after
+          insertion = %(
+          <!-- START FIVERUNS TUNEUP ASSETS -->
+          <link rel="stylesheet" type="text/css" href="/stylesheets/tuneup/tuneup.css" />
+          <script type="text/javascript"> var TuneUp = {}; </script>
+          <script type="text/javascript" src="/javascripts/tuneup/init.js"></script>
+          <!-- END FIVERUNS TUNEUP ASSETS -->
+          )
+          add_content_length(response, insertion.size)
+          body = [before, insertion, '</head>', after].join('').html_safe!
+          response.body.replace(body)
+          log :error, "Inserted asset tags"
+        else
+          log :error, "Could not find closing </head> tag for insertion"
+        end
+      end
+      
+      def show_for?(response)
+        return false unless response.body
+        return true if response.headers['ETag'] && response.headers['Content-Type'].to_s.include?('html')
+        return false unless response.headers['Status'] && response.headers['Status'].include?('200')
+        true
+      end
+      
+      def insert_prototype
+        "<script type='text/javascript' src='/javascripts/tuneup/prototype.js'></script>"
+      end
+      
+      # Modify the Content-Length header, regardless if String/Fixnum, and
+      # keep it the same type
+      def add_content_length(response, delta)
+        length = response.headers["Content-Length"]
+        response.headers["Content-Length"] = case length
+        when Fixnum
+          length + delta
+        when String
+          (length.to_i + delta).to_s
+        else
+          length # Shouldn't happen
+        end
+      end
+      
+    end
+    
+  end
+end

data/lib/fiveruns/tuneup/configuration.rb

@@ -0,0 +1,25 @@
+module Fiveruns::Tuneup
+  
+  class Configuration
+    
+    attr_writer :log_directory
+    def log_directory
+      @log_directory ||= begin
+        rails_log = RAILS_DEFAULT_LOGGER.instance_eval { @log.path rescue @logdev.filename }
+        File.dirname(rails_log)
+      rescue
+        Dir::tmpdir
+      end
+    end
+    
+    def environments
+      @environments ||= %w(development)
+    end
+    
+    def instrument?
+      environments.map(&:to_s).include?(RAILS_ENV)
+    end
+    
+  end
+  
+end 

data/lib/fiveruns/tuneup/custom_methods.rb

@@ -0,0 +1,8 @@
+module Fiveruns::Tuneup::CustomMethods
+  
+  # Manually instrument methods
+  def tuneup(*args)
+    Fiveruns::Tuneup.add_custom_methods(self, *args)
+  end
+  
+end

data/lib/fiveruns/tuneup/environment.rb

@@ -0,0 +1,29 @@
+module Fiveruns
+  module Tuneup
+    module Environment
+      
+      def environment
+        {
+          'application_name' => application_name,
+          'rails_env' => rails_env,
+          'rails_version' => rails_version
+        }
+      end
+      
+      def rails_env
+        RAILS_ENV || 'development'
+      end
+      
+      def rails_version
+        Fiveruns::Tuneup::Version.rails.to_s
+      end
+      
+      def application_name
+        app_name = RAILS_ROOT.split('/').last
+        return app_name unless app_name == 'current'             
+        File.join(RAILS_ROOT, '..').split('/').last
+      end
+      
+    end
+  end
+end

data/lib/fiveruns/tuneup/instrumentation/action_controller/base.rb

@@ -0,0 +1,59 @@
+module Fiveruns
+  module Tuneup
+    module Instrumentation
+      module ActionController
+        module Base
+          def self.included(base)
+            Fiveruns::Tuneup.instrument base, ClassMethods, InstanceMethods
+          end
+          module ClassMethods
+            def cache_page_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Cache page", :controller do
+                cache_page_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def expire_page_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Expire cached page", :controller do
+                expire_page_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+          end
+          module InstanceMethods
+            def perform_action_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.run(self, request) do
+                action = (request.parameters['action'] || 'index').to_s
+                if Fiveruns::Tuneup.recording?
+                  Fiveruns::Tuneup.instrument_filters(self) 
+                  Fiveruns::Tuneup.instrument_action_methods(self) 
+                  Fiveruns::Tuneup.instrument_custom_methods
+                end
+                result = Fiveruns::Tuneup.step "Perform #{action.capitalize} action in #{self.class.name}", :controller, false do
+                  perform_action_without_fiveruns_tuneup(*args, &block)
+                end
+              end
+            end
+            def process_with_fiveruns_tuneup(request, response, *args, &block)
+              result = process_without_fiveruns_tuneup(request, response, *args, &block)
+              if !request.xhr? && response.content_type && response.content_type.include?('html') && controller_name != 'tuneup'
+                Fiveruns::Tuneup.add_asset_tags_to(response)
+              end
+              result
+            end
+            
+            def write_fragment_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Cache fragment", :controller do
+                write_fragment_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def expire_fragment_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Expire fragment cache", :controller do
+                expire_fragment_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fiveruns/tuneup/instrumentation/action_view/base.rb

@@ -0,0 +1,81 @@
+module Fiveruns
+  module Tuneup
+    module Instrumentation
+      module ActionView
+        module Base
+                    
+          BASIC_TEMPLATE_PATH = File.join(RAILS_ROOT, 'app', 'views')           
+          
+          def self.included(base)
+            Fiveruns::Tuneup.instrument base, InstanceMethods
+          end
+
+          def self.normalize_path(path)
+            return path unless path
+            if path[0, BASIC_TEMPLATE_PATH.size] == BASIC_TEMPLATE_PATH
+              path[(BASIC_TEMPLATE_PATH.size + 1)..-1]
+            else
+              if (components = path.split(File::SEPARATOR)).size > 2
+                components[-2, 2].join(File::SEPARATOR)
+              else
+                components.join(File::SEPARATOR)
+              end
+            end
+          end
+          
+          module InstanceMethods
+            def render_file_with_fiveruns_tuneup(path, *args, &block)
+              name = Fiveruns::Tuneup::Instrumentation::ActionView::Base.normalize_path(path)
+              Fiveruns::Tuneup.step "Render file #{name}", :view do
+                render_file_without_fiveruns_tuneup(path, *args, &block)
+              end
+            end
+            def update_page_with_fiveruns_tuneup(*args, &block)
+              path = block.to_s.split('/').last.split(':').first rescue ''
+              name = Fiveruns::Tuneup::Instrumentation::ActionView::Base.normalize_path(path)
+              Fiveruns::Tuneup.step "Render page update #{name}", :view do
+                update_page_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def render_with_fiveruns_tuneup(*args, &block)
+              record = true
+              options = args.first || {}
+              path = case options
+              when String
+                # Pre-Rails 2.1, don't record this as it causes duplicate records 
+                if Fiveruns::Tuneup::Version.rails < Fiveruns::Tuneup::Version.new(2,1,0)
+                  record = false
+                else
+                  "Render #{options}"                  
+                end
+              when :update
+                name = block.to_s.split('/').last.split(':').first rescue ''
+                "Render page update #{name}"
+              when Hash
+                if options[:file]
+                  "Render #{options[:file]}"
+                elsif options[:partial]
+                  record = false
+                elsif options[:inline]
+                  "Render inline"
+                elsif options[:text]
+                  "Render text"
+                end
+              end
+              path ||= 'Render'
+              
+              if record
+                Fiveruns::Tuneup.step path, :view do
+                  render_without_fiveruns_tuneup(*args, &block)
+                end
+              else
+                render_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fiveruns/tuneup/instrumentation/action_view/partial_template.rb

@@ -0,0 +1,28 @@
+module Fiveruns
+  module Tuneup
+    module Instrumentation
+      module ActionView
+        module PartialTemplate       
+          
+          def self.included(base)
+            Fiveruns::Tuneup.instrument base, InstanceMethods
+          end
+          
+          def self.relevant?
+            Fiveruns::Tuneup::Version.rails < Fiveruns::Tuneup::Version.new(2,1,0) ? false : true
+          end
+          
+          module InstanceMethods
+
+            def render_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Render partial #{path}", :view do
+                render_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fiveruns/tuneup/instrumentation/active_record/base.rb

@@ -0,0 +1,126 @@
+module Fiveruns
+  module Tuneup
+    module Instrumentation
+      module ActiveRecord
+        module Base
+          
+          def self.record(model, name, raw_sql=nil, &operation)
+            sql = nil
+            Fiveruns::Tuneup.exclude do
+              model.silence do
+                sql = Fiveruns::Tuneup::Step::SQL.new(raw_sql, model.connection) if raw_sql
+                Fiveruns::Tuneup.add_schema_for(model.table_name, model.connection)
+              end
+            end
+            Fiveruns::Tuneup.step(name, :model, true, sql, model.table_name, &operation)
+          end
+          
+          def self.included(base)
+            Fiveruns::Tuneup.instrument base, InstanceMethods, ClassMethods
+          end
+               
+          module ClassMethods
+            
+            #
+            # FINDS
+            #
+            
+            def find_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Find #{self.name}" do
+                find_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def find_by_sql_with_fiveruns_tuneup(conditions, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Find #{self.name} by SQL", sanitize_sql(conditions) do
+                find_by_sql_without_fiveruns_tuneup(conditions, &block)
+              end
+            end
+            
+            #
+            # CREATE
+            #
+            
+            def create_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Create #{self.name}" do
+                create_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+            #
+            # UPDATES
+            #
+            
+            def update_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Update #{self.name}" do
+                update_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def update_all_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Update #{self.name} (All)" do
+                update_all_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+            #
+            # DELETES
+            #
+            
+            def destroy_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Destroy #{self.name}" do
+                destroy_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def destroy_all_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Destroy #{self.name} (All)" do
+                destroy_all_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def delete_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Delete #{self.name}" do
+                delete_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def delete_all_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self, "Delete #{self.name} (All)" do
+                delete_all_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+          end
+          module InstanceMethods
+            
+            #
+            # UPDATES
+            #
+            
+            def update_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self.class, "Update #{self.class.name}" do
+                update_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def save_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self.class, "Save #{self.class.name}" do
+                save_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def save_with_fiveruns_tuneup!(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self.class, "Save #{self.class.name}" do
+                save_without_fiveruns_tuneup!(*args, &block)
+              end
+            end
+            
+            #
+            # DELETES
+            #
+            
+            def destroy_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup::Instrumentation::ActiveRecord::Base.record self.class, "Destroy #{self.class.name}" do
+                destroy_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fiveruns/tuneup/instrumentation/cgi/session.rb

@@ -0,0 +1,30 @@
+module Fiveruns
+  module Tuneup
+    module Instrumentation
+      module CGI
+        module Session
+          def self.included(base)
+            Fiveruns::Tuneup.instrument base, InstanceMethods
+          end
+          module InstanceMethods
+            def initialize_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Create session", :model do
+                initialize_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def close_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Close session", :model do
+                close_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+            def delete_with_fiveruns_tuneup(*args, &block)
+              Fiveruns::Tuneup.step "Delete session", :model do
+                delete_without_fiveruns_tuneup(*args, &block)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end