Contents

require "stackflame/version"
require "rack/stackflame"
require "stackprof"
require "uri"

class Stackflame
  DEFAULT_OPTIONS = {
    interval: 1000,
    mode:     :cpu,
    raw:      true,
  }.freeze

  def self.profile(options = {}, &block)
    stackflame = self.new
    stackflame.run(options, &block)
    stackflame.open_flamegraph
  end

  def run(options = {}, &block)
    options = DEFAULT_OPTIONS.merge(options)
    result  = StackProf.run(options, &block)

    File.open(temp_js_path, 'w') do |f|
      StackProf::Report.new(result).print_flamegraph(f)
    end
  end

  def open_flamegraph(params = {})
    params = params.merge(data: temp_js_path)
    if system("which osascript > /dev/null")
      # NOTE: `open` can't open path with query string
      `osascript -e 'open location "#{flamegraph_path(params)}"'`
    else
      puts "This OS is not supported. Please open: #{flamegraph_path(params)}"
    end
  end

  private

  def flamegraph_path(params)
    viewer_path = File.expand_path('../../vendor/viewer.html', __FILE__)
    "file://#{viewer_path}#{to_query(params)}"
  end

  def to_query(params)
    return '' if params.size == 0

    queries = []
    params.each do |key, value|
      queries << "#{URI.escape(key.to_s)}=#{URI.escape(value.to_s)}"
    end
    "?#{queries.join('&')}"
  end

  def temp_js_path
    return @js_path if @js_path

    temp = `mktemp /tmp/stackflame-XXXXXXXX`.strip
    @js_path = "#{temp}.js"
    `mv #{temp} #{@js_path}`
    @js_path
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
stackflame-0.2.5	lib/stackflame.rb
stackflame-0.2.4	lib/stackflame.rb