{-# LANGUAGE CPP, QuasiQuotes, TemplateHaskell, TypeFamilies, GeneralizedNewtypeDeriving #-} {-# LANGUAGE TypeSynonymInstances, FlexibleInstances, MultiParamTypeClasses, FlexibleContexts #-} {-| The web app providing a richer interface to hledger's data. -} -- with authentication, registration and persistent storage of user accounts (not yet) module Hledger.Web.App ( App (..) , withApp ) where import Control.Applicative ((<$>), (<*>)) -- import Control.Failure -- import qualified Data.ByteString.Lazy as L import Data.Either -- import System.Directory import System.FilePath ((), takeFileName) import System.IO.Storage (putValue, getValue) import Text.ParserCombinators.Parsec hiding (string) -- import Database.Persist.GenericSql (ConnectionPool, SqlPersist, runMigration, migrate) import Yesod import Yesod.Helpers.Static -- import Yesod.Helpers.Auth -- import Yesod.Mail -- import Yesod.WebRoutes -- import Text.Hamlet (defaultHamletSettings) -- import Text.Hamlet.RT import Hledger.Cli.Add (appendToJournalFile) import Hledger.Cli.Balance import Hledger.Cli.Print import Hledger.Cli.Register import Hledger.Cli.Options hiding (value) import Hledger.Cli.Utils import Hledger.Cli.Version (version) import Hledger.Data hiding (insert, today) import Hledger.Read (journalFromPathAndString) import Hledger.Read.JournalReader (someamount) import Hledger.Web.Settings ( -- withConnectionPool -- , runConnectionPool -- , staticroot datadir -- , hamletFile -- , cassiusFile -- , juliusFile , hledgerorgurl , manualurl , style_css , hledger_js , jquery_js , jquery_url_js , dhtmlxcommon_js , dhtmlxcombo_js , robots_txt ) ---------------------------------------------------------------------- -- define the web app ---------------------------------------------------------------------- -- persistent data schema for the web app. User account info is stored here, -- hledger's main data is stored in the usual places (journal files etc.) -- persist (quasi-quoter from persistent) defines a list of data entities. -- mkPersist (template haskell from persistent) defines persistence-capable data types based on these. -- mkPersist [$persist| -- User -- ident String -- password String null update -- UniqueUser ident -- Email -- email String -- user UserId null update -- verkey String null update -- UniqueEmail email -- |] -- run-time data kept by the web app. data App = App { -- appConnPool :: Maybe ConnectionPool appRoot :: String ,appDataDir :: FilePath ,appOpts :: [Opt] ,appArgs :: [String] ,appJournal :: Journal ,appStatic :: Static } -- parseRoutes (quasi-quoter from web-routes) defines a list of route patterns for the web app. -- mkYesod (template haskell from yesod) defines types for the web app based on the routes. -- /auth AuthR Auth getAuth mkYesod "App" [$parseRoutes| / IndexR GET /favicon.ico FaviconR GET /robots.txt RobotsR GET /static StaticR Static appStatic /journalonly JournalOnlyR GET POST /registeronly RegisterOnlyR GET /accounts AccountsOnlyR GET /journal JournalR GET POST /register RegisterR GET POST |] -- /addformrt AddformRTR GET type Handler = GHandler App App instance Yesod App where approot = appRoot -- defaultLayout widget = do -- mmsg <- getMessage -- pc <- widgetToPageContent $ do -- widget -- addCassius $(cassiusFile "default-layout") -- hamletToRepHtml $(hamletFile "default-layout") -- authRoute _ = Just $ AuthR LoginR -- static file-serving optimisations, disable for the moment -- urlRenderOverride a (StaticR s) = -- Just $ uncurry (joinPath a staticroot) $ format s -- where -- format = formatPathSegments ss -- ss :: Site StaticRoute (String -> Maybe (GHandler Static App ChooseRep)) -- ss = getSubSite urlRenderOverride _ _ = Nothing -- addStaticContent ext' _ content = do -- let fn = base64md5 content ++ '.' : ext' -- let statictmp = datadir ++ "/tmp/" -- liftIO $ createDirectoryIfMissing True statictmp -- liftIO $ L.writeFile (statictmp ++ fn) content -- return $ Just $ Right (StaticR $ StaticRoute ["tmp", fn] [], []) -- instance YesodPersist App where -- type YesodDB App = SqlPersist -- runDB db = do -- y <- getYesod -- let p = appConnPool y -- case p of Just p' -> runConnectionPool db p' -- Nothing -> error "no connection pool, programmer error" -- XXX -- instance YesodAuth App where -- type AuthEntity App = User -- type AuthEmailEntity App = Email -- defaultDest _ = IndexR -- getAuthId creds _extra = runDB $ do -- x <- getBy $ UniqueUser $ credsIdent creds -- case x of -- Just (uid, _) -> return $ Just uid -- Nothing -> do -- fmap Just $ insert $ User (credsIdent creds) Nothing -- openIdEnabled _ = True -- emailSettings _ = Just EmailSettings { -- addUnverified = \email verkey -> runDB $ insert $ Email email Nothing (Just verkey) -- , sendVerifyEmail = sendVerifyEmail' -- , getVerifyKey = runDB . fmap (join . fmap emailVerkey) . get -- , setVerifyKey = \eid key -> runDB $ update eid [EmailVerkey $ Just key] -- , verifyAccount = \eid -> runDB $ do -- me <- get eid -- case me of -- Nothing -> return Nothing -- Just e -> do -- let email = emailEmail e -- case emailUser e of -- Just uid -> return $ Just uid -- Nothing -> do -- uid <- insert $ User email Nothing -- update eid [EmailUser $ Just uid] -- return $ Just uid -- , getPassword = runDB . fmap (join . fmap userPassword) . get -- , setPassword = \uid pass -> runDB $ update uid [UserPassword $ Just pass] -- , getEmailCreds = \email -> runDB $ do -- me <- getBy $ UniqueEmail email -- case me of -- Nothing -> return Nothing -- Just (eid, e) -> return $ Just EmailCreds -- { emailCredsId = eid -- , emailCredsAuthId = emailUser e -- , emailCredsStatus = isJust $ emailUser e -- , emailCredsVerkey = emailVerkey e -- } -- , getEmail = runDB . fmap (fmap emailEmail) . get -- } -- sendVerifyEmail' :: String -> String -> String -> GHandler Auth m () -- sendVerifyEmail' email _ verurl = -- liftIO $ renderSendMail Mail -- { mailHeaders = -- [ ("From", "noreply") -- , ("To", email) -- , ("Subject", "Verify your email address") -- ] -- , mailPlain = verurl -- , mailParts = return Part -- { partType = "text/html; charset=utf-8" -- , partEncoding = None -- , partDisposition = Inline -- , partContent = renderHamlet id [$hamlet| -- %p Please confirm your email address by clicking on the link below. -- %p -- %a!href=$verurl$ $verurl$ -- %p Thank you -- |] -- } -- } -- | Migrate the app's persistent data and run the given yesod/persistent/wai-ish IO action on it. -- withApp :: App -> (Yesod.Application -> IO a) -> IO a -- withApp app f = toPersistentApp app >>= toWaiApp >>= f withApp :: App -> (Yesod.Application -> IO a) -> IO a withApp app f = toWaiApp app >>= f -- -- | Obtain a persistent db connection pool to the app, and run any necessary data migrations. -- toPersistentApp :: App -> IO App -- toPersistentApp app = withConnectionPool $ \p -> do -- flip runConnectionPool p $ runMigration $ do -- migrate (undefined :: User) -- migrate (undefined :: Email) -- return () -- return app{appConnPool=Just p} ---------------------------------------------------------------------- -- handler utilities, common templates ---------------------------------------------------------------------- nulltemplate = [$hamlet||] -- | A bundle of useful data passed to templates. data TemplateData = TD { here :: AppRoute -- ^ the current page's route ,title :: String -- ^ page's title ,msg :: Maybe Html -- ^ transient message ,a :: String -- ^ a (acct/desc filter pattern) parameter ,p :: String -- ^ p (period expression) parameter ,j :: Journal -- ^ the current journal ,today :: Day -- ^ the current day } mktd = TD { here = IndexR ,title = "hledger" ,msg = Nothing ,a = "" ,p = "" ,j = nulljournal ,today = ModifiedJulianDay 0 } -- | Gather the data useful for a hledger web request handler, including: -- initial command-line options, current a and p query string values, a -- journal filter specification based on the above and the current time, -- an up-to-date parsed journal, the current route, and the current ui -- message if any. getHandlerData :: Handler (String, String, [Opt], FilterSpec, Journal, Maybe Html, AppRoute) getHandlerData = do Just here <- getCurrentRoute (a, p, opts, fspec) <- getReportParameters (j, err) <- getLatestJournal opts msg <- getMessage' err return (a, p, opts, fspec, j, msg, here) where -- | Get current report parameters for this request. getReportParameters :: Handler (String, String, [Opt], FilterSpec) getReportParameters = do app <- getYesod t <- liftIO $ getCurrentLocalTime a <- fromMaybe "" <$> lookupGetParam "a" p <- fromMaybe "" <$> lookupGetParam "p" let opts = appOpts app ++ [Period p] args = appArgs app ++ words' a fspec = optsToFilterSpec opts args t return (a, p, opts, fspec) -- | Quote-sensitive words, ie don't split on spaces which are inside quotes. words' :: String -> [String] words' = fromparse . parsewith ((quotedPattern <|> pattern) `sepBy` many1 spacenonewline) where pattern = many (noneOf " \n\r\"") quotedPattern = between (oneOf "'\"") (oneOf "'\"") $ many $ noneOf "'\"" -- | Update our copy of the journal if the file changed. If there is an -- error while reloading, keep the old one and return the error, and set a -- ui message. getLatestJournal :: [Opt] -> Handler (Journal, Maybe String) getLatestJournal opts = do j <- liftIO $ fromJust `fmap` getValue "hledger" "journal" (jE, changed) <- liftIO $ journalReloadIfChanged opts j if not changed then return (j,Nothing) else case jE of Right j' -> do liftIO $ putValue "hledger" "journal" j' return (j',Nothing) Left e -> do setMessage $ string "error while reading" {- ++ ": " ++ e-} return (j, Just e) -- | Helper to work around a yesod feature (can't set and get a message in the same request.) getMessage' :: Maybe String -> Handler (Maybe Html) getMessage' newmsgstr = do oldmsg <- getMessage return $ maybe oldmsg (Just . string) newmsgstr -- | Wrap a template with the standard hledger web ui page layout. pageLayout :: TemplateData -> Hamlet AppRoute -> Hamlet AppRoute pageLayout td@TD{title=basetitle, msg=msg, p=p, j=j, today=today} content = [$hamlet| !!! %html %head %title $title'$ %meta!http-equiv=Content-Type!content=$metacontent$ %script!type=text/javascript!src=@StaticR.jquery_js@ %script!type=text/javascript!src=@StaticR.jquery_url_js@ %script!type=text/javascript!src=@StaticR.dhtmlxcommon_js@ %script!type=text/javascript!src=@StaticR.dhtmlxcombo_js@ %script!type=text/javascript!src=@StaticR.hledger_js@ %link!rel=stylesheet!type=text/css!media=all!href=@StaticR.style_css@ %body ^navbar.td^ #messages $m$ #content ^content^ |] where title' = basetitle ++ " - " ++ journaltitle (journaltitle, _) = journalTitleDesc j p today metacontent = "text/html; charset=utf-8" m = fromMaybe (string "") msg -- | Global toolbar/heading area. navbar :: TemplateData -> Hamlet AppRoute navbar TD{p=p,j=j,today=today} = [$hamlet| #navbar %a.topleftlink!href=$hledgerorgurl$ hledger
$version$ %a.toprightlink!href=$manualurl$!target=hledgerhelp manual %h1 $title$ \ $ %span#journaldesc $desc$ |] where (title, desc) = journalTitleDesc j p today -- | Generate a title and description for the given journal, period -- expression, and date. journalTitleDesc :: Journal -> String -> Day -> (String, String) journalTitleDesc j p today = (title, desc) where title = printf "%s" (takeFileName $ journalFilePath j) :: String desc = printf "%s" (showspan span) :: String span = either (const $ DateSpan Nothing Nothing) snd (parsePeriodExpr today p) showspan (DateSpan Nothing Nothing) = "" showspan s = " (" ++ dateSpanAsText s ++ ")" -- | Links to the main views. navlinks :: TemplateData -> Hamlet AppRoute navlinks td = [$hamlet| #navlinks ^accountsjournallink^ \ | $ ^accountsregisterlink^ \ | $ %a#addformlink!href!onclick="return addformToggle(event)" add transaction %a#importformlink!href!onclick="return importformToggle(event)"!style=display:none; import transactions \ | $ %a#editformlink!href!onclick="return editformToggle(event)" edit journal |] -- \ | $ where accountsjournallink = navlink td "journal" JournalR accountsregisterlink = navlink td "register" RegisterR navlink :: TemplateData -> String -> AppRoute -> Hamlet AppRoute navlink TD{here=here,a=a,p=p} s dest = [$hamlet|%a#$s$link.$style$!href=@?u@ $s$|] where u = (dest, concat [(if null a then [] else [("a", a)]) ,(if null p then [] else [("p", p)])]) style | dest == here = "navlinkcurrent" | otherwise = "navlink" -- | Form controlling journal filtering parameters. filterform :: TemplateData -> Hamlet AppRoute filterform TD{here=here,a=a,p=p} = [$hamlet| #filterformdiv %form#filterform.form!method=GET!style=display:$visible$; %table.form %tr.$filteringperiodclass$ %td filter by period: \ $ %td %input!name=p!size=60!value=$p$ ^phelp^ \ $ %td!align=right ^stopfilteringperiod^ %tr.$filteringclass$ %td filter by account/description: \ $ %td %input!name=a!size=60!value=$a$ ^ahelp^ \ $ %input!type=submit!value=filter $ \ $ %td!align=right ^stopfiltering^ |] where ahelp = helplink "filter-patterns" "?" phelp = helplink "period-expressions" "?" filtering = not $ null a filteringperiod = not $ null p visible = "block" filteringclass = if filtering then "filtering" else "" filteringperiodclass = if filteringperiod then "filtering" else "" stopfiltering = if filtering then [$hamlet|%a#stopfilterlink!href=@?u@ clear filter|] else nulltemplate where u = (here, if filteringperiod then [("p", p)] else []) stopfilteringperiod = if filteringperiod then [$hamlet|%a#stopfilterlink!href=@?u@ clear filter|] else nulltemplate where u = (here, if filtering then [("a", a)] else []) -- | Link to a topic in the manual. helplink :: String -> String -> Hamlet AppRoute helplink topic label = [$hamlet|%a!href=$u$!target=hledgerhelp $label$|] where u = manualurl ++ if null topic then "" else '#':topic {- -- | Render a runtime template with the provided runtime data as html. renderHamletFileRT :: FilePath -> HamletMap AppRoute -> Handler Html renderHamletFileRT hfile hmap = do hrt <- readTemplateFile hfile >>= parseHamletRT defaultHamletSettings renderHamletRT hrt hmap urlParamsToString -- | Read a file from the app's templates directory. readTemplateFile :: FilePath -> Handler String readTemplateFile hfile = do dir <- appDataDir `fmap` getYesod liftIO $ readFile $ dir hfile -- what to do if rendering a runtime template fails. instance Failure HamletException Handler where failure = error' . show renderHamletAsHtml :: Hamlet AppRoute -> Html renderHamletAsHtml h = h urlParamsToString htmlAsHamlet :: Html -> Hamlet AppRoute htmlAsHamlet h = [$hamlet|$h$|] urlParamsToString :: AppRoute -> [(String,String)] -> String urlParamsToString u [] = show u urlParamsToString u ps = show u ++ "?" ++ intercalate "&" [k++"="++v | (k,v) <- ps] -- | Convert a string to a hamlet HDHtml data item. hdstring :: String -> HamletData AppRoute hdstring = HDHtml . string -- | Convert a simple list of strings to hamlet's complicated HDList type. hdstringlist :: [String] -> HamletData AppRoute hdstringlist ss = HDList [ [([], hdstring s)] | s <- ss ] -- renderHamletRT' :: Failure HamletException m => HamletMap AppRoute -> HamletRT -> m Html -- renderHamletRT' m h = renderHamletRT h m urlParamsToString -- parseHamletRT' :: Failure HamletException m => String -> m HamletRT -- parseHamletRT' s = parseHamletRT defaultHamletSettings s -- hamletToHamletRT :: Failure HamletException m => Hamlet AppRoute -> m HamletRT -- hamletToHamletRT h = stringToHamletRT $ show $ unsafeByteString $ renderHamlet show h -} ---------------------------------------------------------------------- -- handlers/views ---------------------------------------------------------------------- getFaviconR :: Handler () getFaviconR = sendFile "image/x-icon" $ datadir "favicon.ico" ---------------------------------------------------------------------- getRobotsR :: Handler RepPlain getRobotsR = return $ RepPlain $ toContent robots_txt ---------------------------------------------------------------------- getIndexR :: Handler () getIndexR = redirect RedirectTemporary defaultroute where defaultroute = JournalR ---------------------------------------------------------------------- -- getDemoR :: Handler RepHtml -- getDemoR = do -- -- mu <- maybeAuth -- defaultLayout $ do -- h2id <- newIdent -- setTitle $ string "hledger front page" -- addBody $(hamletFile "homepage") -- addStyle $(cassiusFile "homepage") -- addJavascript $(juliusFile "homepage") ---------------------------------------------------------------------- -- | A combined accounts and journal view. getJournalR :: Handler RepHtml getJournalR = do (a, p, opts, fspec, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay -- app <- getYesod -- t <- liftIO $ getCurrentLocalTime let -- args = appArgs app -- fspec' = optsToFilterSpec opts args t sidecontent = balanceReportAsHtml opts td $ balanceReport opts fspec j maincontent = journalReportAsHtml opts td $ journalReport opts fspec j td = mktd{here=here, title="hledger journal", msg=msg, a=a, p=p, j=j, today=today} editform' = editform td hamletToRepHtml $ pageLayout td [$hamlet| %div#content %div#sidebar ^sidecontent^ %div#main.journal ^navlinks.td^ %div#transactions ^filterform.td^ ^maincontent^ ^addform.td^ ^editform'^ ^importform^ |] postJournalR :: Handler RepPlain postJournalR = postJournalOnlyR ---------------------------------------------------------------------- -- | A combined accounts and register view. getRegisterR :: Handler RepHtml getRegisterR = do (a, p, opts, fspec, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay -- app <- getYesod -- t <- liftIO $ getCurrentLocalTime let -- args = appArgs app -- opts' = Empty:opts -- fspec' = optsToFilterSpec opts' args t sidecontent = balanceReportAsHtml opts td $ balanceReport opts fspec j maincontent = registerReportAsHtml opts td $ registerReport opts fspec j td = mktd{here=here, title="hledger register", msg=msg, a=a, p=p, j=j, today=today} editform' = editform td hamletToRepHtml $ pageLayout td [$hamlet| %div#content %div#sidebar ^sidecontent^ %div#main.journal ^navlinks.td^ %div#transactions ^filterform.td^ ^maincontent^ ^addform.td^ ^editform'^ ^importform^ |] postRegisterR :: Handler RepPlain postRegisterR = postJournalOnlyR ---------------------------------------------------------------------- -- | A simple accounts and balances view like hledger balance. getAccountsOnlyR :: Handler RepHtml getAccountsOnlyR = do (a, p, opts, fspec, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay let td = mktd{here=here, title="hledger accounts", msg=msg, a=a, p=p, j=j, today=today} hamletToRepHtml $ pageLayout td $ balanceReportAsHtml opts td $ balanceReport opts fspec j -- | Render a balance report as HTML. balanceReportAsHtml :: [Opt] -> TemplateData -> BalanceReport -> Hamlet AppRoute balanceReportAsHtml _ td@TD{here=here,a=a,p=p} (items,total) = [$hamlet| ^accountsheading^ %table.balancereport $forall items i ^itemAsHtml' i^ %tr.totalrule %td!colspan=2 %tr %td %td!align=right $mixedAmountAsHtml.total$ |] where accountsheading = [$hamlet| #accountsheading accounts \ $ ^showlinks^ |] where filteringaccts = not $ null a showlinks = [$hamlet|%span#showmoreaccounts ^showmore^ ^showall^|] showmore = case (filteringaccts, items) of -- cunning parent account logic (True, ((acct, _, _, _):_)) -> let a' = if isAccountRegex a then a else acct a'' = accountNameToAccountRegex $ parentAccountName $ accountRegexToAccountName a' parenturl = (here, [("a",a''), ("p",p)]) in [$hamlet| \ | $ %a!href=@?parenturl@ show more ↑ |] _ -> nulltemplate showall = if filteringaccts then [$hamlet| \ | $ %a!href=@?allurl@ show all |] else nulltemplate where allurl = (here, [("p",p)]) itemAsHtml' = itemAsHtml td itemAsHtml :: TemplateData -> BalanceReportItem -> Hamlet AppRoute itemAsHtml TD{p=p} (acct, adisplay, adepth, abal) = [$hamlet| %tr.item %td.account $indent$ %a!href=$aurl$ $adisplay$ %td.balance!align=right $mixedAmountAsHtml.abal$ |] where -- current = if not (null a) && containsRegex a acct then "current" else "" indent = preEscapedString $ concat $ replicate (2 * adepth) " " aurl = printf ".?a=%s%s" (accountNameToAccountRegex acct) p' :: String p' = if null p then "" else printf "&p=%s" p accountNameToAccountRegex :: String -> String accountNameToAccountRegex "" = "" accountNameToAccountRegex a = printf "^%s(:|$)" a accountRegexToAccountName :: String -> String accountRegexToAccountName = gsubRegexPR "^\\^(.*?)\\(:\\|\\$\\)$" "\\1" isAccountRegex :: String -> Bool isAccountRegex s = take 1 s == "^" && (take 5 $ reverse s) == ")$|:(" ---------------------------------------------------------------------- -- | A basic journal view, like hledger print, with editing. getJournalOnlyR :: Handler RepHtml getJournalOnlyR = do (a, p, opts, fspec, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay let td = mktd{here=here, title="hledger journal", msg=msg, a=a, p=p, j=j, today=today} editform' = editform td txns = journalReportAsHtml opts td $ journalReport opts fspec j hamletToRepHtml $ pageLayout td [$hamlet| %div#journal %div.nav2 %a#addformlink!href!onclick="return addformToggle(event)" add one transaction \ | $ %a#editformlink!href!onclick="return editformToggle(event)" edit the whole journal #transactions ^txns^ ^addform.td^ ^editform'^ |] -- | Render a journal report as HTML. journalReportAsHtml :: [Opt] -> TemplateData -> JournalReport -> Hamlet AppRoute journalReportAsHtml _ td items = [$hamlet| %table.journalreport $forall number.items i ^itemAsHtml' i^ |] where number = zip [1..] itemAsHtml' = itemAsHtml td itemAsHtml :: TemplateData -> (Int, JournalReportItem) -> Hamlet AppRoute itemAsHtml _ (n, t) = [$hamlet| %tr.item.$evenodd$ %td.transaction %pre $txn$ |] where evenodd = if even n then "even" else "odd" txn = trimnl $ showTransaction t where trimnl = reverse . dropWhile (=='\n') . reverse addform :: TemplateData -> Hamlet AppRoute addform td = [$hamlet| %script!type=text/javascript $$(document).ready(function() { /* dhtmlxcombo setup */ window.dhx_globalImgPath="../static/"; var desccombo = new dhtmlXCombo("description"); var acct1combo = new dhtmlXCombo("account1"); var acct2combo = new dhtmlXCombo("account2"); desccombo.enableFilteringMode(true); acct1combo.enableFilteringMode(true); acct2combo.enableFilteringMode(true); desccombo.setSize(300); acct1combo.setSize(300); acct2combo.setSize(300); /* desccombo.enableOptionAutoHeight(true, 20); */ /* desccombo.setOptionHeight(200); */ }); %form#addform!method=POST!style=display:none; %table.form %tr %td!colspan=4 %table %tr#descriptionrow %td Date: %td %input.textinput!size=15!name=date!value=$date$ %td!style=padding-left:1em; Description: %td %select!id=description!name=description %option $forall descriptions d %option!value=$d$ $d$ %tr.helprow %td %td .help $datehelp$ $ %td %td .help $deschelp$ ^postingsfields.td^ %tr#addbuttonrow %td!colspan=4 %input!type=hidden!name=action!value=add %input!type=submit!name=submit!value="add transaction" $if manyfiles \ to: ^journalselect.files.j.td^ |] where -- datehelplink = helplink "dates" "..." datehelp = "eg: 2010/7/20" deschelp = "eg: supermarket (optional)" date = "today" descriptions = sort $ nub $ map tdescription $ jtxns $ j td manyfiles = (length $ files $ j td) > 1 postingsfields :: TemplateData -> Hamlet AppRoute postingsfields td = [$hamlet| ^p1^ ^p2^ |] where p1 = postingfields td 1 p2 = postingfields td 2 postingfields :: TemplateData -> Int -> Hamlet AppRoute postingfields TD{j=j} n = [$hamlet| %tr#postingrow %td!align=right $acctlabel$: %td %select!id=$acctvar$!name=$acctvar$ %option $forall acctnames a %option!value=$a$ $a$ ^amtfield^ %tr.helprow %td %td .help $accthelp$ %td %td .help $amthelp$ |] where numbered = (++ show n) acctvar = numbered "account" amtvar = numbered "amount" acctnames = sort $ journalAccountNamesUsed j (acctlabel, accthelp, amtfield, amthelp) | n == 1 = ("To account" ,"eg: expenses:food" ,[$hamlet| %td!style=padding-left:1em; Amount: %td %input.textinput!size=15!name=$amtvar$!value="" |] ,"eg: $6" ) | otherwise = ("From account" ,"eg: assets:bank:checking" ,nulltemplate ,"" ) editform :: TemplateData -> Hamlet AppRoute editform TD{j=j} = [$hamlet| %form#editform!method=POST!style=display:none; %table.form#editform $if manyfiles %tr %td!colspan=2 Editing ^journalselect.files.j^ %tr %td!colspan=2 $forall files.j f %textarea!id=$fst.f$_textarea!name=text!rows=25!cols=80!style=display:none;!disabled=disabled $snd.f$ %tr#addbuttonrow %td %span.help ^formathelp^ %td!align=right %span.help Are you sure ? This will overwrite the journal. $ %input!type=hidden!name=action!value=edit %input!type=submit!name=submit!value="save journal" \ or $ %a!href!onclick="return editformToggle(event)" cancel |] -- XXX textarea ids are unquoted journal file paths, which is not valid html where manyfiles = (length $ files j) > 1 formathelp = helplink "file-format" "file format help" journalselect :: [(FilePath,String)] -> Hamlet AppRoute journalselect journalfiles = [$hamlet| %select!id=journalselect!name=journal!onchange="editformJournalSelect(event)" $forall journalfiles f %option!value=$fst.f$ $fst.f$ |] importform :: Hamlet AppRoute importform = [$hamlet| %form#importform!method=POST!style=display:none; %table.form %tr %td %input!type=file!name=file %input!type=hidden!name=action!value=import %input!type=submit!name=submit!value="import from file" \ or $ %a!href!onclick="return importformToggle(event)" cancel |] postJournalOnlyR :: Handler RepPlain postJournalOnlyR = do action <- runFormPost' $ maybeStringInput "action" case action of Just "edit" -> postEditForm Just "import" -> postImportForm _ -> postAddForm -- | Handle a journal add form post. postAddForm :: Handler RepPlain postAddForm = do (_, _, _, _, j, _, _) <- getHandlerData today <- liftIO getCurrentDay -- get form input values. M means a Maybe value. (dateM, descM, acct1M, amt1M, acct2M, amt2M, journalM) <- runFormPost' $ (,,,,,,) <$> maybeStringInput "date" <*> maybeStringInput "description" <*> maybeStringInput "account1" <*> maybeStringInput "amount1" <*> maybeStringInput "account2" <*> maybeStringInput "amount2" <*> maybeStringInput "journal" -- supply defaults and parse date and amounts, or get errors. let dateE = maybe (Left "date required") (either (\e -> Left $ showDateParseError e) Right . fixSmartDateStrEither today) dateM descE = Right $ fromMaybe "" descM acct1E = maybe (Left "to account required") Right acct1M acct2E = maybe (Left "from account required") Right acct2M amt1E = maybe (Left "amount required") (either (const $ Left "could not parse amount") Right . parseWithCtx nullctx someamount) amt1M amt2E = maybe (Right missingamt) (either (const $ Left "could not parse amount") Right . parseWithCtx nullctx someamount) amt2M journalE = maybe (Right $ journalFilePath j) (\f -> if f `elem` journalFilePaths j then Right f else Left $ "unrecognised journal file path: " ++ f) journalM strEs = [dateE, descE, acct1E, acct2E, journalE] amtEs = [amt1E, amt2E] errs = lefts strEs ++ lefts amtEs [date,desc,acct1,acct2,journalpath] = rights strEs [amt1,amt2] = rights amtEs -- if no errors so far, generate a transaction and balance it or get the error. tE | not $ null errs = Left errs | otherwise = either (\e -> Left ["unbalanced postings: " ++ (head $ lines e)]) Right (balanceTransaction Nothing $ nulltransaction { -- imprecise balancing tdate=parsedate date ,tdescription=desc ,tpostings=[ Posting False acct1 amt1 "" RegularPosting [] Nothing ,Posting False acct2 amt2 "" RegularPosting [] Nothing ] }) -- display errors or add transaction case tE of Left errs -> do -- save current form values in session setMessage $ string $ intercalate "; " errs redirect RedirectTemporary RegisterR Right t -> do let t' = txnTieKnot t -- XXX move into balanceTransaction liftIO $ appendToJournalFile journalpath $ showTransaction t' setMessage $ string $ printf "Added transaction:\n%s" (show t') redirect RedirectTemporary RegisterR -- | Handle a journal edit form post. postEditForm :: Handler RepPlain postEditForm = do (_, _, _, _, j, _, _) <- getHandlerData -- get form input values, or validation errors. -- getRequest >>= liftIO (reqRequestBody req) >>= mtrace (textM, journalM) <- runFormPost' $ (,) <$> maybeStringInput "text" <*> maybeStringInput "journal" let textE = maybe (Left "No value provided") Right textM journalE = maybe (Right $ journalFilePath j) (\f -> if f `elem` journalFilePaths j then Right f else Left "unrecognised journal file path") journalM strEs = [textE, journalE] errs = lefts strEs [text,journalpath] = rights strEs -- display errors or perform edit if not $ null errs then do setMessage $ string $ intercalate "; " errs redirect RedirectTemporary JournalR else do -- try to avoid unnecessary backups or saving invalid data filechanged' <- liftIO $ journalSpecifiedFileIsNewer j journalpath told <- liftIO $ readFileStrictly journalpath let tnew = filter (/= '\r') text changed = tnew /= told || filechanged' if not changed then do setMessage $ string $ "No change" redirect RedirectTemporary JournalR else do jE <- liftIO $ journalFromPathAndString Nothing journalpath tnew either (\e -> do setMessage $ string e redirect RedirectTemporary JournalR) (const $ do liftIO $ writeFileWithBackup journalpath tnew setMessage $ string $ printf "Saved journal %s\n" (show journalpath) redirect RedirectTemporary JournalR) jE -- | Handle an import page post. postImportForm :: Handler RepPlain postImportForm = do setMessage $ string $ "can't handle file upload yet" redirect RedirectTemporary JournalR -- -- get form input values, or basic validation errors. E means an Either value. -- fileM <- runFormPost' $ maybeFileInput "file" -- let fileE = maybe (Left "No file provided") Right fileM -- -- display errors or import transactions -- case fileE of -- Left errs -> do -- setMessage $ string errs -- redirect RedirectTemporary JournalR -- Right s -> do -- setMessage $ string $ s -- redirect RedirectTemporary JournalR ---------------------------------------------------------------------- -- | A simple postings view like hledger register. getRegisterOnlyR :: Handler RepHtml getRegisterOnlyR = do (a, p, opts, fspec, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay let td = mktd{here=here, title="hledger register", msg=msg, a=a, p=p, j=j, today=today} hamletToRepHtml $ pageLayout td $ registerReportAsHtml opts td $ registerReport opts fspec j -- | Render a register report as HTML. registerReportAsHtml :: [Opt] -> TemplateData -> RegisterReport -> Hamlet AppRoute registerReportAsHtml _ td items = [$hamlet| %table.registerreport %tr.headings ^headings^ $forall number.items i ^itemAsHtml' i^ |] where number = zip [1..] headings = [$hamlet| %th.date!align=left Date %th.description!align=left Description %th.account!align=left Account %th.amount!align=right Amount %th.balance!align=right Balance |] itemAsHtml' = itemAsHtml td itemAsHtml :: TemplateData -> (Int, RegisterReportItem) -> Hamlet AppRoute itemAsHtml TD{p=p} (n, (ds, posting, b)) = [$hamlet| %tr.item.$evenodd$.$firstposting$ %td.date $date$ %td.description $desc$ %td.account %a!href=$aurl$ $acct$ %td.amount!align=right $mixedAmountAsHtml.pamount.posting$ %td.balance!align=right $mixedAmountAsHtml.b$ |] where evenodd = if even n then "even" else "odd" (firstposting, date, desc) = case ds of Just (da, de) -> ("firstposting", show da, de) Nothing -> ("", "", "") acct = paccount posting aurl = printf ".?a=%s%s" (accountNameToAccountRegex acct) p' :: String p' = if null p then "" else printf "&p=%s" p mixedAmountAsHtml b = preEscapedString $ addclass $ intercalate "
" $ lines $ show b where addclass = printf "%s" c c = case isNegativeMixedAmount b of Just True -> "negative amount" _ -> "positive amount" ---------------------------------------------------------------------- -- | A standalone journal edit form page. -- getEditR :: Handler RepHtml -- getEditR = do -- (a, p, _, _, _, msg, here) <- getHandlerData -- today <- liftIO getCurrentDay -- -- reload journal's text without parsing, if changed -- XXX are we doing this right ? -- j <- liftIO $ fromJust `fmap` getValue "hledger" "journal" -- changed <- liftIO $ journalFileIsNewer j -- s <- liftIO $ if changed then readFile (filepath j) else return (jtext j) -- XXX readFile may throw an error -- let td = mktd{here=here, title="hledger journal edit", msg=msg, a=a, p=p, j=j, today=today} -- hamletToRepHtml $ pageLayout td $ editform td s ---------------------------------------------------------------------- {- -- | An add form with template files reloaded at runtime. getAddformRTR :: Handler RepHtml getAddformRTR = do (a, p, _, _, j, msg, here) <- getHandlerData today <- liftIO getCurrentDay let td = mktd{here=here, title="hledger add transaction", msg=msg, a=a, p=p, j=j, today=today} descriptions = sort $ nub $ map tdescription $ jtxns j acctnames = sort $ journalAccountNamesUsed j postingData n = [ (["acctlabel"], hdstring acctlabel) ,(["acctvar"], hdstring acctvar) ,(["acctnames"], hdstringlist acctnames) ,(["amtfield"], HDHtml $ renderHamletAsHtml amtfield) ,(["accthelp"], hdstring accthelp) ,(["amthelp"], hdstring amthelp) ] :: HamletMap AppRoute where numbered = (++ show n) acctvar = numbered "account" amtvar = numbered "amount" (acctlabel, accthelp, amtfield, amthelp) | n == 1 = ("To account" ,"eg: expenses:food" ,[$hamlet| %td!style=padding-left:1em; Amount: %td %input.textinput!size=15!name=$amtvar$!value="" |] ,"eg: $6" ) | otherwise = ("From account" ,"eg: assets:bank:checking" ,nulltemplate ,"" ) pfields1 <- renderHamletFileRT "addformpostingfields.hamlet" (postingData 1) pfields2 <- renderHamletFileRT "addformpostingfields.hamlet" (postingData 2) addform <- renderHamletFileRT "addform.hamlet" ([ (["date"], hdstring "today") ,(["desc"], hdstring "") ,(["descriptions"], hdstringlist descriptions) ,(["datehelp"], hdstring "eg: 2010/7/20") ,(["deschelp"], hdstring "eg: supermarket (optional)") ,(["postingfields1"], HDHtml pfields1) ,(["postingfields2"], HDHtml pfields2) ,(["manyfiles"], HDBool $ (length $ files $ j) > 1) -- ,(["files"], ... $ files $ j) ] :: HamletMap AppRoute) hamletToRepHtml $ pageLayout td $ htmlAsHamlet addform -}