40 lines
932 B
Go
40 lines
932 B
Go
package importers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/Andreychik32/ytdl"
|
|
)
|
|
|
|
type YoutubeImporter struct{}
|
|
|
|
func (i *YoutubeImporter) GetVideoInfo(url string) (videoInfo VideoInfo, err error) {
|
|
if strings.HasPrefix(strings.ToLower(url), "youtube:") {
|
|
url = strings.TrimSpace(strings.SplitN(url, ":", 2)[1])
|
|
}
|
|
|
|
ctx := context.Background()
|
|
info, err := ytdl.GetVideoInfo(ctx, url)
|
|
if err != nil {
|
|
err = fmt.Errorf("error retriving youtube video info: %w", err)
|
|
return
|
|
}
|
|
|
|
ctx = context.Background()
|
|
videoURL, err := ytdl.DefaultClient.GetDownloadURL(ctx, info, info.Formats[0])
|
|
if err != nil {
|
|
err = fmt.Errorf("error retriving youtube video url: %w", err)
|
|
return
|
|
}
|
|
videoInfo.VideoURL = videoURL.String()
|
|
|
|
videoInfo.ThumbnailURL = info.GetThumbnailURL(ytdl.ThumbnailQualityHigh).String()
|
|
|
|
videoInfo.ID = info.ID
|
|
videoInfo.Title = info.Title
|
|
videoInfo.Description = info.Description
|
|
|
|
return
|
|
}
|