main.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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. Avatar 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(&Avatar, "f", "./avatar.jpg", "Avatar File Name")
  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. img, err := ioutil.ReadFile(Avatar)
  49. if err != nil {
  50. fmt.Println(err)
  51. }
  52. base64 := base64.StdEncoding.EncodeToString(img)
  53. avatar := fmt.Sprintf("data:%s;base64,%s", http.DetectContentType(img), string(base64))
  54. _, err = s.UserUpdate("", "", BotUsername, avatar, "")
  55. if err != nil {
  56. fmt.Println(err)
  57. }
  58. }