From 4032c3e28690821e7779bb31b518ca6e258a4993 Mon Sep 17 00:00:00 2001
From: ashmaroli <ashmaroli@users.noreply.github.com>
Date: Thu, 1 Mar 2018 15:27:30 +0530
Subject: [PATCH] Rewrite `script/rubyprof` as a Ruby script (#6813)

Merge pull request 6813
---
 script/rubyprof | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/script/rubyprof b/script/rubyprof
index 18c5777b..51a89d91 100755
--- a/script/rubyprof
+++ b/script/rubyprof
@@ -1,19 +1,23 @@
-#!/usr/bin/env bash
+#!/usr/bin/env ruby
 
-export BENCHMARK=1
+require "ruby-prof"
+require File.expand_path("../lib/jekyll", __dir__)
 
-TEST_SCRIPT="Jekyll::Commands::Build.process({'source' => 'site'})"
+result = RubyProf.profile do
+  Jekyll::Commands::Build.process({
+    "source"      => File.expand_path("../docs", __dir__),
+    "destination" => File.expand_path("../docs/_site", __dir__),
+  })
+end
 
-RUBY=$(cat <<RUBY
-  require 'ruby-prof'
-  result = RubyProf.profile{ ${TEST_SCRIPT} }
-  printer = RubyProf::CallTreePrinter.new(result)
-  filename = "tmp/ruby_prof_#{rand 10000}"
-  puts "Writing profile to #{filename}"
-  file = File.open(filename, "w")
-  printer.print(file, {})
-  file.close
-RUBY
-)
+puts "\nProcessing result.."
 
-bundle exec ruby -r ./lib/jekyll -e "${RUBY}"
+dir_path  = File.expand_path("../tmp", __dir__)
+file_path = File.join(dir_path, "rubyprof-#{Time.now.strftime('%Y%m%d%H%M%S')}")
+
+FileUtils.mkdir_p(dir_path) unless Dir.exist?(dir_path)
+File.open(file_path, "wb") do |file|
+  RubyProf::FlatPrinter.new(result).print(file)
+end
+
+puts "Profile result printed to #{file_path.cyan}"