From 94f91b660e251c92a99643b8bd0574cd04397df6 Mon Sep 17 00:00:00 2001
From: apainintheneck <apainintheneck@gmail.com>
Date: Sat, 24 Aug 2024 10:38:12 -0700
Subject: [PATCH 1/4] Add startup-stats command

---
 cmd/startup-stats.rb | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100755 cmd/startup-stats.rb

diff --git a/cmd/startup-stats.rb b/cmd/startup-stats.rb
new file mode 100755
index 0000000..3425f44
--- /dev/null
+++ b/cmd/startup-stats.rb
@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module Homebrew
+  module Cmd
+    class StartupStats < AbstractCommand
+      cmd_args do
+        usage_banner "`startup-stats` [<option>]"
+        description <<~EOS
+          Get information about the state of brew when a command
+          is called. This includes loaded constants, requires
+          and other such information.
+        EOS
+
+        switch "--requires", description: "List all requires made before this command was run."
+        switch "--constants", description: "List all constants loaded before this command was run."
+
+        conflicts "--requires", "--constants"
+      end
+
+      def run
+        if args.requires?
+          list_requires
+        elsif args.constants?
+          list_constants
+        else
+          odie "No options provided to the command!"
+        end
+      end
+
+      def list_requires
+        puts $LOADED_FEATURES.sort
+      end
+
+      def list_constants
+        puts Object.constants.sort
+      end
+    end
+  end
+end
\ No newline at end of file

From 8a1d9ac1848402278bd33ac7b38031fd85ee7e7e Mon Sep 17 00:00:00 2001
From: apainintheneck <apainintheneck@gmail.com>
Date: Sat, 24 Aug 2024 11:51:00 -0700
Subject: [PATCH 2/4] Add --defined= and --require= flags and rename existing
 switches

- rename --constants flag to --list-constants
- rename --requires flag to --list-requires
- add --defined= switch which returns true or false based on whether
  a constant has already been defined by the time the command is loaded
- add --require= switch which shows the elapsed time for requiring a
  file along with any added constants or additional requires
---
 cmd/startup-stats.rb | 78 +++++++++++++++++++++++++++++++++++++-------
 1 file changed, 67 insertions(+), 11 deletions(-)

diff --git a/cmd/startup-stats.rb b/cmd/startup-stats.rb
index 3425f44..dff10bb 100755
--- a/cmd/startup-stats.rb
+++ b/cmd/startup-stats.rb
@@ -11,28 +11,84 @@ class StartupStats < AbstractCommand
           and other such information.
         EOS
 
-        switch "--requires", description: "List all requires made before this command was run."
-        switch "--constants", description: "List all constants loaded before this command was run."
+        flag "--defined=", description: "Check if a constant is defined."
+        flag "--require=", description: "Benchmark a single require statement."
+        switch "--list-requires", description: "List all requires made before this command was run."
+        switch "--list-constants", description: "List all constants loaded before this command was run."
 
-        conflicts "--requires", "--constants"
+        %w[
+          --defined
+          --require
+          --list-requires
+          --list-constants
+        ].combination(2) do |combo|
+          conflicts *combo
+        end
       end
 
       def run
-        if args.requires?
-          list_requires
-        elsif args.constants?
-          list_constants
+        if args.defined
+          puts Object.const_defined?(args.defined)
+        elsif args.require
+          require_diagnostics
+        elsif args.list_requires?
+          puts require_list
+        elsif args.list_constants?
+          puts constant_list
         else
           odie "No options provided to the command!"
         end
       end
 
-      def list_requires
-        puts $LOADED_FEATURES.sort
+      def require_list
+        $LOADED_FEATURES.sort
+      end
+
+      def constant_list
+        Object.constants.sort
+      end
+
+      def require_diagnostics
+        odie "'brew' cannot be required directly." if args.require == "brew"
+
+        begin
+          before_require_list = require_list
+          before_constant_list = constant_list
+
+          hash = benchmark { require args.require }
+          hash => { result:, elapsed_seconds: }
+
+          new_require_list = require_list - before_require_list
+          new_constant_list = constant_list - before_constant_list
+
+          if result
+            oh1 "require '#{args.require}'"
+            ohai "elapsed seconds"
+            puts "- #{elapsed_seconds}"
+            if new_require_list.any?
+              puts
+              ohai "new requires"
+              new_require_list.each { puts "- #{_1}" }
+            end
+            if new_constant_list.any?
+              puts
+              ohai "new constants"
+              new_constant_list.each { puts "- #{_1}" }
+            end
+          else
+            odie "'#{args.require}' has already been required."
+          end
+        rescue LoadError => e
+          odie e.to_s
+        end
       end
 
-      def list_constants
-        puts Object.constants.sort
+      def benchmark
+        start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+        result = yield
+        end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+        elapsed_seconds = end_time - start_time
+        { result:, elapsed_seconds: }
       end
     end
   end

From 408a4135c19e4c47051f84b4ad80d88c57029de9 Mon Sep 17 00:00:00 2001
From: apainintheneck <apainintheneck@gmail.com>
Date: Sat, 24 Aug 2024 12:00:38 -0700
Subject: [PATCH 3/4] Update readme, fix linter errors and add test

---
 README.md            | 20 ++++++++++++++++++++
 Rakefile             |  5 +++++
 cmd/startup-stats.rb | 11 +++++------
 3 files changed, 30 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 039fb6d..dec785f 100644
--- a/README.md
+++ b/README.md
@@ -102,6 +102,26 @@ running other brew commands simultaneously.
   -h, --help                       Show this message.
 ```
 
+### brew startup-stats
+
+```
+Usage: brew startup-stats [option]
+
+Get information about the state of brew at the time a command is called. This
+includes loaded constants, requires and other such information.
+
+      --defined                    Check if a constant is defined.
+      --require                    Diagnostics about a single require statement.
+      --list-requires              List all requires made before this command
+                                   was run.
+      --list-constants             List all constants loaded before this command
+                                   was run.
+  -d, --debug                      Display any debugging information.
+  -q, --quiet                      Make some output more quiet.
+  -v, --verbose                    Make some output more verbose.
+  -h, --help                       Show this message.
+```
+
 ## Development
 
 Linting and readme checks are run on each pull request while integration tests are only run weekly.
diff --git a/Rakefile b/Rakefile
index d91dabc..228c31c 100644
--- a/Rakefile
+++ b/Rakefile
@@ -112,12 +112,17 @@ namespace "test" do
     end
   end
 
+  task :"startup-stats" do
+    cmd "brew", "startup-stats", "--require", "tap"
+  end
+
   task :all do
     %w[
       test:api-readall-test
       test:branch-compare
       test:generate-api-diff
       test:service-diff
+      test:startup-stats
     ].each_with_index do |task, index|
       puts "--------------------" if index.positive?
       Rake::Task[task].invoke
diff --git a/cmd/startup-stats.rb b/cmd/startup-stats.rb
index dff10bb..54f1733 100755
--- a/cmd/startup-stats.rb
+++ b/cmd/startup-stats.rb
@@ -6,13 +6,12 @@ class StartupStats < AbstractCommand
       cmd_args do
         usage_banner "`startup-stats` [<option>]"
         description <<~EOS
-          Get information about the state of brew when a command
-          is called. This includes loaded constants, requires
-          and other such information.
+          Get information about the state of brew at the time a command is called.
+          This includes loaded constants, requires and other such information.
         EOS
 
         flag "--defined=", description: "Check if a constant is defined."
-        flag "--require=", description: "Benchmark a single require statement."
+        flag "--require=", description: "Diagnostics about a single require statement."
         switch "--list-requires", description: "List all requires made before this command was run."
         switch "--list-constants", description: "List all constants loaded before this command was run."
 
@@ -22,7 +21,7 @@ class StartupStats < AbstractCommand
           --list-requires
           --list-constants
         ].combination(2) do |combo|
-          conflicts *combo
+          conflicts(*combo)
         end
       end
 
@@ -92,4 +91,4 @@ def benchmark
       end
     end
   end
-end
\ No newline at end of file
+end

From 01a0122b39b08a0eaa4b0ee4e74bec9a956ba56c Mon Sep 17 00:00:00 2001
From: apainintheneck <apainintheneck@gmail.com>
Date: Sat, 24 Aug 2024 12:08:21 -0700
Subject: [PATCH 4/4] Add integration test to CI

---
 .github/workflows/integration_tests.yml | 2 ++
 Rakefile                                | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/integration_tests.yml b/.github/workflows/integration_tests.yml
index b33c4d3..7672bf8 100644
--- a/.github/workflows/integration_tests.yml
+++ b/.github/workflows/integration_tests.yml
@@ -40,3 +40,5 @@ jobs:
     - run: rake test:generate-api-diff
 
     - run: rake test:service-diff
+
+    - run: rake test:startup-stats
diff --git a/Rakefile b/Rakefile
index 228c31c..df47ed3 100644
--- a/Rakefile
+++ b/Rakefile
@@ -153,7 +153,7 @@ task :"missing-tests" do
       Missing integration tests for the following commands:
       #{missing_test_list}
 
-      Add tests to the #{INTEGRATON_TESTS} file.
+      Add tests to the #{INTEGRATION_TESTS_FILE} file.
     EOS
   end
 end