Merge pull request #36 from iopred/bans

Modifications to GuildBanAdd, closes #29 #33.  Idiomatic improvements to ChannelMessages func.
This commit is contained in:
Bruce 2015-12-29 13:19:59 -06:00
commit f2ccbd3522

View file

@ -16,6 +16,8 @@ import (
"fmt"
"io/ioutil"
"net/http"
"net/url"
"strconv"
"time"
)
@ -275,12 +277,19 @@ func (s *Session) GuildBans(guildID string) (st []User, err error) {
return
}
// GuildBanAdd bans the given user from the given guild.
// GuildBanCreate bans the given user from the given guild.
// guildID : The ID of a Guild.
// userID : The ID of a User
func (s *Session) GuildBanAdd(guildID, userID string) (err error) {
// days : The number of days of previous comments to delete.
func (s *Session) GuildBanCreate(guildID, userID string, days int) (err error) {
_, err = s.Request("PUT", GUILD_BAN(guildID, userID), nil)
uri := GUILD_BAN(guildID, userID)
if days > 0 {
uri = fmt.Sprintf("%s?delete-message-days=%d", uri, days)
}
_, err = s.Request("PUT", uri, nil)
return
}
@ -458,29 +467,27 @@ func (s *Session) ChannelTyping(channelID string) (err error) {
// afterID : If provided all messages returned will be after given ID.
func (s *Session) ChannelMessages(channelID string, limit int, beforeID int, afterID int) (st []Message, err error) {
var urlStr string
uri := CHANNEL_MESSAGES(channelID)
v := url.Values{}
if limit > 0 {
urlStr = fmt.Sprintf("?limit=%d", limit)
v.Set("limit", strconv.Itoa(limit))
}
if afterID > 0 {
if urlStr != "" {
urlStr = urlStr + fmt.Sprintf("&after=%d", afterID)
} else {
urlStr = fmt.Sprintf("?after=%d", afterID)
v.Set("after", strconv.Itoa(afterID))
}
}
if beforeID > 0 {
if urlStr != "" {
urlStr = urlStr + fmt.Sprintf("&before=%d", beforeID)
} else {
urlStr = fmt.Sprintf("?before=%d", beforeID)
v.Set("before", strconv.Itoa(beforeID))
}
if len(v) > 0 {
uri = fmt.Sprintf("%s?%s", uri, v.Encode())
}
body, err := s.Request("GET", uri, nil)
if err != nil {
return
}
body, err := s.Request("GET", CHANNEL_MESSAGES(channelID)+urlStr, nil)
err = json.Unmarshal(body, &st)
return
}