gleam_http

-module(gleam@http@response).
-compile(no_auto_import).

-export([try_map/2, new/1, get_header/2, set_header/3, prepend_header/3, set_body/2, map/2, redirect/1, get_cookies/1, set_cookie/4, expire_cookie/3]).
-export_type([response/1]).

-type response(EQX) :: {response, integer(), list({binary(), binary()}), EQX}.

-spec try_map(response(ERA), fun((ERA) -> {ok, ERC} | {error, ERD})) -> {ok,
                                                                         response(ERC)} |
    {error, ERD}.
try_map(Response, Transform) ->
    case Transform(erlang:element(4, Response)) of
        {error, _try} -> {error, _try};
        {ok, Body} ->
            {ok, set_body(Response, Body)}
    end.

-spec new(integer()) -> response(binary()).
new(Status) ->
    {response, Status, [], <<""/utf8>>}.

-spec get_header(response(any()), binary()) -> {ok, binary()} | {error, nil}.
get_header(Response, Key) ->
    gleam@list:key_find(
        erlang:element(3, Response),
        gleam@string:lowercase(Key)
    ).

-spec set_header(response(ERO), binary(), binary()) -> response(ERO).
set_header(Response, Key, Value) ->
    Headers = gleam@list:key_set(
        erlang:element(3, Response),
        Key,
        gleam@string:lowercase(Value)
    ),
    erlang:setelement(3, Response, Headers).

-spec prepend_header(response(ERR), binary(), binary()) -> response(ERR).
prepend_header(Response, Key, Value) ->
    Headers = [{gleam@string:lowercase(Key), Value} |
               erlang:element(3, Response)],
    erlang:setelement(3, Response, Headers).

-spec set_body(response(any()), ERW) -> response(ERW).
set_body(Response, Body) ->
    {response, Status, Headers, _@1} = Response,
    {response, Status, Headers, Body}.

-spec map(response(ERY), fun((ERY) -> ESA)) -> response(ESA).
map(Response, Transform) ->
    _pipe = erlang:element(4, Response),
    _pipe@1 = Transform(_pipe),
    set_body(Response, _pipe@1).

-spec redirect(binary()) -> response(binary()).
redirect(Uri) ->
    {response,
     303,
     [{<<"location"/utf8>>, Uri}],
     gleam@string:append(<<"You are being redirected to "/utf8>>, Uri)}.

-spec get_cookies(response(any())) -> list({binary(), binary()}).
get_cookies(Resp) ->
    {response, _@1, Headers, _@2} = Resp,
    _pipe = Headers,
    _pipe@1 = gleam@list:filter_map(
        _pipe,
        fun(Header) ->
            {Name, Value} = Header,
            case Name of
                <<"set-cookie"/utf8>> ->
                    {ok, gleam@http@cookie:parse(Value)};

                _@3 ->
                    {error, nil}
            end
        end
    ),
    gleam@list:flatten(_pipe@1).

-spec set_cookie(
    response(ESF),
    binary(),
    binary(),
    gleam@http@cookie:attributes()
) -> response(ESF).
set_cookie(Response, Name, Value, Attributes) ->
    prepend_header(
        Response,
        <<"set-cookie"/utf8>>,
        gleam@http@cookie:set_header(Name, Value, Attributes)
    ).

-spec expire_cookie(response(ESI), binary(), gleam@http@cookie:attributes()) -> response(ESI).
expire_cookie(Response, Name, Attributes) ->
    Attrs = erlang:setelement(2, Attributes, {some, 0}),
    set_cookie(Response, Name, <<""/utf8>>, Attrs).