{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE TemplateHaskell #-}

module Hledger.Web.Handler.UploadR
  ( getUploadR
  , postUploadR
  ) where

import Control.Monad.Except (runExceptT)
import qualified Data.ByteString.Lazy as BL
import Data.Conduit (connect)
import Data.Conduit.Binary (sinkLbs)
import qualified Data.Text.Encoding as TE

import Hledger.Web.Import
import Hledger.Web.Widget.Common (fromFormSuccess, journalFile404, writeJournalTextIfValidAndChanged)

uploadForm :: FilePath -> Markup -> MForm Handler (FormResult FileInfo, Widget)
uploadForm :: String -> Markup -> MForm Handler (FormResult FileInfo, Widget)
uploadForm String
f =
  forall (m :: * -> *) a.
Monad m =>
Text
-> (Markup -> MForm m (FormResult a, WidgetFor (HandlerSite m) ()))
-> Markup
-> MForm m (FormResult a, WidgetFor (HandlerSite m) ())
identifyForm Text
"upload" forall a b. (a -> b) -> a -> b
$ \Markup
extra -> do
    (FormResult FileInfo
res, FieldView (HandlerSite Handler)
_) <- forall site (m :: * -> *) a.
(RenderMessage site FormMessage, HandlerSite m ~ site,
 MonadHandler m) =>
Field m a
-> FieldSettings site
-> Maybe a
-> MForm m (FormResult a, FieldView site)
mreq forall (m :: * -> *). Monad m => Field m FileInfo
fileField forall {master}. FieldSettings master
fs forall a. Maybe a
Nothing
    -- Ignoring the view - setting the name of the element is enough here
    forall (f :: * -> *) a. Applicative f => a -> f a
pure (FormResult FileInfo
res, $(widgetFile "upload-form"))
  where
    fs :: FieldSettings master
fs = forall master.
SomeMessage master
-> Maybe (SomeMessage master)
-> Maybe Text
-> Maybe Text
-> [(Text, Text)]
-> FieldSettings master
FieldSettings SomeMessage master
"file" forall a. Maybe a
Nothing (forall a. a -> Maybe a
Just Text
"file") (forall a. a -> Maybe a
Just Text
"file") []

getUploadR :: FilePath -> Handler ()
getUploadR :: String -> Handler ()
getUploadR String
f = do
  Handler ()
checkServerSideUiEnabled
  String -> Handler ()
postUploadR String
f

postUploadR :: FilePath -> Handler ()
postUploadR :: String -> Handler ()
postUploadR String
f = do
  Handler ()
checkServerSideUiEnabled
  VD {[Capability]
caps :: ViewData -> [Capability]
caps :: [Capability]
caps, Journal
j :: ViewData -> Journal
j :: Journal
j} <- Handler ViewData
getViewData
  forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (Capability
CapManage forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`notElem` [Capability]
caps) (forall (m :: * -> *) a. MonadHandler m => Text -> m a
permissionDenied Text
"Missing the 'manage' capability")

  (String
f', Text
_) <- forall m. String -> Journal -> HandlerFor m (String, Text)
journalFile404 String
f Journal
j
  ((FormResult FileInfo
res, Widget
view), Enctype
enctype) <- forall (m :: * -> *) a xml.
(RenderMessage (HandlerSite m) FormMessage, MonadResource m,
 MonadHandler m) =>
(Markup -> MForm m (FormResult a, xml))
-> m ((FormResult a, xml), Enctype)
runFormPost (String -> Markup -> MForm Handler (FormResult FileInfo, Widget)
uploadForm String
f')
  FileInfo
fi <- forall (m :: * -> *) a. Applicative m => m a -> FormResult a -> m a
fromFormSuccess (forall {site} {a} {a} {c}.
(Yesod site, ToMarkup a, ToWidget site a) =>
a -> a -> HandlerFor site c
showForm Widget
view Enctype
enctype) FormResult FileInfo
res
  ByteString
lbs <- ByteString -> ByteString
BL.toStrict forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall (m :: * -> *) a r.
Monad m =>
ConduitT () a m () -> ConduitT a Void m r -> m r
connect (forall (m :: * -> *).
MonadResource m =>
FileInfo -> ConduitT () ByteString m ()
fileSource FileInfo
fi) forall (m :: * -> *) o.
Monad m =>
ConduitT ByteString o m ByteString
sinkLbs

  -- Try to parse as UTF-8
  -- XXX Unfortunate - how to parse as system locale?
  Text
newtxt <- case ByteString -> Either UnicodeException Text
TE.decodeUtf8' ByteString
lbs of
    Left UnicodeException
e -> do
      forall (m :: * -> *). MonadHandler m => Markup -> m ()
setMessage forall a b. (a -> b) -> a -> b
$
        Markup
"Encoding error: '" forall a. Semigroup a => a -> a -> a
<> forall a. ToMarkup a => a -> Markup
toHtml (forall a. Show a => a -> String
show UnicodeException
e) forall a. Semigroup a => a -> a -> a
<> Markup
"'. " forall a. Semigroup a => a -> a -> a
<>
        Markup
"If your file is not UTF-8 encoded, try the 'edit form', " forall a. Semigroup a => a -> a -> a
<>
        Markup
"where the transcoding should be handled by the browser."
      forall {site} {a} {a} {c}.
(Yesod site, ToMarkup a, ToWidget site a) =>
a -> a -> HandlerFor site c
showForm Widget
view Enctype
enctype
    Right Text
newtxt -> forall (m :: * -> *) a. Monad m => a -> m a
return Text
newtxt
  forall e (m :: * -> *) a. ExceptT e m a -> m (Either e a)
runExceptT (forall (m :: * -> *).
MonadHandler m =>
String -> Text -> ExceptT String m ()
writeJournalTextIfValidAndChanged String
f Text
newtxt) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
    Left String
e -> do
      forall (m :: * -> *). MonadHandler m => Markup -> m ()
setMessage forall a b. (a -> b) -> a -> b
$ Markup
"Failed to load journal: " forall a. Semigroup a => a -> a -> a
<> forall a. ToMarkup a => a -> Markup
toHtml String
e
      forall {site} {a} {a} {c}.
(Yesod site, ToMarkup a, ToWidget site a) =>
a -> a -> HandlerFor site c
showForm Widget
view Enctype
enctype
    Right () -> do
      forall (m :: * -> *). MonadHandler m => Markup -> m ()
setMessage forall a b. (a -> b) -> a -> b
$ Markup
"File " forall a. Semigroup a => a -> a -> a
<> forall a. ToMarkup a => a -> Markup
toHtml String
f forall a. Semigroup a => a -> a -> a
<> Markup
" uploaded successfully"
      forall (m :: * -> *) url a.
(MonadHandler m, RedirectUrl (HandlerSite m) url) =>
url -> m a
redirect Route App
JournalR
  where
    showForm :: a -> a -> HandlerFor site c
showForm a
view a
enctype =
      forall (m :: * -> *) c a.
(MonadHandler m, ToTypedContent c) =>
c -> m a
sendResponse forall (m :: * -> *) b c a.
Monad m =>
(b -> m c) -> (a -> m b) -> a -> m c
<=< forall site.
Yesod site =>
WidgetFor site () -> HandlerFor site Markup
defaultLayout forall a b. (a -> b) -> a -> b
$ do
        forall (m :: * -> *). MonadWidget m => Markup -> m ()
setTitle Markup
"Upload journal"
        [whamlet|<form method=post enctype=#{enctype}>^{view}|]