app_profiler 0.0.8 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4342f9415d422d77b172d358f968f9bc6c0ae670364e2ed351c7d1daf7cfcd3f
-  data.tar.gz: 4868b701483cc23fef7dad0432437afe61a65ff4ff5bcd752eec37636f4eeddb
+  metadata.gz: 340bc83668cbe8abd27780ef23456a6d98f1f8c46059f2b447a25f1e6784483e
+  data.tar.gz: 06fcbd4abe95e53f8e37595df7ca445bcd59f61cd9ba24af8c82578cdb6bb9c4
 SHA512:
-  metadata.gz: ab3d7f1677cc9f3a645ca23cf5fc2b3125cd498cbaace7640e31aad8b45e8f9c23bda135bf470b105e723cafeb1c0456149618c3268cc9ea5674fe2427b00fe0
-  data.tar.gz: 12189f7de4e869d34939d9c3cdbeeb94d257de0f31c232f2deff2011a879fadc8aaccfde44f9780eec6bd863375f9a098f3c571ed53bc65c9e9825f83be6d96a
+  metadata.gz: 04cf486ef95efd05af19fd241f05b45c0a942c1e701295181fdef898f59cb24ace85c86e047ea63359b5c5ddfb2f8a233afbc3958004a823e0ffcf0daf692830
+  data.tar.gz: bab5ddd1f1dcfb4d06e23b575341a7d7a783e04191b62a74dd553cf78d5726187500931e34c49ebe5e019d60602ceec074d4071d024008c05c8bfe5e5ddb7d52

data/lib/app_profiler/middleware/upload_action.rb

@@ -21,28 +21,20 @@ module AppProfiler
         def append_headers(response, upload:, autoredirect:)
           return unless upload
 
-          response[1][profile_header]      = profile_url(upload)
+          response[1][profile_header]      = AppProfiler.profile_url(upload)
           response[1][profile_data_header] = profile_data_url(upload)
 
           return unless autoredirect
 
           # Automatically redirect to profile if autoredirect is true.
           if response[0].to_i < 500
-            response[1]["Location"] = profile_url(upload)
+            response[1]["Location"] = AppProfiler.profile_url(upload)
             response[0] = 303
           end
         end
 
-        def profile_url(upload)
-          if AppProfiler.profile_url_formatter.nil?
-            "#{AppProfiler.speedscope_host}#profileURL=#{upload.url}"
-          else
-            AppProfiler.profile_url_formatter.call(upload)
-          end
-        end
-
         def profile_data_url(upload)
-          upload.url
+          upload.url.to_s
         end
 
         def profile_header

data/lib/app_profiler/profile.rb

@@ -2,7 +2,7 @@
 
 module AppProfiler
   class Profile
-    INTERNAL_METADATA_KEYS = %i(id context)
+    INTERNAL_METADATA_KEYS = [:id, :context]
     private_constant :INTERNAL_METADATA_KEYS
     class UnsafeFilename < StandardError; end
 
@@ -40,7 +40,13 @@ module AppProfiler
     def upload
       AppProfiler.storage.upload(self).tap do |upload|
         if upload && defined?(upload.url)
-          AppProfiler.logger.info("[Profiler] uploaded profile profile_url=#{upload.url}")
+          AppProfiler.logger.info(
+            <<~INFO.squish
+              [Profiler] data uploaded:
+              profile_url=#{upload.url}
+              profile_viewer_url=#{AppProfiler.profile_url(upload)}
+            INFO
+          )
         end
       end
     rescue => error
@@ -71,7 +77,7 @@ module AppProfiler
         Socket.gethostname,
       ].compact.join("-") << ".json"
 
-      raise UnsafeFilename if /[^0-9A-Za-z.\-\_]/.match(filename)
+      raise UnsafeFilename if /[^0-9A-Za-z.\-\_]/.match?(filename)
 
       AppProfiler.profile_root.join(filename)
     end

data/lib/app_profiler/railtie.rb

@@ -16,6 +16,12 @@ module AppProfiler
       AppProfiler.middleware = app.config.app_profiler.middleware || Middleware
       AppProfiler.middleware.action = app.config.app_profiler.middleware_action || default_middleware_action
       AppProfiler.middleware.disabled = app.config.app_profiler.middleware_disabled || false
+      AppProfiler.server.enabled = app.config.app_profiler.server_enabled || false
+      AppProfiler.server.transport = app.config.app_profiler.server_transport || default_appprofiler_transport
+      AppProfiler.server.port = app.config.app_profiler.server_port || 0
+      AppProfiler.server.duration = app.config.app_profiler.server_duration || 30
+      AppProfiler.server.cors = app.config.app_profiler.server_cors || true
+      AppProfiler.server.cors_host = app.config.app_profiler.server_cors_host || "*"
       AppProfiler.autoredirect = app.config.app_profiler.autoredirect || false
       AppProfiler.speedscope_host = app.config.app_profiler.speedscope_host || ENV.fetch(
         "APP_PROFILER_SPEEDSCOPE_URL", "https://speedscope.app"
@@ -30,10 +36,22 @@ module AppProfiler
 
     initializer "app_profiler.add_middleware" do |app|
       unless AppProfiler.middleware.disabled
+        if AppProfiler.viewer == Viewer::SpeedscopeRemoteViewer
+          app.middleware.insert_before(0, Viewer::SpeedscopeRemoteViewer::Middleware)
+        end
         app.middleware.insert_before(0, AppProfiler.middleware)
       end
     end
 
+    initializer "app_profiler.enable_server" do
+      if AppProfiler.server.enabled
+        AppProfiler::Server.start
+        ActiveSupport::ForkTracker.after_fork do
+          AppProfiler::Server.start
+        end
+      end
+    end
+
     private
 
     def default_middleware_action
@@ -43,5 +61,15 @@ module AppProfiler
         Middleware::UploadAction
       end
     end
+
+    def default_appprofiler_transport
+      if Rails.env.development?
+        # default to TCP server in development so that if wanted users are able to target
+        # the server with speedscope
+        AppProfiler::Server::TRANSPORT_TCP
+      else
+        AppProfiler::Server::TRANSPORT_UNIX
+      end
+    end
   end
 end

data/lib/app_profiler/request_parameters.rb

@@ -37,6 +37,7 @@ module AppProfiler
       {
         mode: mode.to_sym,
         interval: interval.to_i,
+        ignore_gc: !!ignore_gc,
         metadata: {
           id: request_id,
           context: context,
@@ -50,6 +51,10 @@ module AppProfiler
       query_param("profile") || profile_header_param("mode")
     end
 
+    def ignore_gc
+      query_param("ignore_gc") || profile_header_param("ignore_gc")
+    end
+
     def interval
       query_param("interval") || profile_header_param("interval") || DEFAULT_INTERVALS[mode]
     end
@@ -76,6 +81,7 @@ module AppProfiler
 
     def header(name)
       return unless @request.has_header?(name)
+
       @request.get_header(name)
     end
 

data/lib/app_profiler/server.rb

@@ -0,0 +1,307 @@
+# frozen_string_literal: true
+
+require "socket"
+require "rack"
+require "tempfile"
+
+# This module provides a means to start a golang-inspired profile server
+# it is implemented using stdlib and Rack to avoid additional dependencies
+
+module AppProfiler
+  module Server
+    HTTP_OK = 200
+    HTTP_BAD_REQUEST = 400
+    HTTP_NOT_FOUND = 404
+    HTTP_NOT_ALLOWED = 405
+    HTTP_CONFLICT = 409
+
+    TRANSPORT_UNIX = "unix"
+    TRANSPORT_TCP = "tcp"
+
+    DEFAULTS = {
+      enabled: false,
+      transport: TRANSPORT_UNIX,
+      cors: true,
+      cors_host: "*",
+      port: 0,
+      duration: 30,
+    }
+
+    mattr_accessor :enabled, default: DEFAULTS[:enabled]
+    mattr_accessor :transport, default: DEFAULTS[:transport]
+    mattr_accessor :cors, default: DEFAULTS[:cors]
+    mattr_accessor :cors_host, default: DEFAULTS[:cors_host]
+    mattr_accessor :port, default: DEFAULTS[:port]
+    mattr_accessor :duration, default: DEFAULTS[:duration]
+
+    class ProfileApplication
+      class InvalidProfileArgsError < StandardError; end
+
+      def initialize
+        @semaphore = Thread::Mutex.new
+        @profile_running = false
+      end
+
+      def call(env)
+        handle(Rack::Request.new(env)).finish
+      end
+
+      private
+
+      def handle(request)
+        response = Rack::Response.new
+        if request.request_method != "GET"
+          response.status = HTTP_NOT_ALLOWED
+          response.write("Only GET requests are supported")
+          return response
+        end
+        case request.path
+        when "/profile"
+          begin
+            stackprof_args, duration = validate_profile_params(request.params)
+          rescue InvalidProfileArgsError => e
+            response.status = HTTP_BAD_REQUEST
+            response.write("Invalid argument #{e.message}")
+            return response
+          end
+
+          if start_running
+            start_time = Process.clock_gettime(Process::CLOCK_REALTIME, :nanosecond)
+            AppProfiler.start(**stackprof_args)
+            sleep(duration)
+            profile = AppProfiler.stop
+            stop_running
+            response.status = HTTP_OK
+            response.set_header("Content-Type", "application/json")
+            if AppProfiler::Server.cors
+              response.set_header("Access-Control-Allow-Origin", AppProfiler::Server.cors_host)
+            end
+            profile_hash = profile.to_h
+            profile_hash["start_time_nsecs"] = start_time # NOTE: this is not part of the stackprof profile spec
+            response.write(JSON.dump(profile_hash))
+          else
+            response.status = HTTP_CONFLICT
+            response.write("A profile is already running")
+          end
+        else
+          response.status = HTTP_NOT_FOUND
+          response.write("Unsupported endpoint #{request.path}")
+        end
+        response
+      end
+
+      def validate_profile_params(params)
+        params = params.symbolize_keys
+        stackprof_args = {}
+        begin
+          duration = Float(params.key?(:duration) ? params[:duration] : AppProfiler::Server.duration)
+        rescue ArgumentError
+          raise InvalidProfileArgsError, "invalid duration #{params[:duration]}"
+        end
+        if params.key?(:mode)
+          if ["cpu", "wall", "object"].include?(params[:mode])
+            stackprof_args[:mode] = params[:mode].to_sym
+          else
+            raise InvalidProfileArgsError, "invalid mode #{params[:mode]}"
+          end
+        end
+        if params.key?(:interval)
+          stackprof_args[:interval] = params[:interval].to_i
+          raise InvalidProfileArgsError, "invalid interval #{params[:interval]}" if stackprof_args[:interval] <= 0
+        end
+        [stackprof_args, duration]
+      end
+
+      # Prevent multiple concurrent profiles by synchronizing between threads
+      def start_running
+        @semaphore.synchronize do
+          return false if @profile_running
+
+          @profile_running = true
+        end
+      end
+
+      def stop_running
+        @semaphore.synchronize { @profile_running = false }
+      end
+    end
+
+    # This is a minimal, non-compliant "HTTP" server.
+    # It will create an extremely minimal rack environment hash and hand it off
+    # to our application to process
+    class ProfileServer
+      PROFILER_TEMPFILE_PATH = "/tmp/app_profiler" # for tempfile that indicates port in filename or unix sockets
+
+      class Transport
+        attr_reader :socket
+
+        def client
+          raise(NotImplementedError)
+        end
+
+        def stop
+          raise(NotImplementedError)
+        end
+      end
+
+      class UNIX < Transport
+        def initialize
+          super
+          FileUtils.mkdir_p(PROFILER_TEMPFILE_PATH)
+          @socket_file = File.join(PROFILER_TEMPFILE_PATH, "app-profiler-#{Process.pid}.sock")
+          File.unlink(@socket_file) if File.exist?(@socket_file) && File.socket?(@socket_file)
+          @socket = UNIXServer.new(@socket_file)
+        end
+
+        def client
+          UNIXSocket.new(@socket_file)
+        end
+
+        def stop
+          File.unlink(@socket_file) if File.exist?(@socket_file) && File.socket?(@socket_file)
+          @socket.close
+        end
+      end
+
+      class TCP < Transport
+        SERVER_ADDRESS = "127.0.0.1" # it is ONLY safe to run this bound to localhost
+
+        def initialize(port = 0)
+          super()
+          FileUtils.mkdir_p(PROFILER_TEMPFILE_PATH)
+          @socket = TCPServer.new(SERVER_ADDRESS, port)
+          @port = @socket.addr[1]
+          @port_file = Tempfile.new("profileserver-#{Process.pid}-port-#{@port}-", PROFILER_TEMPFILE_PATH)
+        end
+
+        def client
+          TCPSocket.new(SERVER_ADDRESS, @port)
+        end
+
+        def stop
+          @port_file.unlink
+          @socket.close
+        end
+      end
+
+      def initialize(transport)
+        case transport
+        when TRANSPORT_UNIX
+          @transport = ProfileServer::UNIX.new
+        when TRANSPORT_TCP
+          @transport = ProfileServer::TCP.new(AppProfiler::Server.port)
+        else
+          raise "invalid transport #{transport}"
+        end
+
+        @listen_thread = nil
+
+        AppProfiler.logger.info(
+          "[AppProfiler::Server] listening on addr=#{@transport.socket.addr}"
+        )
+        @pid = Process.pid
+        at_exit { stop }
+      end
+
+      def client
+        @transport.client
+      end
+
+      def serve
+        return unless @listen_thread.nil?
+
+        app = ProfileApplication.new
+
+        @listen_thread = Thread.new do
+          loop do
+            Thread.new(@transport.socket.accept) do |session|
+              request = session.gets
+              if request.nil?
+                session.close
+                next
+              end
+              method, path, http_version = request.split(" ")
+              path_info, query_string = path.split("?")
+              env = { # an extremely minimal rack env hash, just enough to get the job done
+                "HTTP_VERSION" => http_version,
+                "REQUEST_METHOD" => method,
+                "PATH_INFO" => path_info,
+                "QUERY_STRING" => query_string,
+                "rack.input" => "",
+              }
+              status, headers, body = app.call(env)
+
+              begin
+                session.print("#{http_version} #{status}\r\n")
+                headers.each do |header, value|
+                  session.print("#{header}: #{value}\r\n")
+                end
+                session.print("\r\n")
+                body.each do |part|
+                  session.print(part)
+                end
+              rescue => e
+                AppProfiler.logger.error(
+                  "[AppProfiler::Server] exception #{e} responding to request #{request}: #{e.message}"
+                )
+              ensure
+                session.close
+              end
+            end
+          end
+        end
+      end
+
+      def stop
+        return unless @pid == Process.pid
+
+        @listen_thread.kill
+        @transport.stop
+      end
+    end
+
+    private_constant :ProfileApplication, :ProfileServer
+
+    class << self
+      def reset
+        profile_servers.clear
+        DEFAULTS.each do |config, value|
+          class_variable_set(:"@@#{config}", value) # rubocop:disable Style/ClassVars
+        end
+      end
+
+      def start
+        return if profile_server
+
+        profile_servers[Process.pid] = ProfileServer.new(AppProfiler::Server.transport)
+        profile_server.serve
+        profile_server
+      end
+
+      def client
+        return unless profile_server
+
+        profile_server.client
+      end
+
+      def stop
+        return unless profile_server
+
+        profile_server.stop
+        profile_servers.delete(Process.pid)
+      end
+
+      private
+
+      def profile_server
+        profile_servers[Process.pid]
+      end
+
+      # It is possible there will be a server pre-fork, this is to distinguish
+      # the child server from its parent via PID
+      def profile_servers
+        @profile_servers ||= {}
+      end
+    end
+  end
+end

data/lib/app_profiler/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AppProfiler
-  VERSION = "0.0.8"
+  VERSION = "0.1.1"
 end

data/lib/app_profiler/viewer/base_viewer.rb

@@ -3,7 +3,13 @@
 module AppProfiler
   module Viewer
     class BaseViewer
-      def self.view(_profile)
+      class << self
+        def view(profile)
+          new(profile).view
+        end
+      end
+
+      def view(_profile)
         raise NotImplementedError
       end
     end

data/lib/app_profiler/viewer/speedscope_remote_viewer/base_middleware.rb

@@ -0,0 +1,101 @@
+# frozen_string_literal: true
+
+require "rails-html-sanitizer"
+
+module AppProfiler
+  module Viewer
+    class SpeedscopeRemoteViewer < BaseViewer
+      class BaseMiddleware
+        class Sanitizer < Rails::Html::SafeListSanitizer
+          self.allowed_tags = Set.new([
+            "strong", "em", "b", "i", "p", "code", "pre", "tt", "samp", "kbd", "var", "sub",
+            "sup", "dfn", "cite", "big", "small", "address", "hr", "br", "div", "span", "h1",
+            "h2", "h3", "h4", "h5", "h6", "ul", "ol", "li", "dl", "dt", "dd", "abbr", "acronym",
+            "a", "img", "blockquote", "del", "ins", "script",
+          ])
+        end
+
+        private_constant(:Sanitizer)
+
+        def self.id(file)
+          file.basename.to_s.delete_suffix(".json")
+        end
+
+        def initialize(app)
+          @app = app
+        end
+
+        def call(env)
+          request = Rack::Request.new(env)
+
+          return index(env)                        if request.path_info =~ %r(\A/app_profiler/?\z)
+          return viewer(env, Regexp.last_match(1)) if request.path_info =~ %r(\A/app_profiler/viewer/(.*)\z)
+          return show(env, Regexp.last_match(1))   if request.path_info =~ %r(\A/app_profiler/(.*)\z)
+
+          @app.call(env)
+        end
+
+        protected
+
+        def id(file)
+          self.class.id(file)
+        end
+
+        def profile_files
+          AppProfiler.profile_root.glob("**/*.json")
+        end
+
+        def render(html)
+          [
+            200,
+            { "Content-Type" => "text/html" },
+            [
+              +<<~HTML,
+                <!doctype html>
+                <html>
+                  <head>
+                    <title>App Profiler</title>
+                  </head>
+                  <body>
+                    #{sanitizer.sanitize(html)}
+                  </body>
+                </html>
+              HTML
+            ],
+          ]
+        end
+
+        def sanitizer
+          @sanitizer ||= Sanitizer.new
+        end
+
+        def viewer(_env, path)
+          raise NotImplementedError
+        end
+
+        def index(_env)
+          render(
+            (+"").tap do |content|
+              content << "<h1>Profiles</h1>"
+              profile_files.each do |file|
+                content << <<~HTML
+                  <p>
+                    <a href="/app_profiler/#{id(file)}">
+                      #{id(file)}
+                    </a>
+                  </p>
+                HTML
+              end
+            end
+          )
+        end
+
+        def show(env, id)
+          raise NotImplementedError
+        end
+      end
+
+      private_constant(:BaseMiddleware)
+    end
+  end
+end

data/lib/app_profiler/viewer/speedscope_remote_viewer/middleware.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require "app_profiler/yarn/command"
+require "app_profiler/yarn/with_speedscope"
+
+module AppProfiler
+  module Viewer
+    class SpeedscopeRemoteViewer < BaseViewer
+      class Middleware < BaseMiddleware
+        include Yarn::WithSpeedscope
+
+        def initialize(app)
+          super
+          @speedscope = Rack::File.new(
+            File.join(AppProfiler.root, "node_modules/speedscope/dist/release")
+          )
+        end
+
+        protected
+
+        attr_reader(:speedscope)
+
+        def viewer(env, path)
+          setup_yarn unless yarn_setup
+          env[Rack::PATH_INFO] = path.delete_prefix("/app_profiler")
+
+          speedscope.call(env)
+        end
+
+        def show(_env, name)
+          profile = profile_files.find do |file|
+            id(file) == name
+          end || raise(ArgumentError)
+
+          render(
+            <<~HTML
+              <script type="text/javascript">
+                var graph = #{profile.read};
+                var json = JSON.stringify(graph);
+                var blob = new Blob([json], { type: 'text/plain' });
+                var objUrl = encodeURIComponent(URL.createObjectURL(blob));
+                var iframe = document.createElement('iframe');
+
+                document.body.style.margin = '0px';
+                document.body.appendChild(iframe);
+
+                iframe.style.width = '100vw';
+                iframe.style.height = '100vh';
+                iframe.style.border = 'none';
+                iframe.setAttribute('src', '/app_profiler/viewer/index.html#profileURL=' + objUrl + '&title=' + 'Flamegraph for #{name}');
+              </script>
+            HTML
+          )
+        end
+      end
+    end
+  end
+end

data/lib/app_profiler/viewer/speedscope_remote_viewer.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require "app_profiler/viewer/speedscope_remote_viewer/base_middleware"
+require "app_profiler/viewer/speedscope_remote_viewer/middleware"
+
+module AppProfiler
+  module Viewer
+    class SpeedscopeRemoteViewer < BaseViewer
+      class << self
+        def view(profile)
+          new(profile).view
+        end
+      end
+
+      def initialize(profile)
+        super()
+        @profile = profile
+      end
+
+      def view
+        id = Middleware.id(@profile.file)
+        AppProfiler.logger.info("[Profiler] Profile available at /app_profiler/#{id}\n")
+      end
+    end
+  end
+end

data/lib/app_profiler/viewer/speedscope_viewer.rb

@@ -1,11 +1,12 @@
 # frozen_string_literal: true
 
+require "app_profiler/yarn/command"
+require "app_profiler/yarn/with_speedscope"
+
 module AppProfiler
   module Viewer
     class SpeedscopeViewer < BaseViewer
-      mattr_accessor :yarn_setup, default: false
-
-      class YarnError < StandardError; end
+      include Yarn::WithSpeedscope
 
       class << self
         def view(profile)
@@ -19,52 +20,7 @@ module AppProfiler
       end
 
       def view
-        yarn("run speedscope \"#{@profile.file}\"")
-      end
-
-      private
-
-      def yarn(command)
-        setup_yarn unless yarn_setup
-        exec("yarn #{command}") do
-          raise YarnError, "Failed to run #{command}."
-        end
-      end
-
-      def setup_yarn
-        ensure_yarn_installed
-        yarn("init --yes") unless package_json_exists?
-        # We currently only support this gem in the root Gemfile.
-        # See https://github.com/Shopify/app_profiler/issues/15
-        # for more information
-        yarn("add --dev --ignore-workspace-root-check speedscope") unless speedscope_added?
-      end
-
-      def ensure_yarn_installed
-        exec("which yarn > /dev/null") do
-          raise(
-            YarnError,
-            <<~MSG.squish
-              `yarn` command not found.
-              Please install `yarn` or make it available in PATH.
-            MSG
-          )
-        end
-        self.yarn_setup = true
-      end
-
-      def package_json_exists?
-        AppProfiler.root.join("package.json").exist?
-      end
-
-      def speedscope_added?
-        AppProfiler.root.join("node_modules/speedscope").exist?
-      end
-
-      def exec(command)
-        system(command).tap do |return_code|
-          yield unless return_code
-        end
+        yarn("run", "speedscope", @profile.file.to_s)
       end
     end
   end

data/lib/app_profiler/yarn/command.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+module AppProfiler
+  module Yarn
+    module Command
+      class YarnError < StandardError; end
+
+      VALID_COMMANDS = [
+        ["which", "yarn"],
+        ["yarn", "init", "--yes"],
+        ["yarn", "add", "speedscope", "--dev", "--ignore-workspace-root-check"],
+        ["yarn", "run", "speedscope", /.*\.json/],
+      ]
+
+      private_constant(:VALID_COMMANDS)
+      mattr_accessor(:yarn_setup, default: false)
+
+      def yarn(command, *options)
+        setup_yarn unless yarn_setup
+
+        exec("yarn", command, *options) do
+          raise YarnError, "Failed to run #{command}."
+        end
+      end
+
+      def setup_yarn
+        ensure_yarn_installed
+
+        yarn("init", "--yes") unless package_json_exists?
+      end
+
+      private
+
+      def ensure_command_valid(command)
+        unless valid_command?(command)
+          raise YarnError, "Illegal command: #{command.join(" ")}."
+        end
+      end
+
+      def valid_command?(command)
+        VALID_COMMANDS.any? do |valid_command|
+          valid_command.zip(command).all? do |valid_part, part|
+            part.match?(valid_part)
+          end
+        end
+      end
+
+      def ensure_yarn_installed
+        exec("which", "yarn", silent: true) do
+          raise(
+            YarnError,
+            <<~MSG.squish
+              `yarn` command not found.
+              Please install `yarn` or make it available in PATH.
+            MSG
+          )
+        end
+        self.yarn_setup = true
+      end
+
+      def package_json_exists?
+        AppProfiler.root.join("package.json").exist?
+      end
+
+      def exec(*command, silent: false)
+        ensure_command_valid(command)
+
+        if silent
+          system(*command, out: File::NULL).tap { |return_code| yield unless return_code }
+        else
+          system(*command).tap { |return_code| yield unless return_code }
+        end
+      end
+    end
+  end
+end

data/lib/app_profiler/yarn/with_speedscope.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module AppProfiler
+  module Yarn
+    module WithSpeedscope
+      include Command
+
+      def setup_yarn
+        super
+        # We currently only support this gem in the root Gemfile.
+        # See https://github.com/Shopify/app_profiler/issues/15
+        # for more information
+        yarn("add", "speedscope", "--dev", "--ignore-workspace-root-check") unless speedscope_added?
+      end
+
+      private
+
+      def speedscope_added?
+        AppProfiler.root.join("node_modules/speedscope").exist?
+      end
+    end
+  end
+end

data/lib/app_profiler.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
-require "active_support"
+require "active_support/core_ext/class"
+require "active_support/core_ext/module"
+require "logger"
 require "app_profiler/version"
 require "app_profiler/railtie" if defined?(Rails::Railtie)
 
@@ -8,6 +10,10 @@ module AppProfiler
   class ConfigurationError < StandardError
   end
 
+  DefaultProfileFormatter = proc do |upload|
+    "#{AppProfiler.speedscope_host}#profileURL=#{upload.url}"
+  end
+
   module Storage
     autoload :BaseStorage, "app_profiler/storage/base_storage"
     autoload :FileStorage, "app_profiler/storage/file_storage"
@@ -17,12 +23,14 @@ module AppProfiler
   module Viewer
     autoload :BaseViewer, "app_profiler/viewer/base_viewer"
     autoload :SpeedscopeViewer, "app_profiler/viewer/speedscope_viewer"
+    autoload :SpeedscopeRemoteViewer, "app_profiler/viewer/speedscope_remote_viewer"
   end
 
-  autoload :Middleware, "app_profiler/middleware"
-  autoload :RequestParameters, "app_profiler/request_parameters"
-  autoload :Profiler, "app_profiler/profiler"
-  autoload :Profile, "app_profiler/profile"
+  require "app_profiler/middleware"
+  require "app_profiler/request_parameters"
+  require "app_profiler/profiler"
+  require "app_profiler/profile"
+  require "app_profiler/server"
 
   mattr_accessor :logger, default: Logger.new($stdout)
   mattr_accessor :root
@@ -32,11 +40,13 @@ module AppProfiler
   mattr_accessor :autoredirect, default: false
   mattr_reader   :profile_header, default: "X-Profile"
   mattr_accessor :context, default: nil
-  mattr_reader   :profile_url_formatter, default: nil
+  mattr_reader   :profile_url_formatter,
+    default: DefaultProfileFormatter
 
   mattr_accessor :storage, default: Storage::FileStorage
   mattr_accessor :viewer, default: Viewer::SpeedscopeViewer
   mattr_accessor :middleware, default: Middleware
+  mattr_accessor :server, default: Server
 
   class << self
     def run(*args, &block)
@@ -59,9 +69,7 @@ module AppProfiler
     end
 
     def request_profile_header
-      @@request_profile_header ||= begin # rubocop:disable Style/ClassVars
-        profile_header.upcase.gsub("-", "_").prepend("HTTP_")
-      end
+      @@request_profile_header ||= profile_header.upcase.tr("-", "_").prepend("HTTP_") # rubocop:disable Style/ClassVars
     end
 
     def profile_data_header
@@ -71,5 +79,9 @@ module AppProfiler
     def profile_url_formatter=(block)
       @@profile_url_formatter = block # rubocop:disable Style/ClassVars
     end
+
+    def profile_url(upload)
+      AppProfiler.profile_url_formatter.call(upload)
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: app_profiler
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.1.1
 platform: ruby
 authors:
 - Gannon McGibbon
@@ -13,7 +13,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-06-09 00:00:00.000000000 Z
+date: 2022-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -30,33 +30,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '5.2'
 - !ruby/object:Gem::Dependency
-  name: railties
+  name: rack
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '5.2'
+        version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '5.2'
+        version: '0'
 - !ruby/object:Gem::Dependency
-  name: rack
+  name: railties
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.2'
 - !ruby/object:Gem::Dependency
   name: stackprof
   requirement: !ruby/object:Gem::Requirement
@@ -86,7 +86,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -100,19 +100,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: minitest
+  name: minitest-stub-const
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.6'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.6'
 - !ruby/object:Gem::Dependency
   name: mocha
   requirement: !ruby/object:Gem::Requirement
@@ -128,19 +128,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: minitest-stub-const
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.6'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.6'
+        version: '0'
 description: 
 email:
 - gems@shopify.com
@@ -157,12 +157,18 @@ files:
 - lib/app_profiler/profiler.rb
 - lib/app_profiler/railtie.rb
 - lib/app_profiler/request_parameters.rb
+- lib/app_profiler/server.rb
 - lib/app_profiler/storage/base_storage.rb
 - lib/app_profiler/storage/file_storage.rb
 - lib/app_profiler/storage/google_cloud_storage.rb
 - lib/app_profiler/version.rb
 - lib/app_profiler/viewer/base_viewer.rb
+- lib/app_profiler/viewer/speedscope_remote_viewer.rb
+- lib/app_profiler/viewer/speedscope_remote_viewer/base_middleware.rb
+- lib/app_profiler/viewer/speedscope_remote_viewer/middleware.rb
 - lib/app_profiler/viewer/speedscope_viewer.rb
+- lib/app_profiler/yarn/command.rb
+- lib/app_profiler/yarn/with_speedscope.rb
 homepage: https://github.com/Shopify/app_profiler
 licenses: []
 metadata:
@@ -175,14 +181,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '2.7'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.17
+rubygems_version: 3.3.3
 signing_key: 
 specification_version: 4
 summary: Collect performance profiles for your Rails application.