-module(gary@array).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).

-export([create/1, create_fixed_size/2, from_list/2, from_dict/2, to_list/1, to_dict/1, to_list_without_defaults/1, to_dict_without_defaults/1, map/2, sparse_map/2, fold/3, fold_right/3, sparse_fold/3, sparse_fold_right/3, get_size/1, get_count/1, get_default/1, is_fixed_size/1, set/3, drop/2, get/2, make_fixed/1, make_extensible/1, resize_to_fit/1, set_size/2]).
-export_type([array_error/0]).

-type array_error() :: index_out_of_range | bad_size.

-spec create(FPJ) -> gary:erlang_array(FPJ).
create(Default) ->
    array:new([{default, Default}]).

-spec create_fixed_size(integer(), FPL) -> {ok, gary:erlang_array(FPL)} |
    {error, array_error()}.
create_fixed_size(Size, Default) ->
    gleam@bool:guard(
        Size < 0,
        {error, bad_size},
        fun() -> {ok, array:new(Size, [{default, Default}])} end
    ).

-spec from_list(list(FPP), FPP) -> gary:erlang_array(FPP).
from_list(List, Default) ->
    array:from_list(List, Default).

-spec from_dict(gleam@dict:dict(integer(), FPS), FPS) -> {ok,
        gary:erlang_array(FPS)} |
    {error, array_error()}.
from_dict(Dict, Default) ->
    gleam@bool:guard(
        gleam@list:any(gleam@dict:keys(Dict), fun(K) -> K < 0 end),
        {error, index_out_of_range},
        fun() -> _pipe = Dict,
            _pipe@1 = maps:to_list(_pipe),
            _pipe@2 = array:from_orddict(_pipe@1, Default),
            {ok, _pipe@2} end
    ).

-spec to_list(gary:erlang_array(FPY)) -> list(FPY).
to_list(Array) ->
    array:to_list(Array).

-spec to_dict(gary:erlang_array(FQB)) -> gleam@dict:dict(integer(), FQB).
to_dict(Array) ->
    _pipe = Array,
    _pipe@1 = array:to_orddict(_pipe),
    maps:from_list(_pipe@1).

-spec to_list_without_defaults(gary:erlang_array(FQF)) -> list(FQF).
to_list_without_defaults(Array) ->
    array:sparse_to_list(Array).

-spec to_dict_without_defaults(gary:erlang_array(FQI)) -> gleam@dict:dict(integer(), FQI).
to_dict_without_defaults(Array) ->
    _pipe = Array,
    _pipe@1 = array:sparse_to_orddict(_pipe),
    maps:from_list(_pipe@1).

-spec map(gary:erlang_array(FQW), fun((integer(), FQW) -> FQY)) -> gary:erlang_array(FQY).
map(Array, Function) ->
    array:map(Function, Array).

-spec sparse_map(gary:erlang_array(FRA), fun((integer(), FRA) -> FRC)) -> gary:erlang_array(FRC).
sparse_map(Array, Function) ->
    array:sparse_map(Function, Array).

-spec fold(gary:erlang_array(FRE), FRG, fun((integer(), FRE, FRG) -> FRG)) -> FRG.
fold(Array, Initial, Function) ->
    array:foldl(Function, Initial, Array).

-spec fold_right(gary:erlang_array(FRH), FRJ, fun((integer(), FRH, FRJ) -> FRJ)) -> FRJ.
fold_right(Array, Initial, Function) ->
    array:foldr(Function, Initial, Array).

-spec sparse_fold(
    gary:erlang_array(FRK),
    FRM,
    fun((integer(), FRK, FRM) -> FRM)
) -> FRM.
sparse_fold(Array, Initial, Function) ->
    array:sparse_foldl(Function, Initial, Array).

-spec sparse_fold_right(
    gary:erlang_array(FRN),
    FRP,
    fun((integer(), FRN, FRP) -> FRP)
) -> FRP.
sparse_fold_right(Array, Initial, Function) ->
    array:sparse_foldr(Function, Initial, Array).

-spec get_size(gary:erlang_array(any())) -> integer().
get_size(Array) ->
    array:size(Array).

-spec get_count(gary:erlang_array(any())) -> integer().
get_count(Array) ->
    sparse_fold(Array, 0, fun(_, _, Acc) -> Acc + 1 end).

-spec get_default(gary:erlang_array(FRY)) -> FRY.
get_default(Array) ->
    array:default(Array).

-spec is_fixed_size(gary:erlang_array(any())) -> boolean().
is_fixed_size(Array) ->
    array:is_fix(Array).

-spec set(gary:erlang_array(FQM), integer(), FQM) -> {ok,
        gary:erlang_array(FQM)} |
    {error, array_error()}.
set(Array, Index, Item) ->
    gleam@bool:guard(
        Index < 0,
        {error, index_out_of_range},
        fun() ->
            gleam@bool:guard(
                is_fixed_size(Array) andalso (Index >= get_size(Array)),
                {error, index_out_of_range},
                fun() -> {ok, array:set(Index, Item, Array)} end
            )
        end
    ).

-spec drop(gary:erlang_array(FQR), integer()) -> {ok, gary:erlang_array(FQR)} |
    {error, array_error()}.
drop(Array, Index) ->
    gleam@bool:guard(
        Index < 0,
        {error, index_out_of_range},
        fun() ->
            gleam@bool:guard(
                is_fixed_size(Array) andalso (Index >= get_size(Array)),
                {error, index_out_of_range},
                fun() -> {ok, array:reset(Index, Array)} end
            )
        end
    ).

-spec get(gary:erlang_array(FRQ), integer()) -> {ok, FRQ} |
    {error, array_error()}.
get(Array, Index) ->
    gleam@bool:guard(
        Index < 0,
        {error, index_out_of_range},
        fun() ->
            gleam@bool:guard(
                is_fixed_size(Array) andalso (Index >= get_size(Array)),
                {error, index_out_of_range},
                fun() -> {ok, array:get(Index, Array)} end
            )
        end
    ).

-spec make_fixed(gary:erlang_array(FSC)) -> gary:erlang_array(FSC).
make_fixed(Array) ->
    array:fix(Array).

-spec make_extensible(gary:erlang_array(FSF)) -> gary:erlang_array(FSF).
make_extensible(Array) ->
    array:relax(Array).

-spec resize_to_fit(gary:erlang_array(FSI)) -> gary:erlang_array(FSI).
resize_to_fit(Array) ->
    array:resize(Array).

-spec set_size(gary:erlang_array(FSL), integer()) -> {ok,
        gary:erlang_array(FSL)} |
    {error, array_error()}.
set_size(Array, Size) ->
    gleam@bool:guard(
        Size < 0,
        {error, bad_size},
        fun() -> {ok, array:resize(Size, Array)} end
    ).