blob: 9f149df5dc1d01a30668e503e9f83c8a1d934e27 (
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
|
open Core
open Async
module RouteMap = Map.Make(String)
type rl = {
limit: int;
remaining: int;
reset: int;
} [@@deriving sexp]
(* TODO improve route getting, use Date header *)
type t = ((rl, read_write) Mvar.t) RouteMap.t
let r_message_delete = Str.regexp "/channel/[0-9]+/messages/"
let r_emoji = Str.regexp "/channel/[0-9]+/messages/[0-9]+/reactions/[A-Za-z0-9_\\-]+/\\(@me|[0-9]+\\)"
let route_of_path meth path =
match meth with
| `Delete -> if Str.string_match r_message_delete path 0 then Str.matched_string path else path
| `Put -> if Str.string_match r_emoji path 0 then Str.matched_string path else path
| _ -> path
let rl_of_header h =
let module C = Cohttp.Header in
match C.get h "X-RateLimit-Limit", C.get h "X-RateLimit-Remaining", C.get h "X-RateLimit-Reset" with
| Some lim, Some rem, Some re ->
let limit = Int.of_string lim in
let remaining = Int.of_string rem in
let reset = Int.of_string re in
Some { limit; remaining; reset; }
| _ -> None
let default = { limit = 1; remaining = 1; reset = 0; }
let empty : t = RouteMap.empty
let update = RouteMap.update
let find = RouteMap.find
let find_exn m s = match find m s with
| Some r -> r
| None -> raise (Not_found_s (String.sexp_of_t s))
let get_rl meth path rl =
let route = route_of_path meth path in
match RouteMap.find rl route with
| Some r -> r, rl
| None ->
let data = Mvar.create () in
Mvar.set data default;
let rl = RouteMap.add_exn rl ~key:route ~data in
data, rl
|