src/gleeps@stdlib@set.erl

-module(gleeps@stdlib@set).
-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch, inline]).
-define(FILEPATH, "src/gleeps/stdlib/set.gleam").
-export([new/0, size/1, is_empty/1, insert/2, contains/2, delete/2, to_list/1, from_list/1, fold/3, filter/2, map/2, drop/2, take/2, union/2, intersection/2, difference/2, is_subset/2, is_disjoint/2, symmetric_difference/2, each/2]).
-export_type([set/1]).

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

-opaque set(CVR) :: {set, gleeps@stdlib@dict:dict(CVR, list(nil))}.

-file("src/gleeps/stdlib/set.gleam", 32).
?DOC(" Creates a new empty set.\n").
-spec new() -> set(any()).
new() ->
    {set, maps:new()}.

-file("src/gleeps/stdlib/set.gleam", 50).
?DOC(
    " Gets the number of members in a set.\n"
    "\n"
    " This function runs in constant time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert new()\n"
    "   |> insert(1)\n"
    "   |> insert(2)\n"
    "   |> size\n"
    "   == 2\n"
    " ```\n"
).
-spec size(set(any())) -> integer().
size(Set) ->
    maps:size(erlang:element(2, Set)).

-file("src/gleeps/stdlib/set.gleam", 66).
?DOC(
    " Determines whether or not the set is empty.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert new() |> is_empty\n"
    " ```\n"
    "\n"
    " ```gleam\n"
    " assert !{ new() |> insert(1) |> is_empty }\n"
    " ```\n"
).
-spec is_empty(set(any())) -> boolean().
is_empty(Set) ->
    Set =:= new().

-file("src/gleeps/stdlib/set.gleam", 84).
?DOC(
    " Inserts a member into the set.\n"
    "\n"
    " This function runs in logarithmic time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert new()\n"
    "   |> insert(1)\n"
    "   |> insert(2)\n"
    "   |> size\n"
    "   == 2\n"
    " ```\n"
).
-spec insert(set(CVZ), CVZ) -> set(CVZ).
insert(Set, Member) ->
    {set, gleeps@stdlib@dict:insert(erlang:element(2, Set), Member, [])}.

-file("src/gleeps/stdlib/set.gleam", 108).
?DOC(
    " Checks whether a set contains a given member.\n"
    "\n"
    " This function runs in logarithmic time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert new()\n"
    "   |> insert(2)\n"
    "   |> contains(2)\n"
    " ```\n"
    "\n"
    " ```gleam\n"
    " assert !{\n"
    "   new()\n"
    "   |> insert(2)\n"
    "   |> contains(1)\n"
    " }\n"
    " ```\n"
).
-spec contains(set(CWC), CWC) -> boolean().
contains(Set, Member) ->
    _pipe = erlang:element(2, Set),
    _pipe@1 = gleam_stdlib:map_get(_pipe, Member),
    gleeps@stdlib@result:is_ok(_pipe@1).

-file("src/gleeps/stdlib/set.gleam", 130).
?DOC(
    " Removes a member from a set. If the set does not contain the member then\n"
    " the set is returned unchanged.\n"
    "\n"
    " This function runs in logarithmic time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert !{\n"
    "   new()\n"
    "   |> insert(2)\n"
    "   |> delete(2)\n"
    "   |> contains(2)\n"
    " }\n"
    " ```\n"
).
-spec delete(set(CWE), CWE) -> set(CWE).
delete(Set, Member) ->
    {set, gleeps@stdlib@dict:delete(erlang:element(2, Set), Member)}.

-file("src/gleeps/stdlib/set.gleam", 147).
?DOC(
    " Converts the set into a list of the contained members.\n"
    "\n"
    " The list has no specific ordering, any unintentional ordering may change in\n"
    " future versions of Gleam or Erlang.\n"
    "\n"
    " This function runs in linear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert new() |> insert(2) |> to_list == [2]\n"
    " ```\n"
).
-spec to_list(set(CWH)) -> list(CWH).
to_list(Set) ->
    maps:keys(erlang:element(2, Set)).

-file("src/gleeps/stdlib/set.gleam", 168).
?DOC(
    " Creates a new set of the members in a given list.\n"
    "\n"
    " This function runs in loglinear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " import gleam/int\n"
    " import gleam/list\n"
    "\n"
    " assert [1, 1, 2, 4, 3, 2]\n"
    "   |> from_list\n"
    "   |> to_list\n"
    "   |> list.sort(by: int.compare)\n"
    "   == [1, 2, 3, 4]\n"
    " ```\n"
).
-spec from_list(list(CWK)) -> set(CWK).
from_list(Members) ->
    Dict = gleeps@stdlib@list:fold(
        Members,
        maps:new(),
        fun(M, K) -> gleeps@stdlib@dict:insert(M, K, []) end
    ),
    {set, Dict}.

-file("src/gleeps/stdlib/set.gleam", 191).
?DOC(
    " Combines all entries into a single value by calling a given function on each\n"
    " one.\n"
    "\n"
    " Sets are not ordered so the values are not returned in any specific order.\n"
    " Do not write code that relies on the order entries are used by this\n"
    " function as it may change in later versions of Gleam or Erlang.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert from_list([1, 3, 9])\n"
    "   |> fold(0, fn(accumulator, member) { accumulator + member })\n"
    "   == 13\n"
    " ```\n"
).
-spec fold(set(CWN), CWP, fun((CWP, CWN) -> CWP)) -> CWP.
fold(Set, Initial, Reducer) ->
    gleeps@stdlib@dict:fold(
        erlang:element(2, Set),
        Initial,
        fun(A, K, _) -> Reducer(A, K) end
    ).

-file("src/gleeps/stdlib/set.gleam", 215).
?DOC(
    " Creates a new set from an existing set, minus any members that a given\n"
    " function returns `False` for.\n"
    "\n"
    " This function runs in loglinear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " import gleam/int\n"
    "\n"
    " assert from_list([1, 4, 6, 3, 675, 44, 67])\n"
    "   |> filter(keeping: int.is_even)\n"
    "   |> to_list\n"
    "   == [4, 6, 44]\n"
    " ```\n"
).
-spec filter(set(CWQ), fun((CWQ) -> boolean())) -> set(CWQ).
filter(Set, Predicate) ->
    {set,
        gleeps@stdlib@dict:filter(
            erlang:element(2, Set),
            fun(M, _) -> Predicate(M) end
        )}.

-file("src/gleeps/stdlib/set.gleam", 234).
?DOC(
    " Creates a new set from a given set with the result of applying the given\n"
    " function to each member.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert from_list([1, 2, 3, 4])\n"
    "   |> map(with: fn(x) { x * 2 })\n"
    "   |> to_list\n"
    "   == [2, 4, 6, 8]\n"
    " ```\n"
).
-spec map(set(CWT), fun((CWT) -> CWV)) -> set(CWV).
map(Set, Fun) ->
    fold(Set, new(), fun(Acc, Member) -> insert(Acc, Fun(Member)) end).

-file("src/gleeps/stdlib/set.gleam", 252).
?DOC(
    " Creates a new set from a given set with all the same entries except any\n"
    " entry found on the given list.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert from_list([1, 2, 3, 4])\n"
    "   |> drop([1, 3])\n"
    "   |> to_list\n"
    "   == [2, 4]\n"
    " ```\n"
).
-spec drop(set(CWX), list(CWX)) -> set(CWX).
drop(Set, Disallowed) ->
    gleeps@stdlib@list:fold(Disallowed, Set, fun delete/2).

-file("src/gleeps/stdlib/set.gleam", 273).
?DOC(
    " Creates a new set from a given set, only including any members which are in\n"
    " a given list.\n"
    "\n"
    " This function runs in loglinear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert from_list([1, 2, 3])\n"
    "   |> take([1, 3, 5])\n"
    "   |> to_list\n"
    "   == [1, 3]\n"
    " ```\n"
).
-spec take(set(CXB), list(CXB)) -> set(CXB).
take(Set, Desired) ->
    {set, gleeps@stdlib@dict:take(erlang:element(2, Set), Desired)}.

-file("src/gleeps/stdlib/set.gleam", 296).
-spec order(set(CXJ), set(CXJ)) -> {set(CXJ), set(CXJ)}.
order(First, Second) ->
    case maps:size(erlang:element(2, First)) > maps:size(
        erlang:element(2, Second)
    ) of
        true ->
            {First, Second};

        false ->
            {Second, First}
    end.

-file("src/gleeps/stdlib/set.gleam", 291).
?DOC(
    " Creates a new set that contains all members of both given sets.\n"
    "\n"
    " This function runs in loglinear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert union(from_list([1, 2]), from_list([2, 3])) |> to_list\n"
    "   == [1, 2, 3]\n"
    " ```\n"
).
-spec union(set(CXF), set(CXF)) -> set(CXF).
union(First, Second) ->
    {Larger, Smaller} = order(First, Second),
    fold(Smaller, Larger, fun insert/2).

-file("src/gleeps/stdlib/set.gleam", 317).
?DOC(
    " Creates a new set that contains members that are present in both given sets.\n"
    "\n"
    " This function runs in loglinear time.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert intersection(from_list([1, 2]), from_list([2, 3])) |> to_list\n"
    "   == [2]\n"
    " ```\n"
).
-spec intersection(set(CXO), set(CXO)) -> set(CXO).
intersection(First, Second) ->
    {Larger, Smaller} = order(First, Second),
    take(Larger, to_list(Smaller)).

-file("src/gleeps/stdlib/set.gleam", 335).
?DOC(
    " Creates a new set that contains members that are present in the first set\n"
    " but not the second.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert difference(from_list([1, 2]), from_list([2, 3, 4])) |> to_list\n"
    "   == [1]\n"
    " ```\n"
).
-spec difference(set(CXS), set(CXS)) -> set(CXS).
difference(First, Second) ->
    drop(First, to_list(Second)).

-file("src/gleeps/stdlib/set.gleam", 354).
?DOC(
    " Determines if a set is fully contained by another.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert is_subset(from_list([1]), from_list([1, 2]))\n"
    " ```\n"
    "\n"
    " ```gleam\n"
    " assert !is_subset(from_list([1, 2, 3]), from_list([3, 4, 5]))\n"
    " ```\n"
).
-spec is_subset(set(CXW), set(CXW)) -> boolean().
is_subset(First, Second) ->
    intersection(First, Second) =:= First.

-file("src/gleeps/stdlib/set.gleam", 370).
?DOC(
    " Determines if two sets contain no common members\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert is_disjoint(from_list([1, 2, 3]), from_list([4, 5, 6]))\n"
    " ```\n"
    "\n"
    " ```gleam\n"
    " assert !is_disjoint(from_list([1, 2, 3]), from_list([3, 4, 5]))\n"
    " ```\n"
).
-spec is_disjoint(set(CXZ), set(CXZ)) -> boolean().
is_disjoint(First, Second) ->
    intersection(First, Second) =:= new().

-file("src/gleeps/stdlib/set.gleam", 385).
?DOC(
    " Creates a new set that contains members that are present in either set, but\n"
    " not both.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " assert symmetric_difference(from_list([1, 2, 3]), from_list([3, 4]))\n"
    "   |> to_list\n"
    "   == [1, 2, 4]\n"
    " ```\n"
).
-spec symmetric_difference(set(CYC), set(CYC)) -> set(CYC).
symmetric_difference(First, Second) ->
    difference(union(First, Second), intersection(First, Second)).

-file("src/gleeps/stdlib/set.gleam", 414).
?DOC(
    " Calls a function for each member in a set, discarding the return\n"
    " value.\n"
    "\n"
    " Useful for producing a side effect for every item of a set.\n"
    "\n"
    " The order of elements in the iteration is an implementation detail that\n"
    " should not be relied upon.\n"
    "\n"
    " ## Examples\n"
    "\n"
    " ```gleam\n"
    " let set = from_list([\"apple\", \"banana\", \"cherry\"])\n"
    "\n"
    " assert each(set, io.println) == Nil\n"
    " // apple\n"
    " // banana\n"
    " // cherry\n"
    " ```\n"
).
-spec each(set(CYG), fun((CYG) -> any())) -> nil.
each(Set, Fun) ->
    fold(
        Set,
        nil,
        fun(Nil, Member) ->
            Fun(Member),
            Nil
        end
    ).