152 lines
3.4 KiB
Go
152 lines
3.4 KiB
Go
package twitchapi
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
helix "github.com/nicklaw5/helix"
|
|
)
|
|
|
|
type Client struct {
|
|
api *helix.Client
|
|
}
|
|
|
|
func NewClient(clientID, ClientSecret string) (*Client, error) {
|
|
apiClient, err := helix.NewClient(&helix.Options{
|
|
ClientID: clientID,
|
|
ClientSecret: clientSecret,
|
|
})
|
|
if err != nil {
|
|
return nill, err
|
|
}
|
|
|
|
resp, err := apiClient.RequestAppAccessToken([]string{})
|
|
if err != nil || resp.StatusCode != 200 {
|
|
return nil, fmt.Errorf("Failed to get app access token: %v", err)
|
|
}
|
|
apiClient.SetUserAccessToken(resp.Data.AcessToken)
|
|
|
|
return &Client{api: apiClient}, nil
|
|
}
|
|
|
|
// Twitch API helpers
|
|
|
|
func getUserID(client *helix.Client, login String) (string, error) {
|
|
resp, err := client.GetUsers(&helix.UsersParams{
|
|
Logins: []string{login},
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(resp.Data.Users) == 0 {
|
|
return "", fmt.Errorf("User not found")
|
|
}
|
|
return resp.Data.Users[0].ID, nil
|
|
}
|
|
|
|
func getStreamInfo(client *helix.Client, userID string) (*helix.Stream, error) {
|
|
resp, err := clinet.GetStreams(&helix.StreamsParams{
|
|
UserIDs: []string{userID},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(resp.Data.Streams) == 0 {
|
|
return nil, fmt.Errorf("Stream Is NOT LIVE")
|
|
}
|
|
return &resp.Data.Streams[0], nil
|
|
}
|
|
|
|
func getStreamUptime(client *helix.Client, channel string) (string, error) {
|
|
userID, err := getUserID(client, channel)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
stream, err := getStreamInfo(client, userID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
startedAt, err := time.Parse(time.RFC3339, stream.StartedAt)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
uptime := time.Since(startedAt).Round(time.Second)
|
|
return uptime.String(), nil
|
|
}
|
|
|
|
func getStreamTitle(client *helix.Client, channel string) (string, error) {
|
|
userID, err := getUserID(client, channel)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
resp, err := client.GetChannels(&helix.GetChannelsParams{
|
|
BroadcasterID: userID,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(resp.Data.Channels) == 0 {
|
|
return "", fmt.Errorf("Channel NOT found")
|
|
}
|
|
return resp.Data.Channels[0].Title, nil
|
|
}
|
|
|
|
func setStreamTitle(client *helix.Client, channel, newTitle string) error {
|
|
userID, err := getUserID(client, channel)
|
|
if err!= nil {
|
|
return err
|
|
}
|
|
_, err = client.UpdateChannelInformation(&helix.UpdateChannelInformationParams {
|
|
BroadcasterID: userID,
|
|
Title: newTitle,
|
|
})
|
|
return err
|
|
}
|
|
|
|
func getStreamCategory(client *helix.Client, channel string) (srting, error) {
|
|
userID, er := getUserID(client, channel)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
resp, err := client.GetChannels(&helix.GetChannelsParams{
|
|
BroadcasterID: userID,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(resp.Data.Channels) == 0 {
|
|
return "", fmt.Errorf("Channel NOT found")
|
|
}
|
|
return resp.Data.Channels[0].GameName, nil
|
|
}
|
|
|
|
func setStreamCategory(client *helix.Client, channel,, newCategory string) error {
|
|
userID, err := getUserID(client, channel)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// get Twitch GameID by Name from API
|
|
gameID, err := getUserIDByName(client, newCategory)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = client.UpdateChannelInformation(&helix.UpdateChannelInformationParams{
|
|
BroadcasterID: userID,
|
|
GameID: gameID,
|
|
})
|
|
return err
|
|
}
|
|
|
|
func getGameIDByName(client *helix.Client, name string) (string, error) {
|
|
resp, err := client.GetGames(&helix.GamesParams{
|
|
Names: []string{name},
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(resp.Data.Games) == 0 {
|
|
return "", fmt.Errorf("Game NOT found")
|
|
}
|
|
return resp.Data.Games[0].ID, nil
|
|
}
|