Configurable header in Email mode

4 files changed, 30 insertions, 17 deletions

diff --git a/Config.hs b/Config.hs
index 96c9125..a0f872a 100644
--- a/Config.hs
+++ b/Config.hs
@@ -22,7 +22,8 @@ defaultConfig = LazymailConfig {
   , showStatusBar      = True
   , initialPath        = ""
   , filterMaildirsHook =  \mds -> return mds
-  , indexDateFormat    = "%m %d"                                
+  , indexDateFormat    = "%m %d"
+  , headersToShow      = ["date", "from", "to", "cc", "bcc", "subject", "reply-to"]                       
 }
 
 --
diff --git a/Email.hs b/Email.hs
index 4d310c4..659d96b 100644
--- a/Email.hs
+++ b/Email.hs
@@ -17,7 +17,7 @@ getBody msg =
     Single c -> c
     Multi mvs -> case firstTextPart mvs of
       Just mv -> unwrapContent . mime_val_content $ mv
-      Nothing -> "Buggity Buggity Buggity!"
+      Nothing -> "This email has no displayable content."
   where
     unwrapContent (Single c) = c
 
diff --git a/Screen.hs b/Screen.hs
index 5150e93..ea90e75 100644
--- a/Screen.hs
+++ b/Screen.hs
@@ -13,6 +13,7 @@ import Codec.MIME.Type(MIMEValue(..))
 import Control.Monad.Trans(liftIO)
 import Control.Monad.Reader
 import Control.Monad.State
+import Data.Char(toUpper)
 import Data.List(isPrefixOf)
 import System.Exit
 import UI.NCurses
@@ -22,7 +23,7 @@ import Config
 import qualified Handlers as EH
 import Lazymail
 import Maildir
-import Email(lookupField, getBody, getHeaders)
+import Email(lookupField, getBody, getHeaders, lookupField')
 import Print
 import Rfc1342
 import State
@@ -132,7 +133,7 @@ clearMain rows columns = do
 
 {- Helper function of drawMode -}
 drawEmailHelper = do
-  drawEmailHeader
+  drawEmailHeaders
 
   st <- get
   let est = emailState st
@@ -145,23 +146,33 @@ drawEmailHelper = do
   resetCurrentRow
 
 {- Draw the email headers -}
-drawEmailHeader = do
+drawEmailHeaders = do
   st <- get
+  cfg <- ask
+  let hs = getHeaders $ currentEmail . emailState $ st
+  let parsedHeaders = parseHeaders hs 0 $ headersToShow cfg
+
   liftUpdate $ do
-    let hs = getHeaders $ currentEmail . emailState $ st
-    let cropWith xs = normalizeLen $ (screenColumns st) - (length xs)
-    let row = curRowAsInteger st
     setColor $ headerColorID . colorStyle $ st
-    moveCursor row (colPadAsInteger st)
-    drawCroppedString st $ ("Date: " ++) . ppField $ lookupField "date" hs
-    moveCursor (row + 1) (colPadAsInteger st)
-    drawCroppedString st $ ("From: " ++) . ppField $ lookupField "from" hs
-    moveCursor (row + 2) (colPadAsInteger st)
-    drawCroppedString st $ ("To: " ++) . ppField $ lookupField "to" hs
-    moveCursor (row + 3) (colPadAsInteger st)
-    drawCroppedString st $ ("Subject: " ++) . ppField $ lookupField "subject" hs
+    drawHeaders st (curRowAsInteger st) parsedHeaders 
     setColor $ baseColorID . colorStyle $ st
-  put $ st { currentRow = (5 + currentRow st) }
+  put $ st { currentRow = 1 + (length parsedHeaders) + (currentRow st) }
+
+  where
+    parseHeaders _ _ [] = []
+    parseHeaders headers row (h:hs)= do
+      let field = lookupField' h headers
+      case field of
+        Nothing -> parseHeaders headers row hs
+        Just f -> let p = capitalize h ++ ": " ++ (ppField f)
+                  in p:parseHeaders headers (row + 1) hs
+          
+    capitalize str = (toUpper . head $ str):(tail str)
+    drawHeaders _ _ []      = return ()
+    drawHeaders st row (h:hs) = do
+      moveCursor row (colPadAsInteger st)
+      drawCroppedString st h
+      drawHeaders st (row + 1) hs
 
 {- Draw the email body -}
 drawBody _ _ _ [] = return ()
diff --git a/Types.hs b/Types.hs
index 82005e2..4a7cc89 100644
--- a/Types.hs
+++ b/Types.hs
@@ -31,6 +31,7 @@ data LazymailConfig = LazymailConfig {
   , initialPath        :: FilePath
   , filterMaildirsHook :: [FilePath] -> IO [FilePath]
   , indexDateFormat    :: String
+  , headersToShow      :: [String]  
 }
 
 data Email = Email {