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
|
package discord
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"net/http"
"github.com/gorilla/websocket"
)
const DISCORD_URl = "https://discord.com/api/v9/"
const USER_AGENT = "DiscordBot (https://jinx.rnrd.eu/, v0.0.0) Jinx"
type Discord struct {
Token string
}
func NewClient(token string) *Discord {
return &Discord{
Token: token,
}
}
func (d *Discord) Connect() error {
gatewayURL, err := d.gateway()
if err != nil {
return err
}
fmt.Printf("Gateway: %s\n", gatewayURL)
connectHeader := http.Header{}
conn, _, err := websocket.DefaultDialer.Dial(gatewayURL, connectHeader)
if err != nil {
return err
}
defer conn.Close()
messageType, message, err := conn.ReadMessage()
if err != nil {
return err
}
fmt.Printf("Type: %d, Message: %s\n", messageType, message)
return nil
}
func (d *Discord) gateway() (string, error) {
url := DISCORD_URl + "gateway"
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return "", err
}
req.Header.Set("Authorization", d.Token)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", USER_AGENT)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return "", err
}
defer resp.Body.Close()
var buf bytes.Buffer
_, err = buf.ReadFrom(resp.Body)
if err != nil {
return "", err
}
switch resp.StatusCode {
case 200:
default:
return "", errors.New("Gateway response status code: " + resp.Status)
}
body := struct {
URL string `json:"url"`
}{}
err = json.Unmarshal(buf.Bytes(), &body)
if err != nil {
return "", err
}
url = body.URL + "?v=9&encoding=json"
return url, nil
}
|
