50 lines
795 B
Go
50 lines
795 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/wader/goutubedl"
|
|
)
|
|
|
|
var dumpFlag = flag.Bool("J", false, "Dump JSON")
|
|
var typeFlag = flag.String("t", "any", "Type")
|
|
|
|
func main() {
|
|
log.SetFlags(0)
|
|
flag.Parse()
|
|
|
|
optType, _ := goutubedl.TypeFromString[*typeFlag]
|
|
result, err := goutubedl.New(context.Background(), flag.Arg(0), goutubedl.Options{Type: optType})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if *dumpFlag {
|
|
json.NewEncoder(os.Stdout).Encode(result.Info)
|
|
return
|
|
}
|
|
|
|
filter := flag.Arg(1)
|
|
if filter == "" {
|
|
filter = "best"
|
|
}
|
|
|
|
dr, err := result.Download(context.Background(), filter)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
f, err := os.Create(filter)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
io.Copy(f, dr)
|
|
dr.Close()
|
|
}
|