module Yesod.Paginator
( paginate
, selectPaginated
, paginationWidget
) where
import Yesod
import Database.Persist.Store (Entity)
import Data.Text (Text)
import qualified Data.Text as T
paginate :: Int
-> [a]
-> GHandler s m ([a], GWidget s m ())
paginate per items = do
let tot = length items
p <- getCurrentPage tot
let xs = take per $ drop ((p 1) * per) items
return (xs, paginationWidget p per tot)
selectPaginated :: ( YesodPersistBackend m ~ PersistEntityBackend v
, YesodPersist m
, PersistEntity v
, PersistQuery (PersistEntityBackend v) (GHandler s m)
)
=> Int -> [Filter v] -> [SelectOpt v]
-> GHandler s m ([Entity v], GWidget s m ())
selectPaginated per filters selectOpts = do
tot <- runDB $ count filters
p <- getCurrentPage tot
xs <- runDB $ selectList filters (selectOpts ++ [OffsetBy ((p1)*per), LimitTo per])
return (xs, paginationWidget p per tot)
paginationWidget :: Int
-> Int
-> Int
-> GWidget s m ()
paginationWidget page per tot = do
let pages = (\(n, r) -> n + (min r 1)) $ tot `divMod` per
if pages <= 1
then return ()
else do
let prev = [1 ..(page1)]
let next = [(page+1)..pages ]
let lim = 9
let prev' = if length prev > lim then drop ((length prev) lim) prev else prev
let next' = if length next > lim then take lim next else next
curParams <- lift $ fmap reqGetParams getRequest
[whamlet|
<ul>
<li .prev :null prev:.disabled>
^{linkTo curParams (page 1) "← Previous"}
$if (/=) prev prev'
<li>^{linkTo curParams 1 "1"}
<li>...
$forall p <- prev'
<li>^{linkTo curParams p (show p)}
<li .active>
<a href="#">#{show page}
$forall n <- next'
<li>^{linkTo curParams n (show n)}
$if (/=) next next'
<li>...
<li>^{linkTo curParams tot (show tot)}
<li .next :null next:.disabled>
^{linkTo curParams (page + 1) "Next →"}
|]
getCurrentPage :: Int -> GHandler s m Int
getCurrentPage tot = do
mp <- lookupGetParam "p"
return $
case mp of
Nothing -> 1
Just "" -> 1
Just p ->
case readIntegral $ T.unpack p of
Just i -> if i > tot then tot else i
_ -> 1
updateGetParam :: [(Text,Text)] -> (Text,Text) -> Text
updateGetParam getParams (p, n) = (T.cons '?') . T.intercalate "&"
. map (\(k,v) -> k `T.append` "=" `T.append` v)
. (++ [(p, n)]) . filter ((/= p) . fst) $ getParams
linkTo :: [(Text,Text)] -> Int -> String -> GWidget s m ()
linkTo params pg txt = do
let param = ("p", T.pack $ show pg)
[whamlet|
<a href="#{updateGetParam params param}">#{txt}
|]