main.go 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. package main
  2. import (
  3. "encoding/base64"
  4. "flag"
  5. "fmt"
  6. "io/ioutil"
  7. "net/http"
  8. "github.com/bwmarrin/discordgo"
  9. )
  10. var (
  11. Email string
  12. Password string
  13. Token string
  14. Url string
  15. BotID string
  16. BotUsername string
  17. )
  18. func init() {
  19. flag.StringVar(&Email, "e", "", "Account Email")
  20. flag.StringVar(&Password, "p", "", "Account Password")
  21. flag.StringVar(&Token, "t", "", "Account Token")
  22. flag.StringVar(&Url, "l", "http://bwmarrin.github.io/discordgo/img/discordgo.png", "Link to the avatar image")
  23. flag.Parse()
  24. }
  25. func main() {
  26. // Create a new Discord session using the provided login information.
  27. // Use discordgo.New(Token) to just use a token for login.
  28. dg, err := discordgo.New(Email, Password, Token)
  29. if err != nil {
  30. fmt.Println("error creating Discord session,", err)
  31. return
  32. }
  33. bot, err := dg.User("@me")
  34. if err != nil {
  35. fmt.Println("error fetching the bot details,", err)
  36. return
  37. }
  38. BotID = bot.ID
  39. BotUsername = bot.Username
  40. changeAvatar(dg)
  41. fmt.Println("Bot is now running. Press CTRL-C to exit.")
  42. // Simple way to keep program running until CTRL-C is pressed.
  43. <-make(chan struct{})
  44. return
  45. }
  46. // Helper function to change the avatar
  47. func changeAvatar(s *discordgo.Session) {
  48. resp, err := http.Get(Url)
  49. if err != nil {
  50. fmt.Println("Error retrieving the file, ", err)
  51. return
  52. }
  53. defer resp.Body.Close()
  54. img, err := ioutil.ReadAll(resp.Body)
  55. if err != nil {
  56. fmt.Println("Error reading the response, ", err)
  57. return
  58. }
  59. base64 := base64.StdEncoding.EncodeToString(img)
  60. avatar := fmt.Sprintf("data:%s;base64,%s", http.DetectContentType(img), string(base64))
  61. _, err = s.UserUpdate("", "", BotUsername, avatar, "")
  62. if err != nil {
  63. fmt.Println("Error setting the avatar, ", err)
  64. }
  65. }