trellis

-module(trellis).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).

-export([style/2, table/1, param/1, with/4, to_string/1]).
-export_type([table/1, column/1, align/0]).

-type table(EZT) :: {table, list(column(EZT)), list(EZT), trellis@style:style()}.

-type column(EZU) :: {column, binary(), align(), fun((EZU) -> binary())}.

-type align() :: left | right | center.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 45).
-spec style(table(EZY), trellis@style:style()) -> table(EZY).
style(Table, Style) ->
    _record = Table,
    {table, erlang:element(2, _record), erlang:element(3, _record), Style}.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 40).
-spec table(list(EZV)) -> table(EZV).
table(Rows) ->
    {table, [], Rows, line}.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 56).
-spec param(fun((FAB) -> FAC)) -> fun((FAB) -> FAC).
param(F) ->
    F.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 61).
-spec with(table(FAD), binary(), align(), fun((FAD) -> binary())) -> table(FAD).
with(Table, Header, Align, Getter) ->
    {table, Columns, _, _} = Table,
    _record = Table,
    {table,
        [{column, Header, Align, Getter} | Columns],
        erlang:element(3, _record),
        erlang:element(4, _record)}.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 75).
-spec calculate_widths(list(binary()), list(binary())) -> list(integer()).
calculate_widths(Headers, Values) ->
    Lengths = gleam@list:sized_chunk(Values, erlang:length(Headers)),
    gleam@list:map2(
        Headers,
        gleam@list:transpose(Lengths),
        fun(Header, Col_values) ->
            Max_value_length = gleam@list:fold(
                Col_values,
                0,
                fun(Acc, Val) -> gleam@int:max(Acc, string:length(Val)) end
            ),
            gleam@int:max(string:length(Header), Max_value_length)
        end
    ).

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 129).
-spec make_separator(
    trellis@style:style(),
    list(integer()),
    boolean(),
    boolean()
) -> binary().
make_separator(Style, Widths, Is_header, Is_top) ->
    {style_guide,
        _,
        Horizontal,
        Top_left,
        Top_middle,
        Top_right,
        Middle_left,
        Middle_right,
        Bottom_left,
        Bottom_middle,
        Bottom_right,
        Cross} = trellis@style:style(Style),
    {Start, Middle, End} = case {Is_top, Is_header} of
        {true, _} ->
            {Top_left, Top_middle, Top_right};

        {false, true} ->
            {Middle_right, Cross, Middle_left};

        {false, false} ->
            {Bottom_left, Bottom_middle, Bottom_right}
    end,
    Line = begin
        _pipe = Widths,
        gleam@list:fold(
            _pipe,
            Start,
            fun(Acc, Col) ->
                <<<<Acc/binary,
                        (gleam@string:repeat(Horizontal, Col + 2))/binary>>/binary,
                    Middle/binary>>
            end
        )
    end,
    <<(gleam@string:slice(Line, 0, string:length(Line) - 1))/binary,
        End/binary>>.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 195).
-spec pad_left(binary(), integer()) -> binary().
pad_left(Text, Width) ->
    Padding = Width - string:length(Text),
    <<(gleam@string:repeat(<<" "/utf8>>, Padding))/binary, Text/binary>>.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 201).
-spec pad_right(binary(), integer()) -> binary().
pad_right(Text, Width) ->
    Padding = Width - string:length(Text),
    <<Text/binary, (gleam@string:repeat(<<" "/utf8>>, Padding))/binary>>.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 207).
-spec pad_center(binary(), integer()) -> binary().
pad_center(Text, Width) ->
    Padding = Width - string:length(Text),
    Left = Padding div 2,
    Right = Padding - Left,
    <<<<(gleam@string:repeat(<<" "/utf8>>, Left))/binary, Text/binary>>/binary,
        (gleam@string:repeat(<<" "/utf8>>, Right))/binary>>.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 165).
-spec make_row(
    trellis@style:style(),
    list(align()),
    list(integer()),
    list(binary())
) -> binary().
make_row(Style, Aligns, Widths, Values) ->
    Vertical = erlang:element(2, trellis@style:style(Style)),
    Row = begin
        _pipe = begin
            gleam@list:map2(
                begin
                    gleam@list:map2(
                        Aligns,
                        Widths,
                        fun(Align, Width) -> {Align, Width} end
                    )
                end,
                Values,
                fun(_use0, Value) ->
                    {Align@1, Width@1} = _use0,
                    Padded = case Align@1 of
                        left ->
                            pad_right(Value, Width@1);

                        right ->
                            pad_left(Value, Width@1);

                        center ->
                            pad_center(Value, Width@1)
                    end,
                    <<<<" "/utf8, Padded/binary>>/binary, " "/utf8>>
                end
            )
        end,
        gleam@string:join(_pipe, Vertical)
    end,
    <<<<Vertical/binary, Row/binary>>/binary, Vertical/binary>>.

-file("/home/ethanthoma/projects/trellis/src/trellis.gleam", 89).
-spec to_string(table(any())) -> binary().
to_string(Table) ->
    {table, Columns, Rows, Style} = Table,
    Columns@1 = lists:reverse(Columns),
    Headers = gleam@list:map(Columns@1, fun(Col) -> erlang:element(2, Col) end),
    Aligns = gleam@list:map(
        Columns@1,
        fun(Col@1) -> erlang:element(3, Col@1) end
    ),
    Values = gleam@list:flat_map(
        Rows,
        fun(Row) ->
            gleam@list:map(
                Columns@1,
                fun(Col@2) -> (erlang:element(4, Col@2))(Row) end
            )
        end
    ),
    Widths = calculate_widths(Headers, Values),
    Separator_top = make_separator(Style, Widths, true, true),
    Header_row = make_row(
        Style,
        gleam@list:map(
            gleam@list:range(1, erlang:length(Columns@1)),
            fun(_) -> center end
        ),
        Widths,
        Headers
    ),
    Separator_header = make_separator(Style, Widths, true, false),
    Body_rows = gleam@list:map(
        Rows,
        fun(Row@1) ->
            Row_values = gleam@list:map(
                Columns@1,
                fun(Col@3) -> (erlang:element(4, Col@3))(Row@1) end
            ),
            make_row(Style, Aligns, Widths, Row_values)
        end
    ),
    Separator_bottom = make_separator(Style, Widths, false, false),
    _pipe = [Separator_top, Header_row, Separator_header],
    _pipe@1 = lists:append(_pipe, Body_rows),
    _pipe@2 = lists:append(_pipe@1, [Separator_bottom]),
    gleam@string:join(_pipe@2, <<"\n"/utf8>>).