1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
open! Import
module Column = struct
type t =
{ header : string
; align : Print_table_ast.Align.t
; cells : Print_table_ast.Cell.t array
; length : int
}
end
type t = { columns : Column.t array }
let of_print_table (Print_table_ast.T { columns; rows }) =
let columns =
List.filter_map columns ~f:(fun { ; align; make_cell } ->
let cells = List.map rows ~f:(fun row -> make_cell row) in
let length =
List.fold_left cells ~init:0 ~f:(fun len (cell : Print_table_ast.Cell.t) ->
Int.max len (String.length cell.text))
in
if length = 0
then None
else (
let length = Int.max length (String.length header) in
let cells = Array.of_list cells in
Some { Column.header; align; cells; length }))
|> Array.of_list
in
{ columns }
;;
let pad ?ansi_code text ~len ~align =
let slen = String.length text in
let text =
match ansi_code with
| None -> text
| Some ansi_code -> Printf.sprintf "\027[%dm%s\027[0m" ansi_code text
in
if slen >= len
then text
else (
let pad = String.make (len - slen) ' ' in
match (align : Print_table_ast.Align.t) with
| Left -> text ^ pad
| Right -> pad ^ text
| Center ->
let left = (len - slen) / 2 in
let right = len - slen - left in
String.make left ' ' ^ text ^ String.make right ' ')
;;