Compare commits
26 Commits
master
...
2c6dbc287c
Author | SHA1 | Date | |
---|---|---|---|
2c6dbc287c | |||
97d509248c | |||
ccaa16a68f | |||
a8cf8cbb8a | |||
20e5594f60 | |||
83f7676b2e | |||
f639545e9b | |||
18d7177099 | |||
07c20f4582 | |||
9b38ea7068 | |||
8ab4b4c280 | |||
0ad8722bf9 | |||
d8c4c343f4 | |||
52fdcc64ef | |||
fa0f1cb48c | |||
63899ac4bd | |||
f5c072360b | |||
7c02cbf0e4 | |||
9b41f9114c | |||
60dc43fd9b | |||
584084c1bc | |||
ed8eaf67a2 | |||
268351f31e | |||
43efe20d8c | |||
8d55bcb5ee | |||
48179c3a67 |
4
.gitignore
vendored
4
.gitignore
vendored
@@ -3,3 +3,7 @@ main.exe
|
||||
logs.log
|
||||
ws-server/deploy.tar
|
||||
downloader/main.log
|
||||
downloader/vendor/golang.org
|
||||
downloader/vendor/github.com/*
|
||||
!downloader/vendor/github.com/kkdai
|
||||
downloader/vendor/modules.txt
|
||||
|
2
dl/dl.go
2
dl/dl.go
@@ -11,7 +11,7 @@ import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
const URL = `http://localhost:5000/download`
|
||||
const URL = `https://nsq.site.quack-lab.dev/pub?topic=ytdqueue`
|
||||
|
||||
type Item struct {
|
||||
Link string `json:"link"`
|
||||
|
@@ -1,71 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"sync"
|
||||
|
||||
"github.com/gen2brain/beeep"
|
||||
"github.com/lrstanley/go-ytdlp"
|
||||
)
|
||||
|
||||
const OUTPUT_DIR = "C:/Users/Administrator/ytdlpVideos"
|
||||
|
||||
type DownloadWorker struct {
|
||||
id int
|
||||
input chan *DownloadTask
|
||||
}
|
||||
|
||||
var ongoingDownloads = make(map[string]struct{})
|
||||
var ongoingDownloadsMutex = &sync.Mutex{}
|
||||
|
||||
var dl = ytdlp.New().
|
||||
// FormatSort("bestvideo[ext=mp4]+bestaudio[ext=m4a]").
|
||||
FormatSort("res,ext:mp4:m4a").
|
||||
Output("C:/Users/Administrator/ytdlpVideos/%(uploader)s/%(title)s.%(ext)s").
|
||||
LimitRate("10M").
|
||||
// HTTPChunkSize("20M").
|
||||
MarkWatched().
|
||||
SponsorblockMark("all").
|
||||
RecodeVideo("mp4").
|
||||
ConcurrentFragments(6)
|
||||
|
||||
func (w *DownloadWorker) Run() {
|
||||
for {
|
||||
task, ok := <-w.input
|
||||
if !ok {
|
||||
log.Printf("DownloadWorker %d: input channel closed, exiting", w.id)
|
||||
return
|
||||
}
|
||||
_, ongoing := ongoingDownloads[task.Url]
|
||||
if ongoing {
|
||||
log.Printf("DownloadWorker %d: Download %s is already ongoing", w.id, task.Url)
|
||||
continue
|
||||
}
|
||||
ongoingDownloadsMutex.Lock()
|
||||
ongoingDownloads[task.Url] = struct{}{}
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
|
||||
log.Printf("DownloadWorker %d: Downloading %s", w.id, task.Url)
|
||||
|
||||
err := beeep.Beep(beeep.DefaultFreq, beeep.DefaultDuration)
|
||||
if err != nil {
|
||||
log.Printf("Failed beeping with %+v", err)
|
||||
}
|
||||
err = beeep.Alert("Download Started", task.Url, "assets/information.png")
|
||||
if err != nil {
|
||||
log.Printf("Failed alerting with %+v", err)
|
||||
}
|
||||
|
||||
_, err = dl.Run(context.TODO(), task.Url)
|
||||
if err != nil {
|
||||
log.Printf("DownloadWorker %d: Failed downloading %s with %+v", w.id, task.Url, err)
|
||||
continue
|
||||
}
|
||||
|
||||
log.Printf("DownloadWorker %d: Downloaded %s", w.id, task.Url)
|
||||
ongoingDownloadsMutex.Lock()
|
||||
delete(ongoingDownloads, task.Url)
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
}
|
||||
}
|
5
downloader/downloaders/downloader.go
Normal file
5
downloader/downloaders/downloader.go
Normal file
@@ -0,0 +1,5 @@
|
||||
package downloaders
|
||||
|
||||
type Downloader interface {
|
||||
Download(url string) error
|
||||
}
|
93
downloader/downloaders/kidai.go
Normal file
93
downloader/downloaders/kidai.go
Normal file
@@ -0,0 +1,93 @@
|
||||
package downloaders
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/gen2brain/beeep"
|
||||
"github.com/kkdai/youtube/v2"
|
||||
ytd "github.com/kkdai/youtube/v2/downloader"
|
||||
)
|
||||
|
||||
type KidaiDownloader struct{}
|
||||
|
||||
func (d *KidaiDownloader) Download(url string) error {
|
||||
_, ongoing := ongoingDownloads[url]
|
||||
if ongoing {
|
||||
// return fmt.Errorf("Download %s is already ongoing", url)
|
||||
log.Printf("Download %s is already ongoing", url)
|
||||
return nil
|
||||
}
|
||||
ongoingDownloadsMutex.Lock()
|
||||
ongoingDownloads[url] = struct{}{}
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
defer func() {
|
||||
ongoingDownloadsMutex.Lock()
|
||||
delete(ongoingDownloads, url)
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
}()
|
||||
|
||||
log.Printf("Kidai downloading %s", url)
|
||||
|
||||
go func() {
|
||||
err := beeep.Beep(beeep.DefaultFreq, beeep.DefaultDuration)
|
||||
if err != nil {
|
||||
log.Printf("Failed beeping with %+v", err)
|
||||
}
|
||||
err = beeep.Alert("Download Started", url, "assets/information.png")
|
||||
if err != nil {
|
||||
log.Printf("Failed alerting with %+v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
client := youtube.Client{}
|
||||
video, err := client.GetVideo(url)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed downloading %s with %+v", url, err)
|
||||
}
|
||||
// $ go run . download -m mp4 -q hd https://www.youtube.com/watch?v=D9trBXaXCgA
|
||||
// This works fine, now I have to figure out how to plug -m and -q into this shit below
|
||||
downloader := ytd.Downloader{
|
||||
OutputDir: OUTPUT_DIR,
|
||||
Client: client,
|
||||
}
|
||||
|
||||
videoTitle := Sanitize(video.Title)
|
||||
videoAuthor := Sanitize(video.Author)
|
||||
|
||||
videoFileRoot := filepath.Join(OUTPUT_DIR, videoAuthor)
|
||||
err = os.MkdirAll(videoFileRoot, 0755)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed creating directory %s with %+v", videoFileRoot, err)
|
||||
}
|
||||
|
||||
fullVideoPath := filepath.Join(OUTPUT_DIR, videoAuthor, videoTitle+".mp4")
|
||||
videoFileHandle, err := os.Open(fullVideoPath)
|
||||
if err == nil {
|
||||
videoFileHandle.Close()
|
||||
log.Printf("File %s already exists, skipping download", fullVideoPath)
|
||||
return nil
|
||||
}
|
||||
|
||||
videoFile := filepath.Join(videoAuthor, videoTitle+".mp4")
|
||||
err = downloader.DownloadComposite(context.Background(), videoFile, video, "hd", "mp4", "")
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed downloading %s with %+v", url, err)
|
||||
}
|
||||
|
||||
log.Printf("Downloaded %s", url)
|
||||
return nil
|
||||
}
|
||||
|
||||
var re = regexp.MustCompile(`[#!$%&/()";:?;,|*]`)
|
||||
|
||||
func Sanitize(s string) string {
|
||||
s = strings.ReplaceAll(s, "&", "and")
|
||||
s = re.ReplaceAllString(s, "")
|
||||
return s
|
||||
}
|
11
downloader/downloaders/utils.go
Normal file
11
downloader/downloaders/utils.go
Normal file
@@ -0,0 +1,11 @@
|
||||
package downloaders
|
||||
|
||||
import "sync"
|
||||
|
||||
const OUTPUT_DIR = "C:/Users/Administrator/ytdlpVideos"
|
||||
//const OUTPUT_DIR = "."
|
||||
|
||||
var ongoingDownloads = make(map[string]struct{})
|
||||
var ongoingDownloadsMutex = &sync.Mutex{}
|
||||
|
||||
const DOWNLOAD_WORKERS = 2
|
59
downloader/downloaders/ytdlp-lib.go
Normal file
59
downloader/downloaders/ytdlp-lib.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package downloaders
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"github.com/gen2brain/beeep"
|
||||
"github.com/lrstanley/go-ytdlp"
|
||||
)
|
||||
|
||||
var dl = ytdlp.New().
|
||||
// FormatSort("bestvideo[ext=mp4]+bestaudio[ext=m4a]").
|
||||
FormatSort("res,ext:mp4:m4a").
|
||||
Output("C:/Users/Administrator/ytdlpVideos/%(uploader)s/%(title)s.%(ext)s").
|
||||
LimitRate(fmt.Sprintf("%dM", 150/DOWNLOAD_WORKERS)).
|
||||
// HTTPChunkSize("20M").
|
||||
MarkWatched().
|
||||
SponsorblockMark("all").
|
||||
RecodeVideo("mp4").
|
||||
ConcurrentFragments(6)
|
||||
|
||||
type YTDLPLibDownloader struct{}
|
||||
|
||||
func (d *YTDLPLibDownloader) Download(url string) error {
|
||||
_, ongoing := ongoingDownloads[url]
|
||||
if ongoing {
|
||||
// return fmt.Errorf("Download %s is already ongoing", url)
|
||||
log.Printf("Download %s is already ongoing", url)
|
||||
return nil
|
||||
}
|
||||
ongoingDownloadsMutex.Lock()
|
||||
ongoingDownloads[url] = struct{}{}
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
|
||||
log.Printf("YTDLPLib downloading %s", url)
|
||||
|
||||
go func() {
|
||||
err := beeep.Beep(beeep.DefaultFreq, beeep.DefaultDuration)
|
||||
if err != nil {
|
||||
log.Printf("Failed beeping with %+v", err)
|
||||
}
|
||||
err = beeep.Alert("Download Started", url, "assets/information.png")
|
||||
if err != nil {
|
||||
log.Printf("Failed alerting with %+v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
_, err := dl.Run(context.TODO(), url)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed downloading %s with %+v", url, err)
|
||||
}
|
||||
|
||||
log.Printf("Downloaded %s", url)
|
||||
ongoingDownloadsMutex.Lock()
|
||||
delete(ongoingDownloads, url)
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
return nil
|
||||
}
|
51
downloader/downloaders/ytdlp-raw.go
Normal file
51
downloader/downloaders/ytdlp-raw.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package downloaders
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
|
||||
"github.com/gen2brain/beeep"
|
||||
)
|
||||
|
||||
type YTDLPRawDownloader struct{}
|
||||
|
||||
func (d *YTDLPRawDownloader) Download(url string) error {
|
||||
_, ongoing := ongoingDownloads[url]
|
||||
if ongoing {
|
||||
// return fmt.Errorf("Download %s is already ongoing", url)
|
||||
log.Printf("Download %s is already ongoing", url)
|
||||
return nil
|
||||
}
|
||||
ongoingDownloadsMutex.Lock()
|
||||
ongoingDownloads[url] = struct{}{}
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
|
||||
log.Printf("YTDLPRaw downloading %s", url)
|
||||
|
||||
go func() {
|
||||
err := beeep.Beep(beeep.DefaultFreq, beeep.DefaultDuration)
|
||||
if err != nil {
|
||||
log.Printf("Failed beeping with %+v", err)
|
||||
}
|
||||
err = beeep.Alert("Download Started", url, "assets/information.png")
|
||||
if err != nil {
|
||||
log.Printf("Failed alerting with %+v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
cmd := exec.Command("yt-dlp", "-o", "C:/Users/Administrator/ytdlpVideos/%(title)s.%(ext)s", "-f", "bestvideo[ext=mp4]+bestaudio[ext=m4a]", "-N", "12", "--sponsorblock-mark", "all", url)
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
err := cmd.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed downloading %s with %+v", url, err)
|
||||
}
|
||||
|
||||
log.Printf("Downloaded %s", url)
|
||||
ongoingDownloadsMutex.Lock()
|
||||
delete(ongoingDownloads, url)
|
||||
ongoingDownloadsMutex.Unlock()
|
||||
return nil
|
||||
}
|
@@ -3,18 +3,32 @@ module main
|
||||
go 1.22.4
|
||||
|
||||
require (
|
||||
github.com/ProtonMail/go-crypto v1.0.0 // indirect
|
||||
github.com/cloudflare/circl v1.3.7 // indirect
|
||||
github.com/ProtonMail/go-crypto v1.1.3 // indirect
|
||||
github.com/VividCortex/ewma v1.2.0 // indirect
|
||||
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d // indirect
|
||||
github.com/bitly/go-simplejson v0.5.1 // indirect
|
||||
github.com/cloudflare/circl v1.5.0 // indirect
|
||||
github.com/dlclark/regexp2 v1.11.4 // indirect
|
||||
github.com/dop251/goja v0.0.0-20241024094426-79f3a7efcdbd // indirect
|
||||
github.com/go-sourcemap/sourcemap v2.1.4+incompatible // indirect
|
||||
github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4 // indirect
|
||||
github.com/godbus/dbus/v5 v5.1.0 // indirect
|
||||
github.com/golang/snappy v0.0.4 // indirect
|
||||
github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad // indirect
|
||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
||||
github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d // indirect
|
||||
github.com/rivo/uniseg v0.4.7 // indirect
|
||||
github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af // indirect
|
||||
golang.org/x/crypto v0.21.0 // indirect
|
||||
golang.org/x/sys v0.18.0 // indirect
|
||||
github.com/vbauerster/mpb/v5 v5.4.0 // indirect
|
||||
golang.org/x/crypto v0.31.0 // indirect
|
||||
golang.org/x/sys v0.28.0 // indirect
|
||||
golang.org/x/text v0.21.0 // indirect
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4
|
||||
github.com/gorilla/websocket v1.5.3
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20240616011628-f35a10876c99
|
||||
github.com/kkdai/youtube/v2 v2.10.2
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20241221063727-6717edbb36dd
|
||||
github.com/nsqio/go-nsq v1.1.0
|
||||
)
|
||||
|
@@ -1,65 +1,70 @@
|
||||
github.com/ProtonMail/go-crypto v1.0.0 h1:LRuvITjQWX+WIfr930YHG2HNfjR1uOfyf5vE0kC2U78=
|
||||
github.com/ProtonMail/go-crypto v1.0.0/go.mod h1:EjAoLdwvbIOoOQr3ihjnSoLZRtE8azugULFRteWMNc0=
|
||||
github.com/bwesterb/go-ristretto v1.2.3/go.mod h1:fUIoIZaG73pV5biE2Blr2xEzDoMj7NFEuV9ekS419A0=
|
||||
github.com/cloudflare/circl v1.3.3/go.mod h1:5XYMA4rFBvNIrhs50XuiBJ15vF2pZn4nnUKZrLbUZFA=
|
||||
github.com/cloudflare/circl v1.3.7 h1:qlCDlTPz2n9fu58M0Nh1J/JzcFpfgkFHHX3O35r5vcU=
|
||||
github.com/cloudflare/circl v1.3.7/go.mod h1:sRTcRWXGLrKw6yIGJ+l7amYJFfAXbZG0kBSc8r4zxgA=
|
||||
github.com/Masterminds/semver/v3 v3.2.1 h1:RN9w6+7QoMeJVGyfmbcgs28Br8cvmnucEXnY0rYXWg0=
|
||||
github.com/Masterminds/semver/v3 v3.2.1/go.mod h1:qvl/7zhW3nngYb5+80sSMF+FG2BjYrf8m9wsX0PNOMQ=
|
||||
github.com/ProtonMail/go-crypto v1.1.3 h1:nRBOetoydLeUb4nHajyO2bKqMLfWQ/ZPwkXqXxPxCFk=
|
||||
github.com/ProtonMail/go-crypto v1.1.3/go.mod h1:rA3QumHc/FZ8pAHreoekgiAbzpNsfQAosU5td4SnOrE=
|
||||
github.com/VividCortex/ewma v1.1.1/go.mod h1:2Tkkvm3sRDVXaiyucHiACn4cqf7DpdyLvmxzcbUokwA=
|
||||
github.com/VividCortex/ewma v1.2.0 h1:f58SaIzcDXrSy3kWaHNvuJgJ3Nmz59Zji6XoJR/q1ow=
|
||||
github.com/VividCortex/ewma v1.2.0/go.mod h1:nz4BbCtbLyFDeC9SUHbtcT5644juEuWfUAUnGx7j5l4=
|
||||
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d h1:licZJFw2RwpHMqeKTCYkitsPqHNxTmd4SNR5r94FGM8=
|
||||
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d/go.mod h1:asat636LX7Bqt5lYEZ27JNDcqxfjdBQuJ/MM4CN/Lzo=
|
||||
github.com/bitly/go-simplejson v0.5.1 h1:xgwPbetQScXt1gh9BmoJ6j9JMr3TElvuIyjR8pgdoow=
|
||||
github.com/bitly/go-simplejson v0.5.1/go.mod h1:YOPVLzCfwK14b4Sff3oP1AmGhI9T9Vsg84etUnlyp+Q=
|
||||
github.com/cloudflare/circl v1.5.0 h1:hxIWksrX6XN5a1L2TI/h53AGPhNHoUBo+TD1ms9+pys=
|
||||
github.com/cloudflare/circl v1.5.0/go.mod h1:uddAzsPgqdMAYatqJ0lsjX1oECcQLIlRpzZh3pJrofs=
|
||||
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
|
||||
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/dlclark/regexp2 v1.11.4 h1:rPYF9/LECdNymJufQKmri9gV604RvvABwgOA8un7yAo=
|
||||
github.com/dlclark/regexp2 v1.11.4/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
|
||||
github.com/dop251/goja v0.0.0-20241024094426-79f3a7efcdbd h1:QMSNEh9uQkDjyPwu/J541GgSH+4hw+0skJDIj9HJ3mE=
|
||||
github.com/dop251/goja v0.0.0-20241024094426-79f3a7efcdbd/go.mod h1:MxLav0peU43GgvwVgNbLAj1s/bSGboKkhuULvq/7hx4=
|
||||
github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4 h1:ygs9POGDQpQGLJPlq4+0LBUmMBNox1N4JSpw+OETcvI=
|
||||
github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4/go.mod h1:0W7dI87PvXJ1Sjs0QPvWXKcQmNERY77e8l7GFhZB/s4=
|
||||
github.com/go-sourcemap/sourcemap v2.1.4+incompatible h1:a+iTbH5auLKxaNwQFg0B+TCYl6lbukKPc7b5x0n1s6Q=
|
||||
github.com/go-sourcemap/sourcemap v2.1.4+incompatible/go.mod h1:F8jJfvm2KbVjc5NqelyYJmf/v5J0dwNLS2mL4sNA1Jg=
|
||||
github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4 h1:qZNfIGkIANxGv/OqtnntR4DfOY2+BgwR60cAcu/i3SE=
|
||||
github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4/go.mod h1:kW3HQ4UdaAyrUCSSDR4xUzBKW6O2iA4uHhk7AtyYp10=
|
||||
github.com/godbus/dbus/v5 v5.1.0 h1:4KLkAxT3aOY8Li4FRJe/KvhoNFFxo0m6fNuFUO8QJUk=
|
||||
github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||
github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||
github.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=
|
||||
github.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||
github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad h1:a6HEuzUHeKH6hwfN/ZoQgRgVIWFJljSWa/zetS2WTvg=
|
||||
github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad/go.mod h1:vavhavw2zAxS5dIdcRluK6cSGGPlZynqzFM8NdvU144=
|
||||
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
||||
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20240616011628-f35a10876c99 h1:ZAo7qJht9PqefOD7C0ZKQ8dEkpJeM955sYw0FtQnzvo=
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20240616011628-f35a10876c99/go.mod h1:75ujbafjqiJugIGw4K6o52/p8C0m/kt+DrYwgClXYT4=
|
||||
github.com/kkdai/youtube/v2 v2.10.2 h1:e3JslUDiKEfjMzxFyrOh3O59C/aLfKNZyrcav00MZV0=
|
||||
github.com/kkdai/youtube/v2 v2.10.2/go.mod h1:4y1MIg7f1o5/kQfkr7nwXFtv8PGSoe4kChOB9/iMA88=
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20241221063727-6717edbb36dd h1:lLajTMgNTs/W4H05uQYnJDRIbIvHk6XXy7DQNFRbvzU=
|
||||
github.com/lrstanley/go-ytdlp v0.0.0-20241221063727-6717edbb36dd/go.mod h1:75ujbafjqiJugIGw4K6o52/p8C0m/kt+DrYwgClXYT4=
|
||||
github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
|
||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/nsqio/go-nsq v1.1.0 h1:PQg+xxiUjA7V+TLdXw7nVrJ5Jbl3sN86EhGCQj4+FYE=
|
||||
github.com/nsqio/go-nsq v1.1.0/go.mod h1:vKq36oyeVXgsS5Q8YEO7WghqidAVXQlcFxzQbQTuDEY=
|
||||
github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d h1:VhgPp6v9qf9Agr/56bj7Y/xa04UccTW04VP0Qed4vnQ=
|
||||
github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d/go.mod h1:YUTz3bUH2ZwIWBy3CJBeOBEugqcmXREj14T+iG/4k4U=
|
||||
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
|
||||
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
|
||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||
github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af h1:6yITBqGTE2lEeTPG04SN9W+iWHCRyHqlVYILiSXziwk=
|
||||
github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af/go.mod h1:4F09kP5F+am0jAwlQLddpoMDM+iewkxxt6nxUQ5nq5o=
|
||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||
golang.org/x/crypto v0.3.1-0.20221117191849-2c476679df9a/go.mod h1:hebNnKkNXi2UzZN1eVRvBB7co0a+JxK6XbPiWVs/3J4=
|
||||
golang.org/x/crypto v0.7.0/go.mod h1:pYwdfH91IfpZVANVyUOhSIPZaFoJGxTFbZhFTx+dXZU=
|
||||
golang.org/x/crypto v0.21.0 h1:X31++rzVUdKhX5sWmSOFZxx8UW/ldWx55cbf08iNAMA=
|
||||
golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||
golang.org/x/net v0.2.0/go.mod h1:KqCZLdyyvdV855qA2rE3GC2aiw5xGR5TEjj8smXukLY=
|
||||
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
||||
golang.org/x/net v0.8.0/go.mod h1:QVkue5JL9kW//ek3r6jTKnTFis1tRmNAW2P1shuFdJc=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
github.com/vbauerster/mpb/v5 v5.4.0 h1:n8JPunifvQvh6P1D1HAl2Ur9YcmKT1tpoUuiea5mlmg=
|
||||
github.com/vbauerster/mpb/v5 v5.4.0/go.mod h1:fi4wVo7BVQ22QcvFObm+VwliQXlV1eBT8JDaKXR4JGI=
|
||||
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
|
||||
golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
|
||||
golang.org/x/net v0.31.0 h1:68CPQngjLL0r2AlUKiSxtQFKvzRVbnzLwMUn5SzcLHo=
|
||||
golang.org/x/net v0.31.0/go.mod h1:P4fl1q7dY2hnZFxEk4pPSkDHF+QqjitcnDjUQyMM+pM=
|
||||
golang.org/x/sys v0.0.0-20201218084310-7d0127a74742/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
|
||||
golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/term v0.2.0/go.mod h1:TVmDHMZPmdnySmBfhjOoOdhjzdE1h4u1VwSiw2l1Nuc=
|
||||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
||||
golang.org/x/term v0.6.0/go.mod h1:m6U89DPEgQRMq3DNkDClhWw02AUbt2daBVO4cn4Hv9U=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||
golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/text v0.8.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
|
||||
golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
|
||||
golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
|
||||
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
@@ -1,13 +1,23 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"main/downloaders"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/nsqio/go-nsq"
|
||||
)
|
||||
|
||||
var Error *log.Logger
|
||||
var Warning *log.Logger
|
||||
|
||||
func init() {
|
||||
log.SetFlags(log.Lmicroseconds | log.Lshortfile)
|
||||
logFile, err := os.Create("main.log")
|
||||
@@ -17,54 +27,91 @@ func init() {
|
||||
}
|
||||
logger := io.MultiWriter(os.Stdout, logFile)
|
||||
log.SetOutput(logger)
|
||||
|
||||
Error = log.New(io.MultiWriter(logFile, os.Stderr, os.Stdout),
|
||||
fmt.Sprintf("%sERROR:%s ", "\033[0;101m", "\033[0m"),
|
||||
log.Lmicroseconds|log.Lshortfile)
|
||||
Warning = log.New(io.MultiWriter(logFile, os.Stdout),
|
||||
fmt.Sprintf("%sWarning:%s ", "\033[0;93m", "\033[0m"),
|
||||
log.Lmicroseconds|log.Lshortfile)
|
||||
}
|
||||
|
||||
const DOWNLOAD_WORKERS = 10
|
||||
//var downloader downloaders.Downloader = &downloaders.YTDLPRawDownloader{}
|
||||
var downloader downloaders.Downloader = &downloaders.KidaiDownloader{}
|
||||
//var downloader downloaders.Downloader = &downloaders.YTDLPLibDownloader{}
|
||||
|
||||
var downloadQueue = make(chan *DownloadTask, 100)
|
||||
type DLHandler struct{}
|
||||
|
||||
func enableCORS(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
|
||||
if r.Method == http.MethodOptions {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func handleDownload(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
http.Error(w, "Invalid request method", http.StatusMethodNotAllowed)
|
||||
return
|
||||
func (*DLHandler) HandleMessage(message *nsq.Message) error {
|
||||
log.Printf("Received message '%s' with %d attempts", message.Body, message.Attempts)
|
||||
data := DownloadRequest{}
|
||||
err := json.Unmarshal(message.Body, &data)
|
||||
if err != nil {
|
||||
Error.Printf("Error unmarshalling message: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
var req DownloadRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
log.Printf("Error parsing JSON: %v", err)
|
||||
http.Error(w, "Error parsing JSON", http.StatusBadRequest)
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
go func() {
|
||||
ticker := time.NewTicker(1 * time.Second)
|
||||
defer ticker.Stop()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ticker.C:
|
||||
message.Touch()
|
||||
case <-ctx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
downloadQueue <- &DownloadTask{Url: req.Link}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
err = downloader.Download(data.Link)
|
||||
if err != nil {
|
||||
Error.Printf("Error downloading %s: %v", data.Link, err)
|
||||
return err
|
||||
}
|
||||
message.Finish()
|
||||
return nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
for i := 0; i < DOWNLOAD_WORKERS; i++ {
|
||||
worker := &DownloadWorker{id: i, input: downloadQueue}
|
||||
go worker.Run()
|
||||
// err := downloader.Download("https://www.youtube.com/watch?v=SiKjprtiPaw")
|
||||
// if err != nil {
|
||||
// Error.Printf("Error downloading: %v", err)
|
||||
// }
|
||||
// return
|
||||
|
||||
config := nsq.NewConfig()
|
||||
config.MaxAttempts = 5
|
||||
config.MaxInFlight = downloaders.DOWNLOAD_WORKERS
|
||||
config.MsgTimeout = 10 * time.Second
|
||||
|
||||
consumer, err := nsq.NewConsumer("ytdqueue", "dl", config)
|
||||
if err != nil {
|
||||
Error.Printf("Error creating consumer: %v", err)
|
||||
return
|
||||
}
|
||||
for i := 0; i < downloaders.DOWNLOAD_WORKERS; i++ {
|
||||
consumer.AddHandler(&DLHandler{})
|
||||
}
|
||||
|
||||
mux := http.NewServeMux()
|
||||
mux.Handle("/download", enableCORS(http.HandlerFunc(handleDownload)))
|
||||
log.Println("Server starting on :5000")
|
||||
err := http.ListenAndServe(":5000", mux)
|
||||
err = consumer.ConnectToNSQD("nsq.site.quack-lab.dev:41505")
|
||||
if err != nil {
|
||||
log.Println("Error starting server:", err)
|
||||
Error.Printf("Error connecting to nsqlookupd: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
sigChan := make(chan os.Signal, 1)
|
||||
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
<-sigChan
|
||||
log.Println("Received signal to terminate. Initiating graceful shutdown...")
|
||||
|
||||
consumer.Stop()
|
||||
<-consumer.StopChan
|
||||
|
||||
log.Println("Graceful shutdown completed.")
|
||||
}
|
||||
|
@@ -1,8 +1,5 @@
|
||||
package main
|
||||
|
||||
type DownloadTask struct {
|
||||
Url string
|
||||
}
|
||||
type DownloadRequest struct {
|
||||
Link string `json:"link"`
|
||||
}
|
8
downloader/vendor/github.com/kkdai/youtube/v2/.gitignore
generated
vendored
Normal file
8
downloader/vendor/github.com/kkdai/youtube/v2/.gitignore
generated
vendored
Normal file
@@ -0,0 +1,8 @@
|
||||
/.idea
|
||||
/.vscode
|
||||
download_test
|
||||
/bin
|
||||
/dist
|
||||
/output
|
||||
*.out
|
||||
.DS_Store
|
8
downloader/vendor/github.com/kkdai/youtube/v2/.golangci.yml
generated
vendored
Normal file
8
downloader/vendor/github.com/kkdai/youtube/v2/.golangci.yml
generated
vendored
Normal file
@@ -0,0 +1,8 @@
|
||||
issues:
|
||||
exclude:
|
||||
- Subprocess launched with function call
|
||||
# Excluding configuration per-path, per-linter, per-text and per-source
|
||||
exclude-rules:
|
||||
# Exclude some linters from running on tests files.
|
||||
- path: cmd/
|
||||
text: InsecureSkipVerify
|
20
downloader/vendor/github.com/kkdai/youtube/v2/.goreleaser.yml
generated
vendored
Normal file
20
downloader/vendor/github.com/kkdai/youtube/v2/.goreleaser.yml
generated
vendored
Normal file
@@ -0,0 +1,20 @@
|
||||
project_name: youtubedr
|
||||
|
||||
env:
|
||||
# Build without CGO to don't depend on specific glibc versions
|
||||
- CGO_ENABLED=0
|
||||
|
||||
builds:
|
||||
- main: ./cmd/youtubedr
|
||||
binary: youtubedr
|
||||
goos:
|
||||
- windows
|
||||
- darwin
|
||||
- linux
|
||||
- freebsd
|
||||
goarch:
|
||||
- amd64
|
||||
- arm
|
||||
- arm64
|
||||
flags:
|
||||
- -trimpath
|
22
downloader/vendor/github.com/kkdai/youtube/v2/LICENSE
generated
vendored
Normal file
22
downloader/vendor/github.com/kkdai/youtube/v2/LICENSE
generated
vendored
Normal file
@@ -0,0 +1,22 @@
|
||||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2015 Evan Lin
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
|
57
downloader/vendor/github.com/kkdai/youtube/v2/Makefile
generated
vendored
Normal file
57
downloader/vendor/github.com/kkdai/youtube/v2/Makefile
generated
vendored
Normal file
@@ -0,0 +1,57 @@
|
||||
FILES_TO_FMT ?= $(shell find . -path ./vendor -prune -o -name '*.go' -print)
|
||||
LOGLEVEL ?= debug
|
||||
|
||||
## help: Show makefile commands
|
||||
.PHONY: help
|
||||
help: Makefile
|
||||
@echo "---- Project: kkdai/youtube ----"
|
||||
@echo " Usage: make COMMAND"
|
||||
@echo
|
||||
@echo " Management Commands:"
|
||||
@sed -n 's/^##//p' $< | column -t -s ':' | sed -e 's/^/ /'
|
||||
@echo
|
||||
|
||||
## build: Build project
|
||||
.PHONY: build
|
||||
build:
|
||||
goreleaser --rm-dist
|
||||
|
||||
## deps: Ensures fresh go.mod and go.sum
|
||||
.PHONY: deps
|
||||
deps:
|
||||
go mod tidy
|
||||
go mod verify
|
||||
|
||||
## lint: Run golangci-lint check
|
||||
.PHONY: lint
|
||||
lint:
|
||||
command -v golangci-lint || curl -sSfL https://raw.githubusercontent.com/golangci/golangci-lint/master/install.sh | sh -s -- -b $$(go env GOPATH)/bin $(GOLANGCI_LINT_VERSION)
|
||||
echo "golangci-lint checking..."
|
||||
golangci-lint run --timeout=30m --enable=misspell --enable=gosec --enable=gofmt --enable=goimports --enable=revive ./cmd/... ./...
|
||||
go vet ./...
|
||||
|
||||
## format: Formats Go code
|
||||
.PHONY: format
|
||||
format:
|
||||
@echo ">> formatting code"
|
||||
@gofmt -s -w $(FILES_TO_FMT)
|
||||
|
||||
## test-unit: Run all Youtube Go unit tests
|
||||
.PHONY: test-unit
|
||||
test-unit:
|
||||
LOGLEVEL=${LOGLEVEL} go test -v -cover ./...
|
||||
|
||||
## test-integration: Run all Youtube Go integration tests
|
||||
.PHONY: test-integration
|
||||
test-integration:
|
||||
mkdir -p output
|
||||
rm -f output/*
|
||||
LOGLEVEL=${LOGLEVEL} ARTIFACTS=output go test -v -race -covermode=atomic -coverprofile=coverage.out -tags=integration ./...
|
||||
|
||||
.PHONY: coverage.out
|
||||
coverage.out:
|
||||
|
||||
## clean: Clean files and downloaded videos from builds during development
|
||||
.PHONY: clean
|
||||
clean:
|
||||
rm -rf dist *.mp4 *.mkv
|
161
downloader/vendor/github.com/kkdai/youtube/v2/README.md
generated
vendored
Normal file
161
downloader/vendor/github.com/kkdai/youtube/v2/README.md
generated
vendored
Normal file
@@ -0,0 +1,161 @@
|
||||
Download Youtube Video in Golang
|
||||
==================
|
||||
|
||||
[](https://raw.githubusercontent.com/kkdai/youtube/master/LICENSE)
|
||||
[](https://pkg.go.dev/github.com/kkdai/youtube/v2)
|
||||
[](https://github.com/kkdai/youtube/actions)
|
||||
[](https://codecov.io/gh/kkdai/youtube)
|
||||
[](https://goreportcard.com/badge/github.com/kkdai/youtube)
|
||||
|
||||
|
||||
This package is a Youtube video download package, for more detail refer [https://github.com/ytdl-org/youtube-dl](https://github.com/ytdl-org/youtube-dl) for more download options.
|
||||
|
||||
This tool is meant to be used to download CC0 licenced content, we do not support nor recommend using it for illegal activities.
|
||||
|
||||
## Overview
|
||||
* [Install](#installation)
|
||||
* [Usage](#usage)
|
||||
* [Example: Download video from \[dotGo 2015 - Rob Pike - Simplicity is Complicated\]](#download-dotGo-2015-rob-pike-video)
|
||||
|
||||
## Installation
|
||||
|
||||
### Install via go get
|
||||
|
||||
Please ensure you have installed Go 1.22 or later.
|
||||
|
||||
```shell
|
||||
go get github.com/kkdai/youtube/v2
|
||||
```
|
||||
|
||||
### From source code
|
||||
|
||||
```shell
|
||||
git clone https://github.com/kkdai/youtube.git
|
||||
cd youtube
|
||||
go run ./cmd/youtubedr
|
||||
```
|
||||
|
||||
### Mac
|
||||
|
||||
```shell
|
||||
brew install youtubedr
|
||||
```
|
||||
|
||||
### in Termux
|
||||
```shell
|
||||
pkg install youtubedr
|
||||
```
|
||||
### You can also find this package in
|
||||
- [archlinux](https://aur.archlinux.org/packages/youtubedr/) (thanks to [cjsthompson](https://github.com/cjsthompson))
|
||||
- [Termux package](https://github.com/termux/termux-packages/tree/master/packages/youtubedr) (thanks to [kcubeterm](https://github.com/kcubeterm))
|
||||
- [Homebrew](https://formulae.brew.sh/formula/youtubedr) (thanks to [kkc](https://github.com/kkc))
|
||||
|
||||
## Usage
|
||||
|
||||
### Use the binary directly
|
||||
It's really simple to use, just get the video id from youtube url - ex: `https://www.youtube.com/watch?v=rFejpH_tAHM`, the video id is `rFejpH_tAHM`
|
||||
|
||||
```shell
|
||||
$ youtubedr download QAGDGja7kbs
|
||||
$ youtubedr download https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
|
||||
### Use this package in your golang program
|
||||
|
||||
Please check out the [example_test.go](example_test.go) for example code.
|
||||
|
||||
|
||||
## Example:
|
||||
* ### Get information of dotGo-2015-rob-pike video for downloading
|
||||
|
||||
`go get github.com/kkdai/youtube/v2/youtubedr`
|
||||
|
||||
Download video from [dotGo 2015 - Rob Pike - Simplicity is Complicated](https://www.youtube.com/watch?v=rFejpH_tAHM)
|
||||
|
||||
```
|
||||
youtubedr info https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
|
||||
Title: dotGo 2015 - Rob Pike - Simplicity is Complicated
|
||||
Author: dotconferences
|
||||
-----available streams-----
|
||||
itag: 18 , quality: medium , type: video/mp4; codecs="avc1.42001E, mp4a.40.2"
|
||||
itag: 22 , quality: hd720 , type: video/mp4; codecs="avc1.64001F, mp4a.40.2"
|
||||
itag: 137 , quality: hd1080 , type: video/mp4; codecs="avc1.640028"
|
||||
itag: 248 , quality: hd1080 , type: video/webm; codecs="vp9"
|
||||
........
|
||||
```
|
||||
* ### Download dotGo-2015-rob-pike-video
|
||||
|
||||
`go get github.com/kkdai/youtube/v2/youtubedr`
|
||||
|
||||
Download video from [dotGo 2015 - Rob Pike - Simplicity is Complicated](https://www.youtube.com/watch?v=rFejpH_tAHM)
|
||||
|
||||
```
|
||||
youtubedr download https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
* ### Download video to specific folder and name
|
||||
|
||||
`go get github.com/kkdai/youtube/v2/youtubedr`
|
||||
|
||||
Download video from [dotGo 2015 - Rob Pike - Simplicity is Complicated](https://www.youtube.com/watch?v=rFejpH_tAHM) to current directory and name the file to simplicity-is-complicated.mp4
|
||||
|
||||
```
|
||||
youtubedr download -d ./ -o simplicity-is-complicated.mp4 https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
* ### Download video with specific quality
|
||||
|
||||
`go get github.com/kkdai/youtube/v2/youtubedr`
|
||||
|
||||
Download video from [dotGo 2015 - Rob Pike - Simplicity is Complicated](https://www.youtube.com/watch?v=rFejpH_tAHM) with specific quality
|
||||
|
||||
```
|
||||
youtubedr download -q medium https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
#### Special case by quality hd1080:
|
||||
Installation of ffmpeg is necessary for hd1080
|
||||
```
|
||||
ffmpeg //check ffmpeg is installed, if not please download ffmpeg and set to your PATH.
|
||||
youtubedr download -q hd1080 https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
|
||||
* ### Download video with specific itag
|
||||
|
||||
`go get github.com/kkdai/youtube/v2/youtubedr`
|
||||
|
||||
Download video from [dotGo 2015 - Rob Pike - Simplicity is Complicated](https://www.youtube.com/watch?v=rFejpH_tAHM)
|
||||
|
||||
```
|
||||
youtubedr download -q 18 https://www.youtube.com/watch?v=rFejpH_tAHM
|
||||
```
|
||||
|
||||
## How it works
|
||||
|
||||
- Parse the video ID you input in URL
|
||||
- ex: `https://www.youtube.com/watch?v=rFejpH_tAHM`, the video id is `rFejpH_tAHM`
|
||||
- Get video information via video id.
|
||||
- Use URL: `http://youtube.com/get_video_info?video_id=`
|
||||
- Parse and decode video information.
|
||||
- Download URL in "url="
|
||||
- title in "title="
|
||||
- Download video from URL
|
||||
- Need the string combination of "url"
|
||||
|
||||
## Inspired
|
||||
- [https://github.com/ytdl-org/youtube-dl](https://github.com/ytdl-org/youtube-dl)
|
||||
- [https://github.com/lepidosteus/youtube-dl](https://github.com/lepidosteus/youtube-dl)
|
||||
- [拆解 Youtube 影片下載位置](http://hkgoldenmra.blogspot.tw/2013/05/youtube.html)
|
||||
- [iawia002/annie](https://github.com/iawia002/annie)
|
||||
- [How to get url from obfuscate video info: youtube video downloader with php](https://stackoverflow.com/questions/60607291/youtube-video-downloader-with-php)
|
||||
|
||||
|
||||
## Project52
|
||||
It is one of my [project 52](https://github.com/kkdai/project52).
|
||||
|
||||
|
||||
## License
|
||||
This package is licensed under MIT license. See LICENSE for details.
|
11
downloader/vendor/github.com/kkdai/youtube/v2/YoutubeChange.md
generated
vendored
Normal file
11
downloader/vendor/github.com/kkdai/youtube/v2/YoutubeChange.md
generated
vendored
Normal file
@@ -0,0 +1,11 @@
|
||||
Tracking Youtube decipher change
|
||||
|
||||
### 2022/01/21:
|
||||
|
||||
#### action objects:
|
||||
|
||||
var $z={fA:function(a){a.reverse()},S2:function(a,b){a.splice(0,b)},l6:function(a,b){var c=a[0];a[0]=a[b%a.length];a[b%a.length]=c}};
|
||||
#### actions function:
|
||||
|
||||
Npa=function(a){a=a.split("");$z.S2(a,3);$z.fA(a,45);$z.l6(a,31);$z.S2(a,1);$z.fA(a,63);$z.S2(a,2);$z.fA(a,68);return a.join("")};
|
||||
|
29
downloader/vendor/github.com/kkdai/youtube/v2/artifacts.go
generated
vendored
Normal file
29
downloader/vendor/github.com/kkdai/youtube/v2/artifacts.go
generated
vendored
Normal file
@@ -0,0 +1,29 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"log/slog"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
// destination for artifacts, used by integration tests
|
||||
var artifactsFolder = os.Getenv("ARTIFACTS")
|
||||
|
||||
func writeArtifact(name string, content []byte) {
|
||||
// Ensure folder exists
|
||||
err := os.MkdirAll(artifactsFolder, os.ModePerm)
|
||||
if err != nil {
|
||||
slog.Error("unable to create artifacts folder", "path", artifactsFolder, "error", err)
|
||||
return
|
||||
}
|
||||
|
||||
path := filepath.Join(artifactsFolder, name)
|
||||
err = os.WriteFile(path, content, 0600)
|
||||
|
||||
log := slog.With("path", path)
|
||||
if err != nil {
|
||||
log.Error("unable to write artifact", "error", err)
|
||||
} else {
|
||||
log.Debug("artifact created")
|
||||
}
|
||||
}
|
632
downloader/vendor/github.com/kkdai/youtube/v2/client.go
generated
vendored
Normal file
632
downloader/vendor/github.com/kkdai/youtube/v2/client.go
generated
vendored
Normal file
@@ -0,0 +1,632 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log/slog"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"sync/atomic"
|
||||
)
|
||||
|
||||
const (
|
||||
Size1Kb = 1024
|
||||
Size1Mb = Size1Kb * 1024
|
||||
Size10Mb = Size1Mb * 10
|
||||
|
||||
playerParams = "CgIQBg=="
|
||||
)
|
||||
|
||||
var ErrNoFormat = errors.New("no video format provided")
|
||||
|
||||
// DefaultClient type to use. No reason to change but you could if you wanted to.
|
||||
var DefaultClient = IOSClient
|
||||
|
||||
// Client offers methods to download video metadata and video streams.
|
||||
type Client struct {
|
||||
// HTTPClient can be used to set a custom HTTP client.
|
||||
// If not set, http.DefaultClient will be used
|
||||
HTTPClient *http.Client
|
||||
|
||||
// MaxRoutines to use when downloading a video.
|
||||
MaxRoutines int
|
||||
|
||||
// ChunkSize to use when downloading videos in chunks. Default is Size10Mb.
|
||||
ChunkSize int64
|
||||
|
||||
// playerCache caches the JavaScript code of a player response
|
||||
playerCache playerCache
|
||||
|
||||
client *clientInfo
|
||||
|
||||
consentID string
|
||||
}
|
||||
|
||||
func (c *Client) assureClient() {
|
||||
if c.client == nil {
|
||||
c.client = &DefaultClient
|
||||
}
|
||||
}
|
||||
|
||||
// GetVideo fetches video metadata
|
||||
func (c *Client) GetVideo(url string) (*Video, error) {
|
||||
return c.GetVideoContext(context.Background(), url)
|
||||
}
|
||||
|
||||
// GetVideoContext fetches video metadata with a context
|
||||
func (c *Client) GetVideoContext(ctx context.Context, url string) (*Video, error) {
|
||||
id, err := ExtractVideoID(url)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("extractVideoID failed: %w", err)
|
||||
}
|
||||
|
||||
return c.videoFromID(ctx, id)
|
||||
}
|
||||
|
||||
func (c *Client) videoFromID(ctx context.Context, id string) (*Video, error) {
|
||||
c.assureClient()
|
||||
|
||||
body, err := c.videoDataByInnertube(ctx, id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
v := Video{
|
||||
ID: id,
|
||||
}
|
||||
|
||||
// return early if all good
|
||||
if err = v.parseVideoInfo(body); err == nil {
|
||||
return &v, nil
|
||||
}
|
||||
|
||||
// If the uploader has disabled embedding the video on other sites, parse video page
|
||||
if errors.Is(err, ErrNotPlayableInEmbed) {
|
||||
// additional parameters are required to access clips with sensitiv content
|
||||
html, err := c.httpGetBodyBytes(ctx, "https://www.youtube.com/watch?v="+id+"&bpctr=9999999999&has_verified=1")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &v, v.parseVideoPage(html)
|
||||
}
|
||||
|
||||
// If the uploader marked the video as inappropriate for some ages, use embed player
|
||||
if errors.Is(err, ErrLoginRequired) {
|
||||
c.client = &EmbeddedClient
|
||||
|
||||
bodyEmbed, errEmbed := c.videoDataByInnertube(ctx, id)
|
||||
if errEmbed == nil {
|
||||
errEmbed = v.parseVideoInfo(bodyEmbed)
|
||||
}
|
||||
|
||||
if errEmbed == nil {
|
||||
return &v, nil
|
||||
}
|
||||
|
||||
// private video clearly not age-restricted and thus should be explicit
|
||||
if errEmbed == ErrVideoPrivate {
|
||||
return &v, errEmbed
|
||||
}
|
||||
|
||||
// wrapping error so its clear whats happened
|
||||
return &v, fmt.Errorf("can't bypass age restriction: %w", errEmbed)
|
||||
}
|
||||
|
||||
// undefined error
|
||||
return &v, err
|
||||
}
|
||||
|
||||
type innertubeRequest struct {
|
||||
VideoID string `json:"videoId,omitempty"`
|
||||
BrowseID string `json:"browseId,omitempty"`
|
||||
Continuation string `json:"continuation,omitempty"`
|
||||
Context inntertubeContext `json:"context"`
|
||||
PlaybackContext *playbackContext `json:"playbackContext,omitempty"`
|
||||
ContentCheckOK bool `json:"contentCheckOk,omitempty"`
|
||||
RacyCheckOk bool `json:"racyCheckOk,omitempty"`
|
||||
Params string `json:"params"`
|
||||
}
|
||||
|
||||
type playbackContext struct {
|
||||
ContentPlaybackContext contentPlaybackContext `json:"contentPlaybackContext"`
|
||||
}
|
||||
|
||||
type contentPlaybackContext struct {
|
||||
// SignatureTimestamp string `json:"signatureTimestamp"`
|
||||
HTML5Preference string `json:"html5Preference"`
|
||||
}
|
||||
|
||||
type inntertubeContext struct {
|
||||
Client innertubeClient `json:"client"`
|
||||
}
|
||||
|
||||
type innertubeClient struct {
|
||||
HL string `json:"hl"`
|
||||
GL string `json:"gl"`
|
||||
ClientName string `json:"clientName"`
|
||||
ClientVersion string `json:"clientVersion"`
|
||||
AndroidSDKVersion int `json:"androidSDKVersion,omitempty"`
|
||||
UserAgent string `json:"userAgent,omitempty"`
|
||||
TimeZone string `json:"timeZone"`
|
||||
UTCOffset int `json:"utcOffsetMinutes"`
|
||||
DeviceModel string `json:"deviceModel,omitempty"`
|
||||
}
|
||||
|
||||
// client info for the innertube API
|
||||
type clientInfo struct {
|
||||
name string
|
||||
key string
|
||||
version string
|
||||
userAgent string
|
||||
androidVersion int
|
||||
deviceModel string
|
||||
}
|
||||
|
||||
var (
|
||||
// WebClient, better to use Android client but go ahead.
|
||||
WebClient = clientInfo{
|
||||
name: "WEB",
|
||||
version: "2.20220801.00.00",
|
||||
key: "AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8",
|
||||
userAgent: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
|
||||
}
|
||||
|
||||
// AndroidClient, download go brrrrrr.
|
||||
AndroidClient = clientInfo{
|
||||
name: "ANDROID",
|
||||
version: "18.11.34",
|
||||
key: "AIzaSyA8eiZmM1FaDVjRy-df2KTyQ_vz_yYM39w",
|
||||
userAgent: "com.google.android.youtube/18.11.34 (Linux; U; Android 11) gzip",
|
||||
androidVersion: 30,
|
||||
}
|
||||
|
||||
// IOSClient Client based brrrr.
|
||||
IOSClient = clientInfo{
|
||||
name: "IOS",
|
||||
version: "19.45.4",
|
||||
key: "AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8",
|
||||
userAgent: "com.google.ios.youtube/19.45.4 (iPhone16,2; U; CPU iOS 18_1_0 like Mac OS X;)",
|
||||
deviceModel: "iPhone16,2",
|
||||
}
|
||||
|
||||
// EmbeddedClient, not really tested.
|
||||
EmbeddedClient = clientInfo{
|
||||
name: "WEB_EMBEDDED_PLAYER",
|
||||
version: "1.19700101",
|
||||
key: "AIzaSyAO_FJ2SlqU8Q4STEHLGCilw_Y9_11qcW8", // seems like same key works for both clients
|
||||
userAgent: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
|
||||
}
|
||||
)
|
||||
|
||||
func (c *Client) videoDataByInnertube(ctx context.Context, id string) ([]byte, error) {
|
||||
data := innertubeRequest{
|
||||
VideoID: id,
|
||||
Context: prepareInnertubeContext(*c.client),
|
||||
ContentCheckOK: true,
|
||||
RacyCheckOk: true,
|
||||
// Params: playerParams,
|
||||
PlaybackContext: &playbackContext{
|
||||
ContentPlaybackContext: contentPlaybackContext{
|
||||
// SignatureTimestamp: sts,
|
||||
HTML5Preference: "HTML5_PREF_WANTS",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
return c.httpPostBodyBytes(ctx, "https://www.youtube.com/youtubei/v1/player?key="+c.client.key, data)
|
||||
}
|
||||
|
||||
func (c *Client) transcriptDataByInnertube(ctx context.Context, id string, lang string) ([]byte, error) {
|
||||
data := innertubeRequest{
|
||||
Context: prepareInnertubeContext(*c.client),
|
||||
Params: transcriptVideoID(id, lang),
|
||||
}
|
||||
|
||||
return c.httpPostBodyBytes(ctx, "https://www.youtube.com/youtubei/v1/get_transcript?key="+c.client.key, data)
|
||||
}
|
||||
|
||||
func prepareInnertubeContext(clientInfo clientInfo) inntertubeContext {
|
||||
return inntertubeContext{
|
||||
Client: innertubeClient{
|
||||
HL: "en",
|
||||
GL: "US",
|
||||
TimeZone: "UTC",
|
||||
DeviceModel: clientInfo.deviceModel,
|
||||
ClientName: clientInfo.name,
|
||||
ClientVersion: clientInfo.version,
|
||||
AndroidSDKVersion: clientInfo.androidVersion,
|
||||
UserAgent: clientInfo.userAgent,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func prepareInnertubePlaylistData(ID string, continuation bool, clientInfo clientInfo) innertubeRequest {
|
||||
context := prepareInnertubeContext(clientInfo)
|
||||
|
||||
if continuation {
|
||||
return innertubeRequest{
|
||||
Context: context,
|
||||
Continuation: ID,
|
||||
ContentCheckOK: true,
|
||||
RacyCheckOk: true,
|
||||
Params: playerParams,
|
||||
}
|
||||
}
|
||||
|
||||
return innertubeRequest{
|
||||
Context: context,
|
||||
BrowseID: "VL" + ID,
|
||||
ContentCheckOK: true,
|
||||
RacyCheckOk: true,
|
||||
Params: playerParams,
|
||||
}
|
||||
}
|
||||
|
||||
// transcriptVideoID encodes the video ID to the param used to fetch transcripts.
|
||||
func transcriptVideoID(videoID string, lang string) string {
|
||||
langCode := encTranscriptLang(lang)
|
||||
|
||||
// This can be optionally appened to the Sprintf str, not sure what it means
|
||||
// *3engagement-panel-searchable-transcript-search-panel\x30\x00\x38\x01\x40\x01
|
||||
return base64Enc(fmt.Sprintf("\n\x0b%s\x12\x12%s\x18\x01", videoID, langCode))
|
||||
}
|
||||
|
||||
func encTranscriptLang(languageCode string) string {
|
||||
s := fmt.Sprintf("\n\x03asr\x12\x02%s\x1a\x00", languageCode)
|
||||
s = base64PadEnc(s)
|
||||
|
||||
return url.QueryEscape(s)
|
||||
}
|
||||
|
||||
// GetPlaylist fetches playlist metadata
|
||||
func (c *Client) GetPlaylist(url string) (*Playlist, error) {
|
||||
return c.GetPlaylistContext(context.Background(), url)
|
||||
}
|
||||
|
||||
// GetPlaylistContext fetches playlist metadata, with a context, along with a list of Videos, and some basic information
|
||||
// for these videos. Playlist entries cannot be downloaded, as they lack all the required metadata, but
|
||||
// can be used to enumerate all IDs, Authors, Titles, etc.
|
||||
func (c *Client) GetPlaylistContext(ctx context.Context, url string) (*Playlist, error) {
|
||||
c.assureClient()
|
||||
|
||||
id, err := extractPlaylistID(url)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("extractPlaylistID failed: %w", err)
|
||||
}
|
||||
|
||||
data := prepareInnertubePlaylistData(id, false, *c.client)
|
||||
body, err := c.httpPostBodyBytes(ctx, "https://www.youtube.com/youtubei/v1/browse?key="+c.client.key, data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
p := &Playlist{ID: id}
|
||||
return p, p.parsePlaylistInfo(ctx, c, body)
|
||||
}
|
||||
|
||||
func (c *Client) VideoFromPlaylistEntry(entry *PlaylistEntry) (*Video, error) {
|
||||
return c.videoFromID(context.Background(), entry.ID)
|
||||
}
|
||||
|
||||
func (c *Client) VideoFromPlaylistEntryContext(ctx context.Context, entry *PlaylistEntry) (*Video, error) {
|
||||
return c.videoFromID(ctx, entry.ID)
|
||||
}
|
||||
|
||||
// GetStream returns the stream and the total size for a specific format
|
||||
func (c *Client) GetStream(video *Video, format *Format) (io.ReadCloser, int64, error) {
|
||||
return c.GetStreamContext(context.Background(), video, format)
|
||||
}
|
||||
|
||||
// GetStreamContext returns the stream and the total size for a specific format with a context.
|
||||
func (c *Client) GetStreamContext(ctx context.Context, video *Video, format *Format) (io.ReadCloser, int64, error) {
|
||||
url, err := c.GetStreamURL(video, format)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
r, w := io.Pipe()
|
||||
contentLength := format.ContentLength
|
||||
|
||||
if contentLength == 0 {
|
||||
// some videos don't have length information
|
||||
contentLength = c.downloadOnce(req, w, format)
|
||||
} else {
|
||||
// we have length information, let's download by chunks!
|
||||
c.downloadChunked(ctx, req, w, format)
|
||||
}
|
||||
|
||||
return r, contentLength, nil
|
||||
}
|
||||
|
||||
func (c *Client) downloadOnce(req *http.Request, w *io.PipeWriter, _ *Format) int64 {
|
||||
resp, err := c.httpDo(req)
|
||||
if err != nil {
|
||||
w.CloseWithError(err) //nolint:errcheck
|
||||
return 0
|
||||
}
|
||||
|
||||
go func() {
|
||||
defer resp.Body.Close()
|
||||
_, err := io.Copy(w, resp.Body)
|
||||
if err == nil {
|
||||
w.Close()
|
||||
} else {
|
||||
w.CloseWithError(err) //nolint:errcheck
|
||||
}
|
||||
}()
|
||||
|
||||
contentLength := resp.Header.Get("Content-Length")
|
||||
length, _ := strconv.ParseInt(contentLength, 10, 64)
|
||||
|
||||
return length
|
||||
}
|
||||
|
||||
func (c *Client) getChunkSize() int64 {
|
||||
if c.ChunkSize > 0 {
|
||||
return c.ChunkSize
|
||||
}
|
||||
|
||||
return Size10Mb
|
||||
}
|
||||
|
||||
func (c *Client) getMaxRoutines(limit int) int {
|
||||
routines := 10
|
||||
|
||||
if c.MaxRoutines > 0 {
|
||||
routines = c.MaxRoutines
|
||||
}
|
||||
|
||||
if limit > 0 && routines > limit {
|
||||
routines = limit
|
||||
}
|
||||
|
||||
return routines
|
||||
}
|
||||
|
||||
func (c *Client) downloadChunked(ctx context.Context, req *http.Request, w *io.PipeWriter, format *Format) {
|
||||
chunks := getChunks(format.ContentLength, c.getChunkSize())
|
||||
maxRoutines := c.getMaxRoutines(len(chunks))
|
||||
|
||||
cancelCtx, cancel := context.WithCancel(ctx)
|
||||
abort := func(err error) {
|
||||
w.CloseWithError(err)
|
||||
cancel()
|
||||
}
|
||||
|
||||
currentChunk := atomic.Uint32{}
|
||||
for i := 0; i < maxRoutines; i++ {
|
||||
go func() {
|
||||
for {
|
||||
chunkIndex := int(currentChunk.Add(1)) - 1
|
||||
if chunkIndex >= len(chunks) {
|
||||
// no more chunks
|
||||
return
|
||||
}
|
||||
|
||||
chunk := &chunks[chunkIndex]
|
||||
err := c.downloadChunk(req.Clone(cancelCtx), chunk)
|
||||
close(chunk.data)
|
||||
|
||||
if err != nil {
|
||||
abort(err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
go func() {
|
||||
// copy chunks into the PipeWriter
|
||||
for i := 0; i < len(chunks); i++ {
|
||||
select {
|
||||
case <-cancelCtx.Done():
|
||||
abort(context.Canceled)
|
||||
return
|
||||
case data := <-chunks[i].data:
|
||||
_, err := io.Copy(w, bytes.NewBuffer(data))
|
||||
if err != nil {
|
||||
abort(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// everything succeeded
|
||||
w.Close()
|
||||
}()
|
||||
}
|
||||
|
||||
// GetStreamURL returns the url for a specific format
|
||||
func (c *Client) GetStreamURL(video *Video, format *Format) (string, error) {
|
||||
return c.GetStreamURLContext(context.Background(), video, format)
|
||||
}
|
||||
|
||||
// GetStreamURLContext returns the url for a specific format with a context
|
||||
func (c *Client) GetStreamURLContext(ctx context.Context, video *Video, format *Format) (string, error) {
|
||||
if format == nil {
|
||||
return "", ErrNoFormat
|
||||
}
|
||||
|
||||
c.assureClient()
|
||||
|
||||
if format.URL != "" {
|
||||
if c.client.androidVersion > 0 {
|
||||
return format.URL, nil
|
||||
}
|
||||
|
||||
return c.unThrottle(ctx, video.ID, format.URL)
|
||||
}
|
||||
|
||||
// TODO: check rest of this function, is it redundant?
|
||||
|
||||
cipher := format.Cipher
|
||||
if cipher == "" {
|
||||
return "", ErrCipherNotFound
|
||||
}
|
||||
|
||||
uri, err := c.decipherURL(ctx, video.ID, cipher)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return uri, err
|
||||
}
|
||||
|
||||
// httpDo sends an HTTP request and returns an HTTP response.
|
||||
func (c *Client) httpDo(req *http.Request) (*http.Response, error) {
|
||||
client := c.HTTPClient
|
||||
if client == nil {
|
||||
client = http.DefaultClient
|
||||
}
|
||||
|
||||
req.Header.Set("User-Agent", c.client.userAgent)
|
||||
req.Header.Set("Origin", "https://youtube.com")
|
||||
req.Header.Set("Sec-Fetch-Mode", "navigate")
|
||||
|
||||
if len(c.consentID) == 0 {
|
||||
c.consentID = strconv.Itoa(rand.Intn(899) + 100) //nolint:gosec
|
||||
}
|
||||
|
||||
req.AddCookie(&http.Cookie{
|
||||
Name: "CONSENT",
|
||||
Value: "YES+cb.20210328-17-p0.en+FX+" + c.consentID,
|
||||
Path: "/",
|
||||
Domain: ".youtube.com",
|
||||
})
|
||||
|
||||
res, err := client.Do(req)
|
||||
|
||||
log := slog.With("method", req.Method, "url", req.URL)
|
||||
|
||||
if err == nil && res.StatusCode != http.StatusOK {
|
||||
err = ErrUnexpectedStatusCode(res.StatusCode)
|
||||
res.Body.Close()
|
||||
res = nil
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
log.Debug("HTTP request failed", "error", err)
|
||||
} else {
|
||||
log.Debug("HTTP request succeeded", "status", res.Status)
|
||||
}
|
||||
|
||||
return res, err
|
||||
}
|
||||
|
||||
// httpGet does a HTTP GET request, checks the response to be a 200 OK and returns it
|
||||
func (c *Client) httpGet(ctx context.Context, url string) (*http.Response, error) {
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
resp, err := c.httpDo(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
resp.Body.Close()
|
||||
return nil, ErrUnexpectedStatusCode(resp.StatusCode)
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// httpGetBodyBytes reads the whole HTTP body and returns it
|
||||
func (c *Client) httpGetBodyBytes(ctx context.Context, url string) ([]byte, error) {
|
||||
resp, err := c.httpGet(ctx, url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
return io.ReadAll(resp.Body)
|
||||
}
|
||||
|
||||
// httpPost does a HTTP POST request with a body, checks the response to be a 200 OK and returns it
|
||||
func (c *Client) httpPost(ctx context.Context, url string, body interface{}) (*http.Response, error) {
|
||||
data, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(data))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req.Header.Set("X-Youtube-Client-Name", "3")
|
||||
req.Header.Set("X-Youtube-Client-Version", c.client.version)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
|
||||
|
||||
resp, err := c.httpDo(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
resp.Body.Close()
|
||||
return nil, ErrUnexpectedStatusCode(resp.StatusCode)
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// httpPostBodyBytes reads the whole HTTP body and returns it
|
||||
func (c *Client) httpPostBodyBytes(ctx context.Context, url string, body interface{}) ([]byte, error) {
|
||||
resp, err := c.httpPost(ctx, url, body)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
return io.ReadAll(resp.Body)
|
||||
}
|
||||
|
||||
// downloadChunk writes the response data into the data channel of the chunk.
|
||||
// Downloading in multiple chunks is much faster:
|
||||
// https://github.com/kkdai/youtube/pull/190
|
||||
func (c *Client) downloadChunk(req *http.Request, chunk *chunk) error {
|
||||
q := req.URL.Query()
|
||||
q.Set("range", fmt.Sprintf("%d-%d", chunk.start, chunk.end))
|
||||
req.URL.RawQuery = q.Encode()
|
||||
|
||||
resp, err := c.httpDo(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return ErrUnexpectedStatusCode(resp.StatusCode)
|
||||
}
|
||||
|
||||
expected := int(chunk.end-chunk.start) + 1
|
||||
data, err := io.ReadAll(resp.Body)
|
||||
n := len(data)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if n != expected {
|
||||
return fmt.Errorf("chunk at offset %d has invalid size: expected=%d actual=%d", chunk.start, expected, n)
|
||||
}
|
||||
|
||||
chunk.data <- data
|
||||
|
||||
return nil
|
||||
}
|
285
downloader/vendor/github.com/kkdai/youtube/v2/decipher.go
generated
vendored
Normal file
285
downloader/vendor/github.com/kkdai/youtube/v2/decipher.go
generated
vendored
Normal file
@@ -0,0 +1,285 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"strconv"
|
||||
|
||||
"github.com/dop251/goja"
|
||||
)
|
||||
|
||||
func (c *Client) decipherURL(ctx context.Context, videoID string, cipher string) (string, error) {
|
||||
params, err := url.ParseQuery(cipher)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
uri, err := url.Parse(params.Get("url"))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
query := uri.Query()
|
||||
|
||||
config, err := c.getPlayerConfig(ctx, videoID)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// decrypt s-parameter
|
||||
bs, err := config.decrypt([]byte(params.Get("s")))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
query.Add(params.Get("sp"), string(bs))
|
||||
|
||||
query, err = c.decryptNParam(config, query)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
uri.RawQuery = query.Encode()
|
||||
|
||||
return uri.String(), nil
|
||||
}
|
||||
|
||||
// see https://github.com/kkdai/youtube/pull/244
|
||||
func (c *Client) unThrottle(ctx context.Context, videoID string, urlString string) (string, error) {
|
||||
config, err := c.getPlayerConfig(ctx, videoID)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
uri, err := url.Parse(urlString)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// for debugging
|
||||
if artifactsFolder != "" {
|
||||
writeArtifact("video-"+videoID+".url", []byte(uri.String()))
|
||||
}
|
||||
|
||||
query, err := c.decryptNParam(config, uri.Query())
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
uri.RawQuery = query.Encode()
|
||||
return uri.String(), nil
|
||||
}
|
||||
|
||||
func (c *Client) decryptNParam(config playerConfig, query url.Values) (url.Values, error) {
|
||||
// decrypt n-parameter
|
||||
nSig := query.Get("v")
|
||||
log := Logger.With("n", nSig)
|
||||
|
||||
if nSig != "" {
|
||||
nDecoded, err := config.decodeNsig(nSig)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("unable to decode nSig: %w", err)
|
||||
}
|
||||
query.Set("v", nDecoded)
|
||||
log = log.With("decoded", nDecoded)
|
||||
}
|
||||
|
||||
log.Debug("nParam")
|
||||
|
||||
return query, nil
|
||||
}
|
||||
|
||||
const (
|
||||
jsvarStr = "[a-zA-Z_\\$][a-zA-Z_0-9]*"
|
||||
reverseStr = ":function\\(a\\)\\{" +
|
||||
"(?:return )?a\\.reverse\\(\\)" +
|
||||
"\\}"
|
||||
spliceStr = ":function\\(a,b\\)\\{" +
|
||||
"a\\.splice\\(0,b\\)" +
|
||||
"\\}"
|
||||
swapStr = ":function\\(a,b\\)\\{" +
|
||||
"var c=a\\[0\\];a\\[0\\]=a\\[b(?:%a\\.length)?\\];a\\[b(?:%a\\.length)?\\]=c(?:;return a)?" +
|
||||
"\\}"
|
||||
)
|
||||
|
||||
var (
|
||||
nFunctionNameRegexp = regexp.MustCompile("\\.get\\(\"n\"\\)\\)&&\\(b=([a-zA-Z0-9$]{0,3})\\[(\\d+)\\](.+)\\|\\|([a-zA-Z0-9]{0,3})")
|
||||
actionsObjRegexp = regexp.MustCompile(fmt.Sprintf(
|
||||
"var (%s)=\\{((?:(?:%s%s|%s%s|%s%s),?\\n?)+)\\};", jsvarStr, jsvarStr, swapStr, jsvarStr, spliceStr, jsvarStr, reverseStr))
|
||||
|
||||
actionsFuncRegexp = regexp.MustCompile(fmt.Sprintf(
|
||||
"function(?: %s)?\\(a\\)\\{"+
|
||||
"a=a\\.split\\(\"\"\\);\\s*"+
|
||||
"((?:(?:a=)?%s\\.%s\\(a,\\d+\\);)+)"+
|
||||
"return a\\.join\\(\"\"\\)"+
|
||||
"\\}", jsvarStr, jsvarStr, jsvarStr))
|
||||
|
||||
reverseRegexp = regexp.MustCompile(fmt.Sprintf("(?m)(?:^|,)(%s)%s", jsvarStr, reverseStr))
|
||||
spliceRegexp = regexp.MustCompile(fmt.Sprintf("(?m)(?:^|,)(%s)%s", jsvarStr, spliceStr))
|
||||
swapRegexp = regexp.MustCompile(fmt.Sprintf("(?m)(?:^|,)(%s)%s", jsvarStr, swapStr))
|
||||
)
|
||||
|
||||
func (config playerConfig) decodeNsig(encoded string) (string, error) {
|
||||
fBody, err := config.getNFunction()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return evalJavascript(fBody, encoded)
|
||||
}
|
||||
|
||||
func evalJavascript(jsFunction, arg string) (string, error) {
|
||||
const myName = "myFunction"
|
||||
|
||||
vm := goja.New()
|
||||
_, err := vm.RunString(myName + "=" + jsFunction)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
var output func(string) string
|
||||
err = vm.ExportTo(vm.Get(myName), &output)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return output(arg), nil
|
||||
}
|
||||
|
||||
func (config playerConfig) getNFunction() (string, error) {
|
||||
nameResult := nFunctionNameRegexp.FindSubmatch(config)
|
||||
if len(nameResult) == 0 {
|
||||
return "", errors.New("unable to extract n-function name")
|
||||
}
|
||||
|
||||
var name string
|
||||
if idx, _ := strconv.Atoi(string(nameResult[2])); idx == 0 {
|
||||
name = string(nameResult[4])
|
||||
} else {
|
||||
name = string(nameResult[1])
|
||||
}
|
||||
|
||||
return config.extraFunction(name)
|
||||
|
||||
}
|
||||
|
||||
func (config playerConfig) extraFunction(name string) (string, error) {
|
||||
// find the beginning of the function
|
||||
def := []byte(name + "=function(")
|
||||
start := bytes.Index(config, def)
|
||||
if start < 1 {
|
||||
return "", fmt.Errorf("unable to extract n-function body: looking for '%s'", def)
|
||||
}
|
||||
|
||||
// start after the first curly bracket
|
||||
pos := start + bytes.IndexByte(config[start:], '{') + 1
|
||||
|
||||
var strChar byte
|
||||
|
||||
// find the bracket closing the function
|
||||
for brackets := 1; brackets > 0; pos++ {
|
||||
b := config[pos]
|
||||
switch b {
|
||||
case '{':
|
||||
if strChar == 0 {
|
||||
brackets++
|
||||
}
|
||||
case '}':
|
||||
if strChar == 0 {
|
||||
brackets--
|
||||
}
|
||||
case '`', '"', '\'':
|
||||
if config[pos-1] == '\\' && config[pos-2] != '\\' {
|
||||
continue
|
||||
}
|
||||
if strChar == 0 {
|
||||
strChar = b
|
||||
} else if strChar == b {
|
||||
strChar = 0
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return string(config[start:pos]), nil
|
||||
}
|
||||
|
||||
func (config playerConfig) decrypt(cyphertext []byte) ([]byte, error) {
|
||||
operations, err := config.parseDecipherOps()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// apply operations
|
||||
bs := []byte(cyphertext)
|
||||
for _, op := range operations {
|
||||
bs = op(bs)
|
||||
}
|
||||
|
||||
return bs, nil
|
||||
}
|
||||
|
||||
/*
|
||||
parses decipher operations from https://youtube.com/s/player/4fbb4d5b/player_ias.vflset/en_US/base.js
|
||||
|
||||
var Mt={
|
||||
splice:function(a,b){a.splice(0,b)},
|
||||
reverse:function(a){a.reverse()},
|
||||
EQ:function(a,b){var c=a[0];a[0]=a[b%a.length];a[b%a.length]=c}};
|
||||
|
||||
a=a.split("");
|
||||
Mt.splice(a,3);
|
||||
Mt.EQ(a,39);
|
||||
Mt.splice(a,2);
|
||||
Mt.EQ(a,1);
|
||||
Mt.splice(a,1);
|
||||
Mt.EQ(a,35);
|
||||
Mt.EQ(a,51);
|
||||
Mt.splice(a,2);
|
||||
Mt.reverse(a,52);
|
||||
return a.join("")
|
||||
*/
|
||||
func (config playerConfig) parseDecipherOps() (operations []DecipherOperation, err error) {
|
||||
objResult := actionsObjRegexp.FindSubmatch(config)
|
||||
funcResult := actionsFuncRegexp.FindSubmatch(config)
|
||||
if len(objResult) < 3 || len(funcResult) < 2 {
|
||||
return nil, fmt.Errorf("error parsing signature tokens (#obj=%d, #func=%d)", len(objResult), len(funcResult))
|
||||
}
|
||||
|
||||
obj := objResult[1]
|
||||
objBody := objResult[2]
|
||||
funcBody := funcResult[1]
|
||||
|
||||
var reverseKey, spliceKey, swapKey string
|
||||
|
||||
if result := reverseRegexp.FindSubmatch(objBody); len(result) > 1 {
|
||||
reverseKey = string(result[1])
|
||||
}
|
||||
if result := spliceRegexp.FindSubmatch(objBody); len(result) > 1 {
|
||||
spliceKey = string(result[1])
|
||||
}
|
||||
if result := swapRegexp.FindSubmatch(objBody); len(result) > 1 {
|
||||
swapKey = string(result[1])
|
||||
}
|
||||
|
||||
regex, err := regexp.Compile(fmt.Sprintf("(?:a=)?%s\\.(%s|%s|%s)\\(a,(\\d+)\\)", regexp.QuoteMeta(string(obj)), regexp.QuoteMeta(reverseKey), regexp.QuoteMeta(spliceKey), regexp.QuoteMeta(swapKey)))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var ops []DecipherOperation
|
||||
for _, s := range regex.FindAllSubmatch(funcBody, -1) {
|
||||
switch string(s[1]) {
|
||||
case reverseKey:
|
||||
ops = append(ops, reverseFunc)
|
||||
case swapKey:
|
||||
arg, _ := strconv.Atoi(string(s[2]))
|
||||
ops = append(ops, newSwapFunc(arg))
|
||||
case spliceKey:
|
||||
arg, _ := strconv.Atoi(string(s[2]))
|
||||
ops = append(ops, newSpliceFunc(arg))
|
||||
}
|
||||
}
|
||||
return ops, nil
|
||||
}
|
27
downloader/vendor/github.com/kkdai/youtube/v2/decipher_operations.go
generated
vendored
Normal file
27
downloader/vendor/github.com/kkdai/youtube/v2/decipher_operations.go
generated
vendored
Normal file
@@ -0,0 +1,27 @@
|
||||
package youtube
|
||||
|
||||
type DecipherOperation func([]byte) []byte
|
||||
|
||||
func newSpliceFunc(pos int) DecipherOperation {
|
||||
return func(bs []byte) []byte {
|
||||
return bs[pos:]
|
||||
}
|
||||
}
|
||||
|
||||
func newSwapFunc(arg int) DecipherOperation {
|
||||
return func(bs []byte) []byte {
|
||||
pos := arg % len(bs)
|
||||
bs[0], bs[pos] = bs[pos], bs[0]
|
||||
return bs
|
||||
}
|
||||
}
|
||||
|
||||
func reverseFunc(bs []byte) []byte {
|
||||
l, r := 0, len(bs)-1
|
||||
for l < r {
|
||||
bs[l], bs[r] = bs[r], bs[l]
|
||||
l++
|
||||
r--
|
||||
}
|
||||
return bs
|
||||
}
|
4
downloader/vendor/github.com/kkdai/youtube/v2/doc.go
generated
vendored
Normal file
4
downloader/vendor/github.com/kkdai/youtube/v2/doc.go
generated
vendored
Normal file
@@ -0,0 +1,4 @@
|
||||
/*
|
||||
Package youtube implement youtube download package in go.
|
||||
*/
|
||||
package youtube
|
203
downloader/vendor/github.com/kkdai/youtube/v2/downloader/downloader.go
generated
vendored
Normal file
203
downloader/vendor/github.com/kkdai/youtube/v2/downloader/downloader.go
generated
vendored
Normal file
@@ -0,0 +1,203 @@
|
||||
package downloader
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"log/slog"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/kkdai/youtube/v2"
|
||||
"github.com/vbauerster/mpb/v5"
|
||||
"github.com/vbauerster/mpb/v5/decor"
|
||||
)
|
||||
|
||||
// Downloader offers high level functions to download videos into files
|
||||
type Downloader struct {
|
||||
youtube.Client
|
||||
OutputDir string // optional directory to store the files
|
||||
}
|
||||
|
||||
func (dl *Downloader) getOutputFile(v *youtube.Video, format *youtube.Format, outputFile string) (string, error) {
|
||||
if outputFile == "" {
|
||||
outputFile = SanitizeFilename(v.Title)
|
||||
outputFile += pickIdealFileExtension(format.MimeType)
|
||||
}
|
||||
|
||||
if dl.OutputDir != "" {
|
||||
if err := os.MkdirAll(dl.OutputDir, 0o755); err != nil {
|
||||
return "", err
|
||||
}
|
||||
outputFile = filepath.Join(dl.OutputDir, outputFile)
|
||||
}
|
||||
|
||||
return outputFile, nil
|
||||
}
|
||||
|
||||
// Download : Starting download video by arguments.
|
||||
func (dl *Downloader) Download(ctx context.Context, v *youtube.Video, format *youtube.Format, outputFile string) error {
|
||||
youtube.Logger.Info(
|
||||
"Downloading video",
|
||||
"id", v.ID,
|
||||
"quality", format.Quality,
|
||||
"mimeType", format.MimeType,
|
||||
)
|
||||
destFile, err := dl.getOutputFile(v, format, outputFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Create output file
|
||||
out, err := os.Create(destFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer out.Close()
|
||||
|
||||
return dl.videoDLWorker(ctx, out, v, format)
|
||||
}
|
||||
|
||||
// DownloadComposite : Downloads audio and video streams separately and merges them via ffmpeg.
|
||||
func (dl *Downloader) DownloadComposite(ctx context.Context, outputFile string, v *youtube.Video, quality string, mimetype, language string) error {
|
||||
videoFormat, audioFormat, err1 := getVideoAudioFormats(v, quality, mimetype, language)
|
||||
if err1 != nil {
|
||||
return err1
|
||||
}
|
||||
|
||||
log := youtube.Logger.With("id", v.ID)
|
||||
|
||||
log.Info(
|
||||
"Downloading composite video",
|
||||
"videoQuality", videoFormat.QualityLabel,
|
||||
"videoMimeType", videoFormat.MimeType,
|
||||
"audioMimeType", audioFormat.MimeType,
|
||||
)
|
||||
|
||||
destFile, err := dl.getOutputFile(v, videoFormat, outputFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
outputDir := filepath.Dir(destFile)
|
||||
|
||||
// Create temporary video file
|
||||
videoFile, err := os.CreateTemp(outputDir, "youtube_*.m4v")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer Delete(videoFile.Name(), log)
|
||||
|
||||
// Create temporary audio file
|
||||
audioFile, err := os.CreateTemp(outputDir, "youtube_*.m4a")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer Delete(audioFile.Name(), log)
|
||||
|
||||
log.Debug("Downloading video file...")
|
||||
err = dl.videoDLWorker(ctx, videoFile, v, videoFormat)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
videoFile.Close()
|
||||
|
||||
log.Debug("Downloading audio file...")
|
||||
err = dl.videoDLWorker(ctx, audioFile, v, audioFormat)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
audioFile.Close()
|
||||
|
||||
//nolint:gosec
|
||||
ffmpegVersionCmd := exec.Command("ffmpeg", "-y",
|
||||
"-i", videoFile.Name(),
|
||||
"-i", audioFile.Name(),
|
||||
"-c", "copy", // Just copy without re-encoding
|
||||
"-shortest", // Finish encoding when the shortest input stream ends
|
||||
destFile,
|
||||
"-loglevel", "warning",
|
||||
)
|
||||
ffmpegVersionCmd.Stderr = os.Stderr
|
||||
ffmpegVersionCmd.Stdout = os.Stdout
|
||||
log.Info("merging video and audio", "output", destFile)
|
||||
|
||||
return ffmpegVersionCmd.Run()
|
||||
}
|
||||
|
||||
func Delete(path string, log *slog.Logger) {
|
||||
err := os.Remove(path)
|
||||
if err != nil {
|
||||
log.Error("Failed deleting file", "path", path, "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func getVideoAudioFormats(v *youtube.Video, quality string, mimetype, language string) (*youtube.Format, *youtube.Format, error) {
|
||||
var videoFormats, audioFormats youtube.FormatList
|
||||
|
||||
formats := v.Formats
|
||||
if mimetype != "" {
|
||||
formats = formats.Type(mimetype)
|
||||
}
|
||||
|
||||
videoFormats = formats.Type("video").AudioChannels(0)
|
||||
audioFormats = formats.Type("audio")
|
||||
|
||||
if quality != "" {
|
||||
videoFormats = videoFormats.Quality(quality)
|
||||
}
|
||||
|
||||
if language != "" {
|
||||
audioFormats = audioFormats.Language(language)
|
||||
}
|
||||
|
||||
if len(videoFormats) == 0 {
|
||||
return nil, nil, errors.New("no video format found after filtering")
|
||||
}
|
||||
|
||||
if len(audioFormats) == 0 {
|
||||
return nil, nil, errors.New("no audio format found after filtering")
|
||||
}
|
||||
|
||||
videoFormats.Sort()
|
||||
audioFormats.Sort()
|
||||
|
||||
return &videoFormats[0], &audioFormats[0], nil
|
||||
}
|
||||
|
||||
func (dl *Downloader) videoDLWorker(ctx context.Context, out *os.File, video *youtube.Video, format *youtube.Format) error {
|
||||
stream, size, err := dl.GetStreamContext(ctx, video, format)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
prog := &progress{
|
||||
contentLength: float64(size),
|
||||
}
|
||||
|
||||
// create progress bar
|
||||
progress := mpb.New(mpb.WithWidth(64))
|
||||
bar := progress.AddBar(
|
||||
int64(prog.contentLength),
|
||||
|
||||
mpb.PrependDecorators(
|
||||
decor.CountersKibiByte("% .2f / % .2f"),
|
||||
decor.Percentage(decor.WCSyncSpace),
|
||||
),
|
||||
mpb.AppendDecorators(
|
||||
decor.EwmaETA(decor.ET_STYLE_GO, 90),
|
||||
decor.Name(" ] "),
|
||||
decor.EwmaSpeed(decor.UnitKiB, "% .2f", 60),
|
||||
),
|
||||
)
|
||||
|
||||
reader := bar.ProxyReader(stream)
|
||||
mw := io.MultiWriter(out, prog)
|
||||
_, err = io.Copy(mw, reader)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
progress.Wait()
|
||||
return nil
|
||||
}
|
65
downloader/vendor/github.com/kkdai/youtube/v2/downloader/file_utils.go
generated
vendored
Normal file
65
downloader/vendor/github.com/kkdai/youtube/v2/downloader/file_utils.go
generated
vendored
Normal file
@@ -0,0 +1,65 @@
|
||||
package downloader
|
||||
|
||||
import (
|
||||
"mime"
|
||||
"regexp"
|
||||
)
|
||||
|
||||
const defaultExtension = ".mov"
|
||||
|
||||
// Rely on hardcoded canonical mime types, as the ones provided by Go aren't exhaustive [1].
|
||||
// This seems to be a recurring problem for youtube downloaders, see [2].
|
||||
// The implementation is based on mozilla's list [3], IANA [4] and Youtube's support [5].
|
||||
// [1] https://github.com/golang/go/blob/ed7888aea6021e25b0ea58bcad3f26da2b139432/src/mime/type.go#L60
|
||||
// [2] https://github.com/ZiTAL/youtube-dl/blob/master/mime.types
|
||||
// [3] https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/MIME_types/Common_types
|
||||
// [4] https://www.iana.org/assignments/media-types/media-types.xhtml#video
|
||||
// [5] https://support.google.com/youtube/troubleshooter/2888402?hl=en
|
||||
var canonicals = map[string]string{
|
||||
"video/quicktime": ".mov",
|
||||
"video/x-msvideo": ".avi",
|
||||
"video/x-matroska": ".mkv",
|
||||
"video/mpeg": ".mpeg",
|
||||
"video/webm": ".webm",
|
||||
"video/3gpp2": ".3g2",
|
||||
"video/x-flv": ".flv",
|
||||
"video/3gpp": ".3gp",
|
||||
"video/mp4": ".mp4",
|
||||
"video/ogg": ".ogv",
|
||||
"video/mp2t": ".ts",
|
||||
}
|
||||
|
||||
func pickIdealFileExtension(mediaType string) string {
|
||||
mediaType, _, err := mime.ParseMediaType(mediaType)
|
||||
if err != nil {
|
||||
return defaultExtension
|
||||
}
|
||||
|
||||
if extension, ok := canonicals[mediaType]; ok {
|
||||
return extension
|
||||
}
|
||||
|
||||
// Our last resort is to ask the operating system, but these give multiple results and are rarely canonical.
|
||||
extensions, err := mime.ExtensionsByType(mediaType)
|
||||
if err != nil || extensions == nil {
|
||||
return defaultExtension
|
||||
}
|
||||
|
||||
return extensions[0]
|
||||
}
|
||||
|
||||
func SanitizeFilename(fileName string) string {
|
||||
// Characters not allowed on mac
|
||||
// :/
|
||||
// Characters not allowed on linux
|
||||
// /
|
||||
// Characters not allowed on windows
|
||||
// <>:"/\|?*
|
||||
|
||||
// Ref https://docs.microsoft.com/en-us/windows/win32/fileio/naming-a-file#naming-conventions
|
||||
|
||||
fileName = regexp.MustCompile(`[:/<>\:"\\|?*]`).ReplaceAllString(fileName, "")
|
||||
fileName = regexp.MustCompile(`\s+`).ReplaceAllString(fileName, " ")
|
||||
|
||||
return fileName
|
||||
}
|
17
downloader/vendor/github.com/kkdai/youtube/v2/downloader/progress.go
generated
vendored
Normal file
17
downloader/vendor/github.com/kkdai/youtube/v2/downloader/progress.go
generated
vendored
Normal file
@@ -0,0 +1,17 @@
|
||||
package downloader
|
||||
|
||||
type progress struct {
|
||||
contentLength float64
|
||||
totalWrittenBytes float64
|
||||
downloadLevel float64
|
||||
}
|
||||
|
||||
func (dl *progress) Write(p []byte) (n int, err error) {
|
||||
n = len(p)
|
||||
dl.totalWrittenBytes = dl.totalWrittenBytes + float64(n)
|
||||
currentPercent := (dl.totalWrittenBytes / dl.contentLength) * 100
|
||||
if (dl.downloadLevel <= currentPercent) && (dl.downloadLevel < 100) {
|
||||
dl.downloadLevel++
|
||||
}
|
||||
return
|
||||
}
|
47
downloader/vendor/github.com/kkdai/youtube/v2/errors.go
generated
vendored
Normal file
47
downloader/vendor/github.com/kkdai/youtube/v2/errors.go
generated
vendored
Normal file
@@ -0,0 +1,47 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
const (
|
||||
ErrCipherNotFound = constError("cipher not found")
|
||||
ErrSignatureTimestampNotFound = constError("signature timestamp not found")
|
||||
ErrInvalidCharactersInVideoID = constError("invalid characters in video id")
|
||||
ErrVideoIDMinLength = constError("the video id must be at least 10 characters long")
|
||||
ErrReadOnClosedResBody = constError("http: read on closed response body")
|
||||
ErrNotPlayableInEmbed = constError("embedding of this video has been disabled")
|
||||
ErrLoginRequired = constError("login required to confirm your age")
|
||||
ErrVideoPrivate = constError("user restricted access to this video")
|
||||
ErrInvalidPlaylist = constError("no playlist detected or invalid playlist ID")
|
||||
)
|
||||
|
||||
type constError string
|
||||
|
||||
func (e constError) Error() string {
|
||||
return string(e)
|
||||
}
|
||||
|
||||
type ErrPlayabiltyStatus struct {
|
||||
Status string
|
||||
Reason string
|
||||
}
|
||||
|
||||
func (err ErrPlayabiltyStatus) Error() string {
|
||||
return fmt.Sprintf("cannot playback and download, status: %s, reason: %s", err.Status, err.Reason)
|
||||
}
|
||||
|
||||
// ErrUnexpectedStatusCode is returned on unexpected HTTP status codes
|
||||
type ErrUnexpectedStatusCode int
|
||||
|
||||
func (err ErrUnexpectedStatusCode) Error() string {
|
||||
return fmt.Sprintf("unexpected status code: %d", err)
|
||||
}
|
||||
|
||||
type ErrPlaylistStatus struct {
|
||||
Reason string
|
||||
}
|
||||
|
||||
func (err ErrPlaylistStatus) Error() string {
|
||||
return fmt.Sprintf("could not load playlist: %s", err.Reason)
|
||||
}
|
35
downloader/vendor/github.com/kkdai/youtube/v2/fetch_testdata_helper.go
generated
vendored
Normal file
35
downloader/vendor/github.com/kkdai/youtube/v2/fetch_testdata_helper.go
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
//go:build fetch
|
||||
// +build fetch
|
||||
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
func init() {
|
||||
FetchTestData()
|
||||
}
|
||||
|
||||
// ran via go generate to fetch and update the playlist response data
|
||||
func FetchTestData() {
|
||||
f, err := os.Create(testPlaylistResponseDataFile)
|
||||
exitOnError(err)
|
||||
requestURL := fmt.Sprintf(playlistFetchURL, testPlaylistID)
|
||||
resp, err := http.Get(requestURL)
|
||||
exitOnError(err)
|
||||
defer resp.Body.Close()
|
||||
n, err := io.Copy(f, resp.Body)
|
||||
exitOnError(err)
|
||||
fmt.Printf("Successfully fetched playlist %s (%d bytes)\n", testPlaylistID, n)
|
||||
}
|
||||
|
||||
func exitOnError(err error) {
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
148
downloader/vendor/github.com/kkdai/youtube/v2/format_list.go
generated
vendored
Normal file
148
downloader/vendor/github.com/kkdai/youtube/v2/format_list.go
generated
vendored
Normal file
@@ -0,0 +1,148 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type FormatList []Format
|
||||
|
||||
// Type returns a new FormatList filtered by itag
|
||||
func (list FormatList) Select(f func(Format) bool) (result FormatList) {
|
||||
for i := range list {
|
||||
if f(list[i]) {
|
||||
result = append(result, list[i])
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// Type returns a new FormatList filtered by itag
|
||||
func (list FormatList) Itag(itagNo int) FormatList {
|
||||
return list.Select(func(f Format) bool {
|
||||
return f.ItagNo == itagNo
|
||||
})
|
||||
}
|
||||
|
||||
// Type returns a new FormatList filtered by mime type
|
||||
func (list FormatList) Type(value string) FormatList {
|
||||
return list.Select(func(f Format) bool {
|
||||
return strings.Contains(f.MimeType, value)
|
||||
})
|
||||
}
|
||||
|
||||
// Type returns a new FormatList filtered by display name
|
||||
func (list FormatList) Language(displayName string) FormatList {
|
||||
return list.Select(func(f Format) bool {
|
||||
return f.LanguageDisplayName() == displayName
|
||||
})
|
||||
}
|
||||
|
||||
// Quality returns a new FormatList filtered by quality, quality label or itag,
|
||||
// but not audio quality
|
||||
func (list FormatList) Quality(quality string) FormatList {
|
||||
itag, _ := strconv.Atoi(quality)
|
||||
|
||||
return list.Select(func(f Format) bool {
|
||||
return itag == f.ItagNo || strings.Contains(f.Quality, quality) || strings.Contains(f.QualityLabel, quality)
|
||||
})
|
||||
}
|
||||
|
||||
// AudioChannels returns a new FormatList filtered by the matching AudioChannels
|
||||
func (list FormatList) AudioChannels(n int) FormatList {
|
||||
return list.Select(func(f Format) bool {
|
||||
return f.AudioChannels == n
|
||||
})
|
||||
}
|
||||
|
||||
// AudioChannels returns a new FormatList filtered by the matching AudioChannels
|
||||
func (list FormatList) WithAudioChannels() FormatList {
|
||||
return list.Select(func(f Format) bool {
|
||||
return f.AudioChannels > 0
|
||||
})
|
||||
}
|
||||
|
||||
// FilterQuality reduces the format list to formats matching the quality
|
||||
func (v *Video) FilterQuality(quality string) {
|
||||
v.Formats = v.Formats.Quality(quality)
|
||||
v.Formats.Sort()
|
||||
}
|
||||
|
||||
// Sort sorts all formats fields
|
||||
func (list FormatList) Sort() {
|
||||
sort.SliceStable(list, func(i, j int) bool {
|
||||
return sortFormat(i, j, list)
|
||||
})
|
||||
}
|
||||
|
||||
// sortFormat sorts video by resolution, FPS, codec (av01, vp9, avc1), bitrate
|
||||
// sorts audio by default, codec (mp4, opus), channels, bitrate, sample rate
|
||||
func sortFormat(i int, j int, formats FormatList) bool {
|
||||
|
||||
// Sort by Width
|
||||
if formats[i].Width == formats[j].Width {
|
||||
// Format 137 downloads slowly, give it less priority
|
||||
// see https://github.com/kkdai/youtube/pull/171
|
||||
switch 137 {
|
||||
case formats[i].ItagNo:
|
||||
return false
|
||||
case formats[j].ItagNo:
|
||||
return true
|
||||
}
|
||||
|
||||
// Sort by FPS
|
||||
if formats[i].FPS == formats[j].FPS {
|
||||
if formats[i].FPS == 0 && formats[i].AudioChannels > 0 && formats[j].AudioChannels > 0 {
|
||||
// Audio
|
||||
// Sort by default
|
||||
if (formats[i].AudioTrack == nil && formats[j].AudioTrack == nil) || (formats[i].AudioTrack != nil && formats[j].AudioTrack != nil && formats[i].AudioTrack.AudioIsDefault == formats[j].AudioTrack.AudioIsDefault) {
|
||||
// Sort by codec
|
||||
codec := map[int]int{}
|
||||
for _, index := range []int{i, j} {
|
||||
if strings.Contains(formats[index].MimeType, "mp4") {
|
||||
codec[index] = 1
|
||||
} else if strings.Contains(formats[index].MimeType, "opus") {
|
||||
codec[index] = 2
|
||||
}
|
||||
}
|
||||
if codec[i] == codec[j] {
|
||||
// Sort by Audio Channel
|
||||
if formats[i].AudioChannels == formats[j].AudioChannels {
|
||||
// Sort by Audio Bitrate
|
||||
if formats[i].Bitrate == formats[j].Bitrate {
|
||||
// Sort by Audio Sample Rate
|
||||
return formats[i].AudioSampleRate > formats[j].AudioSampleRate
|
||||
}
|
||||
return formats[i].Bitrate > formats[j].Bitrate
|
||||
}
|
||||
return formats[i].AudioChannels > formats[j].AudioChannels
|
||||
}
|
||||
return codec[i] < codec[j]
|
||||
} else if formats[i].AudioTrack != nil && formats[i].AudioTrack.AudioIsDefault {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
// Video
|
||||
// Sort by codec
|
||||
codec := map[int]int{}
|
||||
for _, index := range []int{i, j} {
|
||||
if strings.Contains(formats[index].MimeType, "av01") {
|
||||
codec[index] = 1
|
||||
} else if strings.Contains(formats[index].MimeType, "vp9") {
|
||||
codec[index] = 2
|
||||
} else if strings.Contains(formats[index].MimeType, "avc1") {
|
||||
codec[index] = 3
|
||||
}
|
||||
}
|
||||
if codec[i] == codec[j] {
|
||||
// Sort by Audio Bitrate
|
||||
return formats[i].Bitrate > formats[j].Bitrate
|
||||
}
|
||||
return codec[i] < codec[j]
|
||||
}
|
||||
return formats[i].FPS > formats[j].FPS
|
||||
}
|
||||
return formats[i].Width > formats[j].Width
|
||||
}
|
28
downloader/vendor/github.com/kkdai/youtube/v2/logger.go
generated
vendored
Normal file
28
downloader/vendor/github.com/kkdai/youtube/v2/logger.go
generated
vendored
Normal file
@@ -0,0 +1,28 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"os"
|
||||
)
|
||||
|
||||
// The global logger for all Client instances
|
||||
var Logger = getLogger(os.Getenv("LOGLEVEL"))
|
||||
|
||||
func SetLogLevel(value string) {
|
||||
Logger = getLogger(value)
|
||||
}
|
||||
|
||||
func getLogger(logLevel string) *slog.Logger {
|
||||
levelVar := slog.LevelVar{}
|
||||
|
||||
if logLevel != "" {
|
||||
if err := levelVar.UnmarshalText([]byte(logLevel)); err != nil {
|
||||
panic(fmt.Sprintf("Invalid log level %s: %v", logLevel, err))
|
||||
}
|
||||
}
|
||||
|
||||
return slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
|
||||
Level: levelVar.Level(),
|
||||
}))
|
||||
}
|
37
downloader/vendor/github.com/kkdai/youtube/v2/player_cache.go
generated
vendored
Normal file
37
downloader/vendor/github.com/kkdai/youtube/v2/player_cache.go
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
const defaultCacheExpiration = time.Minute * time.Duration(5)
|
||||
|
||||
type playerCache struct {
|
||||
key string
|
||||
expiredAt time.Time
|
||||
config playerConfig
|
||||
}
|
||||
|
||||
// Get : get cache when it has same video id and not expired
|
||||
func (s playerCache) Get(key string) playerConfig {
|
||||
return s.GetCacheBefore(key, time.Now())
|
||||
}
|
||||
|
||||
// GetCacheBefore : can pass time for testing
|
||||
func (s playerCache) GetCacheBefore(key string, time time.Time) playerConfig {
|
||||
if key == s.key && s.expiredAt.After(time) {
|
||||
return s.config
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Set : set cache with default expiration
|
||||
func (s *playerCache) Set(key string, operations playerConfig) {
|
||||
s.setWithExpiredTime(key, operations, time.Now().Add(defaultCacheExpiration))
|
||||
}
|
||||
|
||||
func (s *playerCache) setWithExpiredTime(key string, config playerConfig, time time.Time) {
|
||||
s.key = key
|
||||
s.config = config
|
||||
s.expiredAt = time
|
||||
}
|
59
downloader/vendor/github.com/kkdai/youtube/v2/player_parse.go
generated
vendored
Normal file
59
downloader/vendor/github.com/kkdai/youtube/v2/player_parse.go
generated
vendored
Normal file
@@ -0,0 +1,59 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type playerConfig []byte
|
||||
|
||||
var basejsPattern = regexp.MustCompile(`(/s/player/\w+/player_ias.vflset/\w+/base.js)`)
|
||||
|
||||
func (c *Client) getPlayerConfig(ctx context.Context, videoID string) (playerConfig, error) {
|
||||
embedURL := fmt.Sprintf("https://youtube.com/embed/%s?hl=en", videoID)
|
||||
embedBody, err := c.httpGetBodyBytes(ctx, embedURL)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// example: /s/player/f676c671/player_ias.vflset/en_US/base.js
|
||||
playerPath := string(basejsPattern.Find(embedBody))
|
||||
if playerPath == "" {
|
||||
return nil, errors.New("unable to find basejs URL in playerConfig")
|
||||
}
|
||||
|
||||
// for debugging
|
||||
var artifactName string
|
||||
if artifactsFolder != "" {
|
||||
parts := strings.SplitN(playerPath, "/", 5)
|
||||
artifactName = "player-" + parts[3] + ".js"
|
||||
linkName := filepath.Join(artifactsFolder, "video-"+videoID+".js")
|
||||
if err := os.Symlink(artifactName, linkName); err != nil {
|
||||
log.Printf("unable to create symlink %s: %v", linkName, err)
|
||||
}
|
||||
}
|
||||
|
||||
config := c.playerCache.Get(playerPath)
|
||||
if config != nil {
|
||||
return config, nil
|
||||
}
|
||||
|
||||
config, err = c.httpGetBodyBytes(ctx, "https://youtube.com"+playerPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// for debugging
|
||||
if artifactName != "" {
|
||||
writeArtifact(artifactName, config)
|
||||
}
|
||||
|
||||
c.playerCache.Set(playerPath, config)
|
||||
return config, nil
|
||||
}
|
256
downloader/vendor/github.com/kkdai/youtube/v2/playlist.go
generated
vendored
Normal file
256
downloader/vendor/github.com/kkdai/youtube/v2/playlist.go
generated
vendored
Normal file
@@ -0,0 +1,256 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"regexp"
|
||||
"runtime/debug"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
sjson "github.com/bitly/go-simplejson"
|
||||
)
|
||||
|
||||
var (
|
||||
playlistIDRegex = regexp.MustCompile("^[A-Za-z0-9_-]{13,42}$")
|
||||
playlistInURLRegex = regexp.MustCompile("[&?]list=([A-Za-z0-9_-]{13,42})(&.*)?$")
|
||||
)
|
||||
|
||||
type Playlist struct {
|
||||
ID string
|
||||
Title string
|
||||
Description string
|
||||
Author string
|
||||
Videos []*PlaylistEntry
|
||||
}
|
||||
|
||||
type PlaylistEntry struct {
|
||||
ID string
|
||||
Title string
|
||||
Author string
|
||||
Duration time.Duration
|
||||
Thumbnails Thumbnails
|
||||
}
|
||||
|
||||
func extractPlaylistID(url string) (string, error) {
|
||||
if playlistIDRegex.Match([]byte(url)) {
|
||||
return url, nil
|
||||
}
|
||||
|
||||
matches := playlistInURLRegex.FindStringSubmatch(url)
|
||||
|
||||
if matches != nil {
|
||||
return matches[1], nil
|
||||
}
|
||||
|
||||
return "", ErrInvalidPlaylist
|
||||
}
|
||||
|
||||
// structs for playlist extraction
|
||||
|
||||
// Title: metadata.playlistMetadataRenderer.title | sidebar.playlistSidebarRenderer.items[0].playlistSidebarPrimaryInfoRenderer.title.runs[0].text
|
||||
// Description: metadata.playlistMetadataRenderer.description
|
||||
// Author: sidebar.playlistSidebarRenderer.items[1].playlistSidebarSecondaryInfoRenderer.videoOwner.videoOwnerRenderer.title.runs[0].text
|
||||
|
||||
// Videos: contents.twoColumnBrowseResultsRenderer.tabs[0].tabRenderer.content.sectionListRenderer.contents[0].itemSectionRenderer.contents[0].playlistVideoListRenderer.contents
|
||||
// ID: .videoId
|
||||
// Title: title.runs[0].text
|
||||
// Author: .shortBylineText.runs[0].text
|
||||
// Duration: .lengthSeconds
|
||||
// Thumbnails .thumbnails
|
||||
|
||||
// TODO?: Author thumbnails: sidebar.playlistSidebarRenderer.items[0].playlistSidebarPrimaryInfoRenderer.thumbnailRenderer.playlistVideoThumbnailRenderer.thumbnail.thumbnails
|
||||
func (p *Playlist) parsePlaylistInfo(ctx context.Context, client *Client, body []byte) (err error) {
|
||||
var j *sjson.Json
|
||||
j, err = sjson.NewJson(body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer func() {
|
||||
stack := debug.Stack()
|
||||
if r := recover(); r != nil {
|
||||
err = fmt.Errorf("JSON parsing error: %v\n%s", r, stack)
|
||||
}
|
||||
}()
|
||||
|
||||
renderer := j.GetPath("alerts").GetIndex(0).GetPath("alertRenderer")
|
||||
if renderer != nil && renderer.GetPath("type").MustString() == "ERROR" {
|
||||
message := renderer.GetPath("text", "runs").GetIndex(0).GetPath("text").MustString()
|
||||
|
||||
return ErrPlaylistStatus{Reason: message}
|
||||
}
|
||||
|
||||
// Metadata can be located in multiple places depending on client type
|
||||
var metadata *sjson.Json
|
||||
if node, ok := j.CheckGet("metadata"); ok {
|
||||
metadata = node
|
||||
} else if node, ok := j.CheckGet("header"); ok {
|
||||
metadata = node
|
||||
} else {
|
||||
return fmt.Errorf("no playlist header / metadata found")
|
||||
}
|
||||
|
||||
metadata = metadata.Get("playlistHeaderRenderer")
|
||||
|
||||
p.Title = sjsonGetText(metadata, "title")
|
||||
p.Description = sjsonGetText(metadata, "description", "descriptionText")
|
||||
p.Author = j.GetPath("sidebar", "playlistSidebarRenderer", "items").GetIndex(1).
|
||||
GetPath("playlistSidebarSecondaryInfoRenderer", "videoOwner", "videoOwnerRenderer", "title", "runs").
|
||||
GetIndex(0).Get("text").MustString()
|
||||
|
||||
if len(p.Author) == 0 {
|
||||
p.Author = sjsonGetText(metadata, "owner", "ownerText")
|
||||
}
|
||||
|
||||
contents, ok := j.CheckGet("contents")
|
||||
if !ok {
|
||||
return fmt.Errorf("contents not found in json body")
|
||||
}
|
||||
|
||||
// contents can have different keys with same child structure
|
||||
firstPart := getFirstKeyJSON(contents).GetPath("tabs").GetIndex(0).
|
||||
GetPath("tabRenderer", "content", "sectionListRenderer", "contents").GetIndex(0)
|
||||
|
||||
// This extra nested item is only set with the web client
|
||||
if n := firstPart.GetPath("itemSectionRenderer", "contents").GetIndex(0); isValidJSON(n) {
|
||||
firstPart = n
|
||||
}
|
||||
|
||||
vJSON, err := firstPart.GetPath("playlistVideoListRenderer", "contents").MarshalJSON()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if len(vJSON) <= 4 {
|
||||
return fmt.Errorf("no video data found in JSON")
|
||||
}
|
||||
|
||||
entries, continuation, err := extractPlaylistEntries(vJSON)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if len(continuation) == 0 {
|
||||
continuation = getContinuation(firstPart.Get("playlistVideoListRenderer"))
|
||||
}
|
||||
|
||||
if len(entries) == 0 {
|
||||
return fmt.Errorf("no videos found in playlist")
|
||||
}
|
||||
|
||||
p.Videos = entries
|
||||
|
||||
for continuation != "" {
|
||||
data := prepareInnertubePlaylistData(continuation, true, *client.client)
|
||||
|
||||
body, err := client.httpPostBodyBytes(ctx, "https://www.youtube.com/youtubei/v1/browse?key="+client.client.key, data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
j, err := sjson.NewJson(body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
next := j.GetPath("onResponseReceivedActions").GetIndex(0).
|
||||
GetPath("appendContinuationItemsAction", "continuationItems")
|
||||
|
||||
if !isValidJSON(next) {
|
||||
next = j.GetPath("continuationContents", "playlistVideoListContinuation", "contents")
|
||||
}
|
||||
|
||||
vJSON, err := next.MarshalJSON()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
entries, token, err := extractPlaylistEntries(vJSON)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if len(token) > 0 {
|
||||
continuation = token
|
||||
} else {
|
||||
continuation = getContinuation(j.GetPath("continuationContents", "playlistVideoListContinuation"))
|
||||
}
|
||||
|
||||
p.Videos = append(p.Videos, entries...)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func extractPlaylistEntries(data []byte) ([]*PlaylistEntry, string, error) {
|
||||
var vids []*videosJSONExtractor
|
||||
|
||||
if err := json.Unmarshal(data, &vids); err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
entries := make([]*PlaylistEntry, 0, len(vids))
|
||||
|
||||
var continuation string
|
||||
for _, v := range vids {
|
||||
if v.Renderer == nil {
|
||||
if v.Continuation.Endpoint.Command.Token != "" {
|
||||
continuation = v.Continuation.Endpoint.Command.Token
|
||||
}
|
||||
|
||||
continue
|
||||
}
|
||||
|
||||
entries = append(entries, v.PlaylistEntry())
|
||||
}
|
||||
|
||||
return entries, continuation, nil
|
||||
}
|
||||
|
||||
type videosJSONExtractor struct {
|
||||
Renderer *struct {
|
||||
ID string `json:"videoId"`
|
||||
Title withRuns `json:"title"`
|
||||
Author withRuns `json:"shortBylineText"`
|
||||
Duration string `json:"lengthSeconds"`
|
||||
Thumbnail struct {
|
||||
Thumbnails []Thumbnail `json:"thumbnails"`
|
||||
} `json:"thumbnail"`
|
||||
} `json:"playlistVideoRenderer"`
|
||||
Continuation struct {
|
||||
Endpoint struct {
|
||||
Command struct {
|
||||
Token string `json:"token"`
|
||||
} `json:"continuationCommand"`
|
||||
} `json:"continuationEndpoint"`
|
||||
} `json:"continuationItemRenderer"`
|
||||
}
|
||||
|
||||
func (vje videosJSONExtractor) PlaylistEntry() *PlaylistEntry {
|
||||
ds, err := strconv.Atoi(vje.Renderer.Duration)
|
||||
if err != nil {
|
||||
panic("invalid video duration: " + vje.Renderer.Duration)
|
||||
}
|
||||
return &PlaylistEntry{
|
||||
ID: vje.Renderer.ID,
|
||||
Title: vje.Renderer.Title.String(),
|
||||
Author: vje.Renderer.Author.String(),
|
||||
Duration: time.Second * time.Duration(ds),
|
||||
Thumbnails: vje.Renderer.Thumbnail.Thumbnails,
|
||||
}
|
||||
}
|
||||
|
||||
type withRuns struct {
|
||||
Runs []struct {
|
||||
Text string `json:"text"`
|
||||
} `json:"runs"`
|
||||
}
|
||||
|
||||
func (wr withRuns) String() string {
|
||||
if len(wr.Runs) > 0 {
|
||||
return wr.Runs[0].Text
|
||||
}
|
||||
return ""
|
||||
}
|
153
downloader/vendor/github.com/kkdai/youtube/v2/response_data.go
generated
vendored
Normal file
153
downloader/vendor/github.com/kkdai/youtube/v2/response_data.go
generated
vendored
Normal file
@@ -0,0 +1,153 @@
|
||||
package youtube
|
||||
|
||||
type playerResponseData struct {
|
||||
Captions struct {
|
||||
PlayerCaptionsTracklistRenderer struct {
|
||||
CaptionTracks []CaptionTrack `json:"captionTracks"`
|
||||
AudioTracks []struct {
|
||||
CaptionTrackIndices []int `json:"captionTrackIndices"`
|
||||
} `json:"audioTracks"`
|
||||
TranslationLanguages []struct {
|
||||
LanguageCode string `json:"languageCode"`
|
||||
LanguageName struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"languageName"`
|
||||
} `json:"translationLanguages"`
|
||||
DefaultAudioTrackIndex int `json:"defaultAudioTrackIndex"`
|
||||
} `json:"playerCaptionsTracklistRenderer"`
|
||||
} `json:"captions"`
|
||||
|
||||
PlayabilityStatus struct {
|
||||
Status string `json:"status"`
|
||||
Reason string `json:"reason"`
|
||||
PlayableInEmbed bool `json:"playableInEmbed"`
|
||||
Miniplayer struct {
|
||||
MiniplayerRenderer struct {
|
||||
PlaybackMode string `json:"playbackMode"`
|
||||
} `json:"miniplayerRenderer"`
|
||||
} `json:"miniplayer"`
|
||||
ContextParams string `json:"contextParams"`
|
||||
} `json:"playabilityStatus"`
|
||||
StreamingData struct {
|
||||
ExpiresInSeconds string `json:"expiresInSeconds"`
|
||||
Formats []Format `json:"formats"`
|
||||
AdaptiveFormats []Format `json:"adaptiveFormats"`
|
||||
DashManifestURL string `json:"dashManifestUrl"`
|
||||
HlsManifestURL string `json:"hlsManifestUrl"`
|
||||
} `json:"streamingData"`
|
||||
VideoDetails struct {
|
||||
VideoID string `json:"videoId"`
|
||||
Title string `json:"title"`
|
||||
LengthSeconds string `json:"lengthSeconds"`
|
||||
Keywords []string `json:"keywords"`
|
||||
ChannelID string `json:"channelId"`
|
||||
IsOwnerViewing bool `json:"isOwnerViewing"`
|
||||
ShortDescription string `json:"shortDescription"`
|
||||
IsCrawlable bool `json:"isCrawlable"`
|
||||
Thumbnail struct {
|
||||
Thumbnails []Thumbnail `json:"thumbnails"`
|
||||
} `json:"thumbnail"`
|
||||
AverageRating float64 `json:"averageRating"`
|
||||
AllowRatings bool `json:"allowRatings"`
|
||||
ViewCount string `json:"viewCount"`
|
||||
Author string `json:"author"`
|
||||
IsPrivate bool `json:"isPrivate"`
|
||||
IsUnpluggedCorpus bool `json:"isUnpluggedCorpus"`
|
||||
IsLiveContent bool `json:"isLiveContent"`
|
||||
} `json:"videoDetails"`
|
||||
Microformat struct {
|
||||
PlayerMicroformatRenderer struct {
|
||||
Thumbnail struct {
|
||||
Thumbnails []struct {
|
||||
URL string `json:"url"`
|
||||
Width int `json:"width"`
|
||||
Height int `json:"height"`
|
||||
} `json:"thumbnails"`
|
||||
} `json:"thumbnail"`
|
||||
Title struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"title"`
|
||||
Description struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"description"`
|
||||
LengthSeconds string `json:"lengthSeconds"`
|
||||
OwnerProfileURL string `json:"ownerProfileUrl"`
|
||||
ExternalChannelID string `json:"externalChannelId"`
|
||||
IsFamilySafe bool `json:"isFamilySafe"`
|
||||
AvailableCountries []string `json:"availableCountries"`
|
||||
IsUnlisted bool `json:"isUnlisted"`
|
||||
HasYpcMetadata bool `json:"hasYpcMetadata"`
|
||||
ViewCount string `json:"viewCount"`
|
||||
Category string `json:"category"`
|
||||
PublishDate string `json:"publishDate"`
|
||||
OwnerChannelName string `json:"ownerChannelName"`
|
||||
UploadDate string `json:"uploadDate"`
|
||||
} `json:"playerMicroformatRenderer"`
|
||||
} `json:"microformat"`
|
||||
}
|
||||
|
||||
type Format struct {
|
||||
ItagNo int `json:"itag"`
|
||||
URL string `json:"url"`
|
||||
MimeType string `json:"mimeType"`
|
||||
Quality string `json:"quality"`
|
||||
Cipher string `json:"signatureCipher"`
|
||||
Bitrate int `json:"bitrate"`
|
||||
FPS int `json:"fps"`
|
||||
Width int `json:"width"`
|
||||
Height int `json:"height"`
|
||||
LastModified string `json:"lastModified"`
|
||||
ContentLength int64 `json:"contentLength,string"`
|
||||
QualityLabel string `json:"qualityLabel"`
|
||||
ProjectionType string `json:"projectionType"`
|
||||
AverageBitrate int `json:"averageBitrate"`
|
||||
AudioQuality string `json:"audioQuality"`
|
||||
ApproxDurationMs string `json:"approxDurationMs"`
|
||||
AudioSampleRate string `json:"audioSampleRate"`
|
||||
AudioChannels int `json:"audioChannels"`
|
||||
|
||||
// InitRange is only available for adaptive formats
|
||||
InitRange *struct {
|
||||
Start string `json:"start"`
|
||||
End string `json:"end"`
|
||||
} `json:"initRange"`
|
||||
|
||||
// IndexRange is only available for adaptive formats
|
||||
IndexRange *struct {
|
||||
Start string `json:"start"`
|
||||
End string `json:"end"`
|
||||
} `json:"indexRange"`
|
||||
|
||||
// AudioTrack is only available for videos with multiple audio track languages
|
||||
AudioTrack *struct {
|
||||
DisplayName string `json:"displayName"`
|
||||
ID string `json:"id"`
|
||||
AudioIsDefault bool `json:"audioIsDefault"`
|
||||
}
|
||||
}
|
||||
|
||||
func (f *Format) LanguageDisplayName() string {
|
||||
if f.AudioTrack == nil {
|
||||
return ""
|
||||
}
|
||||
return f.AudioTrack.DisplayName
|
||||
}
|
||||
|
||||
type Thumbnails []Thumbnail
|
||||
|
||||
type Thumbnail struct {
|
||||
URL string
|
||||
Width uint
|
||||
Height uint
|
||||
}
|
||||
|
||||
type CaptionTrack struct {
|
||||
BaseURL string `json:"baseUrl"`
|
||||
Name struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"name"`
|
||||
VssID string `json:"vssId"`
|
||||
LanguageCode string `json:"languageCode"`
|
||||
Kind string `json:"kind,omitempty"`
|
||||
IsTranslatable bool `json:"isTranslatable"`
|
||||
}
|
214
downloader/vendor/github.com/kkdai/youtube/v2/transcript.go
generated
vendored
Normal file
214
downloader/vendor/github.com/kkdai/youtube/v2/transcript.go
generated
vendored
Normal file
@@ -0,0 +1,214 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrTranscriptDisabled = errors.New("transcript is disabled on this video")
|
||||
)
|
||||
|
||||
// TranscriptSegment is a single transcipt segment spanning a few milliseconds.
|
||||
type TranscriptSegment struct {
|
||||
// Text is the transcipt text.
|
||||
Text string `json:"text"`
|
||||
|
||||
// StartMs is the start timestamp in ms.
|
||||
StartMs int `json:"offset"`
|
||||
|
||||
// OffsetText e.g. '4:00'.
|
||||
OffsetText string `json:"offsetText"`
|
||||
|
||||
// Duration the transcript segment spans in ms.
|
||||
Duration int `json:"duration"`
|
||||
}
|
||||
|
||||
func (tr TranscriptSegment) String() string {
|
||||
return tr.OffsetText + " - " + strings.TrimSpace(tr.Text)
|
||||
}
|
||||
|
||||
type VideoTranscript []TranscriptSegment
|
||||
|
||||
func (vt VideoTranscript) String() string {
|
||||
var str string
|
||||
for _, tr := range vt {
|
||||
str += tr.String() + "\n"
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
// GetTranscript fetches the video transcript if available.
|
||||
//
|
||||
// Not all videos have transcripts, only relatively new videos.
|
||||
// If transcripts are disabled or not available, ErrTranscriptDisabled is returned.
|
||||
func (c *Client) GetTranscript(video *Video, lang string) (VideoTranscript, error) {
|
||||
return c.GetTranscriptCtx(context.Background(), video, lang)
|
||||
}
|
||||
|
||||
// GetTranscriptCtx fetches the video transcript if available.
|
||||
//
|
||||
// Not all videos have transcripts, only relatively new videos.
|
||||
// If transcripts are disabled or not available, ErrTranscriptDisabled is returned.
|
||||
func (c *Client) GetTranscriptCtx(ctx context.Context, video *Video, lang string) (VideoTranscript, error) {
|
||||
c.assureClient()
|
||||
|
||||
if video == nil || video.ID == "" {
|
||||
return nil, fmt.Errorf("no video provided")
|
||||
}
|
||||
|
||||
body, err := c.transcriptDataByInnertube(ctx, video.ID, lang)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
transcript, err := parseTranscript(body)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return transcript, nil
|
||||
}
|
||||
|
||||
func parseTranscript(body []byte) (VideoTranscript, error) {
|
||||
var resp transcriptResp
|
||||
if err := json.Unmarshal(body, &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(resp.Actions) > 0 {
|
||||
// Android client response
|
||||
if app := resp.Actions[0].AppSegment; app != nil {
|
||||
return getSegments(app)
|
||||
}
|
||||
|
||||
// Web client response
|
||||
if web := resp.Actions[0].WebSegment; web != nil {
|
||||
return nil, fmt.Errorf("not implemented")
|
||||
}
|
||||
}
|
||||
|
||||
return nil, ErrTranscriptDisabled
|
||||
}
|
||||
|
||||
type segmenter interface {
|
||||
ParseSegments() []TranscriptSegment
|
||||
}
|
||||
|
||||
func getSegments(f segmenter) (VideoTranscript, error) {
|
||||
if segments := f.ParseSegments(); len(segments) > 0 {
|
||||
return segments, nil
|
||||
}
|
||||
|
||||
return nil, ErrTranscriptDisabled
|
||||
}
|
||||
|
||||
// transcriptResp is the JSON structure as returned by the transcript API.
|
||||
type transcriptResp struct {
|
||||
Actions []struct {
|
||||
AppSegment *appData `json:"elementsCommand"`
|
||||
WebSegment *webData `json:"updateEngagementPanelAction"`
|
||||
} `json:"actions"`
|
||||
}
|
||||
|
||||
type appData struct {
|
||||
TEC struct {
|
||||
Args struct {
|
||||
ListArgs struct {
|
||||
Ow struct {
|
||||
InitialSeg []struct {
|
||||
TranscriptSegment struct {
|
||||
StartMs string `json:"startMs"`
|
||||
EndMs string `json:"endMs"`
|
||||
Text struct {
|
||||
String struct {
|
||||
// Content is the actual transctipt text
|
||||
Content string `json:"content"`
|
||||
} `json:"elementsAttributedString"`
|
||||
} `json:"snippet"`
|
||||
StartTimeText struct {
|
||||
String struct {
|
||||
// Content is the fomratted timestamp, e.g. '4:00'
|
||||
Content string `json:"content"`
|
||||
} `json:"elementsAttributedString"`
|
||||
} `json:"startTimeText"`
|
||||
} `json:"transcriptSegmentRenderer"`
|
||||
} `json:"initialSegments"`
|
||||
} `json:"overwrite"`
|
||||
} `json:"transformTranscriptSegmentListArguments"`
|
||||
} `json:"arguments"`
|
||||
} `json:"transformEntityCommand"`
|
||||
}
|
||||
|
||||
func (s *appData) ParseSegments() []TranscriptSegment {
|
||||
rawSegments := s.TEC.Args.ListArgs.Ow.InitialSeg
|
||||
segments := make([]TranscriptSegment, 0, len(rawSegments))
|
||||
|
||||
for _, segment := range rawSegments {
|
||||
startMs, _ := strconv.Atoi(segment.TranscriptSegment.StartMs)
|
||||
endMs, _ := strconv.Atoi(segment.TranscriptSegment.EndMs)
|
||||
|
||||
segments = append(segments, TranscriptSegment{
|
||||
Text: segment.TranscriptSegment.Text.String.Content,
|
||||
StartMs: startMs,
|
||||
OffsetText: segment.TranscriptSegment.StartTimeText.String.Content,
|
||||
Duration: endMs - startMs,
|
||||
})
|
||||
}
|
||||
|
||||
return segments
|
||||
}
|
||||
|
||||
type webData struct {
|
||||
Content struct {
|
||||
TR struct {
|
||||
Body struct {
|
||||
TBR struct {
|
||||
Cues []struct {
|
||||
Transcript struct {
|
||||
FormattedStartOffset struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"formattedStartOffset"`
|
||||
Cues []struct {
|
||||
TranscriptCueRenderer struct {
|
||||
Cue struct {
|
||||
SimpleText string `json:"simpleText"`
|
||||
} `json:"cue"`
|
||||
StartOffsetMs string `json:"startOffsetMs"`
|
||||
DurationMs string `json:"durationMs"`
|
||||
} `json:"transcriptCueRenderer"`
|
||||
} `json:"cues"`
|
||||
} `json:"transcriptCueGroupRenderer"`
|
||||
} `json:"cueGroups"`
|
||||
} `json:"transcriptSearchPanelRenderer"`
|
||||
} `json:"content"`
|
||||
} `json:"transcriptRenderer"`
|
||||
} `json:"content"`
|
||||
}
|
||||
|
||||
func (s *webData) ParseSegments() []TranscriptSegment {
|
||||
// TODO: doesn't actually work now, check json.
|
||||
cues := s.Content.TR.Body.TBR.Cues
|
||||
segments := make([]TranscriptSegment, 0, len(cues))
|
||||
|
||||
for _, s := range cues {
|
||||
formatted := s.Transcript.FormattedStartOffset.SimpleText
|
||||
segment := s.Transcript.Cues[0].TranscriptCueRenderer
|
||||
start, _ := strconv.Atoi(segment.StartOffsetMs)
|
||||
duration, _ := strconv.Atoi(segment.DurationMs)
|
||||
|
||||
segments = append(segments, TranscriptSegment{
|
||||
Text: segment.Cue.SimpleText,
|
||||
StartMs: start,
|
||||
OffsetText: formatted,
|
||||
Duration: duration,
|
||||
})
|
||||
}
|
||||
|
||||
return segments
|
||||
}
|
97
downloader/vendor/github.com/kkdai/youtube/v2/utils.go
generated
vendored
Normal file
97
downloader/vendor/github.com/kkdai/youtube/v2/utils.go
generated
vendored
Normal file
@@ -0,0 +1,97 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
|
||||
sjson "github.com/bitly/go-simplejson"
|
||||
)
|
||||
|
||||
type chunk struct {
|
||||
start int64
|
||||
end int64
|
||||
data chan []byte
|
||||
}
|
||||
|
||||
func getChunks(totalSize, chunkSize int64) []chunk {
|
||||
var chunks []chunk
|
||||
|
||||
for start := int64(0); start < totalSize; start += chunkSize {
|
||||
end := chunkSize + start - 1
|
||||
if end > totalSize-1 {
|
||||
end = totalSize - 1
|
||||
}
|
||||
|
||||
chunks = append(chunks, chunk{start, end, make(chan []byte, 1)})
|
||||
}
|
||||
|
||||
return chunks
|
||||
}
|
||||
|
||||
func getFirstKeyJSON(j *sjson.Json) *sjson.Json {
|
||||
m, err := j.Map()
|
||||
if err != nil {
|
||||
return j
|
||||
}
|
||||
|
||||
for key := range m {
|
||||
return j.Get(key)
|
||||
}
|
||||
|
||||
return j
|
||||
}
|
||||
|
||||
func isValidJSON(j *sjson.Json) bool {
|
||||
b, err := j.MarshalJSON()
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
if len(b) <= 4 {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func sjsonGetText(j *sjson.Json, paths ...string) string {
|
||||
for _, path := range paths {
|
||||
if isValidJSON(j.Get(path)) {
|
||||
j = j.Get(path)
|
||||
}
|
||||
}
|
||||
|
||||
if text, err := j.String(); err == nil {
|
||||
return text
|
||||
}
|
||||
|
||||
if isValidJSON(j.Get("text")) {
|
||||
return j.Get("text").MustString()
|
||||
}
|
||||
|
||||
if p := j.Get("runs"); isValidJSON(p) {
|
||||
var text string
|
||||
|
||||
for i := 0; i < len(p.MustArray()); i++ {
|
||||
if textNode := p.GetIndex(i).Get("text"); isValidJSON(textNode) {
|
||||
text += textNode.MustString()
|
||||
}
|
||||
}
|
||||
|
||||
return text
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func getContinuation(j *sjson.Json) string {
|
||||
return j.GetPath("continuations").
|
||||
GetIndex(0).GetPath("nextContinuationData", "continuation").MustString()
|
||||
}
|
||||
|
||||
func base64PadEnc(str string) string {
|
||||
return base64.StdEncoding.EncodeToString([]byte(str))
|
||||
}
|
||||
|
||||
func base64Enc(str string) string {
|
||||
return base64.RawStdEncoding.EncodeToString([]byte(str))
|
||||
}
|
147
downloader/vendor/github.com/kkdai/youtube/v2/video.go
generated
vendored
Normal file
147
downloader/vendor/github.com/kkdai/youtube/v2/video.go
generated
vendored
Normal file
@@ -0,0 +1,147 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Video struct {
|
||||
ID string
|
||||
Title string
|
||||
Description string
|
||||
Author string
|
||||
ChannelID string
|
||||
ChannelHandle string
|
||||
Views int
|
||||
Duration time.Duration
|
||||
PublishDate time.Time
|
||||
Formats FormatList
|
||||
Thumbnails Thumbnails
|
||||
DASHManifestURL string // URI of the DASH manifest file
|
||||
HLSManifestURL string // URI of the HLS manifest file
|
||||
CaptionTracks []CaptionTrack
|
||||
}
|
||||
|
||||
const dateFormat = "2006-01-02"
|
||||
|
||||
func (v *Video) parseVideoInfo(body []byte) error {
|
||||
var prData playerResponseData
|
||||
if err := json.Unmarshal(body, &prData); err != nil {
|
||||
return fmt.Errorf("unable to parse player response JSON: %w", err)
|
||||
}
|
||||
|
||||
if err := v.isVideoFromInfoDownloadable(prData); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return v.extractDataFromPlayerResponse(prData)
|
||||
}
|
||||
|
||||
func (v *Video) isVideoFromInfoDownloadable(prData playerResponseData) error {
|
||||
return v.isVideoDownloadable(prData, false)
|
||||
}
|
||||
|
||||
var playerResponsePattern = regexp.MustCompile(`var ytInitialPlayerResponse\s*=\s*(\{.+?\});`)
|
||||
|
||||
func (v *Video) parseVideoPage(body []byte) error {
|
||||
initialPlayerResponse := playerResponsePattern.FindSubmatch(body)
|
||||
if initialPlayerResponse == nil || len(initialPlayerResponse) < 2 {
|
||||
return errors.New("no ytInitialPlayerResponse found in the server's answer")
|
||||
}
|
||||
|
||||
var prData playerResponseData
|
||||
if err := json.Unmarshal(initialPlayerResponse[1], &prData); err != nil {
|
||||
return fmt.Errorf("unable to parse player response JSON: %w", err)
|
||||
}
|
||||
|
||||
if err := v.isVideoFromPageDownloadable(prData); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return v.extractDataFromPlayerResponse(prData)
|
||||
}
|
||||
|
||||
func (v *Video) isVideoFromPageDownloadable(prData playerResponseData) error {
|
||||
return v.isVideoDownloadable(prData, true)
|
||||
}
|
||||
|
||||
func (v *Video) isVideoDownloadable(prData playerResponseData, isVideoPage bool) error {
|
||||
// Check if video is downloadable
|
||||
switch prData.PlayabilityStatus.Status {
|
||||
case "OK":
|
||||
return nil
|
||||
case "LOGIN_REQUIRED":
|
||||
// for some reason they use same status message for age-restricted and private videos
|
||||
if strings.HasPrefix(prData.PlayabilityStatus.Reason, "This video is private") {
|
||||
return ErrVideoPrivate
|
||||
}
|
||||
return ErrLoginRequired
|
||||
}
|
||||
|
||||
if !isVideoPage && !prData.PlayabilityStatus.PlayableInEmbed {
|
||||
return ErrNotPlayableInEmbed
|
||||
}
|
||||
|
||||
return &ErrPlayabiltyStatus{
|
||||
Status: prData.PlayabilityStatus.Status,
|
||||
Reason: prData.PlayabilityStatus.Reason,
|
||||
}
|
||||
}
|
||||
|
||||
func (v *Video) extractDataFromPlayerResponse(prData playerResponseData) error {
|
||||
v.Title = prData.VideoDetails.Title
|
||||
v.Description = prData.VideoDetails.ShortDescription
|
||||
v.Author = prData.VideoDetails.Author
|
||||
v.Thumbnails = prData.VideoDetails.Thumbnail.Thumbnails
|
||||
v.ChannelID = prData.VideoDetails.ChannelID
|
||||
v.CaptionTracks = prData.Captions.PlayerCaptionsTracklistRenderer.CaptionTracks
|
||||
|
||||
if views, _ := strconv.Atoi(prData.VideoDetails.ViewCount); views > 0 {
|
||||
v.Views = views
|
||||
}
|
||||
|
||||
if seconds, _ := strconv.Atoi(prData.VideoDetails.LengthSeconds); seconds > 0 {
|
||||
v.Duration = time.Duration(seconds) * time.Second
|
||||
}
|
||||
|
||||
if seconds, _ := strconv.Atoi(prData.Microformat.PlayerMicroformatRenderer.LengthSeconds); seconds > 0 {
|
||||
v.Duration = time.Duration(seconds) * time.Second
|
||||
}
|
||||
|
||||
if str := prData.Microformat.PlayerMicroformatRenderer.PublishDate; str != "" {
|
||||
v.PublishDate, _ = time.Parse(dateFormat, str)
|
||||
}
|
||||
|
||||
if profileURL, err := url.Parse(prData.Microformat.PlayerMicroformatRenderer.OwnerProfileURL); err == nil && len(profileURL.Path) > 1 {
|
||||
v.ChannelHandle = profileURL.Path[1:]
|
||||
}
|
||||
|
||||
// Assign Streams
|
||||
v.Formats = append(prData.StreamingData.Formats, prData.StreamingData.AdaptiveFormats...)
|
||||
if len(v.Formats) == 0 {
|
||||
return errors.New("no formats found in the server's answer")
|
||||
}
|
||||
|
||||
// Sort formats by bitrate
|
||||
sort.SliceStable(v.Formats, v.SortBitrateDesc)
|
||||
|
||||
v.HLSManifestURL = prData.StreamingData.HlsManifestURL
|
||||
v.DASHManifestURL = prData.StreamingData.DashManifestURL
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (v *Video) SortBitrateDesc(i int, j int) bool {
|
||||
return v.Formats[i].Bitrate > v.Formats[j].Bitrate
|
||||
}
|
||||
|
||||
func (v *Video) SortBitrateAsc(i int, j int) bool {
|
||||
return v.Formats[i].Bitrate < v.Formats[j].Bitrate
|
||||
}
|
34
downloader/vendor/github.com/kkdai/youtube/v2/video_id.go
generated
vendored
Normal file
34
downloader/vendor/github.com/kkdai/youtube/v2/video_id.go
generated
vendored
Normal file
@@ -0,0 +1,34 @@
|
||||
package youtube
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var videoRegexpList = []*regexp.Regexp{
|
||||
regexp.MustCompile(`(?:v|embed|shorts|watch\?v)(?:=|/)([^"&?/=%]{11})`),
|
||||
regexp.MustCompile(`(?:=|/)([^"&?/=%]{11})`),
|
||||
regexp.MustCompile(`([^"&?/=%]{11})`),
|
||||
}
|
||||
|
||||
// ExtractVideoID extracts the videoID from the given string
|
||||
func ExtractVideoID(videoID string) (string, error) {
|
||||
if strings.Contains(videoID, "youtu") || strings.ContainsAny(videoID, "\"?&/<%=") {
|
||||
for _, re := range videoRegexpList {
|
||||
if isMatch := re.MatchString(videoID); isMatch {
|
||||
subs := re.FindStringSubmatch(videoID)
|
||||
videoID = subs[1]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if strings.ContainsAny(videoID, "?&/<%=") {
|
||||
return "", ErrInvalidCharactersInVideoID
|
||||
}
|
||||
|
||||
if len(videoID) < 10 {
|
||||
return "", ErrVideoIDMinLength
|
||||
}
|
||||
|
||||
return videoID, nil
|
||||
}
|
@@ -22,7 +22,7 @@ if (!(titleNode && titleNode.matches(selector))) {
|
||||
link = window.location.origin + link;
|
||||
}
|
||||
console.log(link);
|
||||
fetch("http://localhost:5000/download", {
|
||||
fetch("https://nsq.site.quack-lab.dev/pub?topic=ytdqueue", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
|
Reference in New Issue
Block a user