-module(gleam@otp@supervisor).
-compile(no_auto_import).

-export([add/2, supervisor/1, worker/1, returning/2, start_spec/1, start/1, application_stopped/0, to_erlang_start_result/1]).
-export_type([spec/2, children/1, child_spec/3, child_start_error/0, message/0, instruction/0, state/1, starter/1, child/1, handle_exit_error/0, application_start_mode/0, application_stop/0]).

-type spec(FFA, FFB) :: {spec,
        FFA,
        integer(),
        integer(),
        fun((children(FFA)) -> children(FFB))}.

-opaque children(FFC) :: {ready, starter(FFC)} | {failed, child_start_error()}.

-opaque child_spec(FFD, FFE, FFF) :: {child_spec,
        fun((FFE) -> {ok, gleam@erlang@process:subject(FFD)} |
            {error, gleam@otp@actor:start_error()}),
        fun((FFE, gleam@erlang@process:subject(FFD)) -> FFF)}.

-type child_start_error() :: {child_start_error,
        gleam@option:option(gleam@erlang@process:pid_()),
        gleam@otp@actor:start_error()}.

-opaque message() :: {exit, gleam@erlang@process:exit_message()} |
    {retry_restart, gleam@erlang@process:pid_()}.

-type instruction() :: start_all | {start_from, gleam@erlang@process:pid_()}.

-type state(FFG) :: {state,
        gleam@otp@intensity_tracker:intensity_tracker(),
        starter(FFG),
        gleam@erlang@process:subject(gleam@erlang@process:pid_())}.

-type starter(FFH) :: {starter,
        FFH,
        gleam@option:option(fun((instruction()) -> {ok,
                {starter(FFH), instruction()}} |
            {error, child_start_error()}))}.

-type child(FFI) :: {child, gleam@erlang@process:pid_(), FFI}.

-type handle_exit_error() :: {restart_failed,
        gleam@erlang@process:pid_(),
        gleam@otp@intensity_tracker:intensity_tracker()} |
    too_many_restarts.

-type application_start_mode() :: normal |
    {takeover, gleam@otp@node:node_()} |
    {failover, gleam@otp@node:node_()}.

-type application_stop() :: any().

-spec start_child(child_spec(any(), FGL, FGM), FGL) -> {ok, child(FGM)} |
    {error, child_start_error()}.
start_child(Child_spec, Argument) ->
    case begin
        _pipe = (erlang:element(2, Child_spec))(Argument),
        gleam@result:map_error(
            _pipe,
            fun(_capture) -> {child_start_error, none, _capture} end
        )
    end of
        {error, _try} -> {error, _try};
        {ok, Subject} ->
            {ok,
                {child,
                    gleam@erlang@process:subject_owner(Subject),
                    (erlang:element(3, Child_spec))(Argument, Subject)}}
    end.

-spec shutdown_child(
    gleam@erlang@process:pid_(),
    child_spec(any(), any(), any())
) -> nil.
shutdown_child(Pid, _) ->
    gleam@erlang@process:send_exit(Pid).

-spec perform_instruction_for_child(
    FGZ,
    instruction(),
    child_spec(any(), FGZ, FHB),
    child(FHB)
) -> {ok, {child(FHB), instruction()}} | {error, child_start_error()}.
perform_instruction_for_child(Argument, Instruction, Child_spec, Child) ->
    Current = erlang:element(2, Child),
    case Instruction of
        {start_from, Target} when Target =/= Current ->
            {ok, {Child, Instruction}};

        _@1 ->
            shutdown_child(Current, Child_spec),
            case start_child(Child_spec, Argument) of
                {error, _try} -> {error, _try};
                {ok, Child@1} ->
                    {ok, {Child@1, start_all}}
            end
    end.

-spec add_child_to_starter(
    starter(FHJ),
    child_spec(any(), FHJ, FHM),
    child(FHM)
) -> starter(FHM).
add_child_to_starter(Starter, Child_spec, Child) ->
    Starter@3 = fun(Instruction) -> case case erlang:element(3, Starter) of
            {some, Start} ->
                Start(Instruction);

            none ->
                {ok, {Starter, Instruction}}
        end of
            {error, _try} -> {error, _try};
            {ok, {Starter@1, Instruction@1}} ->
                case perform_instruction_for_child(
                    erlang:element(2, Starter@1),
                    Instruction@1,
                    Child_spec,
                    Child
                ) of
                    {error, _try@1} -> {error, _try@1};
                    {ok, {Child@1, Instruction@2}} ->
                        Starter@2 = add_child_to_starter(
                            Starter@1,
                            Child_spec,
                            Child@1
                        ),
                        {ok, {Starter@2, Instruction@2}}
                end
        end end,
    {starter, erlang:element(3, Child), {some, Starter@3}}.

-spec start_and_add_child(starter(FHS), child_spec(any(), FHS, FHV)) -> children(FHV).
start_and_add_child(State, Child_spec) ->
    case start_child(Child_spec, erlang:element(2, State)) of
        {ok, Child} ->
            {ready, add_child_to_starter(State, Child_spec, Child)};

        {error, Reason} ->
            {failed, Reason}
    end.

-spec add(children(FIA), child_spec(any(), FIA, FID)) -> children(FID).
add(Children, Child_spec) ->
    case Children of
        {failed, Fail} ->
            {failed, Fail};

        {ready, State} ->
            start_and_add_child(State, Child_spec)
    end.

-spec supervisor(
    fun((FII) -> {ok, gleam@erlang@process:subject(FIJ)} |
        {error, gleam@otp@actor:start_error()})
) -> child_spec(FIJ, FII, FII).
supervisor(Start) ->
    {child_spec, Start, fun(Argument, _) -> Argument end}.

-spec worker(
    fun((FIQ) -> {ok, gleam@erlang@process:subject(FIR)} |
        {error, gleam@otp@actor:start_error()})
) -> child_spec(FIR, FIQ, FIQ).
worker(Start) ->
    {child_spec, Start, fun(Argument, _) -> Argument end}.

-spec returning(
    child_spec(FIY, FIZ, any()),
    fun((FIZ, gleam@erlang@process:subject(FIY)) -> FJF)
) -> child_spec(FIY, FIZ, FJF).
returning(Child, Updater) ->
    {child_spec, erlang:element(2, Child), Updater}.

-spec init(spec(any(), FJK)) -> gleam@otp@actor:init_result(state(FJK), message()).
init(Spec) ->
    Retry = gleam@erlang@process:new_subject(),
    gleam_erlang_ffi:trap_exits(true),
    Selector = begin
        _pipe = gleam_erlang_ffi:new_selector(),
        _pipe@1 = gleam@erlang@process:selecting(
            _pipe,
            Retry,
            fun(Field@0) -> {retry_restart, Field@0} end
        ),
        gleam@erlang@process:selecting_trapped_exits(
            _pipe@1,
            fun(Field@0) -> {exit, Field@0} end
        )
    end,
    Result = begin
        _pipe@2 = {starter, erlang:element(2, Spec), none},
        _pipe@3 = {ready, _pipe@2},
        (erlang:element(5, Spec))(_pipe@3)
    end,
    case Result of
        {ready, Starter} ->
            Restarts = gleam@otp@intensity_tracker:new(
                erlang:element(3, Spec),
                erlang:element(4, Spec)
            ),
            State = {state, Restarts, Starter, Retry},
            {ready, State, Selector};

        {failed, Error} ->
            {failed, case erlang:element(3, Error) of
                    init_timeout ->
                        <<"Child initialisation timed out"/utf8>>;

                    {init_crashed, Reason} ->
                        gleam@string:append(
                            <<"Child crashed during initialisation: "/utf8>>,
                            gleam@string:inspect(Reason)
                        );

                    {init_failed, Reason@1} ->
                        gleam@string:append(
                            <<"Child failed to start during initialisation: "/utf8>>,
                            gleam@string:inspect(Reason@1)
                        )
                end}
    end.

-spec handle_exit(gleam@erlang@process:pid_(), state(FJQ)) -> gleam@otp@actor:next(state(FJQ)).
handle_exit(Pid, State) ->
    Outcome = begin
        {some, Start@1} = case erlang:element(3, erlang:element(3, State)) of
            {some, Start} -> {some, Start};
            _try ->
                erlang:error(#{gleam_error => assert,
                            message => <<"Assertion pattern match failed"/utf8>>,
                            value => _try,
                            module => <<"gleam/otp/supervisor"/utf8>>,
                            function => <<"handle_exit"/utf8>>,
                            line => 293})
        end,
        case begin
            _pipe = erlang:element(2, State),
            _pipe@1 = gleam@otp@intensity_tracker:add_event(_pipe),
            gleam@result:map_error(_pipe@1, fun(_) -> too_many_restarts end)
        end of
            {error, _try@1} -> {error, _try@1};
            {ok, Restarts} ->
                case begin
                    _pipe@2 = Start@1({start_from, Pid}),
                    gleam@result:map_error(
                        _pipe@2,
                        fun(E) ->
                            {restart_failed,
                                gleam@option:unwrap(erlang:element(2, E), Pid),
                                Restarts}
                        end
                    )
                end of
                    {error, _try@2} -> {error, _try@2};
                    {ok, {Starter, _@1}} ->
                        {ok,
                            erlang:setelement(
                                2,
                                erlang:setelement(3, State, Starter),
                                Restarts
                            )}
                end
        end
    end,
    case Outcome of
        {ok, State@1} ->
            {continue, State@1};

        {error, {restart_failed, Failed_child, Restarts@1}} ->
            gleam@erlang@process:send(erlang:element(4, State), Failed_child),
            State@2 = erlang:setelement(2, State, Restarts@1),
            {continue, State@2};

        {error, too_many_restarts} ->
            {stop,
                {abnormal,
                    <<"Child processes restarted too many times within allowed period"/utf8>>}}
    end.

-spec loop(message(), state(FJU)) -> gleam@otp@actor:next(state(FJU)).
loop(Message, State) ->
    case Message of
        {exit, Exit_message} ->
            handle_exit(erlang:element(2, Exit_message), State);

        {retry_restart, Pid} ->
            handle_exit(Pid, State)
    end.

-spec start_spec(spec(any(), any())) -> {ok,
        gleam@erlang@process:subject(message())} |
    {error, gleam@otp@actor:start_error()}.
start_spec(Spec) ->
    gleam@otp@actor:start_spec(
        {spec, fun() -> init(Spec) end, 60000, fun loop/2}
    ).

-spec start(fun((children(nil)) -> children(any()))) -> {ok,
        gleam@erlang@process:subject(message())} |
    {error, gleam@otp@actor:start_error()}.
start(Init) ->
    start_spec({spec, nil, 5, 1, Init}).

-spec application_stopped() -> application_stop().
application_stopped() ->
    gleam_otp_external:application_stopped().

-spec to_erlang_start_result(
    {ok, gleam@erlang@process:subject(any())} |
        {error, gleam@otp@actor:start_error()}
) -> {ok, gleam@erlang@process:pid_()} | {error, gleam@dynamic:dynamic()}.
to_erlang_start_result(Res) ->
    gleam@otp@actor:to_erlang_start_result(Res).