midas

-module (pg_client_native).
-export ([start_link/1, query/2]).

start_link(OptionsList) ->
  Options = maps:from_list(OptionsList),
  pgo_pool:start_link(default, Options).


query(String, GleamArgs) ->
  Args = [Value || {_, Value} <- GleamArgs],
  Return = pgo:query(String, Args),
  case Return of
    #{command := select, num_rows := Count, rows := PgRows} ->
      Rows = [map_row(PgRow) || PgRow <- PgRows],
      {ok, {select, Count, Rows}};
    #{command := insert, num_rows := Count, rows := PgRows} ->
      Rows = [map_row(PgRow) || PgRow <- PgRows],
      {ok, {insert, Count, Rows}}
  end.

map_row(PgRow) ->
  [map_element(PgElement) || PgElement <- erlang:tuple_to_list(PgRow)].

map_element(PgElement) ->
  case PgElement of
    E when is_binary(E) ->
      {pg_string, E};
    E when is_integer(E) ->
      {pg_int, E};
    E when is_boolean(E) ->
      {pg_bool, E};
    {Date, Time}  ->
      {pg_date_time, {Date, Time}}
  end.