-- This file is part of hs-notmuch - Haskell Notmuch binding
-- Copyright (C) 2017  Fraser Tweedale
--
-- hs-notmuch is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program.  If not, see <http://www.gnu.org/licenses/>.

{-

Add or remove a tag from all messages matching query.

-}

import Control.Monad.Except (runExceptT)
import Data.Foldable (traverse_)
import Data.String (fromString)
import System.Environment (getArgs)
import System.Exit (die)

import Notmuch

{-# ANN module "HLint: ignore Use lambda-case" #-}


main :: IO ()
main = getArgs >>= \args -> case args of
  [dbDir, search, '+':tag] -> go dbDir search tag messageAddTag
  [dbDir, search, '-':tag] -> go dbDir search tag messageRemoveTag
  _ -> die "usage: hs-notmuch-tag-set DB-DIR SEARCH-TERM +TAG|-TAG"
  where
    go dbDir searchTerm tag f =
      runExceptT (do
        db <- databaseOpen dbDir
        query db (Bare searchTerm) >>= messages >>= traverse_ (f (fromString tag))
      ) >>= either (die . (show :: Status -> String)) pure