-module(testcontainer_formulas@postgres).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch, inline]).
-define(FILEPATH, "src/testcontainer_formulas/postgres.gleam").
-export([new/0, with_image/2, with_version/2, with_database/2, with_username/2, with_password/2, with_secret_password/2, with_extra_wait/2, on_network/2, on_network_name/2, with_name/2, formula/1]).
-export_type([postgres_container/0, postgres_config/0]).

-if(?OTP_RELEASE >= 27).
-define(MODULEDOC(Str), -moduledoc(Str)).
-define(DOC(Str), -doc(Str)).
-else.
-define(MODULEDOC(Str), -compile([])).
-define(DOC(Str), -compile([])).
-endif.

?MODULEDOC(
    " Postgres formula. Defaults to `postgres:16-alpine`, database `postgres`,\n"
    " user `postgres`, password `postgres`.\n"
    "\n"
    "     use pg <- testcontainer.with_formula(\n"
    "       postgres.new()\n"
    "       |> postgres.with_database(\"myapp_test\")\n"
    "       |> postgres.with_password(\"secret\")\n"
    "       |> postgres.formula(),\n"
    "     )\n"
    "\n"
    "     // pg.connection_url, pg.host, pg.port, pg.database, pg.username\n"
).

-type postgres_container() :: {postgres_container,
        testcontainer@container:container(),
        binary(),
        binary(),
        integer(),
        binary(),
        binary()}.

-opaque postgres_config() :: {postgres_config,
        binary(),
        binary(),
        binary(),
        cowl:secret(binary()),
        gleam@option:option(testcontainer@wait:wait_strategy()),
        gleam@option:option(binary()),
        gleam@option:option(binary())}.

-file("src/testcontainer_formulas/postgres.gleam", 51).
?DOC(" Sensible defaults: `postgres:16-alpine` / `postgres` / `postgres` / `postgres`.\n").
-spec new() -> postgres_config().
new() ->
    {postgres_config,
        <<"postgres:16-alpine"/utf8>>,
        <<"postgres"/utf8>>,
        <<"postgres"/utf8>>,
        cowl:secret(<<"postgres"/utf8>>),
        none,
        none,
        none}.

-file("src/testcontainer_formulas/postgres.gleam", 64).
?DOC(" Replaces the image entirely (e.g. for hardened images or different bases).\n").
-spec with_image(postgres_config(), binary()) -> postgres_config().
with_image(C, Image) ->
    {postgres_config,
        Image,
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 69).
?DOC(" Shorthand: keep `postgres:` prefix, override only the tag.\n").
-spec with_version(postgres_config(), binary()) -> postgres_config().
with_version(C, Version) ->
    {postgres_config,
        <<"postgres:"/utf8, Version/binary>>,
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 73).
-spec with_database(postgres_config(), binary()) -> postgres_config().
with_database(C, Db) ->
    {postgres_config,
        erlang:element(2, C),
        Db,
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 77).
-spec with_username(postgres_config(), binary()) -> postgres_config().
with_username(C, User) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        User,
        erlang:element(5, C),
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 81).
-spec with_password(postgres_config(), binary()) -> postgres_config().
with_password(C, Pass) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        cowl:secret(Pass),
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 85).
-spec with_secret_password(postgres_config(), cowl:secret(binary())) -> postgres_config().
with_secret_password(C, Pass) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        Pass,
        erlang:element(6, C),
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 94).
?DOC(
    " Adds an extra wait strategy on top of the default\n"
    " `log(\"database system is ready to accept connections\")`.\n"
).
-spec with_extra_wait(postgres_config(), testcontainer@wait:wait_strategy()) -> postgres_config().
with_extra_wait(C, S) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        {some, S},
        erlang:element(7, C),
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 103).
?DOC(
    " Attaches the container to the given network (typically built by\n"
    " `testcontainer.with_network/2` or `testcontainer.with_stack/2`).\n"
).
-spec on_network(postgres_config(), testcontainer@network:network()) -> postgres_config().
on_network(C, Net) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        {some, testcontainer@network:name(Net)},
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 108).
?DOC(" Same as `on_network/2` when you already have the Docker network name.\n").
-spec on_network_name(postgres_config(), binary()) -> postgres_config().
on_network_name(C, Net) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        {some, Net},
        erlang:element(8, C)}.

-file("src/testcontainer_formulas/postgres.gleam", 114).
?DOC(
    " Names the container so other containers on the same network can reach\n"
    " it (e.g. `\"db\"` → `db:5432`).\n"
).
-spec with_name(postgres_config(), binary()) -> postgres_config().
with_name(C, N) ->
    {postgres_config,
        erlang:element(2, C),
        erlang:element(3, C),
        erlang:element(4, C),
        erlang:element(5, C),
        erlang:element(6, C),
        erlang:element(7, C),
        {some, N}}.

-file("src/testcontainer_formulas/postgres.gleam", 120).
?DOC(
    " Builds the `Formula(PostgresContainer)` ready to pass to\n"
    " `testcontainer.with_formula/2`.\n"
).
-spec formula(postgres_config()) -> testcontainer@formula:formula(postgres_container()).
formula(C) ->
    Pg_port = testcontainer@port:tcp(5432),
    Base_wait = testcontainer@wait:log(
        <<"database system is ready to accept connections"/utf8>>
    ),
    Wait_strategy = case erlang:element(6, C) of
        none ->
            Base_wait;

        {some, Extra} ->
            testcontainer@wait:all_of([Base_wait, Extra])
    end,
    With_creds = begin
        _pipe = testcontainer@container:new(erlang:element(2, C)),
        _pipe@1 = testcontainer@container:expose_port(_pipe, Pg_port),
        _pipe@2 = testcontainer@container:with_env(
            _pipe@1,
            <<"POSTGRES_DB"/utf8>>,
            erlang:element(3, C)
        ),
        _pipe@3 = testcontainer@container:with_env(
            _pipe@2,
            <<"POSTGRES_USER"/utf8>>,
            erlang:element(4, C)
        ),
        _pipe@4 = testcontainer@container:with_secret_env(
            _pipe@3,
            <<"POSTGRES_PASSWORD"/utf8>>,
            erlang:element(5, C)
        ),
        testcontainer@container:wait_for(_pipe@4, Wait_strategy)
    end,
    With_net = case erlang:element(7, C) of
        none ->
            With_creds;

        {some, N} ->
            _pipe@5 = With_creds,
            testcontainer@container:on_network(_pipe@5, N)
    end,
    Final_spec = case erlang:element(8, C) of
        none ->
            With_net;

        {some, N@1} ->
            _pipe@6 = With_net,
            testcontainer@container:with_name(_pipe@6, N@1)
    end,
    testcontainer@formula:new(
        Final_spec,
        fun(Running) ->
            gleam@result:'try'(
                testcontainer@container:host_port(Running, Pg_port),
                fun(Mapped_port) ->
                    Host = testcontainer@container:host(Running),
                    Pass = cowl:reveal(erlang:element(5, C)),
                    {ok,
                        {postgres_container,
                            Running,
                            <<<<<<<<<<<<<<<<<<"postgresql://"/utf8,
                                                                (gleam_stdlib:percent_encode(
                                                                    erlang:element(
                                                                        4,
                                                                        C
                                                                    )
                                                                ))/binary>>/binary,
                                                            ":"/utf8>>/binary,
                                                        (gleam_stdlib:percent_encode(
                                                            Pass
                                                        ))/binary>>/binary,
                                                    "@"/utf8>>/binary,
                                                Host/binary>>/binary,
                                            ":"/utf8>>/binary,
                                        (erlang:integer_to_binary(Mapped_port))/binary>>/binary,
                                    "/"/utf8>>/binary,
                                (gleam_stdlib:percent_encode(
                                    erlang:element(3, C)
                                ))/binary>>,
                            Host,
                            Mapped_port,
                            erlang:element(3, C),
                            erlang:element(4, C)}}
                end
            )
        end
    ).