Current section
Files
Jump to
Current section
Files
src/squirrel@internal@database@postgres.erl
-module(squirrel@internal@database@postgres).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).
-export([main/2]).
-export_type([pg_type/0, context/0, nullability/0, plan/0, join_type/0, connection_options/0]).
-type pg_type() :: {p_base, binary()} |
{p_array, pg_type()} |
{p_enum, binary(), list(binary())} |
{p_option, pg_type()}.
-type context() :: {context,
squirrel@internal@database@postgres_protocol:connection(),
gleam@dict:dict(integer(), squirrel@internal@gleam:type()),
gleam@dict:dict({integer(), integer()}, nullability())}.
-type nullability() :: nullable | not_nullable.
-type plan() :: {plan,
gleam@option:option(join_type()),
list(binary()),
list(plan())}.
-type join_type() :: full_join | left_join | right_join | inner_join.
-type connection_options() :: {connection_options,
binary(),
integer(),
binary(),
binary(),
binary(),
integer()}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 46).
-spec find_postgres_type_query() -> squirrel@internal@query:untyped_query().
find_postgres_type_query() ->
_assert_subject = squirrel@internal@gleam:value_identifier(
<<"find_postgres_type_query"/utf8>>
),
{ok, Name} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_postgres_type_query"/utf8>>,
line => 47})
end,
{untyped_query,
<<""/utf8>>,
1,
Name,
[],
<<"
select
-- The name of the type or, if the type is an array, the name of its
-- elements' type.
case
when elem.typname is null then type.typname
else elem.typname
end as type,
-- Tells us how to interpret the first column: if this is true then the first
-- column is the type of the elements of the array type.
-- Otherwise it means we've found a base type.
case
when elem.typname is null then false
else true
end as is_array,
-- The type of the type/array item.
-- It will be 'e' if the thing is an enum.
case
when elem.typname is null then type.typtype
else elem.typtype
end as kind
from
pg_type as type
left join pg_type as elem on type.typelem = elem.oid
where
type.oid = $1
"/utf8>>}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 86).
-spec find_enum_variants_query() -> squirrel@internal@query:untyped_query().
find_enum_variants_query() ->
_assert_subject = squirrel@internal@gleam:value_identifier(
<<"find_enum_variants_query"/utf8>>
),
{ok, Name} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_enum_variants_query"/utf8>>,
line => 87})
end,
{untyped_query,
<<""/utf8>>,
1,
Name,
[],
<<"
select
enumlabel
from
pg_enum
where
enumtypid = $1
order by
enumsortorder asc
"/utf8>>}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 107).
-spec find_column_nullability_query() -> squirrel@internal@query:untyped_query().
find_column_nullability_query() ->
_assert_subject = squirrel@internal@gleam:value_identifier(
<<"find_column_nullability_query"/utf8>>
),
{ok, Name} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_column_nullability_query"/utf8>>,
line => 108})
end,
{untyped_query,
<<""/utf8>>,
1,
Name,
[],
<<"
select
-- Whether the column has a not-null constraint.
attnotnull
from
pg_attribute
where
-- The oid of the table the column comes from.
attrelid = $1
-- The index of the column we're looking for.
and attnum = $2
"/utf8>>}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 474).
-spec check_sasl_final_message(bitstring(), bitstring(), binary()) -> eval:eval(nil, squirrel@internal@error:error(), context()).
check_sasl_final_message(Msg, Expected_server_proof, User) ->
case squirrel@internal@scram:parse_server_final(Msg) of
{error, nil} ->
eval:throw(pg_invalid_sha256_server_proof);
{ok, {failed, _}} ->
eval:throw({pg_invalid_password, User});
{ok, {successful, Server_proof}} ->
case Server_proof =:= Expected_server_proof of
true ->
eval:return(nil);
false ->
eval:throw(pg_invalid_sha256_server_proof)
end
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 817).
-spec plan_outputs_indices(plan(), gleam@dict:dict(binary(), integer())) -> gleam@set:set(integer()).
plan_outputs_indices(Plan, Query_outputs) ->
gleam@list:fold(
erlang:element(3, Plan),
gleam@set:new(),
fun(Nullables, Output) ->
case gleam_stdlib:map_get(Query_outputs, Output) of
{ok, I} ->
gleam@set:insert(Nullables, I);
{error, _} ->
Nullables
end
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 768).
-spec do_nullables_from_plan(
plan(),
gleam@dict:dict(binary(), integer()),
gleam@set:set(integer())
) -> gleam@set:set(integer()).
do_nullables_from_plan(Plan, Query_outputs, Nullables) ->
case {erlang:element(2, Plan), erlang:element(4, Plan)} of
{{some, full_join}, _} ->
_pipe = plan_outputs_indices(Plan, Query_outputs),
gleam@set:union(_pipe, Nullables);
{{some, right_join}, [Left, Right]} ->
Nullables@1 = begin
_pipe@1 = plan_outputs_indices(Left, Query_outputs),
gleam@set:union(_pipe@1, Nullables)
end,
do_nullables_from_plan(Right, Query_outputs, Nullables@1);
{{some, left_join}, [Left@1, Right@1]} ->
Nullables@2 = begin
_pipe@2 = plan_outputs_indices(Right@1, Query_outputs),
gleam@set:union(_pipe@2, Nullables)
end,
do_nullables_from_plan(Left@1, Query_outputs, Nullables@2);
{{some, right_join}, Plans} ->
gleam@list:fold(
Plans,
Nullables,
fun(Nullables@3, Plan@1) ->
do_nullables_from_plan(Plan@1, Query_outputs, Nullables@3)
end
);
{{some, left_join}, Plans} ->
gleam@list:fold(
Plans,
Nullables,
fun(Nullables@3, Plan@1) ->
do_nullables_from_plan(Plan@1, Query_outputs, Nullables@3)
end
);
{none, Plans} ->
gleam@list:fold(
Plans,
Nullables,
fun(Nullables@3, Plan@1) ->
do_nullables_from_plan(Plan@1, Query_outputs, Nullables@3)
end
);
{{some, inner_join}, Plans@1} ->
gleam@list:fold(
Plans@1,
Nullables,
fun(Nullables@4, Plan@2) ->
do_nullables_from_plan(Plan@2, Query_outputs, Nullables@4)
end
)
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 763).
-spec nullables_from_plan(plan()) -> gleam@set:set(integer()).
nullables_from_plan(Plan) ->
Outputs = gleam@list:index_fold(
erlang:element(3, Plan),
maps:new(),
fun gleam@dict:insert/3
),
do_nullables_from_plan(Plan, Outputs, gleam@set:new()).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1067).
-spec fields_to_permission_denied_error(
binary(),
gleam@set:set(squirrel@internal@database@postgres_protocol:error_or_notice_field())
) -> {ok, squirrel@internal@error:error()} | {error, nil}.
fields_to_permission_denied_error(Query_file, Fields) ->
{Code@2, Reason@2} = (gleam@set:fold(
Fields,
{none, none},
fun(_use0, Field) ->
{Code, Reason} = _use0,
case Field of
{code, Code@1} ->
{{some, Code@1}, Reason};
{message, Reason@1} ->
{Code, {some, Reason@1}};
_ ->
{Code, Reason}
end
end
)),
case {Code@2, Reason@2} of
{{some, <<"42501"/utf8>>}, {some, Reason@3}} ->
{ok, {pg_permission_denied, Query_file, Reason@3}};
{_, _} ->
{error, nil}
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1087).
-spec 'receive'() -> eval:eval(squirrel@internal@database@postgres_protocol:backend_message(), squirrel@internal@error:error(), context()).
'receive'() ->
eval:from(
fun(_use0) ->
{context, Db, _, _} = Context = _use0,
case squirrel@internal@database@postgres_protocol:'receive'(Db) of
{ok, {Db@1, Msg}} ->
{erlang:setelement(2, Context, Db@1), {ok, Msg}};
{error, {read_decode_error, Error}} ->
{Context,
{error,
{pg_cannot_decode_received_message,
gleam@string:inspect(Error)}}};
{error, {socket_error, Error@1}} ->
{Context,
{error,
{pg_cannot_receive_message,
gleam@string:inspect(Error@1)}}}
end
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 997).
-spec do_accumulate_data_rows_until_command_complete(
binary(),
list(list(bitstring()))
) -> eval:eval(list(list(bitstring())), squirrel@internal@error:error(), context()).
do_accumulate_data_rows_until_command_complete(Query_file, Acc) ->
eval:'try'('receive'(), fun(Msg) -> case Msg of
{be_command_complete, _, _} ->
eval:return(lists:reverse(Acc));
{be_message_data_row, Data} ->
do_accumulate_data_rows_until_command_complete(
Query_file,
[Data | Acc]
);
{be_error_response, Fields} ->
case fields_to_permission_denied_error(Query_file, Fields) of
{ok, Error} ->
eval:throw(Error);
{error, _} ->
erlang:error(#{gleam_error => panic,
message => gleam@string:inspect(Msg),
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"do_accumulate_data_rows_until_command_complete"/utf8>>,
line => 1009})
end;
_ ->
erlang:error(#{gleam_error => panic,
message => gleam@string:inspect(Msg),
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"do_accumulate_data_rows_until_command_complete"/utf8>>,
line => 1011})
end end).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 993).
-spec accumulate_data_rows_until_command_complete(binary()) -> eval:eval(list(list(bitstring())), squirrel@internal@error:error(), context()).
accumulate_data_rows_until_command_complete(Query_file) ->
do_accumulate_data_rows_until_command_complete(Query_file, []).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1104).
-spec send(squirrel@internal@database@postgres_protocol:frontend_message()) -> eval:eval(nil, squirrel@internal@error:error(), context()).
send(Message) ->
eval:from(
fun(_use0) ->
{context, Db, _, _} = Context = _use0,
Result = begin
_pipe = Message,
_pipe@1 = squirrel@internal@database@postgres_protocol:encode_frontend_message(
_pipe
),
squirrel@internal@database@postgres_protocol:send(Db, _pipe@1)
end,
{Db@2, Result@1} = case Result of
{ok, Db@1} ->
{Db@1, {ok, nil}};
{error, Error} ->
{Db,
{error,
{pg_cannot_send_message,
gleam@string:inspect(Error)}}}
end,
{erlang:setelement(2, Context, Db@2), Result@1}
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1122).
-spec send_all(
list(squirrel@internal@database@postgres_protocol:frontend_message())
) -> eval:eval(nil, squirrel@internal@error:error(), context()).
send_all(Messages) ->
squirrel@internal@eval_extra:try_fold(
Messages,
nil,
fun(Acc, Msg) ->
eval:'try'(send(Msg), fun(_) -> eval:return(Acc) end)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1136).
-spec do_wait_until_ready() -> eval:eval(nil, squirrel@internal@error:error(), context()).
do_wait_until_ready() ->
eval:'try'('receive'(), fun(Msg) -> case Msg of
{be_ready_for_query, _} ->
eval:return(nil);
_ ->
do_wait_until_ready()
end end).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1131).
-spec wait_until_ready() -> eval:eval(nil, squirrel@internal@error:error(), context()).
wait_until_ready() ->
eval:'try'(send(fe_flush), fun(_) -> do_wait_until_ready() end).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1147).
-spec unexpected_message(
fun((binary(), binary()) -> squirrel@internal@error:error()),
binary(),
squirrel@internal@database@postgres_protocol:backend_message()
) -> eval:eval(any(), squirrel@internal@error:error(), any()).
unexpected_message(Builder, Expected, Got) ->
_pipe = Builder(Expected, gleam@string:inspect(Got)),
eval:throw(_pipe).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 367).
-spec cleartext_authenticate(binary(), binary()) -> eval:eval(nil, squirrel@internal@error:error(), context()).
cleartext_authenticate(User, Password) ->
eval:'try'(
send({fe_ambigous, {fe_password_message, Password}}),
fun(_) -> eval:'try'('receive'(), fun(Msg) -> case Msg of
be_authentication_ok ->
eval:return(nil);
{be_error_response, _} ->
eval:throw({pg_invalid_password, User});
_ ->
unexpected_message(
fun(Field@0, Field@1) -> {pg_unexpected_cleartext_auth_message, Field@0, Field@1} end,
<<"AuthenticationOk ok ErrorRespose"/utf8>>,
Msg
)
end end) end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 434).
-spec expect_sasl_continue_message(
squirrel@internal@database@postgres_protocol:backend_message()
) -> eval:eval(bitstring(), squirrel@internal@error:error(), context()).
expect_sasl_continue_message(Msg) ->
case Msg of
{be_authentication_sasl_continue, Server_first} ->
eval:return(Server_first);
_ ->
unexpected_message(
fun(Field@0, Field@1) -> {pg_unexpected_sha256_auth_message, Field@0, Field@1} end,
<<"AuthenticationSASLContinue(server-first)"/utf8>>,
Msg
)
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 452).
-spec expect_sasl_final_message(
squirrel@internal@database@postgres_protocol:backend_message(),
binary()
) -> eval:eval(bitstring(), squirrel@internal@error:error(), any()).
expect_sasl_final_message(Msg, User) ->
Unexpected_message = unexpected_message(
fun(Field@0, Field@1) -> {pg_unexpected_sha256_auth_message, Field@0, Field@1} end,
<<"AuthenticationSASLFinal or BeErrorResponse"/utf8>>,
Msg
),
case Msg of
{be_authentication_sasl_final, Msg@1} ->
eval:return(Msg@1);
{be_error_response, Fields} ->
case gleam@set:contains(Fields, {code, <<"28P01"/utf8>>}) of
true ->
eval:throw({pg_invalid_password, User});
false ->
Unexpected_message
end;
_ ->
Unexpected_message
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 385).
-spec sha_256_authenticate(binary(), binary()) -> eval:eval(nil, squirrel@internal@error:error(), context()).
sha_256_authenticate(User, Password) ->
Nonce = squirrel@internal@scram:nonce(),
Client_first_msg = {client_first, User, Nonce},
eval:'try'(
begin
_pipe = squirrel@internal@scram:encode_client_first(
Client_first_msg
),
_pipe@1 = {fe_sasl_initial_response,
<<"SCRAM-SHA-256"/utf8>>,
_pipe},
_pipe@2 = {fe_ambigous, _pipe@1},
send(_pipe@2)
end,
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg) ->
eval:'try'(
expect_sasl_continue_message(Msg),
fun(Raw_server_first_msg) ->
_assert_subject = squirrel@internal@scram:parse_server_first(
Raw_server_first_msg,
Nonce
),
{ok, Server_first_msg} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"sha_256_authenticate"/utf8>>,
line => 404})
end,
Client_last_msg = {client_last,
Client_first_msg,
Server_first_msg,
Password},
{Client_last_msg@1, Expected_server_proof} = squirrel@internal@scram:encode_client_last(
Client_last_msg
),
eval:'try'(
begin
_pipe@3 = Client_last_msg@1,
_pipe@4 = {fe_sasl_response, _pipe@3},
_pipe@5 = {fe_ambigous, _pipe@4},
send(_pipe@5)
end,
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg@1) ->
eval:'try'(
expect_sasl_final_message(
Msg@1,
User
),
fun(Msg@2) ->
check_sasl_final_message(
Msg@2,
Expected_server_proof,
User
)
end
)
end
)
end
)
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1059).
-spec expect_ready_for_query_then_throw(
squirrel@internal@error:error(),
fun((binary(), binary()) -> squirrel@internal@error:error())
) -> eval:eval(any(), squirrel@internal@error:error(), context()).
expect_ready_for_query_then_throw(Error, To_error) ->
eval:'try'('receive'(), fun(Msg) -> case Msg of
{be_ready_for_query, _} ->
eval:throw(Error);
_ ->
unexpected_message(
To_error,
<<"BeReadyForQuery(_)"/utf8>>,
Msg
)
end end).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1157).
-spec unsupported_authentication(binary()) -> eval:eval(any(), squirrel@internal@error:error(), any()).
unsupported_authentication(Auth) ->
eval:throw({pg_unsupported_authentication, Auth}).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 319).
-spec authenticate(connection_options()) -> eval:eval(nil, squirrel@internal@error:error(), context()).
authenticate(Connection) ->
{connection_options, _, _, User, Password, Database, _} = Connection,
Params = [{<<"user"/utf8>>, User}, {<<"database"/utf8>>, Database}],
eval:'try'(
send({fe_startup_message, Params}),
fun(_) -> eval:'try'('receive'(), fun(Msg) -> eval:'try'(case Msg of
be_authentication_ok ->
eval:return(nil);
be_authentication_cleartext_password ->
cleartext_authenticate(User, Password);
{be_authentication_md5_password, _} ->
unsupported_authentication(<<"md5"/utf8>>);
be_authentication_gss ->
unsupported_authentication(<<"GSS"/utf8>>);
{be_authentication_sasl, Methods} ->
case gleam@list:contains(
Methods,
<<"SCRAM-SHA-256"/utf8>>
) of
true ->
sha_256_authenticate(User, Password);
_ ->
_pipe = case gleam@list:filter(
Methods,
fun(Method) ->
Method /= <<""/utf8>>
end
) of
[_ | _] ->
<<<<"SASL("/utf8,
(gleam@string:join(
Methods,
<<","/utf8>>
))/binary>>/binary,
")"/utf8>>;
[] ->
<<"SASL"/utf8>>
end,
unsupported_authentication(_pipe)
end;
be_authentication_sspi ->
unsupported_authentication(<<"SSPI"/utf8>>);
be_authentication_kerberos_v5 ->
unsupported_authentication(
<<"KerberosV5"/utf8>>
);
_ ->
unexpected_message(
fun(Field@0, Field@1) -> {pg_unexpected_auth_method_message, Field@0, Field@1} end,
<<"AuthMethod"/utf8>>,
Msg
)
end, fun(_) ->
eval:'try'(
begin
_pipe@1 = wait_until_ready(),
eval:replace_error(
_pipe@1,
{pg_invalid_user_database,
erlang:element(4, Connection),
erlang:element(6, Connection)}
)
end,
fun(_) -> eval:return(nil) end
)
end) end) end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1166).
-spec with_cached_gleam_type(
integer(),
fun(() -> eval:eval(squirrel@internal@gleam:type(), squirrel@internal@error:error(), context()))
) -> eval:eval(squirrel@internal@gleam:type(), squirrel@internal@error:error(), context()).
with_cached_gleam_type(Oid, Do) ->
eval:from(
fun(Context) ->
case gleam_stdlib:map_get(erlang:element(3, Context), Oid) of
{ok, Type_} ->
{Context, {ok, Type_}};
{error, _} ->
case eval:step(Do(), Context) of
{_, {error, _}} = Result ->
Result;
{{context, _, Gleam_types, _} = Context@1,
{ok, Type_@1}} ->
Gleam_types@1 = gleam@dict:insert(
Gleam_types,
Oid,
Type_@1
),
New_context = erlang:setelement(
3,
Context@1,
Gleam_types@1
),
{New_context, {ok, Type_@1}}
end
end
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1190).
-spec with_cached_column(
integer(),
integer(),
fun(() -> eval:eval(nullability(), squirrel@internal@error:error(), context()))
) -> eval:eval(nullability(), squirrel@internal@error:error(), context()).
with_cached_column(Table_oid, Column, Do) ->
eval:from(
fun(Context) ->
Key = {Table_oid, Column},
case gleam_stdlib:map_get(erlang:element(4, Context), Key) of
{ok, Type_} ->
{Context, {ok, Type_}};
{error, _} ->
case eval:step(Do(), Context) of
{_, {error, _}} = Result ->
Result;
{{context, _, _, Column_nullability} = Context@1,
{ok, Type_@1}} ->
Column_nullability@1 = gleam@dict:insert(
Column_nullability,
Key,
Type_@1
),
New_context = erlang:setelement(
4,
Context@1,
Column_nullability@1
),
{New_context, {ok, Type_@1}}
end
end
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1213).
-spec unsupported_type_error(squirrel@internal@query:untyped_query(), binary()) -> squirrel@internal@error:error().
unsupported_type_error(Query, Type_) ->
{untyped_query, File, Starting_line, Name, _, Content} = Query,
{query_has_unsupported_type,
File,
squirrel@internal@gleam:value_identifier_to_string(Name),
Content,
Starting_line,
Type_}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1224).
-spec invalid_enum_error(
squirrel@internal@query:untyped_query(),
binary(),
squirrel@internal@error:enum_error()
) -> squirrel@internal@error:error().
invalid_enum_error(Query, Enum_name, Reason) ->
{untyped_query, File, Starting_line, _, _, Content} = Query,
{query_has_invalid_enum, File, Content, Starting_line, Enum_name, Reason}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 241).
-spec pg_to_gleam_type(squirrel@internal@query:untyped_query(), pg_type()) -> {ok,
squirrel@internal@gleam:type()} |
{error, squirrel@internal@error:error()}.
pg_to_gleam_type(Query, Type_) ->
case Type_ of
{p_array, Inner} ->
_pipe = pg_to_gleam_type(Query, Inner),
gleam@result:map(_pipe, fun(Field@0) -> {list, Field@0} end);
{p_option, Inner@1} ->
_pipe@1 = pg_to_gleam_type(Query, Inner@1),
gleam@result:map(_pipe@1, fun(Field@0) -> {option, Field@0} end);
{p_base, Name} ->
case Name of
<<"bool"/utf8>> ->
{ok, bool};
<<"text"/utf8>> ->
{ok, string};
<<"char"/utf8>> ->
{ok, string};
<<"bpchar"/utf8>> ->
{ok, string};
<<"varchar"/utf8>> ->
{ok, string};
<<"float4"/utf8>> ->
{ok, float};
<<"float8"/utf8>> ->
{ok, float};
<<"numeric"/utf8>> ->
{ok, float};
<<"int2"/utf8>> ->
{ok, int};
<<"int4"/utf8>> ->
{ok, int};
<<"int8"/utf8>> ->
{ok, int};
<<"json"/utf8>> ->
{ok, json};
<<"jsonb"/utf8>> ->
{ok, json};
<<"uuid"/utf8>> ->
{ok, uuid};
<<"bytea"/utf8>> ->
{ok, bit_array};
<<"date"/utf8>> ->
{ok, date};
<<"timestamp"/utf8>> ->
{ok, timestamp};
_ ->
{error, unsupported_type_error(Query, Name)}
end;
{p_enum, Name@1, Variants} ->
_pipe@2 = squirrel@internal@gleam:try_make_enum(Name@1, Variants),
gleam@result:map_error(
_pipe@2,
fun(_capture) -> invalid_enum_error(Query, Name@1, _capture) end
)
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1229).
-spec cannot_parse_error(
squirrel@internal@query:untyped_query(),
gleam@option:option(binary()),
gleam@option:option(binary()),
gleam@option:option(binary()),
gleam@option:option(squirrel@internal@error:pointer())
) -> squirrel@internal@error:error().
cannot_parse_error(Query, Error_code, Hint, Additional_error_message, Pointer) ->
{untyped_query, File, Starting_line, Name, _, Content} = Query,
{cannot_parse_query,
File,
squirrel@internal@gleam:value_identifier_to_string(Name),
Content,
Starting_line,
Error_code,
Pointer,
Additional_error_message,
Hint}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 594).
-spec error_fields_to_parse_error(
squirrel@internal@query:untyped_query(),
gleam@set:set(squirrel@internal@database@postgres_protocol:error_or_notice_field())
) -> squirrel@internal@error:error().
error_fields_to_parse_error(Query, Errors) ->
{Error_code, Message@2, Position@3, Hint@2} = (gleam@set:fold(
Errors,
{none, none, none, none},
fun(Acc, Error_field) ->
{Code, Message, Position, Hint} = Acc,
case Error_field of
{code, Code@1} ->
{{some, Code@1}, Message, Position, Hint};
{message, Message@1} ->
{Code, {some, Message@1}, Position, Hint};
{hint, Hint@1} ->
{Code, Message, Position, {some, Hint@1}};
{position, Position@1} ->
case gleam_stdlib:parse_int(Position@1) of
{ok, Position@2} ->
{Code, Message, {some, Position@2}, Hint};
{error, _} ->
Acc
end;
_ ->
Acc
end
end
)),
{Pointer, Additional_error_message} = case {Message@2, Position@3} of
{{some, Message@3}, {some, Position@4}} ->
{{some, {pointer, {byte_index, Position@4}, Message@3}}, none};
{{some, Message@4}, none} ->
{none, {some, Message@4}};
{_, _} ->
{none, none}
end,
cannot_parse_error(
Query,
Error_code,
Hint@2,
Additional_error_message,
Pointer
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 522).
-spec parameters_and_returns(squirrel@internal@query:untyped_query()) -> eval:eval({list(integer()),
list(squirrel@internal@database@postgres_protocol:row_description_field())}, squirrel@internal@error:error(), context()).
parameters_and_returns(Query) ->
eval:'try'(
send_all(
[{fe_parse, <<""/utf8>>, erlang:element(6, Query), []},
{fe_describe, prepared_statement, <<""/utf8>>},
fe_sync]
),
fun(_) ->
Cannot_describe = fun(Expected, Got) ->
{pg_cannot_describe_query,
erlang:element(2, Query),
squirrel@internal@gleam:value_identifier_to_string(
erlang:element(4, Query)
),
Expected,
Got}
end,
eval:'try'('receive'(), fun(Msg) -> case Msg of
{be_error_response, Errors} ->
eval:'try'('receive'(), fun(Msg@1) -> case Msg@1 of
{be_ready_for_query, _} ->
eval:throw(
error_fields_to_parse_error(
Query,
Errors
)
);
_ ->
unexpected_message(
Cannot_describe,
<<"BeReadyForQuery(_)"/utf8>>,
Msg@1
)
end end);
be_parse_complete ->
eval:'try'(
'receive'(),
fun(Msg@2) -> eval:'try'(case Msg@2 of
{be_parameter_description,
Parameters} ->
eval:return(Parameters);
be_no_data ->
eval:return([]);
_ ->
unexpected_message(
Cannot_describe,
<<"ParameterDescription"/utf8>>,
Msg@2
)
end, fun(Parameters@1) ->
eval:'try'(
'receive'(),
fun(Msg@3) ->
eval:'try'(case Msg@3 of
{be_row_descriptions,
Rows} ->
eval:return(
Rows
);
be_no_data ->
eval:return([]);
_ ->
unexpected_message(
Cannot_describe,
<<"RowDescriptions"/utf8>>,
Msg@3
)
end, fun(Rows@1) ->
eval:'try'(
'receive'(),
fun(Msg@4) ->
eval:'try'(
case Msg@4 of
{be_ready_for_query,
_} ->
eval:return(
nil
);
_ ->
unexpected_message(
Cannot_describe,
<<"ReadyForQuery"/utf8>>,
Msg@4
)
end,
fun(_) ->
eval:return(
{Parameters@1,
Rows@1}
)
end
)
end
)
end)
end
)
end) end
);
_ ->
unexpected_message(
Cannot_describe,
<<"ParseComplete or ErrorResponse"/utf8>>,
Msg
)
end end)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1015).
-spec expect_parse_complete(
squirrel@internal@database@postgres_protocol:backend_message(),
squirrel@internal@query:untyped_query()
) -> eval:eval(nil, squirrel@internal@error:error(), context()).
expect_parse_complete(Msg, Query) ->
case Msg of
be_parse_complete ->
eval:return(nil);
{be_error_response, Fields} ->
{untyped_query, File, _, Name, _, _} = Query,
Query_name = squirrel@internal@gleam:value_identifier_to_string(
Name
),
Unexpected = fun(Expected, Got) ->
{pg_cannot_describe_query, File, Query_name, Expected, Got}
end,
Parse_error = error_fields_to_parse_error(Query, Fields),
expect_ready_for_query_then_throw(Parse_error, Unexpected);
_ ->
erlang:error(#{gleam_error => panic,
message => gleam@string:inspect(Msg),
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"expect_parse_complete"/utf8>>,
line => 1029})
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 933).
-spec run_query(
squirrel@internal@query:untyped_query(),
list(squirrel@internal@database@postgres_protocol:parameter_value()),
list(integer())
) -> eval:eval(list(list(bitstring())), squirrel@internal@error:error(), context()).
run_query(Query, Parameters, Parameters_object_ids) ->
eval:'try'(
send_all(
[{fe_parse,
<<""/utf8>>,
erlang:element(6, Query),
Parameters_object_ids},
{fe_bind,
<<""/utf8>>,
<<""/utf8>>,
{format_all, binary},
Parameters,
{format_all, binary}},
{fe_execute, <<""/utf8>>, 0},
{fe_close, prepared_statement, <<""/utf8>>},
{fe_close, portal, <<""/utf8>>},
fe_sync]
),
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg) ->
eval:'try'(
expect_parse_complete(Msg, Query),
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg@1) ->
be_bind_complete = case Msg@1 of
be_bind_complete -> Msg@1;
_assert_fail ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_query"/utf8>>,
line => 978}
)
end,
eval:'try'(
accumulate_data_rows_until_command_complete(
erlang:element(2, Query)
),
fun(Res) ->
eval:'try'(
'receive'(),
fun(Msg@2) ->
be_close_complete = case Msg@2 of
be_close_complete -> Msg@2;
_assert_fail@1 ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@1,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_query"/utf8>>,
line => 981}
)
end,
eval:'try'(
'receive'(),
fun(Msg@3) ->
be_close_complete = case Msg@3 of
be_close_complete -> Msg@3;
_assert_fail@2 ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@2,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_query"/utf8>>,
line => 983}
)
end,
eval:'try'(
'receive'(),
fun(Msg@4) ->
{be_ready_for_query,
_} = case Msg@4 of
{be_ready_for_query,
_} -> Msg@4;
_assert_fail@3 ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@3,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_query"/utf8>>,
line => 985}
)
end,
eval:return(
Res
)
end
)
end
)
end
)
end
)
end
)
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 693).
-spec resolve_enum_type(binary(), integer()) -> eval:eval(pg_type(), squirrel@internal@error:error(), context()).
resolve_enum_type(Name, Oid) ->
Params = [{parameter, <<Oid:32>>}],
eval:'try'(
run_query(find_enum_variants_query(), Params, [23]),
fun(Rows) ->
Variants = gleam@list:map(
Rows,
fun(Row) ->
[Variant] = case Row of
[_] -> Row;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"resolve_enum_type"/utf8>>,
line => 700})
end,
_assert_subject = gleam@bit_array:to_string(Variant),
{ok, Variant@1} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail@1 ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@1,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"resolve_enum_type"/utf8>>,
line => 701})
end,
Variant@1
end
),
eval:return({p_enum, Name, Variants})
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1249).
-spec invalid_column_error(
squirrel@internal@query:untyped_query(),
binary(),
squirrel@internal@error:value_identifier_error()
) -> squirrel@internal@error:error().
invalid_column_error(Query, Column_name, Reason) ->
{untyped_query, File, Starting_line, _, _, Content} = Query,
{query_has_invalid_column,
File,
Column_name,
begin
_pipe = squirrel@internal@gleam:similar_value_identifier_string(
Column_name
),
gleam@option:from_result(_pipe)
end,
Content,
Starting_line,
Reason}.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1289).
-spec adjust_parse_error_for_explain(squirrel@internal@error:error()) -> squirrel@internal@error:error().
adjust_parse_error_for_explain(Error) ->
case Error of
{cannot_parse_query, _, _, _, _, _, Pointer, _, _} ->
Pointer@1 = case Pointer of
{some, {pointer, {byte_index, Index}, Message}} ->
{some, {pointer, {byte_index, Index - 45}, Message}};
_ ->
Pointer
end,
case Pointer@1 of
{some, {pointer, _, Message@1}} ->
Message@2 = string:lowercase(Message@1),
Is_explain_error = gleam_stdlib:contains_string(
Message@2,
<<"explain"/utf8>>
)
andalso gleam_stdlib:contains_string(
Message@2,
<<"generic_plan"/utf8>>
),
case Is_explain_error of
true ->
postgres_version_is_too_old;
false ->
erlang:setelement(7, Error, Pointer@1)
end;
_ ->
erlang:setelement(7, Error, Pointer@1)
end;
_ ->
Error
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1033).
-spec expect_query_plan_row_description(
squirrel@internal@database@postgres_protocol:backend_message(),
squirrel@internal@query:untyped_query()
) -> eval:eval(nil, squirrel@internal@error:error(), context()).
expect_query_plan_row_description(Msg, Query) ->
case Msg of
{be_row_descriptions,
[{row_description_field, <<"QUERY PLAN"/utf8>>, _, _, _, _, _, _}]} ->
eval:return(nil);
{be_error_response, Fields} ->
{untyped_query, File, _, Name, _, _} = Query,
Query_name = squirrel@internal@gleam:value_identifier_to_string(
Name
),
Unexpected = fun(Expected, Got) ->
{pg_cannot_explain_query, File, Query_name, Expected, Got}
end,
Parse_error = begin
_pipe = error_fields_to_parse_error(Query, Fields),
adjust_parse_error_for_explain(_pipe)
end,
expect_ready_for_query_then_throw(Parse_error, Unexpected);
_ ->
erlang:error(#{gleam_error => panic,
message => gleam@string:inspect(Msg),
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"expect_query_plan_row_description"/utf8>>,
line => 1055})
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 743).
-spec run_explain_query(squirrel@internal@query:untyped_query()) -> eval:eval(bitstring(), squirrel@internal@error:error(), context()).
run_explain_query(Query) ->
Explain_query = <<"explain (format json, verbose, generic_plan) "/utf8,
(erlang:element(6, Query))/binary>>,
eval:'try'(
send_all([{fe_query, Explain_query}]),
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg) ->
eval:'try'(
expect_query_plan_row_description(Msg, Query),
fun(_) ->
eval:'try'(
'receive'(),
fun(Msg@1) ->
{be_message_data_row, [Query_plan]} = case Msg@1 of
{be_message_data_row, [_]} -> Msg@1;
_assert_fail ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_explain_query"/utf8>>,
line => 752}
)
end,
eval:'try'(
'receive'(),
fun(Msg@2) ->
{be_command_complete, _, _} = case Msg@2 of
{be_command_complete, _, _} -> Msg@2;
_assert_fail@1 ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@1,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_explain_query"/utf8>>,
line => 754}
)
end,
eval:'try'(
'receive'(),
fun(Msg@3) ->
{be_ready_for_query, _} = case Msg@3 of
{be_ready_for_query, _} -> Msg@3;
_assert_fail@2 ->
erlang:error(
#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@2,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"run_explain_query"/utf8>>,
line => 756}
)
end,
eval:return(Query_plan)
end
)
end
)
end
)
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1338).
-spec join_type_decoder() -> decode@zero:decoder(gleam@option:option(join_type())).
join_type_decoder() ->
decode@zero:then(
{decoder, fun decode@zero:decode_string/1},
fun(Data) -> case Data of
<<"Full"/utf8>> ->
decode@zero:success({some, full_join});
<<"Left"/utf8>> ->
decode@zero:success({some, left_join});
<<"Right"/utf8>> ->
decode@zero:success({some, right_join});
<<"Inner"/utf8>> ->
decode@zero:success({some, inner_join});
_ ->
decode@zero:failure(none, <<"a join type"/utf8>>)
end end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1331).
-spec plan_decoder() -> decode@zero:decoder(plan()).
plan_decoder() ->
decode@zero:optional_field(
<<"Join Type"/utf8>>,
none,
join_type_decoder(),
fun(Join_type) ->
decode@zero:optional_field(
<<"Output"/utf8>>,
[],
decode@zero:list({decoder, fun decode@zero:decode_string/1}),
fun(Output) ->
decode@zero:optional_field(
<<"Plans"/utf8>>,
[],
decode@zero:list(plan_decoder()),
fun(Plans) ->
decode@zero:success(
{plan, Join_type, Output, Plans}
)
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1327).
-spec json_plans_decoder() -> decode@zero:decoder(list(plan())).
json_plans_decoder() ->
decode@zero:list(decode@zero:at([<<"Plan"/utf8>>], plan_decoder())).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 711).
-spec query_plan(squirrel@internal@query:untyped_query()) -> eval:eval(plan(), squirrel@internal@error:error(), context()).
query_plan(Query) ->
eval:'try'(
run_explain_query(Query),
fun(Plan) ->
case gleam@json:decode_bits(
Plan,
fun(_capture) ->
decode@zero:run(_capture, json_plans_decoder())
end
) of
{ok, [Plan@1 | _]} ->
eval:return(Plan@1);
{ok, []} ->
erlang:error(#{gleam_error => panic,
message => <<"unreachable: no query plan"/utf8>>,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"query_plan"/utf8>>,
line => 718});
{error, Reason} ->
eval:throw(
{cannot_parse_plan_for_query,
erlang:element(2, Query),
Reason}
)
end
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 1354).
-spec bit_array_to_bool(bitstring()) -> boolean().
bit_array_to_bool(Bit_array) ->
case Bit_array of
<<0, Rest/bitstring>> ->
bit_array_to_bool(Rest);
<<>> ->
false;
_ ->
true
end.
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 657).
-spec find_gleam_type(squirrel@internal@query:untyped_query(), integer()) -> eval:eval(squirrel@internal@gleam:type(), squirrel@internal@error:error(), context()).
find_gleam_type(Query, Oid) ->
with_cached_gleam_type(
Oid,
fun() ->
Params = [{parameter, <<Oid:32>>}],
eval:'try'(
run_query(find_postgres_type_query(), Params, [23]),
fun(Res) ->
[[Name, Is_array, Kind]] = case Res of
[[_, _, _]] -> Res;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_gleam_type"/utf8>>,
line => 671})
end,
_assert_subject = gleam@bit_array:to_string(Name),
{ok, Name@1} = case _assert_subject of
{ok, _} -> _assert_subject;
_assert_fail@1 ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@1,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_gleam_type"/utf8>>,
line => 672})
end,
_assert_subject@1 = gleam@bit_array:to_string(Kind),
{ok, Kind@1} = case _assert_subject@1 of
{ok, _} -> _assert_subject@1;
_assert_fail@2 ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail@2,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"find_gleam_type"/utf8>>,
line => 673})
end,
eval:'try'(case Kind@1 of
<<"e"/utf8>> ->
resolve_enum_type(Name@1, Oid);
_ ->
eval:return({p_base, Name@1})
end, fun(Type_) ->
Type_@1 = case bit_array_to_bool(Is_array) of
true ->
{p_array, Type_};
false ->
Type_
end,
_pipe = pg_to_gleam_type(Query, Type_@1),
eval:from_result(_pipe)
end)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 643).
-spec resolve_parameters(
squirrel@internal@query:untyped_query(),
list(integer())
) -> eval:eval(list(squirrel@internal@gleam:type()), squirrel@internal@error:error(), context()).
resolve_parameters(Query, Parameters) ->
squirrel@internal@eval_extra:try_map(
Parameters,
fun(Oid) -> find_gleam_type(Query, Oid) end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 891).
-spec column_nullability(integer(), integer()) -> eval:eval(nullability(), squirrel@internal@error:error(), context()).
column_nullability(Table, Column) ->
with_cached_column(
Table,
Column,
fun() ->
gleam@bool:guard(
Table =:= 0,
eval:return(not_nullable),
fun() ->
Params = [{parameter, <<Table:32>>},
{parameter, <<Column:32>>}],
Run_query = run_query(
find_column_nullability_query(),
Params,
[23, 23]
),
eval:'try'(
Run_query,
fun(Res) ->
[[Has_non_null_constraint]] = case Res of
[[_]] -> Res;
_assert_fail ->
erlang:error(#{gleam_error => let_assert,
message => <<"Pattern match failed, no pattern matched the value."/utf8>>,
value => _assert_fail,
module => <<"squirrel/internal/database/postgres"/utf8>>,
function => <<"column_nullability"/utf8>>,
line => 909})
end,
case bit_array_to_bool(Has_non_null_constraint) of
true ->
eval:return(not_nullable);
false ->
eval:return(nullable)
end
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 839).
-spec resolve_returns(
squirrel@internal@query:untyped_query(),
list(squirrel@internal@database@postgres_protocol:row_description_field()),
gleam@set:set(integer())
) -> eval:eval(list(squirrel@internal@gleam:field()), squirrel@internal@error:error(), context()).
resolve_returns(Query, Returns, Nullables) ->
squirrel@internal@eval_extra:try_index_map(
Returns,
fun(Column, I) ->
{row_description_field, Name, Table, Column@1, Type_oid, _, _, _} = Column,
eval:'try'(
find_gleam_type(Query, Type_oid),
fun(Type_) ->
Ends_with_exclamation_mark = gleam_stdlib:string_ends_with(
Name,
<<"!"/utf8>>
),
Ends_with_question_mark = gleam_stdlib:string_ends_with(
Name,
<<"?"/utf8>>
),
eval:'try'(case Ends_with_exclamation_mark of
true ->
eval:return(not_nullable);
false ->
case Ends_with_question_mark of
true ->
eval:return(nullable);
false ->
case gleam@set:contains(Nullables, I) of
true ->
eval:return(nullable);
false ->
column_nullability(
Table,
Column@1
)
end
end
end, fun(Nullability) ->
Type_@1 = case Nullability of
nullable ->
{option, Type_};
not_nullable ->
Type_
end,
Try_convert_name = begin
_pipe = case Ends_with_exclamation_mark orelse Ends_with_question_mark of
true ->
gleam@string:drop_end(Name, 1);
false ->
Name
end,
_pipe@1 = squirrel@internal@gleam:value_identifier(
_pipe
),
gleam@result:map_error(
_pipe@1,
fun(_capture) ->
invalid_column_error(
Query,
Name,
_capture
)
end
)
end,
eval:'try'(
eval:from_result(Try_convert_name),
fun(Name@1) ->
Field = {field, Name@1, Type_@1},
eval:return(Field)
end
)
end)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 492).
-spec infer_types(squirrel@internal@query:untyped_query()) -> eval:eval(squirrel@internal@query:typed_query(), squirrel@internal@error:error(), context()).
infer_types(Query) ->
eval:'try'(
parameters_and_returns(Query),
fun(_use0) ->
{Parameters, Returns} = _use0,
eval:'try'(
resolve_parameters(Query, Parameters),
fun(Parameters@1) ->
eval:'try'(
query_plan(Query),
fun(Plan) ->
Nullables = nullables_from_plan(Plan),
eval:'try'(
resolve_returns(Query, Returns, Nullables),
fun(Returns@1) -> _pipe = Query,
_pipe@1 = squirrel@internal@query:add_types(
_pipe,
Parameters@1,
Returns@1
),
eval:from_result(_pipe@1) end
)
end
)
end
)
end
).
-file("/home/runner/work/squirrel/squirrel/src/squirrel/internal/database/postgres.gleam", 284).
-spec main(list(squirrel@internal@query:untyped_query()), connection_options()) -> {ok,
{list(squirrel@internal@query:typed_query()),
list(squirrel@internal@error:error())}} |
{error, squirrel@internal@error:error()}.
main(Queries, Connection) ->
gleam@result:'try'(
begin
_pipe = squirrel@internal@database@postgres_protocol:connect(
erlang:element(2, Connection),
erlang:element(3, Connection),
erlang:element(7, Connection)
),
gleam@result:map_error(
_pipe,
fun(_capture) ->
{pg_cannot_establish_tcp_connection,
erlang:element(2, Connection),
erlang:element(3, Connection),
_capture}
end
)
end,
fun(Db) ->
Context = {context, Db, maps:new(), maps:new()},
{Context@1, Connection@1} = eval:step(
authenticate(Connection),
Context
),
case Connection@1 of
{error, Error} ->
{error, Error};
{ok, _} ->
_pipe@1 = gleam@list:map(Queries, fun infer_types/1),
_pipe@2 = squirrel@internal@eval_extra:run_all(
_pipe@1,
Context@1
),
_pipe@3 = gleam@result:partition(_pipe@2),
{ok, _pipe@3}
end
end
).