blob: 932c3b792a685e665f143b5209800da05c6c1069 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
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
}
|
