lib/client.ml


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

open Async

module Make(T : S.Token) = struct
    include T

    module Http = Http.Make(T)
    module Sharder = Sharder.Make(Http)

    type t = {
        sharder: Sharder.t Ivar.t;
        handler: string Pipe.Writer.t;
        token: string;
    }

    let init ~handler () =
        {
            sharder = Ivar.create ();
            handler;
            token;
        }

    let start ?count client =
        Sharder.start ?count ()
        >>| fun sharder ->
        Ivar.fill_if_empty client.sharder sharder;
        client

    let set_status ~status client =
        Ivar.read client.sharder
        >>= fun sharder ->
        Sharder.set_status ~status sharder

    let set_status_with ~f client =
        Ivar.read client.sharder
        >>= fun sharder ->
        Sharder.set_status_with ~f sharder

    let request_guild_members ~guild ?query ?limit client =
        Ivar.read client.sharder
        >>= fun sharder ->
        Sharder.request_guild_members ~guild ?query ?limit sharder
end