mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-11-08 16:20:02 +00:00
112 lines
3.1 KiB
Go
112 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/Luzifer/twitch-bot/v3/pkg/twitch"
|
|
"github.com/Luzifer/twitch-bot/v3/plugins"
|
|
)
|
|
|
|
func init() {
|
|
tplFuncs.Register("displayName", plugins.GenericTemplateFunctionGetter(tplTwitchDisplayName))
|
|
tplFuncs.Register("doesFollow", plugins.GenericTemplateFunctionGetter(tplTwitchDoesFollow))
|
|
tplFuncs.Register("followAge", plugins.GenericTemplateFunctionGetter(tplTwitchFollowAge))
|
|
tplFuncs.Register("followDate", plugins.GenericTemplateFunctionGetter(tplTwitchFollowDate))
|
|
tplFuncs.Register("doesFollowLongerThan", plugins.GenericTemplateFunctionGetter(tplTwitchDoesFollowLongerThan))
|
|
tplFuncs.Register("recentGame", plugins.GenericTemplateFunctionGetter(tplTwitchRecentGame))
|
|
tplFuncs.Register("recentTitle", plugins.GenericTemplateFunctionGetter(tplTwitchRecentTitle))
|
|
tplFuncs.Register("streamUptime", plugins.GenericTemplateFunctionGetter(tplTwitchStreamUptime))
|
|
}
|
|
|
|
func tplTwitchDisplayName(username string, v ...string) (string, error) {
|
|
displayName, err := twitchClient.GetDisplayNameForUser(strings.TrimLeft(username, "#"))
|
|
if len(v) > 0 && (err != nil || displayName == "") {
|
|
return v[0], nil
|
|
}
|
|
|
|
return displayName, err
|
|
}
|
|
|
|
func tplTwitchDoesFollow(from, to string) (bool, error) {
|
|
_, err := twitchClient.GetFollowDate(from, to)
|
|
switch {
|
|
case err == nil:
|
|
return true, nil
|
|
|
|
case errors.Is(err, twitch.ErrUserDoesNotFollow):
|
|
return false, nil
|
|
|
|
default:
|
|
return false, errors.Wrap(err, "getting follow date")
|
|
}
|
|
}
|
|
|
|
func tplTwitchFollowAge(from, to string) (time.Duration, error) {
|
|
since, err := twitchClient.GetFollowDate(from, to)
|
|
return time.Since(since), errors.Wrap(err, "getting follow date")
|
|
}
|
|
|
|
func tplTwitchFollowDate(from, to string) (time.Time, error) {
|
|
return twitchClient.GetFollowDate(from, to)
|
|
}
|
|
|
|
func tplTwitchDoesFollowLongerThan(from, to string, t any) (bool, error) {
|
|
var (
|
|
age time.Duration
|
|
err error
|
|
)
|
|
|
|
switch v := t.(type) {
|
|
case int64:
|
|
age = time.Duration(v) * time.Second
|
|
|
|
case string:
|
|
if age, err = time.ParseDuration(v); err != nil {
|
|
return false, errors.Wrap(err, "parsing duration")
|
|
}
|
|
|
|
default:
|
|
return false, errors.Errorf("unexpected input for duration %t", t)
|
|
}
|
|
|
|
fd, err := twitchClient.GetFollowDate(from, to)
|
|
switch {
|
|
case err == nil:
|
|
return time.Since(fd) > age, nil
|
|
|
|
case errors.Is(err, twitch.ErrUserDoesNotFollow):
|
|
return false, nil
|
|
|
|
default:
|
|
return false, errors.Wrap(err, "getting follow date")
|
|
}
|
|
}
|
|
|
|
func tplTwitchRecentGame(username string, v ...string) (string, error) {
|
|
game, _, err := twitchClient.GetRecentStreamInfo(strings.TrimLeft(username, "#"))
|
|
if len(v) > 0 && (err != nil || game == "") {
|
|
return v[0], nil
|
|
}
|
|
|
|
return game, err
|
|
}
|
|
|
|
func tplTwitchRecentTitle(username string, v ...string) (string, error) {
|
|
_, title, err := twitchClient.GetRecentStreamInfo(strings.TrimLeft(username, "#"))
|
|
if len(v) > 0 && (err != nil || title == "") {
|
|
return v[0], nil
|
|
}
|
|
|
|
return title, err
|
|
}
|
|
|
|
func tplTwitchStreamUptime(username string) (time.Duration, error) {
|
|
si, err := twitchClient.GetCurrentStreamInfo(strings.TrimLeft(username, "#"))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return time.Since(si.StartedAt), nil
|
|
}
|