main.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. package main
  2. import (
  3. "fmt"
  4. "log"
  5. "os"
  6. "time"
  7. "github.com/joho/godotenv"
  8. "github.com/nicklaw5/helix"
  9. "github.com/zveinn/twitch-bot/mongowrapper"
  10. )
  11. var (
  12. monitor = make(chan int, 100)
  13. MAIN_CLIENT *IRC_CLIENT
  14. HELIX_CLIENT *helix.Client
  15. )
  16. var (
  17. TextCommands = make(map[string]string)
  18. MP3Map = make(map[string]string)
  19. EmoteMap = make(map[string]helix.Emote)
  20. )
  21. var SoundQueue = make(chan SoundEvent, 1000)
  22. type SoundEvent struct {
  23. // mp3/tts
  24. T string
  25. Data string
  26. }
  27. func ProcessSoundEvents() {
  28. defer func() {
  29. r := recover()
  30. if r != nil {
  31. log.Println(r)
  32. }
  33. monitor <- 7
  34. }()
  35. for s := range SoundQueue {
  36. fmt.Printf("SOUND QUEUE: len(%d), max(%d)", len(SoundQueue), cap(SoundQueue))
  37. switch s.T {
  38. case "mp3":
  39. PlayMP3(s.Data)
  40. case "tts":
  41. PlayTTS(s.Data)
  42. default:
  43. fmt.Println("UKNOWN SOUND EVENT", s)
  44. }
  45. }
  46. }
  47. var TWITCH_CLIENT = new(IRC_CLIENT)
  48. func main() {
  49. err := godotenv.Load()
  50. if err != nil {
  51. log.Fatal("Error loading .env file")
  52. }
  53. // MakeNewToken()
  54. // os.Exit(1)
  55. fmt.Println("MONGO CONNECTING:", os.Getenv("DB"))
  56. err = mongowrapper.Connect(os.Getenv("DB"))
  57. if err != nil {
  58. log.Fatal(err)
  59. os.Exit(1)
  60. }
  61. mongowrapper.InitCollections()
  62. InitTwitchClient()
  63. INIT_MSG()
  64. InitCommands()
  65. InitMP3Map()
  66. err = RenewTokens()
  67. if err != nil {
  68. fmt.Println(err)
  69. os.Exit(1)
  70. }
  71. CreateAPIClient()
  72. GetGlobalEmotes()
  73. TWITCH_CLIENT.GetAllChannelEmotes()
  74. go ProcessSoundEvents()
  75. // go RenewTokensLoop()
  76. go TWITCH_CLIENT.Connect()
  77. for {
  78. select {
  79. case ID := <-monitor:
  80. log.Println("ID RETURNED: ", ID)
  81. if ID == 10 {
  82. TWITCH_CLIENT.Connect()
  83. } else if ID == 7 {
  84. go RenewTokensLoop()
  85. } else if ID == 1337 {
  86. go TWITCH_CLIENT.POST_INFO()
  87. } else if ID == 11 {
  88. go ProcessSoundEvents()
  89. }
  90. default:
  91. time.Sleep(500 * time.Millisecond)
  92. }
  93. }
  94. }