Update command_live_handler.go

handler/command_live_handler.go

@@ -1,9 +1,13 @@
 package handler
 
 import (
+	"encoding/json"
 	"fmt"
+	"io/ioutil"
+	"net/http"
 	"strings"
 
+	"github.com/MiguelMachado-dev/disc-go-bot/config"
 	"github.com/bwmarrin/discordgo"
 	"golang.org/x/text/cases"
 	"golang.org/x/text/language"
@@ -75,14 +79,48 @@ func (h *LiveHandler) Handler(s *discordgo.Session, i *discordgo.InteractionCrea
 	username := options[0].StringValue()
 	platform := strings.ToLower(options[1].StringValue())
 
-	var streamURL, thumbnailURL string
+	var streamURL, thumbnailURL, streamTitle string
 	var color int
+	var isLive bool
 
 	switch platform {
 	case "twitch":
 		streamURL = fmt.Sprintf("https://www.twitch.tv/%s", username)
-		thumbnailURL = fmt.Sprintf("https://static-cdn.jtvnw.net/previews-ttv/live_user_%s-1920x1080.jpg", username)
 		color = 0x6441A4 // Twitch purple
+
+		// Fetch Twitch channel info
+		accessToken, err := h.getTwitchAccessToken()
+		if err != nil {
+			log.Errorf("Error getting Twitch access token: %v", err)
+			break
+		}
+
+		channelInfo, err := h.getTwitchChannelInfo(username, accessToken)
+		if err != nil {
+			log.Errorf("Error getting Twitch channel info: %v", err)
+			break
+		}
+
+		if channelInfo == nil {
+			log.Infof("No channel found for user: %s", username)
+			break
+		}
+
+		log.Infof("Channel info for %s: %+v", username, channelInfo)
+
+		// Check if the channel is live
+		streamInfo, err := h.getTwitchStreamInfo(username, accessToken)
+		if err != nil {
+			log.Errorf("Error getting Twitch stream info: %v", err)
+		}
+
+		isLive := streamInfo != nil
+		if isLive {
+			streamTitle = streamInfo.Title
+			thumbnailURL = streamInfo.ThumbnailURL
+		} else {
+			thumbnailURL = channelInfo.ProfileImageURL
+		}
 	case "youtube":
 		streamURL = fmt.Sprintf("https://www.youtube.com/channel/%s/live", username)
 		thumbnailURL = fmt.Sprintf("https://img.youtube.com/vi/%s/maxresdefault.jpg", username)
@@ -103,21 +141,43 @@ func (h *LiveHandler) Handler(s *discordgo.Session, i *discordgo.InteractionCrea
 
 	// Create an embed message
 	embed := &discordgo.MessageEmbed{
-		Title:       fmt.Sprintf("%s está ao vivo no %s!", username, platformTitle),
-		Description: fmt.Sprintf("Clique no link abaixo para assistir a live de %s.", username),
+		Title:       fmt.Sprintf("%s no %s", username, platformTitle),
+		Description: fmt.Sprintf("Confira o canal de %s", username),
 		URL:         streamURL,
 		Color:       color,
-		Image: &discordgo.MessageEmbedImage{
+		Thumbnail: &discordgo.MessageEmbedThumbnail{
 			URL: thumbnailURL,
 		},
 		Fields: []*discordgo.MessageEmbedField{
 			{
-				Name:  "Link da live",
+				Name:  "Link do canal",
 				Value: streamURL,
 			},
 		},
 	}
 
+	if platform == "twitch" {
+		if isLive {
+			embed.Title = fmt.Sprintf("%s está ao vivo no %s!", username, platformTitle)
+			embed.Description = fmt.Sprintf("Clique no link abaixo para assistir a live de %s.", username)
+			embed.Image = &discordgo.MessageEmbedImage{
+				URL: thumbnailURL,
+			}
+			embed.Thumbnail = nil
+			if streamTitle != "" {
+				embed.Fields = append(embed.Fields, &discordgo.MessageEmbedField{
+					Name:  "Título da Stream",
+					Value: streamTitle,
+				})
+			}
+		} else {
+			embed.Fields = append(embed.Fields, &discordgo.MessageEmbedField{
+				Name:  "Status",
+				Value: "Offline",
+			})
+		}
+	}
+
 	// Send embed message to a specific channel
 	channelID := "1187123213382201486"
 	_, err := s.ChannelMessageSendEmbed(channelID, embed)
@@ -134,3 +194,109 @@ func (h *LiveHandler) Handler(s *discordgo.Session, i *discordgo.InteractionCrea
 		},
 	})
 }
+
+func (h *LiveHandler) getTwitchAccessToken() (string, error) {
+	TwitchClientID := config.GetEnv().TWITCH_CLIENT_ID
+	TwitchClientSecret := config.GetEnv().TWITCH_CLIENT_SECRET
+
+	url := "https://id.twitch.tv/oauth2/token"
+	payload := strings.NewReader(fmt.Sprintf("client_id=%s&client_secret=%s&grant_type=client_credentials", TwitchClientID, TwitchClientSecret))
+	req, _ := http.NewRequest("POST", url, payload)
+	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", fmt.Errorf("failed to send request: %v", err)
+	}
+	defer res.Body.Close()
+
+	body, _ := ioutil.ReadAll(res.Body)
+
+	var result map[string]interface{}
+	if err := json.Unmarshal(body, &result); err != nil {
+		return "", fmt.Errorf("failed to parse response: %v", err)
+	}
+
+	if accessToken, ok := result["access_token"].(string); ok {
+		return accessToken, nil
+	}
+	return "", fmt.Errorf("access token not found in response: %v", result)
+}
+
+type TwitchChannelInfo struct {
+	ID              string `json:"id"`
+	Login           string `json:"login"`
+	DisplayName     string `json:"display_name"`
+	Type            string `json:"type"`
+	BroadcasterType string `json:"broadcaster_type"`
+	Description     string `json:"description"`
+	ProfileImageURL string `json:"profile_image_url"`
+	OfflineImageURL string `json:"offline_image_url"`
+}
+
+func (h *LiveHandler) getTwitchChannelInfo(username, accessToken string) (*TwitchChannelInfo, error) {
+	TwitchClientID := config.GetEnv().TWITCH_CLIENT_ID
+
+	url := fmt.Sprintf("https://api.twitch.tv/helix/users?login=%s", username)
+	req, _ := http.NewRequest("GET", url, nil)
+	req.Header.Add("Client-ID", TwitchClientID)
+	req.Header.Add("Authorization", "Bearer "+accessToken)
+
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("failed to send request: %v", err)
+	}
+	defer res.Body.Close()
+
+	body, _ := ioutil.ReadAll(res.Body)
+
+	var result struct {
+		Data []TwitchChannelInfo `json:"data"`
+	}
+	if err := json.Unmarshal(body, &result); err != nil {
+		return nil, fmt.Errorf("failed to parse response: %v", err)
+	}
+
+	log.Infof("Raw Twitch API response for channel info: %s", string(body))
+
+	if len(result.Data) > 0 {
+		return &result.Data[0], nil
+	}
+
+	return nil, nil
+}
+
+type TwitchStreamInfo struct {
+	Title        string `json:"title"`
+	ThumbnailURL string `json:"thumbnail_url"`
+}
+
+func (h *LiveHandler) getTwitchStreamInfo(username, accessToken string) (*TwitchStreamInfo, error) {
+	url := fmt.Sprintf("https://api.twitch.tv/helix/streams?user_login=%s", username)
+	req, _ := http.NewRequest("GET", url, nil)
+	req.Header.Add("Client-ID", "2ld69qispua7q167rst5mz7p150efj")
+	req.Header.Add("Authorization", "Bearer "+accessToken)
+
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("failed to send request: %v", err)
+	}
+	defer res.Body.Close()
+
+	body, _ := ioutil.ReadAll(res.Body)
+
+	var result struct {
+		Data []TwitchStreamInfo `json:"data"`
+	}
+	if err := json.Unmarshal(body, &result); err != nil {
+		return nil, fmt.Errorf("failed to parse response: %v", err)
+	}
+
+	log.Infof("Raw Twitch API response: %s", string(body))
+
+	if len(result.Data) > 0 {
+		return &result.Data[0], nil
+	}
+
+	return nil, nil
+}