feat: add `--todo` to `list`

1 files changed, 29 insertions, 12 deletions

diff --git a/app/Main.hs b/app/Main.hs
index 32f4037..e9189d4 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -19,7 +19,6 @@ import Data.Attoparsec.Text qualified as A
 import Data.ByteString.Lazy qualified as LB
 import Data.Default
 import Data.Digest.Pure.SHA (sha256, showDigest)
-import Data.Function ((&))
 import Data.List
 import Data.Map qualified as M
 import Data.Maybe (catMaybes, fromMaybe)
@@ -48,8 +47,14 @@ data Args = Args
   }
 
 data Cmd
-  = Consume Bool [FilePath]
-  | List [Filter]
+  = Consume
+      { keep :: Bool,
+        inputs :: [FilePath]
+      }
+  | List
+      { filters :: [Filter],
+        todo :: Bool
+      }
   | Todo
 
 args :: O.Parser Args
@@ -71,19 +76,20 @@ consumeCmd :: O.Parser Cmd
 consumeCmd =
   Consume
     <$> keepArg
-    <*> filePathsArg
+    <*> inputsArg
 
 listCmd :: O.Parser Cmd
 listCmd =
   List
-    <$> filterArg
+    <$> filtersArg
+    <*> todoArg
 
 todoCmd :: O.Parser Cmd
 todoCmd =
   pure Todo
 
-filePathsArg :: O.Parser [FilePath]
-filePathsArg =
+inputsArg :: O.Parser [FilePath]
+inputsArg =
   O.many (O.strArgument (O.metavar "FILE" <> O.action "file"))
 
 keepArg :: O.Parser Bool
@@ -93,8 +99,8 @@ keepArg =
         <> O.help "Keep input document"
     )
 
-filterArg :: O.Parser [Filter]
-filterArg =
+filtersArg :: O.Parser [Filter]
+filtersArg =
   O.many $
     O.option
       (O.maybeReader parse)
@@ -106,6 +112,13 @@ filterArg =
     parse ('@' : tagKey) = Just (FilterByTag (T.pack tagKey))
     parse _ = Nothing
 
+todoArg :: O.Parser Bool
+todoArg =
+  O.switch
+    ( O.long "todo"
+        <> O.help "Run command `todo` on listed documents."
+    )
+
 data Filter
   = FilterByTag T.Text
 
@@ -118,10 +131,10 @@ main = do
   ensureDir "index"
 
   O.execParser (O.info (args O.<**> O.helper) O.idm) >>= \case
-    Args {cmd = Consume keep filePaths} ->
+    Args {cmd = Consume {keep, inputs}} ->
       mapM_ putStrLn
-        =<< parMapM (consume1 keep) (map (cwd </>) filePaths)
-    Args {cmd = List filters} -> do
+        =<< parMapM (consume1 keep) (map (cwd </>) inputs)
+    Args {cmd = List {filters, todo = False}} -> do
       mapM_
         ( \(Document {iFileName, index}) -> do
             if hasTag (Tag "todo" Nothing) index
@@ -130,6 +143,10 @@ main = do
         )
         . applyFilters filters
         =<< getDocuments
+    Args {cmd = List {filters, todo = True}} -> do
+      processDocuments
+        . applyFilters filters
+        =<< getDocuments
     Args {cmd = Todo} -> do
       processDocuments
         . applyFilters [FilterByTag "todo"]