Initial add of client.go
This commit is contained in:
Bruce Marriner
parent
6f4049d383
commit
1e1ad5940b
1 changed files with 395 additions and 0 deletions
395
client.go
Normal file
395
client.go
Normal file
|
|
@ -0,0 +1,395 @@
|
|||
/******************************************************************************
|
||||
|
||||
Known API Commands:
|
||||
|
||||
Login - POST http://discordapp.com/api/auth/login
|
||||
Send Message - POST http://discordapp.com/api/channels/107877361818570752/messages
|
||||
|
||||
About Self - GET http://discordapp.com/api/users/@me
|
||||
Guild List - GET http://discordapp.com/api/users/90975935880241152/guilds
|
||||
Channel List - GET http://discordapp.com/api/guilds/107877361818570752/channels
|
||||
Get Messages - GET http://discordapp.com/api/channels/107877361818570752/messages
|
||||
Get PM Channels - GET http://discordapp.com/api/users/@me/channels
|
||||
Get Guild Members - GET http://discordapp.com/api/guilds/107877361818570752/members
|
||||
|
||||
|
||||
*/
|
||||
|
||||
package discordgo
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
var discordUrl = "http://discordapp.com/api"
|
||||
|
||||
type RestClient struct {
|
||||
Url string
|
||||
Session *Session
|
||||
client *http.Client
|
||||
Debug bool
|
||||
}
|
||||
|
||||
type Session struct {
|
||||
Id string
|
||||
Email string
|
||||
Password string
|
||||
Token string
|
||||
}
|
||||
|
||||
type Guild struct {
|
||||
Afk_timeout int
|
||||
Joined_at string
|
||||
// Afk_channel_id int `json:",string"`
|
||||
Id int `json:",string"`
|
||||
Icon string
|
||||
Name string
|
||||
// Roles []Role
|
||||
Region string
|
||||
//Embed_channel_id int `json:",string"`
|
||||
// Embed_channel_id string
|
||||
// Embed_enabled bool
|
||||
Owner_id int `json:",string"`
|
||||
}
|
||||
|
||||
type Role struct {
|
||||
Permissions int
|
||||
Id int `json:",string"`
|
||||
Name string
|
||||
}
|
||||
|
||||
type Channel struct {
|
||||
Guild_id int `json:",string"`
|
||||
Id int `json:",string"`
|
||||
Name string
|
||||
Last_message_id string
|
||||
Is_private string
|
||||
|
||||
// Permission_overwrites string
|
||||
// Position int `json:",string"`
|
||||
// Type string
|
||||
}
|
||||
|
||||
type Message struct {
|
||||
Attachments []Attachment
|
||||
Tts bool
|
||||
Embeds []Embed
|
||||
Timestamp string
|
||||
Mention_everyone bool
|
||||
Id int `json:",string"`
|
||||
Edited_timestamp string
|
||||
Author *Author
|
||||
Content string
|
||||
Channel_id int `json:",string"`
|
||||
Mentions []Mention
|
||||
}
|
||||
|
||||
type Mention struct {
|
||||
}
|
||||
|
||||
type Attachment struct {
|
||||
}
|
||||
|
||||
type Embed struct {
|
||||
}
|
||||
|
||||
type Author struct {
|
||||
Username string
|
||||
Discriminator int `json:",string"`
|
||||
Id int `json:",string"`
|
||||
Avatar string
|
||||
}
|
||||
|
||||
// Create takes an email and password then prepares a RestClient with the given data,
|
||||
// which is a simple object used for future requests.
|
||||
func Create(email string, password string) (restClient *RestClient, err error) {
|
||||
if len(email) < 3 {
|
||||
err = errors.New("email too short")
|
||||
return
|
||||
}
|
||||
if len(password) < 3 {
|
||||
err = errors.New("password too short")
|
||||
return
|
||||
}
|
||||
session := &Session{"0", email, password, ""}
|
||||
httpClient := &http.Client{Timeout: (20 * time.Second)}
|
||||
restClient = &RestClient{discordUrl, session, httpClient, false}
|
||||
restClient.Session.Token, err = requestToken(restClient)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
restClient.Session.Id, err = requestSelf(restClient)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// RequestToken asks the Rest server for a token by provided email/password
|
||||
func requestToken(restClient *RestClient) (token string, err error) {
|
||||
|
||||
if restClient == nil {
|
||||
err = errors.New("Empty restClient, Create() one first")
|
||||
return
|
||||
}
|
||||
|
||||
if restClient.Session == nil || len(restClient.Session.Email) == 0 || len(restClient.Session.Password) == 0 {
|
||||
err = errors.New("Empty restClient.Session data, Create() to set email/password")
|
||||
return
|
||||
}
|
||||
|
||||
var urlStr string = fmt.Sprintf("%s/%s", restClient.Url, "auth/login")
|
||||
req, err := http.NewRequest("POST", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(`{"email":"%s", "password":"%s"}`, restClient.Session.Email, restClient.Session.Password))))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
resp, err := restClient.client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, _ := ioutil.ReadAll(resp.Body)
|
||||
if resp.StatusCode != 200 {
|
||||
err = errors.New(fmt.Sprintf("StatusCode: %d, %s", resp.StatusCode, string(body)))
|
||||
return
|
||||
}
|
||||
session := &Session{}
|
||||
|
||||
err = json.Unmarshal(body, &session)
|
||||
token = session.Token
|
||||
return
|
||||
}
|
||||
|
||||
// Identify user himself
|
||||
func requestSelf(restClient *RestClient) (clientId string, err error) {
|
||||
|
||||
body, err := Request(restClient, fmt.Sprintf("%s/%s", restClient.Url, "users/@me"))
|
||||
session := &Session{} // what's this for?
|
||||
err = json.Unmarshal(body, &session)
|
||||
clientId = session.Id
|
||||
return
|
||||
}
|
||||
|
||||
func ListGuilds(restClient *RestClient) (guilds []Guild, err error) {
|
||||
|
||||
body, err := Request(restClient, fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("users/%s/guilds", restClient.Session.Id)))
|
||||
err = json.Unmarshal(body, &guilds)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func ListChannels(restClient *RestClient, guildId int) (channels []Channel, err error) {
|
||||
|
||||
body, err := Request(restClient, fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("guilds/%d/channels", guildId)))
|
||||
err = json.Unmarshal(body, &channels)
|
||||
|
||||
body, err = Request(restClient, fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("users/@me/channels", guildId)))
|
||||
err = json.Unmarshal(body, &channels)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func GetMessages(restClient *RestClient, channelId int, before int, limit int) (messages []Message, err error) {
|
||||
// var urlStr = fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("channels/%d/messages?limit=%d&after=%d", channelId,limit,before))
|
||||
|
||||
var urlStr = fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("channels/%d/messages", channelId))
|
||||
|
||||
if limit > 0 {
|
||||
urlStr = urlStr + fmt.Sprintf("?limit=%d", limit)
|
||||
} else {
|
||||
urlStr = urlStr + "?limit=1"
|
||||
}
|
||||
|
||||
if before > 0 {
|
||||
urlStr = urlStr + fmt.Sprintf("&after=%d", before)
|
||||
|
||||
}
|
||||
|
||||
body, err := Request(restClient, urlStr)
|
||||
err = json.Unmarshal(body, &messages)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func CreateChannelUser(restClient *RestClient, userId int) (channelId int, err error) {
|
||||
|
||||
var urlStr string = fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("users/@me/channels"))
|
||||
req, err := http.NewRequest("POST", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(`{"recipient_id":"%d"}`, userId))))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("authorization", restClient.Session.Token)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := restClient.client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
err = errors.New(fmt.Sprintf("StatusCode: %d, %s", resp.StatusCode, string(body)))
|
||||
return
|
||||
}
|
||||
|
||||
// something something if debug
|
||||
var prettyJSON bytes.Buffer
|
||||
error := json.Indent(&prettyJSON, body, "", "\t")
|
||||
if error != nil {
|
||||
fmt.Print("JSON parse error: ", error)
|
||||
return
|
||||
}
|
||||
fmt.Println(urlStr+" Response:\n", string(prettyJSON.Bytes()))
|
||||
|
||||
// err = json.Unmarshal(body, &responseMessage)
|
||||
return
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func SendMessage(restClient *RestClient, channelId int, message string) (responseMessage Message, err error) {
|
||||
var urlStr string = fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("channels/%d/messages", channelId))
|
||||
req, err := http.NewRequest("POST", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(`{"content":"%s"}`, message))))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("authorization", restClient.Session.Token)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := restClient.client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
err = errors.New(fmt.Sprintf("StatusCode: %d, %s", resp.StatusCode, string(body)))
|
||||
return
|
||||
}
|
||||
|
||||
// something something if debug
|
||||
var prettyJSON bytes.Buffer
|
||||
error := json.Indent(&prettyJSON, body, "", "\t")
|
||||
if error != nil {
|
||||
fmt.Print("JSON parse error: ", error)
|
||||
return
|
||||
}
|
||||
fmt.Println(urlStr+" Response:\n", string(prettyJSON.Bytes()))
|
||||
|
||||
err = json.Unmarshal(body, &responseMessage)
|
||||
return
|
||||
}
|
||||
|
||||
func Close(restClient *RestClient) (err error) {
|
||||
req, err := http.NewRequest("POST", fmt.Sprintf("%s/%s", restClient.Url, fmt.Sprintf("auth/logout")), bytes.NewBuffer([]byte(fmt.Sprintf(``))))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("authorization", restClient.Session.Token)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := restClient.client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 204 && resp.StatusCode != 200 {
|
||||
err = errors.New(fmt.Sprintf("StatusCode: %d, %s", resp.StatusCode, string(body)))
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func ReadLoop(restClient *RestClient, channelId int) {
|
||||
|
||||
var lastMessageId int = 0
|
||||
|
||||
var i int = 0
|
||||
|
||||
for i < 1000 {
|
||||
|
||||
messages, err := GetMessages(restClient, channelId, lastMessageId, 10)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
|
||||
var i int = len(messages) - 1
|
||||
// fmt.Println("loop ", i, " ", lastMessageId);
|
||||
|
||||
if i > -1 { // seems poorly wrote..
|
||||
|
||||
for i >= 0 {
|
||||
var message Message = messages[i]
|
||||
fmt.Println("\n", message.Id, ":", message.Timestamp, ":\n", message.Author.Username, " > ", message.Content)
|
||||
lastMessageId = message.Id
|
||||
i--
|
||||
}
|
||||
}
|
||||
|
||||
time.Sleep(2000 * time.Millisecond)
|
||||
i++
|
||||
}
|
||||
}
|
||||
|
||||
// Request makes a API GET Request. This is a general purpose function
|
||||
// and is used by all API functions. It is exposed currently so it can
|
||||
// also be used outside of this library.
|
||||
func Request(restClient *RestClient, urlStr string) (body []byte, err error) {
|
||||
|
||||
req, err := http.NewRequest("GET", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(``))))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
req.Header.Set("authorization", restClient.Session.Token)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := restClient.client.Do(req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
body, err = ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
err = errors.New(fmt.Sprintf("StatusCode: %d, %s", resp.StatusCode, string(body)))
|
||||
return
|
||||
}
|
||||
|
||||
if restClient.Debug {
|
||||
var prettyJSON bytes.Buffer
|
||||
error := json.Indent(&prettyJSON, body, "", "\t")
|
||||
if error != nil {
|
||||
fmt.Print("JSON parse error: ", error)
|
||||
return
|
||||
}
|
||||
fmt.Println(urlStr+" Response:\n", string(prettyJSON.Bytes()))
|
||||
}
|
||||
return
|
||||
}
|
||||
Loading…
Reference in a new issue