aboutsummaryrefslogtreecommitdiff
path: root/lib/models/guild.ml
blob: 73643276b5237682d174740a865ee95232d96ffd (plain) (blame)
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
module Make(Http : S.Http) = struct
    open Core
    open Async
    open Guild_t

    let ban_user ~id ?(reason="") ?(days=0) guild =
        Http.guild_ban_add guild.id id (`Assoc [
            ("delete-message-days", `Int days);
            ("reason", `String reason);
        ])

    let create_emoji ~name ~image guild =
        Http.create_emoji guild.id (`Assoc [
            ("name", `String name);
            ("image", `String image);
            ("roles", `List []);
        ])

    let create_role ~name ?colour ?permissions ?hoist ?mentionable guild =
        let payload = ("name", `String name) :: [] in
        let payload = match permissions with
        | Some p -> ("permissions", `Int p) :: payload
        | None -> payload
        in let payload = match colour with
        | Some c -> ("color", `Int c) :: payload
        | None -> payload
        in let payload = match hoist with
        | Some h -> ("hoist", `Bool h) :: payload
        | None -> payload
        in let payload = match mentionable with
        | Some m -> ("mentionable", `Bool m) :: payload
        | None -> payload
        in Http.guild_role_add guild.id (`Assoc payload)
    
    let create_channel ~mode ~name guild =
        let kind = match mode with
        | `Text -> 0
        | `Voice -> 2
        | `Category -> 4
        in Http.create_guild_channel guild.id (`Assoc [
            ("name", `String name);
            ("type", `Int kind);
        ])
    
    let delete guild = Http.delete_guild guild.id
    let get_ban ~id guild = Http.get_ban guild.id id
    let get_bans guild = Http.get_bans guild.id

    let get_channel ~id guild =
        match List.find ~f:(fun c -> c.id = id) guild.channels with
        | Some c -> return c
        | None -> Http.get_channel id >>| fun c ->
            c |> Yojson.Safe.to_string |> Channel_j.t_of_string
    
    let get_emoji ~id guild = Http.get_emoji guild.id id
    let get_invites guild = Http.get_guild_invites guild.id

    let get_member ~id guild =
        match List.find ~f:(fun m -> m.user.id = id) guild.members with
        | Some m -> return m
        | None -> Http.get_member guild.id id >>| fun m ->
             m |> Yojson.Safe.to_string |> Member_j.t_of_string

    let get_prune_count ~days guild = Http.guild_prune_count guild.id days

    let get_role ~id guild =
        let role = List.find ~f:(fun r -> r.id = id) guild.roles in
        Option.(role >>| fun role -> Role.wrap { role; id=guild.id; })
    
    let get_webhooks guild = Http.get_guild_webhooks guild.id

    let kick_user ~id ?reason guild =
        let payload = match reason with
        | Some r -> `Assoc [("reason", `String r)]
        | None -> `Null
        in Http.remove_member guild.id id payload

    let leave guild = Http.leave_guild guild.id
    let list_voice_regions guild = Http.get_guild_voice_regions guild.id
    let prune ~days guild = Http.guild_prune_start guild.id days
    let request_members guild = Http.get_members guild.id

    let set_afk_channel ~id guild = Http.edit_guild guild.id (`Assoc [
        ("afk_channel_id", `Int id);
    ])

    let set_afk_timeout ~timeout guild = Http.edit_guild guild.id (`Assoc [
        ("afk_timeout", `Int timeout);
    ])

    let set_name ~name guild = Http.edit_guild guild.id (`Assoc [
        ("name", `String name);
    ])

    let set_icon ~icon guild = Http.edit_guild guild.id (`Assoc [
        ("icon", `String icon);
    ])

    let unban_user ~id ?reason guild =
        let payload = match reason with
        | Some r -> `Assoc [("reason", `String r)]
        | None -> `Null
        in Http.guild_ban_remove guild.id id payload
end