Current section

Files

Jump to
glisten src glisten.erl
Raw

src/glisten.erl

-module(glisten).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).
-export([get_port/1, get_supervisor/1, send/2, handler/2, with_close/2, with_pool_size/2, with_http2/1, serve/2, serve_ssl/4]).
-export_type([start_error/0, message/1, server_port/0, server/0, connection/1, handler/2]).
-type start_error() :: listener_closed |
listener_timeout |
acceptor_timeout |
{acceptor_failed, gleam@erlang@process:exit_reason()} |
{acceptor_crashed, gleam@dynamic:dynamic_()} |
{system_error, glisten@socket:socket_reason()}.
-type message(IEI) :: {packet, bitstring()} | {user, IEI}.
-type server_port() :: {provided, integer()} | assigned.
-opaque server() :: {server,
gleam@erlang@process:subject(gleam@otp@supervisor:message()),
server_port(),
glisten@socket:listen_socket()}.
-type connection(IEJ) :: {connection,
{ok, {{integer(), integer(), integer(), integer()}, integer()}} |
{error, nil},
glisten@socket:socket(),
glisten@transport:transport(),
gleam@erlang@process:subject(glisten@internal@handler:message(IEJ))}.
-opaque handler(IEK, IEL) :: {handler,
fun((connection(IEK)) -> {IEL,
gleam@option:option(gleam@erlang@process:selector(IEK))}),
fun((message(IEK), IEL, connection(IEK)) -> gleam@otp@actor:next(message(IEK), IEL)),
gleam@option:option(fun((IEL) -> nil)),
integer(),
boolean()}.
-spec get_port(server()) -> {ok, integer()} | {error, nil}.
get_port(Server) ->
case erlang:element(3, Server) of
{provided, Value} ->
{ok, Value};
assigned ->
_pipe = erlang:element(4, Server),
_pipe@1 = inet:port(_pipe),
gleam@result:nil_error(_pipe@1)
end.
-spec get_supervisor(server()) -> gleam@erlang@process:subject(gleam@otp@supervisor:message()).
get_supervisor(Server) ->
erlang:element(2, Server).
-spec send(connection(any()), gleam@bytes_builder:bytes_builder()) -> {ok, nil} |
{error, glisten@socket:socket_reason()}.
send(Conn, Msg) ->
glisten@transport:send(
erlang:element(4, Conn),
erlang:element(3, Conn),
Msg
).
-spec convert_on_init(
fun((connection(IFL)) -> {IFN,
gleam@option:option(gleam@erlang@process:selector(IFL))})
) -> fun((glisten@internal@handler:connection(IFL)) -> {IFN,
gleam@option:option(gleam@erlang@process:selector(IFL))}).
convert_on_init(On_init) ->
fun(Conn) ->
Connection = {connection,
erlang:element(2, Conn),
erlang:element(3, Conn),
erlang:element(4, Conn),
erlang:element(5, Conn)},
On_init(Connection)
end.
-spec handler(
fun((connection(IFT)) -> {IFV,
gleam@option:option(gleam@erlang@process:selector(IFT))}),
fun((message(IFT), IFV, connection(IFT)) -> gleam@otp@actor:next(message(IFT), IFV))
) -> handler(IFT, IFV).
handler(On_init, Loop) ->
{handler, On_init, Loop, none, 10, false}.
-spec map_user_selector(gleam@erlang@process:selector(message(IFA))) -> gleam@erlang@process:selector(glisten@internal@handler:loop_message(IFA)).
map_user_selector(Selector) ->
gleam_erlang_ffi:map_selector(Selector, fun(Value) -> case Value of
{packet, Msg} ->
{packet, Msg};
{user, Msg@1} ->
{custom, Msg@1}
end end).
-spec convert_loop(
fun((message(IFF), IFG, connection(IFF)) -> gleam@otp@actor:next(message(IFF), IFG))
) -> fun((glisten@internal@handler:loop_message(IFF), IFG, glisten@internal@handler:connection(IFF)) -> gleam@otp@actor:next(glisten@internal@handler:loop_message(IFF), IFG)).
convert_loop(Loop) ->
fun(Msg, Data, Conn) ->
Conn@1 = {connection,
erlang:element(2, Conn),
erlang:element(3, Conn),
erlang:element(4, Conn),
erlang:element(5, Conn)},
case Msg of
{packet, Msg@1} ->
case Loop({packet, Msg@1}, Data, Conn@1) of
{continue, Data@1, Selector} ->
{continue,
Data@1,
gleam@option:map(Selector, fun map_user_selector/1)};
{stop, Reason} ->
{stop, Reason}
end;
{custom, Msg@2} ->
case Loop({user, Msg@2}, Data, Conn@1) of
{continue, Data@2, Selector@1} ->
{continue,
Data@2,
gleam@option:map(
Selector@1,
fun map_user_selector/1
)};
{stop, Reason@1} ->
{stop, Reason@1}
end
end
end.
-spec with_close(handler(IGC, IGD), fun((IGD) -> nil)) -> handler(IGC, IGD).
with_close(Handler, On_close) ->
erlang:setelement(4, Handler, {some, On_close}).
-spec with_pool_size(handler(IGI, IGJ), integer()) -> handler(IGI, IGJ).
with_pool_size(Handler, Size) ->
erlang:setelement(5, Handler, Size).
-spec with_http2(handler(IGO, IGP)) -> handler(IGO, IGP).
with_http2(Handler) ->
erlang:setelement(6, Handler, true).
-spec serve(handler(any(), any()), integer()) -> {ok, server()} |
{error, start_error()}.
serve(Handler, Port) ->
_pipe = glisten@tcp:listen(Port, []),
_pipe@1 = gleam@result:map_error(_pipe, fun(Err) -> case Err of
closed ->
listener_closed;
timeout ->
listener_timeout;
Err@1 ->
{system_error, Err@1}
end end),
gleam@result:then(
_pipe@1,
fun(Socket) ->
_pipe@2 = {pool,
Socket,
convert_loop(erlang:element(3, Handler)),
erlang:element(5, Handler),
convert_on_init(erlang:element(2, Handler)),
erlang:element(4, Handler),
tcp},
_pipe@3 = glisten@internal@acceptor:start_pool(_pipe@2),
_pipe@4 = gleam@result:map_error(
_pipe@3,
fun(Err@2) -> case Err@2 of
init_timeout ->
acceptor_timeout;
{init_failed, Reason} ->
{acceptor_failed, Reason};
{init_crashed, Reason@1} ->
{acceptor_crashed, Reason@1}
end end
),
gleam@result:map(_pipe@4, fun(Pool) -> {server, Pool, case Port of
0 ->
assigned;
Val ->
{provided, Val}
end, Socket} end)
end
).
-spec serve_ssl(handler(any(), any()), integer(), binary(), binary()) -> {ok,
server()} |
{error, start_error()}.
serve_ssl(Handler, Port, Certfile, Keyfile) ->
_assert_subject = glisten_ssl_ffi:start_ssl(),
{ok, _} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Assertion pattern match failed"/utf8>>,
value => _assert_fail,
module => <<"glisten"/utf8>>,
function => <<"serve_ssl"/utf8>>,
line => 260})
end,
Ssl_options = [{certfile, Certfile}, {keyfile, Keyfile}],
Protocol_options = case erlang:element(6, Handler) of
true ->
[{alpn_preferred_protocols, [<<"h2"/utf8>>, <<"http/1.1"/utf8>>]}];
false ->
[{alpn_preferred_protocols, [<<"http/1.1"/utf8>>]}]
end,
_pipe = glisten@ssl:listen(
Port,
gleam@list:concat([Ssl_options, Protocol_options])
),
_pipe@1 = gleam@result:map_error(_pipe, fun(Err) -> case Err of
closed ->
listener_closed;
timeout ->
listener_timeout;
Err@1 ->
{system_error, Err@1}
end end),
gleam@result:then(
_pipe@1,
fun(Socket) ->
_pipe@2 = {pool,
Socket,
convert_loop(erlang:element(3, Handler)),
erlang:element(5, Handler),
convert_on_init(erlang:element(2, Handler)),
erlang:element(4, Handler),
ssl},
_pipe@3 = glisten@internal@acceptor:start_pool(_pipe@2),
_pipe@4 = gleam@result:map_error(
_pipe@3,
fun(Err@2) -> case Err@2 of
init_timeout ->
acceptor_timeout;
{init_failed, Reason} ->
{acceptor_failed, Reason};
{init_crashed, Reason@1} ->
{acceptor_crashed, Reason@1}
end end
),
gleam@result:map(_pipe@4, fun(Pool) -> {server, Pool, case Port of
0 ->
assigned;
Val ->
{provided, Val}
end, Socket} end)
end
).