60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/julienschmidt/httprouter"
|
||
|
|
"google.golang.org/api/option"
|
||
|
|
"google.golang.org/api/youtube/v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
type YoutubeStats struct {
|
||
|
|
Subscribers int `json:"subscribers"`
|
||
|
|
ChannelName string `json:"channelName"`
|
||
|
|
Views int `json:"views"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func getChannelStats(apiKey string, channelID string) httprouter.Handle {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
yts, err := youtube.NewService(ctx, option.WithAPIKey(apiKey))
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error creating new YouTube service: %v\n", err)
|
||
|
|
w.WriteHeader(http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
call := yts.Channels.List([]string{"snippet,contentDetails,statistics"})
|
||
|
|
response, err := call.Id(channelID).Do()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error calling YouTube API: %v\n", err)
|
||
|
|
w.WriteHeader(http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(response.Items) > 0 {
|
||
|
|
yt := YoutubeStats{
|
||
|
|
Subscribers: int(response.Items[0].Statistics.SubscriberCount),
|
||
|
|
ChannelName: response.Items[0].Snippet.Title,
|
||
|
|
Views: int(response.Items[0].Statistics.ViewCount),
|
||
|
|
}
|
||
|
|
|
||
|
|
w.Header().Set("Content-Type", "application/json")
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
if err := json.NewEncoder(w).Encode(yt); err != nil {
|
||
|
|
w.WriteHeader(http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
w.WriteHeader(http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|