69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"regexp"
|
|
"sync"
|
|
//"path/filepath"
|
|
|
|
. "github.com/kkdai/youtube"
|
|
)
|
|
|
|
type Batch struct {
|
|
batchfile, destDir string
|
|
}
|
|
|
|
func NewBatch(batchFile, destinationDir string) *Batch {
|
|
return &Batch{batchfile: batchFile, destDir: destinationDir}
|
|
}
|
|
|
|
func (b *Batch) Start() {
|
|
//Start downloading all videos
|
|
log.Println("Start downloading...")
|
|
//open the file containing yt links
|
|
linkfile, err := os.Open(b.batchfile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer linkfile.Close()
|
|
|
|
//Create scanner to go line by line through the file
|
|
//This scanner allows 65536 characters per line.
|
|
//That should be enough for yt links.
|
|
scanner := bufio.NewScanner(linkfile)
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
for scanner.Scan() {
|
|
//Call download
|
|
log.Println(scanner.Text())
|
|
//add go routine to wait for
|
|
wg.Add(1)
|
|
go downloadVideo(b.destDir, scanner.Text(), wg)
|
|
}
|
|
|
|
//wait for all go routines to finish
|
|
wg.Wait()
|
|
if err := scanner.Err(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func downloadVideo(destDir, url string, wg sync.WaitGroup) {
|
|
// NewYoutube(debug) if debug parameter will set true we can log of messages
|
|
//defer Done() so that WaitGroup knows when the routine finishes
|
|
defer wg.Done()
|
|
y := NewYoutube(false)
|
|
y.DecodeURL(url)
|
|
re := regexp.MustCompile("(mp4|webm|3gpp)")
|
|
fileext := re.FindString(y.StreamList[0]["type"])
|
|
filename := fmt.Sprintf("%s/%s.%s", destDir, y.StreamList[0]["title"], fileext)
|
|
|
|
log.Println("Start downloading:", filename)
|
|
y.StartDownload(filename)
|
|
log.Println("Finished downloading:", filename)
|
|
}
|