-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(FPI) -> gary:erlang_array(FPI).
create(Default) ->
    array:new([{default, Default}]).

-spec create_fixed_size(integer(), FPK) -> {ok, gary:erlang_array(FPK)} |
    {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(FPO), FPO) -> gary:erlang_array(FPO).
from_list(List, Default) ->
    array:from_list(List, Default).

-spec from_dict(gleam@dict:dict(integer(), FPR), FPR) -> {ok,
        gary:erlang_array(FPR)} |
    {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(FPX)) -> list(FPX).
to_list(Array) ->
    array:to_list(Array).

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

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

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

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

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

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

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

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

-spec sparse_fold_right(
    gary:erlang_array(FRM),
    FRO,
    fun((integer(), FRM, FRO) -> FRO)
) -> FRO.
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(FRX)) -> FRX.
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(FQL), integer(), FQL) -> {ok,
        gary:erlang_array(FQL)} |
    {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(FQQ), integer()) -> {ok, gary:erlang_array(FQQ)} |
    {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(FRP), integer()) -> {ok, FRP} |
    {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(FSB)) -> gary:erlang_array(FSB).
make_fixed(Array) ->
    array:fix(Array).

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

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

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