From 53c2a2f49a3677f3b3ac7ee17bf514f24f1c5893 Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@snoyman.com>
Date: Fri, 20 Aug 2010 09:01:58 +0300
Subject: [PATCH] Included experimental Yesod.Mail

---
 Yesod/Mail.hs | 116 ++++++++++++++++++++++++++++++++++++++++++++++++++
 mail.hs       |  14 ++++++
 yesod.cabal   |   4 +-
 3 files changed, 133 insertions(+), 1 deletion(-)
 create mode 100644 Yesod/Mail.hs
 create mode 100644 mail.hs

diff --git a/Yesod/Mail.hs b/Yesod/Mail.hs
new file mode 100644
index 00000000..7411110c
--- /dev/null
+++ b/Yesod/Mail.hs
@@ -0,0 +1,116 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Yesod.Mail
+    ( Boundary (..)
+    , Mail (..)
+    , Part (..)
+    , Encoding (..)
+    , renderMail
+    , renderMail'
+    , sendmail
+    , Disposition (..)
+    , renderSendMail
+    ) where
+
+import qualified Data.ByteString.Lazy as L
+import Text.Blaze.Builder.Utf8
+import Text.Blaze.Builder.Core
+import Data.Monoid
+import System.Random
+import Control.Arrow
+import System.Process
+import System.IO
+import System.Exit
+import Codec.Binary.Base64 (encode)
+import Control.Monad ((<=<))
+
+newtype Boundary = Boundary { unBoundary :: String }
+instance Random Boundary where
+    randomR = const random
+    random =
+        first (Boundary . map toChar) . sequence' (replicate 10 (randomR (0, 61)))
+      where
+        sequence' [] g = ([], g)
+        sequence' (f:fs) g =
+            let (f', g') = f g
+                (fs', g'') = sequence' fs g'
+             in (f' : fs', g'')
+        toChar i
+            | i < 26 = toEnum $ i + fromEnum 'A'
+            | i < 52 = toEnum $ i + fromEnum 'a' - 26
+            | otherwise = toEnum $ i + fromEnum '0' - 52
+
+data Mail = Mail
+    { mailHeaders :: [(String, String)]
+    , mailPlain :: String
+    , mailParts :: [Part]
+    }
+
+data Encoding = None | Base64
+
+data Part = Part
+    { partType :: String -- ^ content type
+    , partEncoding :: Encoding
+    , partDisposition :: Disposition
+    , partContent :: L.ByteString
+    }
+
+data Disposition = Inline | Attachment String
+
+renderMail :: Boundary -> Mail -> L.ByteString
+renderMail (Boundary b) (Mail headers plain parts) = toLazyByteString $ mconcat
+    [ mconcat $ map showHeader headers
+    , mconcat $ map showHeader
+        [ ("MIME-Version", "1.0")
+        , ("Content-Type", "multipart/mixed; boundary=\""
+            ++ b ++ "\"")
+        ]
+    , fromByteString "\n"
+    , fromString plain
+    , mconcat $ map showPart parts
+    , fromByteString "\n--"
+    , fromString b
+    , fromByteString "--"
+    ]
+  where
+    showHeader (k, v) = mconcat
+        [ fromString k
+        , fromByteString ": "
+        , fromString v
+        , fromByteString "\n"
+        ]
+    showPart (Part contentType encoding disposition content) = mconcat
+        [ fromByteString "\n--"
+        , fromString b
+        , fromByteString "\n"
+        , showHeader ("Content-Type", contentType)
+        , case encoding of
+            None -> mempty
+            Base64 -> showHeader ("Content-Transfer-Encoding", "base64")
+        , case disposition of
+            Inline -> mempty
+            Attachment filename ->
+                showHeader ("Content-Disposition", "attachment; filename=" ++ filename)
+        , fromByteString "\n"
+        , case encoding of
+            None -> writeList writeByteString $ L.toChunks content
+            Base64 -> fromString $ encode $ L.unpack content
+        ]
+
+renderMail' :: Mail -> IO L.ByteString
+renderMail' m = do
+    b <- randomIO
+    return $ renderMail b m
+
+sendmail :: L.ByteString -> IO ()
+sendmail lbs = do
+    (Just hin, _, _, phandle) <- createProcess $ (proc
+        "/usr/sbin/sendmail" ["-t"]) { std_in = CreatePipe }
+    L.hPut hin lbs
+    hClose hin
+    exitCode <- waitForProcess phandle
+    case exitCode of
+        ExitSuccess -> return ()
+        _ -> error $ "sendmail exited with error code " ++ show exitCode
+
+renderSendMail :: Mail -> IO ()
+renderSendMail = sendmail <=< renderMail'
diff --git a/mail.hs b/mail.hs
new file mode 100644
index 00000000..8e39e0e2
--- /dev/null
+++ b/mail.hs
@@ -0,0 +1,14 @@
+import Yesod.Mail
+import qualified Data.ByteString.Lazy.Char8 as L
+import System.Environment
+
+main = do
+    [dest] <- getArgs
+    let p1 = Part "text/html" None Inline $ L.pack "<h1>Hello World!!!</h1>"
+    lbs <- L.readFile "mail.hs"
+    let p2 = Part "text/plain" Base64 (Attachment "mail.hs") lbs
+    let mail = Mail
+            [("To", dest), ("Subject", "mail quine")]
+            "Plain stuff. Mime-clients should not show it."
+            [p1, p2]
+    renderSendMail mail
diff --git a/yesod.cabal b/yesod.cabal
index 424d8a8b..6153ea9c 100644
--- a/yesod.cabal
+++ b/yesod.cabal
@@ -46,7 +46,8 @@ library
                      MonadCatchIO-transformers >= 0.2.2.0 && < 0.3,
                      data-object >= 0.3.1 && < 0.4,
                      network >= 2.2.1.5 && < 2.3,
-                     email-validate >= 0.2.5 && < 0.3
+                     email-validate >= 0.2.5 && < 0.3,
+                     process >= 1.0.1 && < 1.1
     exposed-modules: Yesod
                      Yesod.Content
                      Yesod.Dispatch
@@ -61,6 +62,7 @@ library
                      Yesod.Handler
                      Yesod.Internal
                      Yesod.Json
+                     Yesod.Mail
                      Yesod.Request
                      Yesod.Widget
                      Yesod.Yesod