From 33c39662b994a2bf85c0a04d7b45bb64af8e9f82 Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@snoyman.com>
Date: Wed, 11 Jul 2012 15:58:36 +0300
Subject: [PATCH] Reflect wai changes for #262

---
 yesod-core/Yesod/Handler.hs          | 7 +++----
 yesod-core/Yesod/Internal/Core.hs    | 6 +++---
 yesod-core/Yesod/Internal/Request.hs | 6 +++---
 3 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/yesod-core/Yesod/Handler.hs b/yesod-core/Yesod/Handler.hs
index 0e355161..4640fba8 100644
--- a/yesod-core/Yesod/Handler.hs
+++ b/yesod-core/Yesod/Handler.hs
@@ -178,7 +178,6 @@ import Control.Monad.Trans.Resource
 import Control.Monad.Base
 import Yesod.Routes.Class
 import Data.Word (Word64)
-import Data.Conduit (Sink)
 import Language.Haskell.TH.Syntax (Loc)
 
 class YesodSubRoute s y where
@@ -343,12 +342,12 @@ rbHelper upload =
         FileUploadDisk s -> rbHelper' s mkFileInfoFile
         FileUploadSource s -> rbHelper' s mkFileInfoSource
 
-rbHelper' :: Sink S8.ByteString (ResourceT IO) x
+rbHelper' :: NWP.BackEnd x
           -> (Text -> Text -> x -> FileInfo)
           -> W.Request
           -> ResourceT IO ([(Text, Text)], [(Text, FileInfo)])
-rbHelper' sink mkFI req =
-    (map fix1 *** map fix2) <$> (NWP.parseRequestBody sink req)
+rbHelper' backend mkFI req =
+    (map fix1 *** map fix2) <$> (NWP.parseRequestBody backend req)
   where
     fix1 = go *** go
     fix2 (x, NWP.FileInfo a b c) =
diff --git a/yesod-core/Yesod/Internal/Core.hs b/yesod-core/Yesod/Internal/Core.hs
index 2ae32d5c..0eb78497 100644
--- a/yesod-core/Yesod/Internal/Core.hs
+++ b/yesod-core/Yesod/Internal/Core.hs
@@ -83,7 +83,7 @@ import Data.Aeson (Value (Array, String))
 import Data.Aeson.Encode (encode)
 import qualified Data.Vector as Vector
 import Network.Wai.Middleware.Gzip (GzipSettings, def)
-import Network.Wai.Parse (tempFileSink, lbsSink)
+import Network.Wai.Parse (tempFileBackEnd, lbsBackEnd)
 import qualified Paths_yesod_core
 import Data.Version (showVersion)
 import System.Log.FastLogger (Logger, mkLogger, loggerDate, LogStr (..), loggerPutStr)
@@ -340,8 +340,8 @@ $doctype 5
                -> Word64 -- ^ request body size
                -> FileUpload
     fileUpload _ size
-        | size > 50000 = FileUploadDisk tempFileSink
-        | otherwise = FileUploadMemory lbsSink
+        | size > 50000 = FileUploadDisk tempFileBackEnd
+        | otherwise = FileUploadMemory lbsBackEnd
 
 formatLogMessage :: IO ZonedDate
                  -> Loc
diff --git a/yesod-core/Yesod/Internal/Request.hs b/yesod-core/Yesod/Internal/Request.hs
index d23baaa7..3fe09758 100644
--- a/yesod-core/Yesod/Internal/Request.hs
+++ b/yesod-core/Yesod/Internal/Request.hs
@@ -145,6 +145,6 @@ mkFileInfoFile name ct fp = FileInfo name ct (sourceFile fp) (\dst -> runResourc
 mkFileInfoSource :: Text -> Text -> Source (ResourceT IO) ByteString -> FileInfo
 mkFileInfoSource name ct src = FileInfo name ct src (\dst -> runResourceT $ src $$ sinkFile dst)
 
-data FileUpload = FileUploadMemory (Sink ByteString (ResourceT IO) L.ByteString)
-                | FileUploadDisk (Sink ByteString (ResourceT IO) FilePath)
-                | FileUploadSource (Sink ByteString (ResourceT IO) (Source (ResourceT IO) ByteString))
+data FileUpload = FileUploadMemory (NWP.BackEnd L.ByteString)
+                | FileUploadDisk (NWP.BackEnd FilePath)
+                | FileUploadSource (NWP.BackEnd (Source (ResourceT IO) ByteString))