123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257 |
- package discordgo
- import (
- "bytes"
- "encoding/json"
- "errors"
- "fmt"
- "io/ioutil"
- "net/http"
- "time"
- )
- func Request(session *Session, 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", session.Token)
- req.Header.Set("Content-Type", "application/json")
- client := &http.Client{Timeout: (20 * time.Second)}
- resp, err := 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 session.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
- }
- func Login(session *Session, email string, password string) (token string, err error) {
- var urlStr string = fmt.Sprintf("%s/%s", discordApi, "auth/login")
- req, err := http.NewRequest("POST", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(`{"email":"%s", "password":"%s"}`, email, password))))
- if err != nil {
- return
- }
- req.Header.Set("Content-Type", "application/json")
- client := &http.Client{Timeout: (20 * time.Second)}
- resp, err := 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
- }
- if session.Debug {
- var prettyJSON bytes.Buffer
- error := json.Indent(&prettyJSON, body, "", "\t")
- if error != nil {
- fmt.Print("JSON parse error: ", error)
- return
- }
- fmt.Println("requestToken Response:\n", string(prettyJSON.Bytes()))
- }
- temp := &Session{}
- err = json.Unmarshal(body, &temp)
- token = temp.Token
- return
- }
- func Users(session *Session, userId string) (user User, err error) {
- body, err := Request(session, fmt.Sprintf("%s/users/%s", discordApi, userId))
- err = json.Unmarshal(body, &user)
- return
- }
- func PrivateChannels(session *Session, userId string) (channels []Channel, err error) {
- body, err := Request(session, fmt.Sprintf("%s/%s", discordApi, fmt.Sprintf("users/%s/channels", userId)))
- err = json.Unmarshal(body, &channels)
- return
- }
- func Servers(session *Session, userId string) (servers []Server, err error) {
- body, err := Request(session, fmt.Sprintf("%s/users/%s/guilds", discordApi, userId))
- err = json.Unmarshal(body, &servers)
- return
- }
- func Members(session *Session, serverId int) (members []Member, err error) {
- body, err := Request(session, fmt.Sprintf("%s/guilds/%d/members", discordApi, serverId))
- err = json.Unmarshal(body, &members)
- return
- }
- func Channels(session *Session, serverId int) (channels []Channel, err error) {
- body, err := Request(session, fmt.Sprintf("%s/guilds/%d/channels", discordApi, serverId))
- err = json.Unmarshal(body, &channels)
- return
- }
- func Messages(session *Session, channelId int, limit int, afterId int, beforeId int) (messages []Message, err error) {
- var urlStr string
- if limit > 0 {
- urlStr = fmt.Sprintf("%s/channels/%d/messages?limit=%d", discordApi, channelId, limit)
- }
- if afterId > 0 {
- if urlStr != "" {
- urlStr = urlStr + fmt.Sprintf("&after=%d", afterId)
- } else {
- urlStr = fmt.Sprintf("%s/channels/%d/messages?after=%d", discordApi, channelId, afterId)
- }
- }
- if beforeId > 0 {
- if urlStr != "" {
- urlStr = urlStr + fmt.Sprintf("&before=%d", beforeId)
- } else {
- urlStr = fmt.Sprintf("%s/channels/%d/messages?after=%d", discordApi, channelId, beforeId)
- }
- }
- if urlStr == "" {
- urlStr = fmt.Sprintf("%s/channels/%d/messages", discordApi, channelId)
- }
- fmt.Println(urlStr)
- body, err := Request(session, urlStr)
- err = json.Unmarshal(body, &messages)
- return
- }
- func SendMessage(session *Session, channelId int, message string) (response Message, err error) {
- var urlStr string = fmt.Sprintf("%s/channels/%d/messages", discordApi, channelId)
- req, err := http.NewRequest("POST", urlStr, bytes.NewBuffer([]byte(fmt.Sprintf(`{"content":"%s"}`, message))))
- if err != nil {
- return
- }
- req.Header.Set("authorization", session.Token)
- req.Header.Set("Content-Type", "application/json")
- client := &http.Client{Timeout: (20 * time.Second)}
- resp, err := 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
- }
- if session.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, &response)
- return
- }
- func Logout(session *Session) (err error) {
- req, err := http.NewRequest("POST", fmt.Sprintf("%s/%s", discordApi, fmt.Sprintf("auth/logout")), bytes.NewBuffer([]byte(fmt.Sprintf(``))))
- if err != nil {
- return
- }
- req.Header.Set("authorization", session.Token)
- req.Header.Set("Content-Type", "application/json")
- client := &http.Client{Timeout: (20 * time.Second)}
- resp, err := 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
- }
|