From 74acf17a3158938c8034fdf58e5b05cfdabb8b9f Mon Sep 17 00:00:00 2001
From: Joey Hess <joeyh@joeyh.name>
Date: Wed, 21 Apr 2021 14:19:58 -0400
Subject: [PATCH] refactoring

---
 Annex/Branch.hs | 27 +++++++++++++++++++++++++++
 CmdLine/Seek.hs | 17 ++++-------------
 2 files changed, 31 insertions(+), 13 deletions(-)

diff --git a/Annex/Branch.hs b/Annex/Branch.hs
index 9b20c7a907..73ab2ac10d 100644
--- a/Annex/Branch.hs
+++ b/Annex/Branch.hs
@@ -31,6 +31,7 @@ module Annex.Branch (
 	performTransitions,
 	withIndex,
 	precache,
+	overBranchFileContents,
 ) where
 
 import qualified Data.ByteString as B
@@ -66,6 +67,7 @@ import Annex.HashObject
 import Git.Types (Ref(..), fromRef, fromRef', RefDate, TreeItemType(..))
 import Git.FilePath
 import Annex.CatFile
+import Git.CatFile (catObjectStreamLsTree)
 import Annex.Perms
 import Logs
 import Logs.Transitions
@@ -752,3 +754,28 @@ rememberTreeishLocked treeish graftpoint jl = do
 	-- say that the index contains c'.
 	setIndexSha c'
 
+{- Runs an action on the content of selected files from the branch.
+ - This is much faster than reading the content of each file in turn,
+ - because it lets git cat-file stream content as fast as it can run.
+ -
+ - The action is passed an IO action that it can repeatedly call to read
+ - the next file and its contents. When there are no more files, that
+ - action will return Nothing.
+ -}
+overBranchFileContents
+	:: (RawFilePath -> Maybe v)
+	-> (IO (Maybe (v, RawFilePath, Maybe L.ByteString)) -> Annex ())
+	-> Annex ()
+overBranchFileContents select go = do
+	void update
+	g <- Annex.gitRepo
+	(l, cleanup) <- inRepo $ Git.LsTree.lsTree
+		Git.LsTree.LsTreeRecursive
+		(Git.LsTree.LsTreeLong False)
+		fullname
+	let select' f = fmap (\v -> (v, f)) (select f)
+	let go' reader = go $ reader >>= \case
+		Nothing -> return Nothing
+		Just ((v, f), content) -> return (Just (v, f, content))
+	catObjectStreamLsTree l (select' . getTopFilePath . Git.LsTree.file) g go'
+	liftIO $ void cleanup
diff --git a/CmdLine/Seek.hs b/CmdLine/Seek.hs
index 1245ec4d4d..3f0bfa8ce3 100644
--- a/CmdLine/Seek.hs
+++ b/CmdLine/Seek.hs
@@ -9,8 +9,6 @@
  - Licensed under the GNU AGPL version 3 or higher.
  -}
 
-{-# LANGUAGE TupleSections #-}
-
 module CmdLine.Seek where
 
 import Annex.Common
@@ -273,25 +271,18 @@ withKeyOptions' ko auto mkkeyaction fallbackaction worktreeitems = do
 		checktimelimit <- mkCheckTimeLimit
 		keyaction <- mkkeyaction
 		config <- Annex.getGitConfig
-		g <- Annex.gitRepo
 		
-		void Annex.Branch.update
-		(l, cleanup) <- inRepo $ LsTree.lsTree
-			LsTree.LsTreeRecursive
-			(LsTree.LsTreeLong False)
-			Annex.Branch.fullname
-		let getk f = fmap (,f) (locationLogFileKey config f)
+		let getk = locationLogFileKey config
 		let discard reader = reader >>= \case
 			Nothing -> noop
 			Just _ -> discard reader
 		let go reader = liftIO reader >>= \case
-			Nothing -> return ()
-			Just ((k, f), content) -> checktimelimit (discard reader) $ do
+			Just (k, f, content) -> checktimelimit (discard reader) $ do
 				maybe noop (Annex.Branch.precache f) content
 				keyaction Nothing (SeekInput [], k, mkActionItem k)
 				go reader
-		catObjectStreamLsTree l (getk . getTopFilePath . LsTree.file) g go
-		liftIO $ void cleanup
+			Nothing -> return ()
+		Annex.Branch.overBranchFileContents getk go
 
 	runkeyaction getks = do
 		keyaction <- mkkeyaction
-- 
2.30.2