From f23d4d4c75049f58f34cdd5bc29e2dbdf1ce7b98 Mon Sep 17 00:00:00 2001
From: Connor Mann <connor@cnnrmnn.com>
Date: Tue, 18 May 2021 11:00:17 -0400
Subject: [PATCH] Switch to using `to_keg` and `to_kegs`

---
 Library/Homebrew/cmd/link.rb | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/Library/Homebrew/cmd/link.rb b/Library/Homebrew/cmd/link.rb
index bf6a179021..a003ad6d8b 100644
--- a/Library/Homebrew/cmd/link.rb
+++ b/Library/Homebrew/cmd/link.rb
@@ -42,15 +42,18 @@ module Homebrew
       verbose:   args.verbose?,
     }
 
-    args.named.to_formulae_and_casks(only: :formula, method: :kegs).freeze.each do |keg|
-      head = keg.version.head?
-      next if args.HEAD? != head
+    kegs = if args.HEAD?
+      args.named.to_kegs.filter { |keg| keg.version.head? }
+    else
+      args.named.to_keg
+    end
 
+    kegs.freeze.each do |keg|
       keg_only = Formulary.keg_only?(keg.rack)
 
       if keg.linked?
         opoo "Already linked: #{keg}"
-        name_and_flag = "#{"--HEAD " if head}#{"--force " if keg_only}#{keg.name}"
+        name_and_flag = "#{"--HEAD " if args.HEAD?}#{"--force " if keg_only}#{keg.name}"
         puts <<~EOS
           To relink:
             brew unlink #{keg.name} && brew link #{name_and_flag}