blob: a42fd1633c9e22f2359c7bfeeb92d726b14d83ef (
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
|
open Async
open Core
open Disml
let check_command (msg:Message.t) =
let cmd, rest = match String.split ~on:' ' msg.content with
| hd::tl -> hd, tl
| [] -> "", []
in match cmd with
| "!ping" ->
Message.reply msg "Pong!" >>= begin fun msg ->
let msg = match msg with Ok m -> m | Error e -> Error.raise e in
let diff = Time.diff (Time.now ()) (Time.of_string msg.timestamp) in
Message.set_content msg (Printf.sprintf "Pong! `%d ms`" (Time.Span.to_ms diff |> Float.to_int))
end >>> ignore
| "spam" ->
let count = Option.((List.hd rest >>| Int.of_string) |> value ~default:0) in
List.range 0 count
|> List.iter ~f:(fun i -> Message.reply msg (string_of_int i) >>> ignore)
| "!list" ->
let count = Option.((List.hd rest >>| Int.of_string) |> value ~default:0) in
let list = List.range 0 count
|> List.sexp_of_t Int.sexp_of_t
|> Sexp.to_string_hum in
Message.reply msg list >>> ignore
| "!fold" ->
let count = Option.((List.hd rest >>| Int.of_string) |> value ~default:0) in
let list = List.range 0 count
|> List.fold ~init:0 ~f:(+)
|> Int.to_string in
Message.reply msg list >>> ignore
| _ -> ()
let main () =
let token = match Sys.getenv "DISCORD_TOKEN" with
| Some t -> t
| None -> failwith "No token in env"
in
Client.start token
>>> ignore
let _ =
Client.message_create := check_command;
Scheduler.go_main ~main ()
|