twitch-bot/functions_twitch.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(func(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
	}))

	tplFuncs.Register("doesFollow", plugins.GenericTemplateFunctionGetter(func(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")
		}
	}))

	tplFuncs.Register("followAge", plugins.GenericTemplateFunctionGetter(func(from, to string) (time.Duration, error) {
		since, err := twitchClient.GetFollowDate(from, to)
		return time.Since(since), errors.Wrap(err, "getting follow date")
	}))
	tplFuncs.Register("followDate", plugins.GenericTemplateFunctionGetter(func(from, to string) (time.Time, error) { return twitchClient.GetFollowDate(from, to) }))

	tplFuncs.Register("doesFollowLongerThan", plugins.GenericTemplateFunctionGetter(func(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")
		}
	}))

	tplFuncs.Register("recentGame", plugins.GenericTemplateFunctionGetter(func(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
	}))

	tplFuncs.Register("streamUptime", plugins.GenericTemplateFunctionGetter(func(username string) (time.Duration, error) {
		si, err := twitchClient.GetCurrentStreamInfo(strings.TrimLeft(username, "#"))
		if err != nil {
			return 0, err
		}
		return time.Since(si.StartedAt), nil
	}))
}