package main import ( "context" "flag" "fmt" "io" "log" "net/http" "os" "strings" "sync" "time" "git.netflux.io/rob/clipper/media" "git.netflux.io/rob/clipper/youtube" youtubev2 "github.com/kkdai/youtube/v2" ) func main() { var ( verbose bool printMode bool downloadMode bool audioOnly bool videoOnly bool ) flag.BoolVar(&verbose, "v", false, "verbose output") flag.BoolVar(&printMode, "print", true, "print format info") flag.BoolVar(&downloadMode, "download", false, "download all media to ./debug") flag.BoolVar(&audioOnly, "audio", false, "only print audio formats") flag.BoolVar(&videoOnly, "video", false, "only print video formats") flag.Parse() videoID := flag.Arg(0) ctx := context.Background() var youtubeClient youtubev2.Client video, err := youtubeClient.GetVideoContext(ctx, videoID) if err != nil { log.Fatal(err) } formats := video.Formats if downloadMode { downloadAll(formats) return } switch { case audioOnly: formats = media.FilterYoutubeAudio(formats) case videoOnly: formats = media.FilterYoutubeVideo(formats) } fmt.Println("In descending order of preference:") for n, f := range formats { fmt.Printf("%d: %s\n", n+1, youtube.FormatDebugString(&f, verbose)) } } func downloadAll(formats youtubev2.FormatList) { var wg sync.WaitGroup for i := range formats { format := formats[i] wg.Add(1) go func() { defer wg.Done() start := time.Now() outpath := fmt.Sprintf("./debug/%s.%s-itag-%d", strings.ReplaceAll(format.MimeType, "/", "'"), format.Quality, format.ItagNo) output, err := os.Create(outpath) if err != nil { log.Fatalf("error opening output file: %v", err) } resp, err := http.Get(format.URL) if err != nil { log.Fatalf("error fetching media: %v", err) } defer resp.Body.Close() n, err := io.Copy(output, resp.Body) if err != nil { log.Fatalf("error reading media: %v", err) } dur := time.Since(start) log.Printf("downloaded itag %d, %d bytes in %v secs", format.ItagNo, n, dur.Seconds()) }() } wg.Wait() }