newsbot-api/services/input/twitch.go

package input

import (
	"database/sql"
	"errors"
	"fmt"
	"strings"
	"time"

	"github.com/jtom38/newsbot/collector/database"
	"github.com/jtom38/newsbot/collector/services/config"
	"github.com/nicklaw5/helix/v2"
)

type TwitchClient struct {
	SourceRecord database.Source

	// config
	monitorClips string
	monitorVod   string

	// API Connection
	api *helix.Client
}

var (
	ErrTwitchClientIdMissing     = errors.New("unable to find the client id for auth")
	ErrTwitchClientSecretMissing = errors.New("unable to find the client secret for auth")

	twitchScopes = "user:read:email"
)

func NewTwitchClient() (TwitchClient, error) {
	c := config.New()

	id := c.GetConfig(config.TWITCH_CLIENT_ID)
	if id == "" {
		return TwitchClient{}, ErrTwitchClientIdMissing
	}

	secret := c.GetConfig(config.TWITCH_CLIENT_SECRET)
	if secret == "" {
		return TwitchClient{}, ErrTwitchClientSecretMissing
	}

	api, err := initTwitchApi(id, secret)
	if err != nil {
		return TwitchClient{}, nil
	}

	client := TwitchClient{
		//SourceRecord: &source,
		monitorClips: c.GetConfig(config.TWITCH_MONITOR_CLIPS),
		monitorVod:   c.GetConfig(config.TWITCH_MONITOR_VOD),
		api:          &api,
	}

	return client, nil
}

// Sets up the API connection to Twitch.
func initTwitchApi(ClientId string, ClientSecret string) (helix.Client, error) {
	api, err := helix.NewClient(&helix.Options{
		ClientID:     ClientId,
		ClientSecret: ClientSecret,
	})
	if err != nil {
		return helix.Client{}, err
	}

	return *api, nil
}

// This will let you replace the bound source record to keep the same session alive.
func (tc *TwitchClient) ReplaceSourceRecord(source database.Source) {
	tc.SourceRecord = source
}

// Invokes Logon request to the API
func (tc *TwitchClient) Login() error {
	token, err := tc.api.RequestAppAccessToken([]string{twitchScopes})
	if err != nil {
		return err
	}

	tc.api.SetAppAccessToken(token.Data.AccessToken)
	return nil
}

func (tc *TwitchClient) GetContent() ([]database.Article, error) {
	var items []database.Article

	user, err := tc.GetUserDetails()
	if err != nil {
		return items, err
	}

	posts, err := tc.GetPosts(user)
	if err != nil {
		return items, err
	}

	for _, video := range posts {
		var article database.Article

		AuthorName, err := tc.ExtractAuthor(video)
		if err != nil {
			return items, err
		}
		article.Authorname = sql.NullString{String: AuthorName}

		Authorimage, err := tc.ExtractAuthorImage(user)
		if err != nil {
			return items, err
		}
		article.Authorimage = sql.NullString{String: Authorimage}

		article.Description, err = tc.ExtractDescription(video)
		if err != nil {
			return items, err
		}

		article.Pubdate, err = tc.ExtractPubDate(video)
		if err != nil {
			return items, err
		}

		article.Sourceid = tc.SourceRecord.ID
		article.Tags, err = tc.ExtractTags(video, user)
		if err != nil {
			return items, err
		}

		article.Thumbnail, err = tc.ExtractThumbnail(video)
		if err != nil {
			return items, err
		}

		article.Title, err = tc.ExtractTitle(video)
		if err != nil {
			return items, err
		}

		article.Url, err = tc.ExtractUrl(video)
		if err != nil {
			return items, err
		}

		items = append(items, article)
	}

	return items, nil
}

func (tc *TwitchClient) GetUserDetails() (helix.User, error) {
	var blank helix.User

	users, err := tc.api.GetUsers(&helix.UsersParams{
		Logins: []string{tc.SourceRecord.Name},
	})
	if err != nil {
		return blank, err
	}

	if len(users.Data.Users) == 0 {
		return blank, errors.New("no results have been returned")
	}

	return users.Data.Users[0], nil
}

// This will reach out and collect the posts made by the user.
func (tc *TwitchClient) GetPosts(user helix.User) ([]helix.Video, error) {
	var blank []helix.Video

	videos, err := tc.api.GetVideos(&helix.VideosParams{
		UserID: user.ID,
	})
	if err != nil {
		return blank, err
	}

	//log.Println(videos.Data.Videos)
	return videos.Data.Videos, nil
}

func (tc *TwitchClient) ExtractAuthor(post helix.Video) (string, error) {
	if post.UserName == "" {
		return "", ErrMissingAuthorName
	}
	return post.UserName, nil
}

func (tc *TwitchClient) ExtractThumbnail(post helix.Video) (string, error) {
	if post.ThumbnailURL == "" {
		return "", ErrMissingThumbnail
	}
	var thumb = post.ThumbnailURL
	thumb = strings.Replace(thumb, "%{width}", "600", -1)
	thumb = strings.Replace(thumb, "%{height}", "400", -1)
	return thumb, nil
}

func (tc *TwitchClient) ExtractPubDate(post helix.Video) (time.Time, error) {
	if post.PublishedAt == "" {
		return time.Now(), ErrMissingPublishDate
	}
	pubDate, err := time.Parse("2006-01-02T15:04:05Z", post.PublishedAt)
	if err != nil {
		return time.Now(), err
	}
	return pubDate, nil
}

func (tc *TwitchClient) ExtractDescription(post helix.Video) (string, error) {
	// Check if the description is null but we have a title.
	// The poster didnt add a description but this isnt an error.
	if post.Description == "" && post.Title == "" {
		return "", ErrMissingDescription
	}
	if post.Description == "" {
		return "No description was given", nil
	}
	return post.Description, nil
}

// Extracts the avatar of the author with some validation.
func (tc *TwitchClient) ExtractAuthorImage(user helix.User) (string, error) {
	if user.ProfileImageURL == "" {
		return "", ErrMissingAuthorImage
	}
	if !strings.Contains(user.ProfileImageURL, "-profile_image-") {
		return "", ErrInvalidAuthorImage
	}
	return user.ProfileImageURL, nil
}

// Generate tags based on the video metadata.
// TODO Figure out how to query what game is played
func (tc *TwitchClient) ExtractTags(post helix.Video, user helix.User) (string, error) {
	res := fmt.Sprintf("twitch,%v,%v", post.Title, user.DisplayName)
	return res, nil
}

// Extracts the title from a post with some validation.
func (tc *TwitchClient) ExtractTitle(post helix.Video) (string, error) {
	if post.Title == "" {
		return "", errors.New("unable to find the title on the requested post")
	}
	return post.Title, nil
}

func (tc *TwitchClient) ExtractUrl(post helix.Video) (string, error) {
	if post.URL == "" {
		return "", ErrMissingUrl
	}
	return post.URL, nil
}