From d20b946cfe86dc464abd5fab34915d565d29e2e3 Mon Sep 17 00:00:00 2001
From: Mike McQuaid <mike@mikemcquaid.com>
Date: Mon, 28 Dec 2020 09:49:54 +0000
Subject: [PATCH] dev-cmd/unbottled: improve output.

Handle a few more edge cases:
- improve the unbottled output heading so it makes sense when we're
  outputting bottles that are ready to bottle
- output already bottled formulae if named
- handle Linux/Mac requiring formulae and output if named
- output bottle unneeded/disabled formulae if named
- use green/yellow/red for bottles ready to be built now/later/never
---
 Library/Homebrew/dev-cmd/unbottled.rb | 53 ++++++++++++++++++++-------
 1 file changed, 39 insertions(+), 14 deletions(-)

diff --git a/Library/Homebrew/dev-cmd/unbottled.rb b/Library/Homebrew/dev-cmd/unbottled.rb
index 2c440808b9027..f570b5e48a2d4 100644
--- a/Library/Homebrew/dev-cmd/unbottled.rb
+++ b/Library/Homebrew/dev-cmd/unbottled.rb
@@ -41,8 +41,8 @@ def unbottled
       raise UsageError, "cannot specify `<formula>` and `--total`."
     end
 
-    formulae, all_formulae, sort, formula_installs =
-      formulae_all_sort_installs_from_args(args)
+    formulae, all_formulae, formula_installs =
+      formulae_all_installs_from_args(args)
     deps_hash, uses_hash = deps_uses_from_formulae(all_formulae)
 
     if args.dependents?
@@ -65,10 +65,11 @@ def unbottled
     else
       ["installs", formula_installs]
     end
-    output_unbottled(sort, formulae, deps_hash, noun, hash)
+
+    output_unbottled(formulae, deps_hash, noun, hash, args.named.present?)
   end
 
-  def formulae_all_sort_installs_from_args(args)
+  def formulae_all_installs_from_args(args)
     if args.named.present?
       formulae = all_formulae = args.named.to_formulae
     elsif args.total?
@@ -76,7 +77,7 @@ def formulae_all_sort_installs_from_args(args)
     elsif args.dependents?
       formulae = all_formulae = Formula.to_a
 
-      sort = " (sorted by installs in the last 90 days)"
+      @sort = " (sorted by installs in the last 90 days)"
     else
       formula_installs = {}
 
@@ -101,12 +102,12 @@ def formulae_all_sort_installs_from_args(args)
           nil
         end
       end.compact
-      sort = " (sorted by installs in the last 90 days)"
+      @sort = " (sorted by installs in the last 90 days)"
 
       all_formulae = Formula
     end
 
-    [formulae, all_formulae, sort, formula_installs]
+    [formulae, all_formulae, formula_installs]
   end
 
   def deps_uses_from_formulae(all_formulae)
@@ -146,30 +147,54 @@ def output_total(formulae)
     puts "#{unbottled_formulae}/#{formulae.length} remaining."
   end
 
-  def output_unbottled(sort, formulae, deps_hash, noun, hash)
-    ohai "Unbottled :#{@bottle_tag} dependencies#{sort}"
+  def output_unbottled(formulae, deps_hash, noun, hash, any_named_args)
+    ohai ":#{@bottle_tag} bottle status#{@sort}"
     any_found = T.let(false, T::Boolean)
 
     formulae.each do |f|
-      next if f.bottle_specification.tag?(@bottle_tag)
+      name = f.name.downcase
+      if f.bottle_specification.tag?(@bottle_tag)
+        puts "#{Tty.bold}#{Tty.green}#{name}#{Tty.reset}: already bottled" if any_named_args
+        next
+      end
+
+      requirement_classes = f.recursive_requirements.map(&:class)
+      if @bottle_tag.to_s.end_with?("_linux")
+        if requirement_classes.include?(MacOSRequirement)
+          puts "#{Tty.bold}#{Tty.red}#{name}#{Tty.reset}: requires macOS" if any_named_args
+          next
+        end
+      elsif requirement_classes.include?(LinuxRequirement)
+        puts "#{Tty.bold}#{Tty.red}#{name}#{Tty.reset}: requires Linux" if any_named_args
+        next
+      end
+
+      if f.bottle_unneeded? || f.bottle_disabled?
+        reason = if f.bottle_unneeded?
+          "unneeded"
+        else
+          "disabled"
+        end
+        puts "#{Tty.bold}#{Tty.red}#{name}#{Tty.reset}: bottle #{reason}" if any_named_args
+        next
+      end
 
       deps = Array(deps_hash[f.name]).reject do |dep|
         dep.bottle_specification.tag?(@bottle_tag) || dep.bottle_unneeded?
       end
 
       if deps.blank?
-        next if f.bottle_unneeded?
-
         count = " (#{hash[f.name]} #{noun})" if noun
-        puts "#{f.name}#{count}: ready to bottle"
+        puts "#{Tty.bold}#{Tty.green}#{name}#{Tty.reset}#{count}: ready to bottle"
         next
       end
 
       any_found ||= true
       count = " (#{hash[f.name]} #{noun})" if noun
-      puts "#{f.name}#{count}: #{deps.join(" ")}"
+      puts "#{Tty.bold}#{Tty.yellow}#{name}#{Tty.reset}#{count}: unbottled deps: #{deps.join(" ")}"
     end
     return if any_found
+    return if any_named_args
 
     puts "No unbottled dependencies found!"
   end