cli/Hcompta/CLI/Lib/Leijen/Table.hs

   1 {-# LANGUAGE FlexibleInstances #-}
   2 {-# LANGUAGE MultiParamTypeClasses #-}
   3 {-# LANGUAGE NamedFieldPuns #-}
   4 module Hcompta.CLI.Lib.Leijen.Table where
   5
   6 import qualified Data.List
   7 import           Data.Maybe (fromMaybe)
   8 import qualified Data.Text.Lazy as TL
   9 import           Data.Text (Text)
  10 import qualified Data.Text as Text
  11
  12 import qualified Hcompta.Lib.Leijen as W
  13 import           Hcompta.Lib.Leijen ((<>), toDoc, ToDoc(..))
  14
  15 -- * The 'Table' type
  16
  17 type Table = [Column]
  18
  19 -- * The 'Column' type
  20
  21 data Column
  22  =   Column
  23  { column_title :: Text
  24  , column_width :: Int
  25  , column_align :: Align
  26  , column_rows  :: [Cell]
  27  }
  28 instance ToDoc m [Column] where
  29         toDoc _m cols = do
  30                 let rows = Data.List.transpose $ map column_rows cols
  31                 let has_title = any (not . Text.null . column_title) cols
  32                 let titles =
  33                         W.intercalate (W.bold $ W.dullblack $ W.char '|')
  34                          (\col@Column{column_title} -> do
  35                                 let cell_width = Text.length column_title
  36                                 let under = W.bold $ W.dullblack $ W.char '_'
  37                                 let cell_content = W.enclose under under $
  38                                         W.hcat $ map
  39                                          (\c -> case c of { ' ' -> under; _ -> W.char c })
  40                                          (Text.unpack column_title)
  41                                 let pad len = W.bold $ W.dullblack $
  42                                         W.text $ TL.pack $ replicate len '_'
  43                                 align (Just pad) col
  44                                  Cell{cell_width, cell_content, cell_align=Just Align_Center}
  45                          ) cols
  46                 W.vsep (
  47                         (if has_title then (:) titles else id) $
  48                         map
  49                          ( W.intercalate (W.space <> do W.bold $ W.dullblack $ W.char '|') id
  50                          . map (W.space <>)
  51                          . zipWith toDoc cols
  52                          ) rows
  53                  ) <> do
  54                 (if null cols then W.empty else W.line)
  55 column :: Text -> Align -> [Cell] -> Column
  56 column column_title column_align column_rows =
  57         Column
  58          { column_title
  59          , column_width = max (Text.length column_title) $
  60                           foldr (max . cell_width) 0 column_rows
  61          , column_align
  62          , column_rows
  63          }
  64
  65 -- ** The 'Align' type
  66
  67 data Align
  68  = Align_Left
  69  | Align_Center
  70  | Align_Right
  71 align :: Maybe (Int -> W.Doc) -> Column -> Cell -> W.Doc
  72 align filling
  73  Column{column_width, column_align}
  74  Cell{cell_width, cell_content, cell_align} =
  75         let pad = column_width - cell_width in
  76         let fill = fromMaybe (`W.fill` W.empty) filling in
  77         case fromMaybe column_align cell_align of
  78          Align_Left   -> cell_content <> fill pad
  79          Align_Center ->
  80                 let half = fromInteger $ quot (toInteger pad) 2 in
  81                 fill half <> cell_content <> fill (pad - half)
  82          Align_Right  -> fill pad <> cell_content
  83 align _filling
  84  Column{column_width}
  85  (Cell_Line {cell_pad}) =
  86         W.bold $ W.dullblack $ W.text $
  87         TL.replicate (fromIntegral column_width) $
  88         TL.singleton cell_pad
  89
  90 -- * The 'Cell' type
  91
  92 data Cell
  93  = Cell
  94          { cell_align   :: Maybe Align
  95          , cell_width   :: Int
  96          , cell_content :: W.Doc
  97          }
  98  | Cell_Line
  99          { cell_pad     :: Char
 100          , cell_width   :: Int
 101          }
 102 cell :: Cell
 103 cell =
 104         Cell
 105          { cell_width   = 0
 106          , cell_content = W.empty
 107          , cell_align   = Nothing
 108          }
 109 instance ToDoc Column Cell where
 110         toDoc = align Nothing