Compare commits

...

2 Commits

Author SHA1 Message Date
dependabot[bot] 12fe98536a
build(deps): bump github.com/stretchr/testify from 1.8.0 to 1.8.1
Bumps [github.com/stretchr/testify](https://github.com/stretchr/testify) from 1.8.0 to 1.8.1.
- [Release notes](https://github.com/stretchr/testify/releases)
- [Commits](https://github.com/stretchr/testify/compare/v1.8.0...v1.8.1)

---
updated-dependencies:
- dependency-name: github.com/stretchr/testify
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
2023-01-04 16:28:02 +00:00
Chris Sexton f70eb46c5d twitch: redo with webhooks 2023-01-04 11:27:03 -05:00
7 changed files with 267 additions and 276 deletions

View File

@ -159,6 +159,7 @@ func (b *bot) ListenAndServe() {
}() }()
<-stop <-stop
b.DefaultConnector().Shutdown() b.DefaultConnector().Shutdown()
b.Receive(b.DefaultConnector(), Shutdown, msg.Message{})
} }
func (b *bot) RegisterWeb(r http.Handler, root string) { func (b *bot) RegisterWeb(r http.Handler, root string) {

View File

@ -40,6 +40,8 @@ const (
Delete Delete
// Startup is triggered after the connector has run the Serve function // Startup is triggered after the connector has run the Serve function
Startup Startup
// Shutdown is triggered after an OS interrupt
Shutdown
) )
type EphemeralID string type EphemeralID string

6
go.mod
View File

@ -26,10 +26,11 @@ require (
github.com/mattn/go-sqlite3 v2.0.3+incompatible github.com/mattn/go-sqlite3 v2.0.3+incompatible
github.com/mmcdole/gofeed v1.1.3 github.com/mmcdole/gofeed v1.1.3
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646
github.com/nicklaw5/helix v1.25.0
github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254 github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254
github.com/rs/zerolog v1.28.0 github.com/rs/zerolog v1.28.0
github.com/slack-go/slack v0.11.3 github.com/slack-go/slack v0.11.3
github.com/stretchr/testify v1.8.0 github.com/stretchr/testify v1.8.1
github.com/trubitsyn/go-zero-width v1.0.1 github.com/trubitsyn/go-zero-width v1.0.1
github.com/velour/velour v0.0.0-20160303155839-8e090e68d158 github.com/velour/velour v0.0.0-20160303155839-8e090e68d158
golang.org/x/crypto v0.0.0-20220829220503-c86fa9a7ed90 golang.org/x/crypto v0.0.0-20220829220503-c86fa9a7ed90
@ -53,6 +54,7 @@ require (
github.com/garyburd/go-oauth v0.0.0-20180319155456-bca2e7f09a17 // indirect github.com/garyburd/go-oauth v0.0.0-20180319155456-bca2e7f09a17 // indirect
github.com/go-stack/stack v1.8.0 // indirect github.com/go-stack/stack v1.8.0 // indirect
github.com/gobwas/glob v0.2.3 // indirect github.com/gobwas/glob v0.2.3 // indirect
github.com/golang-jwt/jwt v3.2.1+incompatible // indirect
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
github.com/golang/groupcache v0.0.0-20191027212112-611e8accdfc9 // indirect github.com/golang/groupcache v0.0.0-20191027212112-611e8accdfc9 // indirect
github.com/golang/protobuf v1.3.1 // indirect github.com/golang/protobuf v1.3.1 // indirect
@ -79,7 +81,7 @@ require (
github.com/robertkrimen/otto v0.0.0-20180617131154-15f95af6e78d // indirect github.com/robertkrimen/otto v0.0.0-20180617131154-15f95af6e78d // indirect
github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca // indirect github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca // indirect
github.com/spaolacci/murmur3 v1.1.0 // indirect github.com/spaolacci/murmur3 v1.1.0 // indirect
github.com/stretchr/objx v0.4.0 // indirect github.com/stretchr/objx v0.5.0 // indirect
github.com/temoto/robotstxt v1.1.1 // indirect github.com/temoto/robotstxt v1.1.1 // indirect
github.com/ttacon/builder v0.0.0-20170518171403-c099f663e1c2 // indirect github.com/ttacon/builder v0.0.0-20170518171403-c099f663e1c2 // indirect
github.com/ttacon/libphonenumber v1.1.0 // indirect github.com/ttacon/libphonenumber v1.1.0 // indirect

10
go.sum
View File

@ -76,6 +76,8 @@ github.com/gobwas/glob v0.2.3/go.mod h1:d3Ez4x06l9bZtSvzIay5+Yzi0fmZzPgnTbPcKjJA
github.com/gocolly/colly v1.2.0 h1:qRz9YAn8FIH0qzgNUw+HT9UN7wm1oF9OBAilwEWpyrI= github.com/gocolly/colly v1.2.0 h1:qRz9YAn8FIH0qzgNUw+HT9UN7wm1oF9OBAilwEWpyrI=
github.com/gocolly/colly v1.2.0/go.mod h1:Hof5T3ZswNVsOHYmba1u03W65HDWgpV5HifSuueE0EA= github.com/gocolly/colly v1.2.0/go.mod h1:Hof5T3ZswNVsOHYmba1u03W65HDWgpV5HifSuueE0EA=
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
github.com/golang-jwt/jwt v3.2.1+incompatible h1:73Z+4BJcrTC+KczS6WvTPvRGOp1WmfEP4Q1lOd9Z/+c=
github.com/golang-jwt/jwt v3.2.1+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g= github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g=
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=
github.com/golang/groupcache v0.0.0-20191027212112-611e8accdfc9 h1:uHTyIjqVhYRhLbJ8nIiOJHkEZZ+5YoOsAbD3sk82NiE= github.com/golang/groupcache v0.0.0-20191027212112-611e8accdfc9 h1:uHTyIjqVhYRhLbJ8nIiOJHkEZZ+5YoOsAbD3sk82NiE=
@ -140,6 +142,8 @@ github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLD
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ= github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8= github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
github.com/nicklaw5/helix v1.25.0 h1:Mrz537izZVsGdM3I46uGAAlslj61frgkhS/9xQqyT/M=
github.com/nicklaw5/helix v1.25.0/go.mod h1:yvXZFapT6afIoxnAvlWiJiUMsYnoHl7tNs+t0bloAMw=
github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254 h1:JYoQR67E1vv1WGoeW8DkdFs7vrIEe/5wP+qJItd5tUE= github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254 h1:JYoQR67E1vv1WGoeW8DkdFs7vrIEe/5wP+qJItd5tUE=
github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254/go.mod h1:DPucAeQGDPUzYUt+NaWw6qsF5SFapWWToxEiVDh2aV0= github.com/olebedev/when v0.0.0-20190311101825-c3b538a97254/go.mod h1:DPucAeQGDPUzYUt+NaWw6qsF5SFapWWToxEiVDh2aV0=
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
@ -163,13 +167,15 @@ github.com/slack-go/slack v0.11.3/go.mod h1:hlGi5oXA+Gt+yWTPP0plCdRKmjsDxecdHxYQ
github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI= github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0 h1:M2gUjqZET1qApGOWNSnZ49BAIMX4F/1plDv3+l31EJ4=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/objx v0.5.0 h1:1zr/of2m5FGMsad5YfcqgdqdWrIhu+EBEJRhR1U7z/c=
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
github.com/temoto/robotstxt v1.1.1 h1:Gh8RCs8ouX3hRSxxK7B1mO5RFByQ4CmJZDwgom++JaA= github.com/temoto/robotstxt v1.1.1 h1:Gh8RCs8ouX3hRSxxK7B1mO5RFByQ4CmJZDwgom++JaA=
github.com/temoto/robotstxt v1.1.1/go.mod h1:+1AmkuG3IYkh1kv0d2qEB9Le88ehNO0zwOr3ujewlOo= github.com/temoto/robotstxt v1.1.1/go.mod h1:+1AmkuG3IYkh1kv0d2qEB9Le88ehNO0zwOr3ujewlOo=
github.com/trubitsyn/go-zero-width v1.0.1 h1:AAZhtyGXW79T5BouAF0R9FtDhGcp7IGbLZo2Id3N+m8= github.com/trubitsyn/go-zero-width v1.0.1 h1:AAZhtyGXW79T5BouAF0R9FtDhGcp7IGbLZo2Id3N+m8=

View File

@ -3,8 +3,10 @@ package twitch
import ( import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"io/ioutil" "github.com/nicklaw5/helix"
"io"
"net/http" "net/http"
"net/url" "net/url"
"regexp" "regexp"
@ -36,17 +38,24 @@ type Twitch struct {
irc *IRC irc *IRC
ircLock sync.Mutex ircLock sync.Mutex
bridgeMap map[string]string bridgeMap map[string]string
helix *helix.Client
subs map[string]bool
} }
type Twitcher struct { type Twitcher struct {
name string id string
gameID string gameID string
isStreaming bool online bool
Name string
Game string
URL string
} }
func (t Twitcher) URL() string { func (t Twitcher) url() string {
u, _ := url.Parse("https://twitch.tv/") u, _ := url.Parse("https://twitch.tv/")
u2, _ := url.Parse(t.name) u2, _ := url.Parse(t.Name)
return u.ResolveReference(u2).String() return u.ResolveReference(u2).String()
} }
@ -76,21 +85,13 @@ func New(b bot.Bot) *Twitch {
bridgeMap: map[string]string{}, bridgeMap: map[string]string{},
} }
for _, ch := range p.c.GetArray("Twitch.Channels", []string{}) { for _, twitcherName := range p.c.GetArray("Twitch.Users", []string{}) {
for _, twitcherName := range p.c.GetArray("Twitch."+ch+".Users", []string{}) { twitcherName = strings.ToLower(twitcherName)
twitcherName = strings.ToLower(twitcherName) p.twitchList[twitcherName] = &Twitcher{
if _, ok := p.twitchList[twitcherName]; !ok { Name: twitcherName,
p.twitchList[twitcherName] = &Twitcher{
name: twitcherName,
gameID: "",
}
}
} }
go p.twitchChannelLoop(b.DefaultConnector(), ch)
} }
go p.twitchAuthLoop(b.DefaultConnector())
p.register() p.register()
p.registerWeb() p.registerWeb()
@ -99,38 +100,9 @@ func New(b bot.Bot) *Twitch {
func (p *Twitch) registerWeb() { func (p *Twitch) registerWeb() {
r := chi.NewRouter() r := chi.NewRouter()
r.HandleFunc("/{user}", p.serveStreaming) r.HandleFunc("/online", p.onlineCB)
p.b.RegisterWeb(r, "/isstreaming") r.HandleFunc("/offline", p.offlineCB)
} p.b.RegisterWeb(r, "/twitch")
func (p *Twitch) serveStreaming(w http.ResponseWriter, r *http.Request) {
userName := strings.ToLower(chi.URLParam(r, "user"))
if userName == "" {
fmt.Fprint(w, "User not found.")
return
}
twitcher := p.twitchList[userName]
if twitcher == nil {
fmt.Fprint(w, "User not found.")
return
}
status := "NO."
if twitcher.gameID != "" {
status = "YES."
}
context := map[string]any{"Name": twitcher.name, "Status": status}
t, err := template.New("streaming").Parse(page)
if err != nil {
log.Error().Err(err).Msg("Could not parse template!")
return
}
err = t.Execute(w, context)
if err != nil {
log.Error().Err(err).Msg("Could not execute template!")
}
} }
func (p *Twitch) register() { func (p *Twitch) register() {
@ -170,11 +142,118 @@ func (p *Twitch) register() {
Regex: regexp.MustCompile(`.*`), Regex: regexp.MustCompile(`.*`),
Handler: p.bridgeMsg, Handler: p.bridgeMsg,
}, },
{
Kind: bot.Startup, IsCmd: false,
Regex: regexp.MustCompile(`.*`),
Handler: p.startup,
},
{
Kind: bot.Shutdown, IsCmd: false,
Regex: regexp.MustCompile(`.*`),
Handler: p.shutdown,
},
} }
p.b.Register(p, bot.Help, p.help) p.b.Register(p, bot.Help, p.help)
p.b.RegisterTable(p, p.tbl) p.b.RegisterTable(p, p.tbl)
} }
func (p *Twitch) shutdown(r bot.Request) bool {
return false
}
func (p *Twitch) startup(r bot.Request) bool {
var err error
clientID := p.c.Get("twitch.clientid", "")
clientSecret := p.c.Get("twitch.secret", "")
if clientID == "" || clientSecret == "" {
log.Info().Msg("No clientID/secret, twitch disabled")
return false
}
p.helix, err = helix.NewClient(&helix.Options{
ClientID: clientID,
ClientSecret: clientSecret,
})
if err != nil {
log.Printf("Login error: %v", err)
return false
}
access, err := p.helix.RequestAppAccessToken([]string{"user:read:email"})
if err != nil {
log.Printf("Login error: %v", err)
return false
}
fmt.Printf("%+v\n", access)
// Set the access token on the client
p.helix.SetAppAccessToken(access.Data.AccessToken)
p.subs = map[string]bool{}
for _, t := range p.twitchList {
if err := p.follow(t); err != nil {
log.Error().Err(err).Msg("")
}
}
return false
}
func (p *Twitch) follow(twitcher *Twitcher) error {
if twitcher.id == "" {
users, err := p.helix.GetUsers(&helix.UsersParams{
Logins: []string{twitcher.Name},
})
if err != nil {
return err
}
if users.Error != "" {
return errors.New(users.Error)
}
twitcher.id = users.Data.Users[0].ID
}
base := p.c.Get("baseurl", "") + "/twitch"
_, err := p.helix.CreateEventSubSubscription(&helix.EventSubSubscription{
Type: helix.EventSubTypeStreamOnline,
Version: "1",
Condition: helix.EventSubCondition{
BroadcasterUserID: twitcher.id,
},
Transport: helix.EventSubTransport{
Method: "webhook",
Callback: base + "/online",
Secret: "s3cre7w0rd",
},
})
if err != nil {
return err
}
_, err = p.helix.CreateEventSubSubscription(&helix.EventSubSubscription{
Type: helix.EventSubTypeStreamOffline,
Version: "1",
Condition: helix.EventSubCondition{
BroadcasterUserID: twitcher.id,
},
Transport: helix.EventSubTransport{
Method: "webhook",
Callback: base + "/offline",
Secret: "s3cre7w0rd",
},
})
if err != nil {
return err
}
return nil
}
func (p *Twitch) twitchStatus(r bot.Request) bool { func (p *Twitch) twitchStatus(r bot.Request) bool {
channel := r.Msg.Channel channel := r.Msg.Channel
if users := p.c.GetArray("Twitch."+channel+".Users", []string{}); len(users) > 0 { if users := p.c.GetArray("Twitch."+channel+".Users", []string{}); len(users) > 0 {
@ -182,9 +261,8 @@ func (p *Twitch) twitchStatus(r bot.Request) bool {
twitcherName = strings.ToLower(twitcherName) twitcherName = strings.ToLower(twitcherName)
// we could re-add them here instead of needing to restart the bot. // we could re-add them here instead of needing to restart the bot.
if t, ok := p.twitchList[twitcherName]; ok { if t, ok := p.twitchList[twitcherName]; ok {
err := p.checkTwitch(r.Conn, channel, t, true) if t.online {
if err != nil { p.streaming(r.Conn, r.Msg.Channel, t)
log.Error().Err(err).Msgf("error in checking twitch")
} }
} }
} }
@ -195,10 +273,8 @@ func (p *Twitch) twitchStatus(r bot.Request) bool {
func (p *Twitch) twitchUserStatus(r bot.Request) bool { func (p *Twitch) twitchUserStatus(r bot.Request) bool {
who := strings.ToLower(r.Values["who"]) who := strings.ToLower(r.Values["who"])
if t, ok := p.twitchList[who]; ok { if t, ok := p.twitchList[who]; ok {
err := p.checkTwitch(r.Conn, r.Msg.Channel, t, true) if t.online {
if err != nil { p.streaming(r.Conn, r.Msg.Channel, t)
log.Error().Err(err).Msgf("error in checking twitch")
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "I had trouble with that.")
} }
} else { } else {
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "I don't know who that is.") p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "I don't know who that is.")
@ -225,163 +301,11 @@ func (p *Twitch) help(c bot.Connector, kind bot.Kind, message msg.Message, args
return true return true
} }
func (p *Twitch) twitchAuthLoop(c bot.Connector) { func (p *Twitch) connectBridge(c bot.Connector, ch string, info *Twitcher) {
frequency := p.c.GetInt("Twitch.AuthFreq", 60*60) msg := fmt.Sprintf("This post is tracking #%s\n<%s>", info.Name, info.url())
cid := p.c.Get("twitch.clientid", "")
secret := p.c.Get("twitch.secret", "")
if cid == "" || secret == "" {
log.Info().Msgf("Disabling twitch autoauth.")
return
}
log.Info().Msgf("Checking auth every %d seconds", frequency)
if err := p.validateCredentials(); err != nil {
log.Error().Err(err).Msgf("error checking twitch validity")
}
for {
select {
case <-time.After(time.Duration(frequency) * time.Second):
if err := p.validateCredentials(); err != nil {
log.Error().Err(err).Msgf("error checking twitch validity")
}
}
}
}
func (p *Twitch) twitchChannelLoop(c bot.Connector, channel string) {
frequency := p.c.GetInt("Twitch.Freq", 60)
if p.c.Get("twitch.clientid", "") == "" || p.c.Get("twitch.secret", "") == "" {
log.Info().Msgf("Disabling twitch autochecking.")
return
}
log.Info().Msgf("Checking channels every %d seconds", frequency)
for {
time.Sleep(time.Duration(frequency) * time.Second)
for _, twitcherName := range p.c.GetArray("Twitch."+channel+".Users", []string{}) {
log.Debug().Msgf("checking %s on twiwch", twitcherName)
twitcherName = strings.ToLower(twitcherName)
if err := p.checkTwitch(c, channel, p.twitchList[twitcherName], false); err != nil {
log.Error().Err(err).Msgf("error in twitch loop")
}
}
}
}
func getRequest(url, clientID, token string) ([]byte, int, bool) {
bearer := fmt.Sprintf("Bearer %s", token)
var body []byte
var resp *http.Response
client := &http.Client{}
req, err := http.NewRequest("GET", url, nil)
if err != nil {
goto errCase
}
req.Header.Add("Client-ID", clientID)
req.Header.Add("Authorization", bearer)
resp, err = client.Do(req)
if err != nil {
goto errCase
}
body, err = ioutil.ReadAll(resp.Body)
if err != nil {
goto errCase
}
return body, resp.StatusCode, true
errCase:
log.Error().Err(err)
return []byte{}, resp.StatusCode, false
}
type twitchInfo struct {
Name string
Game string
URL string
}
func (p *Twitch) checkTwitch(c bot.Connector, channel string, twitcher *Twitcher, alwaysPrintStatus bool) error {
baseURL, err := url.Parse("https://api.twitch.tv/helix/streams")
if err != nil {
err := fmt.Errorf("error parsing twitch stream URL")
log.Error().Msg(err.Error())
return err
}
query := baseURL.Query()
query.Add("user_login", twitcher.name)
baseURL.RawQuery = query.Encode()
cid := p.c.Get("twitch.clientid", "")
token := p.c.Get("twitch.token", "")
if cid == token && cid == "" {
log.Info().Msgf("Twitch plugin not enabled.")
return nil
}
body, status, ok := getRequest(baseURL.String(), cid, token)
if !ok {
return fmt.Errorf("got status %d: %s", status, string(body))
}
var s stream
err = json.Unmarshal(body, &s)
if err != nil {
log.Error().Err(err).Msgf("error reading twitch data")
return err
}
games := s.Data
gameID, title := "", ""
if len(games) == 0 {
p.twitchList[twitcher.name].isStreaming = false
} else {
p.twitchList[twitcher.name].isStreaming = true
gameID = games[0].GameID
if gameID == "" {
gameID = "unknown"
}
title = games[0].Title
}
info := twitchInfo{
twitcher.name,
title,
twitcher.URL(),
}
log.Debug().Interface("info", info).Interface("twitcher", twitcher).Msgf("checking twitch")
if alwaysPrintStatus && twitcher.isStreaming {
p.streaming(c, channel, info)
} else if alwaysPrintStatus && !twitcher.isStreaming {
p.notStreaming(c, channel, info)
} else if twitcher.gameID != "" && !twitcher.isStreaming {
twitcher.gameID = ""
p.disconnectBridge(c, twitcher)
p.stopped(c, channel, info)
twitcher.gameID = ""
} else if twitcher.gameID != gameID && twitcher.isStreaming {
p.streaming(c, channel, info)
p.connectBridge(c, channel, info, twitcher)
twitcher.gameID = gameID
}
return nil
}
func (p *Twitch) connectBridge(c bot.Connector, ch string, info twitchInfo, twitcher *Twitcher) {
msg := fmt.Sprintf("This post is tracking #%s\n<%s>", twitcher.name, twitcher.URL())
err := p.startBridgeMsg( err := p.startBridgeMsg(
fmt.Sprintf("%s-%s-%s", info.Name, info.Game, time.Now().Format("2006-01-02-15:04")), fmt.Sprintf("%s-%s-%s", info.Name, info.Game, time.Now().Format("2006-01-02-15:04")),
twitcher.name, info.Name,
msg, msg,
) )
if err != nil { if err != nil {
@ -391,16 +315,16 @@ func (p *Twitch) connectBridge(c bot.Connector, ch string, info twitchInfo, twit
} }
func (p *Twitch) disconnectBridge(c bot.Connector, twitcher *Twitcher) { func (p *Twitch) disconnectBridge(c bot.Connector, twitcher *Twitcher) {
log.Debug().Msgf("Disconnecting bridge: %s -> %+v", twitcher.name, p.bridgeMap) log.Debug().Msgf("Disconnecting bridge: %s -> %+v", twitcher.Name, p.bridgeMap)
for threadID, ircCh := range p.bridgeMap { for threadID, ircCh := range p.bridgeMap {
if strings.HasSuffix(ircCh, twitcher.name) { if strings.HasSuffix(ircCh, twitcher.Name) {
delete(p.bridgeMap, threadID) delete(p.bridgeMap, threadID)
p.b.Send(c, bot.Message, threadID, "Stopped tracking #"+twitcher.name) p.b.Send(c, bot.Message, threadID, "Stopped tracking #"+twitcher.Name)
} }
} }
} }
func (p *Twitch) stopped(c bot.Connector, ch string, info twitchInfo) { func (p *Twitch) stopped(c bot.Connector, ch string, info *Twitcher) {
notStreamingTpl := p.c.Get("Twitch.StoppedTpl", stoppedStreamingTplFallback) notStreamingTpl := p.c.Get("Twitch.StoppedTpl", stoppedStreamingTplFallback)
buf := bytes.Buffer{} buf := bytes.Buffer{}
t, err := template.New("notStreaming").Parse(notStreamingTpl) t, err := template.New("notStreaming").Parse(notStreamingTpl)
@ -413,7 +337,7 @@ func (p *Twitch) stopped(c bot.Connector, ch string, info twitchInfo) {
p.b.Send(c, bot.Message, ch, buf.String()) p.b.Send(c, bot.Message, ch, buf.String())
} }
func (p *Twitch) streaming(c bot.Connector, channel string, info twitchInfo) { func (p *Twitch) streaming(c bot.Connector, channel string, info *Twitcher) {
isStreamingTpl := p.c.Get("Twitch.IsTpl", isStreamingTplFallback) isStreamingTpl := p.c.Get("Twitch.IsTpl", isStreamingTplFallback)
buf := bytes.Buffer{} buf := bytes.Buffer{}
t, err := template.New("isStreaming").Parse(isStreamingTpl) t, err := template.New("isStreaming").Parse(isStreamingTpl)
@ -426,7 +350,7 @@ func (p *Twitch) streaming(c bot.Connector, channel string, info twitchInfo) {
p.b.Send(c, bot.Message, channel, buf.String()) p.b.Send(c, bot.Message, channel, buf.String())
} }
func (p *Twitch) notStreaming(c bot.Connector, ch string, info twitchInfo) { func (p *Twitch) notStreaming(c bot.Connector, ch string, info *Twitcher) {
notStreamingTpl := p.c.Get("Twitch.NotTpl", notStreamingTplFallback) notStreamingTpl := p.c.Get("Twitch.NotTpl", notStreamingTplFallback)
buf := bytes.Buffer{} buf := bytes.Buffer{}
t, err := template.New("notStreaming").Parse(notStreamingTpl) t, err := template.New("notStreaming").Parse(notStreamingTpl)
@ -439,44 +363,120 @@ func (p *Twitch) notStreaming(c bot.Connector, ch string, info twitchInfo) {
p.b.Send(c, bot.Message, ch, buf.String()) p.b.Send(c, bot.Message, ch, buf.String())
} }
func (p *Twitch) validateCredentials() error { type twitchCB struct {
cid := p.c.Get("twitch.clientid", "") Challenge string `json:"challenge"`
token := p.c.Get("twitch.token", "") Subscription struct {
if token == "" { ID string `json:"id"`
return p.reAuthenticate() Type string `json:"type"`
} Version string `json:"version"`
_, status, ok := getRequest("https://id.twitch.tv/oauth2/validate", cid, token) Status string `json:"status"`
if !ok || status == http.StatusUnauthorized { Cost int `json:"cost"`
return p.reAuthenticate() Condition struct {
} BroadcasterUserID string `json:"broadcaster_user_id"`
log.Debug().Msgf("checked credentials and they were valid") } `json:"condition"`
return nil CreatedAt time.Time `json:"created_at"`
Transport struct {
Method string `json:"method"`
Callback string `json:"callback"`
} `json:"transport"`
} `json:"subscription"`
Event struct {
BroadcasterUserID string `json:"broadcaster_user_id"`
BroadcasterUserLogin string `json:"broadcaster_user_login"`
BroadcasterUserName string `json:"broadcaster_user_name"`
} `json:"event"`
} }
func (p *Twitch) reAuthenticate() error { func (p *Twitch) offlineCB(w http.ResponseWriter, r *http.Request) {
cid := p.c.Get("twitch.clientid", "") body, err := io.ReadAll(r.Body)
secret := p.c.Get("twitch.secret", "") if err != nil {
if cid == "" || secret == "" { log.Error().Err(err).Msg("")
return fmt.Errorf("could not request a new token without config values set") return
} }
resp, err := http.PostForm("https://id.twitch.tv/oauth2/token", url.Values{ defer r.Body.Close()
"client_id": {cid}, // verify that the notification came from twitch using the secret.
"client_secret": {secret}, if !helix.VerifyEventSubNotification("s3cre7w0rd", r.Header, string(body)) {
"grant_type": {"client_credentials"}, log.Error().Msg("no valid signature on subscription")
return
} else {
log.Info().Msg("verified signature for subscription")
}
var vals twitchCB
if err = json.Unmarshal(body, &vals); err != nil {
log.Error().Err(err).Msg("")
return
}
if vals.Challenge != "" {
w.Write([]byte(vals.Challenge))
return
}
log.Printf("got offline webhook: %v\n", vals)
w.WriteHeader(200)
w.Write([]byte("ok"))
twitcher := p.twitchList[vals.Event.BroadcasterUserLogin]
if !twitcher.online {
return
}
twitcher.online = false
twitcher.URL = twitcher.url()
if ch := p.c.Get("twitch.channel", ""); ch != "" {
p.stopped(p.b.DefaultConnector(), ch, twitcher)
p.disconnectBridge(p.b.DefaultConnector(), twitcher)
}
}
func (p *Twitch) onlineCB(w http.ResponseWriter, r *http.Request) {
body, err := io.ReadAll(r.Body)
if err != nil {
log.Error().Err(err).Msg("")
return
}
defer r.Body.Close()
// verify that the notification came from twitch using the secret.
if !helix.VerifyEventSubNotification("s3cre7w0rd", r.Header, string(body)) {
log.Error().Msg("no valid signature on subscription")
return
} else {
log.Info().Msg("verified signature for subscription")
}
var vals twitchCB
if err = json.Unmarshal(body, &vals); err != nil {
log.Error().Err(err).Msg("")
return
}
if vals.Challenge != "" {
w.Write([]byte(vals.Challenge))
return
}
log.Printf("got online webhook: %v\n", vals)
w.WriteHeader(200)
w.Write([]byte("ok"))
streams, err := p.helix.GetStreams(&helix.StreamsParams{
UserIDs: []string{vals.Event.BroadcasterUserID},
}) })
if err != nil { if err != nil {
return err log.Error().Err(err).Msg("")
return
} }
body, err := ioutil.ReadAll(resp.Body)
if err != nil { twitcher := p.twitchList[vals.Event.BroadcasterUserLogin]
return err
if twitcher.online {
return
}
twitcher.online = true
twitcher.URL = twitcher.url()
twitcher.gameID = streams.Data.Streams[0].GameID
twitcher.Game = streams.Data.Streams[0].GameName
if ch := p.c.Get("twitch.channel", ""); ch != "" {
p.streaming(p.b.DefaultConnector(), ch, twitcher)
p.connectBridge(p.b.DefaultConnector(), ch, twitcher)
} }
credentials := struct {
AccessToken string `json:"access_token"`
ExpiresIn int `json:"expires_in"`
TokenType string `json:"token_type"`
}{}
err = json.Unmarshal(body, &credentials)
log.Debug().Int("expires", credentials.ExpiresIn).Msgf("setting new twitch token")
return p.c.RegisterSecret("twitch.token", credentials.AccessToken)
} }

View File

@ -47,7 +47,7 @@ func makeTwitchPlugin(t *testing.T) (*Twitch, *bot.MockBot) {
assert.NotNil(t, c) assert.NotNil(t, c)
c.twitchList["drseabass"] = &Twitcher{ c.twitchList["drseabass"] = &Twitcher{
name: "drseabass", Name: "drseabass",
gameID: "", gameID: "",
} }

View File

@ -1,20 +0,0 @@
package twitch
var page = `
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title>Is {{.Name}} streaming?</title>
</head>
<body style="text-align: center; padding-top: 200px;">
<a style="font-weight: bold; font-size: 120pt;
font-family: Arial, sans-serif; text-decoration: none; color: black;"
title="{{.Status}}">{{.Status}}</a>
</body>
</html>
`