From 15646fa9e492ded6978620e8e17239c636cfbb16 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kier=C3=A1n=20Meinhardt?= <kieran.meinhardt@gmail.com>
Date: Wed, 30 Sep 2020 10:56:37 +0200
Subject: kmein config: add saveAttachment action

---
 config/kmein.hs | 44 ++++++++++++++++++++++++++++++++++++++++++--
 much.cabal      | 11 ++++++++---
 src/Notmuch.hs  | 10 ++++++++++
 3 files changed, 60 insertions(+), 5 deletions(-)

diff --git a/config/kmein.hs b/config/kmein.hs
index 77d14d7..294bc3b 100644
--- a/config/kmein.hs
+++ b/config/kmein.hs
@@ -1,4 +1,5 @@
 {-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE LambdaCase #-}
 {-# LANGUAGE RecordWildCards #-}
 
 module Main (main) where
@@ -7,17 +8,25 @@ import Much.Action
 import Much.Core
 import Much.State
 import Much.TreeView
+import qualified Notmuch
 import qualified Notmuch.Message as Notmuch
+import qualified Data.ByteString.Lazy.Char8 as LBS8
 
 import Blessings.String
 import Control.Monad
 import Data.Maybe
+import Data.Time.Format
 import Scanner
+import System.Exit
+import System.FilePath
+import System.IO
 import System.Posix.Signals
+import System.Process
 import Text.Hyphenation
 import Text.LineBreak
-import qualified Data.Tree as Tree
 import qualified Data.Map as M
+import qualified Data.Text as T
+import qualified Data.Tree as Tree
 import qualified Data.Tree.Zipper as Z
 
 {- notmuch's special tags are:
@@ -26,9 +35,12 @@ import qualified Data.Tree.Zipper as Z
   automatic: attachment signed encrypted
   cli default tags: unread inbox deleted spam
 
-ref: https://notmuchmail.org/special-tags/
+  ref: https://notmuchmail.org/special-tags/
 -}
 
+attachmentDestination :: FilePath
+attachmentDestination = "/tmp"
+
 main :: IO ()
 main =
     mainWithState emptyState
@@ -61,6 +73,28 @@ main =
       , query = "tag:inbox"
       }
 
+
+saveAttachment :: State -> IO (Maybe FilePath, State)
+saveAttachment q
+  | TVMessagePart message part <- Z.label (cursor q) = do
+    let query = Notmuch.unMessageID $ Notmuch.messageId message
+        defaultFilename = "much_part_" <> show (Notmuch.partID part) <> "_" <> formatTime defaultTimeLocale "%s" (Notmuch.messageTime message)
+        destination =
+          attachmentDestination </>
+            maybe defaultFilename T.unpack (Notmuch.partContentFilename part)
+    Notmuch.notmuchShowPartRaw query (Notmuch.partID part) >>= \case
+      Right byteString -> do
+        LBS8.writeFile destination byteString
+        return
+          ( Just destination
+          , q { flashMessage = Plain "Attachment saved to " <> SGR [1] (Plain destination) <> Plain "." }
+          )
+      Left err -> return
+        ( Nothing
+        , q { flashMessage = SGR [38,5,9] $ Plain err }
+        )
+  | otherwise = return (Nothing, q { flashMessage = SGR [38,5,9] $ Plain "Cursor not on attachment." })
+
 myKeymap :: String -> State -> IO State
 myKeymap "h" = closeFold
 myKeymap "l" = openFold
@@ -77,6 +111,12 @@ myKeymap "\ESC[D" = moveTreeRight 10 -- right
 
 myKeymap "r" = notmuchSearch
 
+myKeymap "S" = fmap snd . saveAttachment
+myKeymap "o" = saveAttachment >=> \case
+  (Nothing, q') -> return q'
+  (Just filePath, q') ->
+    q' <$ runCommand ("xdg-open " <> filePath)
+
 myKeymap "q" = \q -> q <$ raiseSignal sigINT
 
 myKeymap "*" = toggleTagAtCursor "flagged"
diff --git a/much.cabal b/much.cabal
index 5f3c18f..3d5181e 100644
--- a/much.cabal
+++ b/much.cabal
@@ -37,13 +37,18 @@ executable much-kmein
   ghc-options:      -O2 -threaded -with-rtsopts=-N
   build-depends:    much
                   , base
-                  , unix
-                  , scanner
                   , blessings
+                  , bytestring
+                  , containers
+                  , filepath
                   , hyphenation
                   , linebreak
-                  , containers
+                  , process
                   , rosezipper
+                  , scanner
+                  , text
+                  , time
+                  , unix
 
 library
   hs-source-dirs:  src
diff --git a/src/Notmuch.hs b/src/Notmuch.hs
index f86bd3d..0781650 100644
--- a/src/Notmuch.hs
+++ b/src/Notmuch.hs
@@ -170,6 +170,16 @@ notmuchShow term = do
         either error id (eitherDecodeLenient' c')
 
 
+notmuchShowPartRaw :: String -> Int -> IO (Either String LBS8.ByteString)
+notmuchShowPartRaw term partId = do
+    (exitCode, out, err) <-
+        notmuch' [ "show", "--format=raw"
+                 , "--part=" <> show partId
+                 , term ]
+    return $ case exitCode of
+        ExitSuccess -> Right out
+        _ -> Left $ show exitCode <> ": " <> LBS8.unpack err
+
 notmuchShowPart :: String -> Int -> IO (Either String MessagePart)
 notmuchShowPart term partId = do
     -- TODO handle partId == 0 and partId > N
-- 
cgit v1.2.3