-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 ).