stackprof-webnav 0.0.2 → 1.0.1

data/lib/stackprof-webnav/server.rb

@@ -1,73 +1,147 @@
-require 'nyny'
+require 'sinatra'
 require 'haml'
 require "stackprof"
-require 'sprockets/nyny'
-require 'net/http'
 require_relative 'presenter'
+require_relative 'dump'
+require 'pry'
+require "sinatra/reloader" if development?
+require 'ruby-graphviz'
 
 module StackProf
   module Webnav
-    class Server < NYNY::App
-      register Sprockets::NYNY
-      config.assets.paths << File.join(__dir__, 'css')
-
+    class Server < Sinatra::Application
       class << self
-        attr_accessor :report_dump_path, :report_dump_url
+        attr_accessor :cmd_options
+      end
 
-        def presenter
-          return @presenter unless @presenter.nil?
-          content = if report_dump_path.nil?
-                      Net::HTTP.get(URI.parse(report_dump_url))
-                    else
-                      File.open(report_dump_path).read
-                    end
-
-          report = StackProf::Report.new(Marshal.load(content))
-          @presenter ||= Presenter.new(report)
+      configure :development do
+        register Sinatra::Reloader
+      end
+
+      configure :production do
+        set :show_exceptions, false
+      end
+
+      error do
+        @error = env['sinatra.error']
+        haml :error
+      end
+
+      before do
+        unless request.path_info == '/'
+          if params[:dump] && params[:dump] != current_dump.path
+            current_dump.path = params[:dump]
+          end
         end
       end
 
       helpers do
-        def template_path name
-          File.join(__dir__, name)
+        def current_dump
+          Thread.current[:cache] = {}
+          Thread.current[params[:dump]] ||= Dump.new(params[:dump])
         end
 
-        def render_with_layout *args
-          args[0] = template_path("views/#{args[0]}.haml")
-          render(template_path('views/layout.haml')) { render(*args) }
+        def current_report
+          StackProf::Report.new(
+            Marshal.load(current_dump.content)
+          )
         end
 
         def presenter
-          Server.presenter
+          Presenter.new(current_report)
         end
 
-        def method_url name
-          "/method?name=#{URI.escape(name)}"
+        def ensure_file_generated(path, &block)
+          return if File.exist?(path)
+          File.open(path, 'wb', &block)
         end
 
-        def file_url path
-          "/file?path=#{URI.escape(path)}"
+        def url_for(path, options={})
+          query = URI.encode_www_form({dump: params[:dump]}.merge(options))
+          path + "?" + query
         end
       end
 
       get '/' do
-        @file = Server.report_dump_path || Server.report_dump_url
+        if Server.cmd_options[:filepath]
+          query = URI.encode_www_form(dump: Server.cmd_options[:filepath])
+          next redirect to("/overview?#{query}")
+        end
+
+        @directory = File.expand_path(Server.cmd_options[:directory] || '.')
+        @files = Dir.entries(@directory).sort.map do |file|
+          path = File.expand_path(file, @directory)
+
+          OpenStruct.new(
+            name: file,
+            path: path,
+            modified: File.mtime(path)
+          )
+        end.select do |file|
+          File.file?(file.path)
+        end
+
+        haml :index
+      end
+
+      get '/overview' do
         @action = "overview"
-        @frames = presenter.overview_frames
-        render_with_layout :overview
+
+        begin
+          @frames = presenter.overview_frames
+          haml :overview
+        rescue => error
+          haml :invalid_dump
+        end
+      end
+
+      get '/flames.json' do
+        ensure_file_generated(current_dump.flame_graph_path) do |file|
+          current_report.print_flamegraph(file, true, true)
+        end
+
+        send_file(current_dump.flame_graph_path, type: 'text/javascript')
+      end
+
+      get '/graph.png' do
+        ensure_file_generated(current_dump.graph_path) do |file|
+          current_report.print_graphviz({}, file)
+        end
+
+        ensure_file_generated(current_dump.graph_image_path) do |file|
+          GraphViz
+            .parse(current_dump.graph_path)
+            .output(png: current_dump.graph_image_path)
+        end
+
+        send_file(current_dump.graph_image_path, type: 'image/png')
+      end
+
+      get '/graph' do
+        haml :graph
+      end
+
+      get '/flamegraph' do
+        haml :flamegraph
       end
 
       get '/method' do
         @action = params[:name]
         @frames = presenter.method_info(params[:name])
-        render_with_layout :method
+        haml :method
       end
 
       get '/file' do
         path = params[:path]
-        @path = path
-        @data = presenter.file_overview(path)
-        render_with_layout :file
+
+        if File.exist?(path)
+          @path = path
+          @data = presenter.file_overview(path)
+        else
+          @data = nil
+        end
+
+        haml :file
       end
     end
   end

data/lib/stackprof-webnav/version.rb

@@ -1,5 +1,5 @@
 module StackProf
   module Webnav
-    VERSION = '0.0.2'
+    VERSION = '1.0.1'
   end
 end

data/lib/stackprof-webnav/views/error.haml

@@ -0,0 +1,8 @@
+%h1 Oops, something went wrong
+
+%h2
+  = @error.class
+  \-
+  = @error.message
+
+%h3 Are you looking at a valid stackprof dump?

data/lib/stackprof-webnav/views/file.haml

@@ -5,7 +5,10 @@
 %br
 %br
 
-.code_linenums
-  - @data[:lineinfo].each do |l|
-    %span= l
-!= @data[:code]
+- if @data
+  .code_linenums
+    - @data[:lineinfo].each do |l|
+      %span= l
+  != @data[:code]
+- else
+  %h3 Unable to load file, was the dump captured on another machine?

data/lib/stackprof-webnav/views/flamegraph.haml

@@ -0,0 +1,77 @@
+%a{:href => url_for("/overview")}
+  %button Back to overview
+
+%br
+%br
+
+:css
+  body {
+    margin: 0;
+    padding: 0;
+    font-family: Consolas, "Liberation Mono", Menlo, Courier, monospace;
+    font-size: 10pt;
+  }
+  .overview-container {
+    position: relative;
+  }
+  .overview {
+    cursor: col-resize;
+  }
+  .overview-viewport-overlay {
+    position: absolute;
+    top: 0;
+    left: 0;
+    width: 1;
+    height: 1;
+    background-color: rgba(0, 0, 0, 0.25);
+    transform-origin: top left;
+    cursor: -moz-grab;
+    cursor: -webkit-grab;
+    cursor: grab;
+  }
+  .moving {
+    cursor: -moz-grabbing;
+    cursor: -webkit-grabbing;
+    cursor: grabbing;
+  }
+  .info {
+    display: block;
+    height: 40px;
+    margin: 3px 6px;
+    margin-right: 206px;
+    padding: 3px 6px;
+    line-height: 18px;
+  }
+  .legend {
+    display: block;
+    float: right;
+    width: 195px;
+    max-height: 100%;
+    overflow-y: scroll;
+  }
+  .legend > div {
+    padding: 6px;
+    clear: right;
+  }
+  .legend > div span {
+    opacity: 0.75;
+    display: block;
+    text-align: right;
+  }
+  .legend > div .name {
+    max-width: 70%;
+    word-wrap: break-word;
+  }
+%script{:src => "flamegraph.js"}
+.legend
+.overview-container
+  %canvas.overview
+  .overview-viewport-overlay
+.info
+  %div{:style => "float: right; text-align: right"}
+    .samples
+    .exclusive
+  .frame
+  .file
+%canvas.flamegraph
+%script{:src => "/flames.json?dump=#{params[:dump]}"}

data/lib/stackprof-webnav/views/graph.haml

@@ -0,0 +1,4 @@
+%a{:href => url_for("/overview")}
+  %button Back to overview
+
+%img{src: url_for("/graph.png")}

data/lib/stackprof-webnav/views/index.haml

@@ -0,0 +1,18 @@
+%h3 StackProf Navigator
+%hr
+
+%p
+  Listing dumps in
+  %b= @directory
+
+%table.centered
+  %thead
+    %th Filename
+    %th Modified
+
+  %tbody
+    - @files.each do |file|
+      %tr
+        %td
+          %a{href: url_for("/overview", dump: file.path)}= file.name
+        %td= file.modified

data/lib/stackprof-webnav/views/invalid_dump.haml

@@ -0,0 +1,4 @@
+%h3 StackProf Navigator - #{@action}
+%hr
+
+%h4 Unable to open the specified file as a dump

data/lib/stackprof-webnav/views/layout.haml

@@ -1,9 +1,10 @@
 %html
   %head
     %title Stackprof navigator
-    %link(rel="stylesheet" href="/assets/application.css")
+    %link(rel="stylesheet" href="/css/normalize.css")
+    %link(rel="stylesheet" href="/css/foundation.min.css")
+    %link(rel="stylesheet" href="/css/code.css")
+    %link(rel="stylesheet" href="/css/application.css")
 
   %body
-    .row
-      .large-12.columns
-        = yield
+    = yield

data/lib/stackprof-webnav/views/method.haml

@@ -1,12 +1,15 @@
 %h3 StackProf Navigator - #{@action} (#{@frames.count} frames)
 %hr
 
-%a{:href => '/'} ← Back to overview
-%br
+%p
+  %a{:href => url_for("/overview")} ← Back to overview
 
 - @frames.each do |frame|
-  %h4
-    %a{:href => file_url(frame[:location])}= frame[:location]
+  %a{:href => url_for("/file", path: frame[:location])}
+    %button.btn
+      Browse
+      = frame[:location]
+
   - if frame[:callers].any?
     %table
       %thead
@@ -21,8 +24,8 @@
             %td= caller[:weight]
             %td= caller[:pct]
             %td
-              %a{:href => method_url(caller[:method])}
-                = caller[:method]
+              %a{:href => url_for("/method", name: caller[:method])}
+                &= caller[:method]
 
   - if frame[:callees].any?
     %table
@@ -38,8 +41,8 @@
             %td= caller[:weight]
             %td= caller[:pct]
             %td
-              %a{:href => method_url(caller[:method])}
-                = caller[:method]
+              %a{:href => url_for("/method", name: caller[:method])}
+                &= caller[:method]
 
   %h4 Code
   != frame[:source]

data/lib/stackprof-webnav/views/overview.haml

@@ -3,7 +3,21 @@
 
 %p
   Viewing dump
-  %b= @file
+  %b= current_dump.path
+
+%a{href: "/"}
+  %button.btn Browse directory
+
+
+%a{href: url_for("/graph")}
+  %button.btn View call graph
+
+- if current_report.data[:raw]
+  %a{href: url_for("/flamegraph")}
+    %button View flamegraph
+- else
+  %a{href: "https://github.com/tmm1/stackprof#all-options"}
+    %button.secondary Flamegraph is not available
 
 %table.centered
   %thead
@@ -21,5 +35,5 @@
         %td= frame[:samples]
         %td= frame[:samples_pct]
         %td
-          %a{:href => method_url(frame[:method])}
-            = frame[:method]
+          %a{:href => url_for("/method", name: frame[:method])}
+            &= frame[:method]

data/spec/helpers.rb

@@ -0,0 +1,17 @@
+require 'rack/test'
+
+module Helpers
+  def build_app(options={})
+    StackProf::Webnav::Server.cmd_options = options
+    Rack::Test::Session.new(Rack::MockSession.new(StackProf::Webnav::Server))
+  end
+
+  def fixture_path(name)
+    File.join(File.dirname(__FILE__), "fixtures", name)
+  end
+
+  def build_presenter(path)
+    report = StackProf::Report.new(Marshal.load(File.read(path)))
+    StackProf::Webnav::Presenter.new(report)
+  end
+end