mirrors/vivo
1
0
Fork 0
mirror of https://github.com/bytedream/vivo.git synced 2025-05-10 04:35:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Added quiet (-q) option

Browse Source
This commit is contained in:
ByteDream 2021-07-19 01:30:11 +02:00
parent 8690dbd303
commit 40898b6d1a
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
cmd/main.go
View File

@ -3,6 +3,7 @@ package main
import ( import (
"flag" "flag"
"fmt" "fmt"
"io/ioutil"
"log" "log"
"net/http" "net/http"
"net/url" "net/url"
@ -20,6 +21,7 @@ var (
output string output string
proxy string proxy string
clean bool clean bool
quiet bool
) )
const ( const (
@ -55,10 +57,19 @@ func main() {
flag.StringVar(&output, "o", ".", "Destination of the file") flag.StringVar(&output, "o", ".", "Destination of the file")
flag.StringVar(&proxy, "p", "", "Proxy to use") flag.StringVar(&proxy, "p", "", "Proxy to use")
flag.BoolVar(&clean, "c", false, "Show clean output / disable all additions (colors and separator between multiple downloads)") flag.BoolVar(&clean, "c", false, "Show clean output / disable all additions (colors and separator between multiple downloads)")
flag.BoolVar(&quiet, "q", false, "Disable the output")
flag.Parse() flag.Parse()
e := log.New(os.Stderr, "", 0) var o, e *log.Logger
if quiet {
o = log.New(ioutil.Discard, "", 0)
e = log.New(ioutil.Discard, "", 0)
} else {
o = log.New(os.Stdout, "", 0)
e = log.New(os.Stderr, "", 0)
}
if flag.NArg() == 0 { if flag.NArg() == 0 {
e.Fatalln(colorize(RED, "At least one url must be specified")) e.Fatalln(colorize(RED, "At least one url must be specified"))
@ -96,17 +107,17 @@ func main() {
colorize(MAGENTA, video.Mime)) colorize(MAGENTA, video.Mime))
if infos { if infos {
fmt.Println(videoInformation) o.Println(videoInformation)
} }
if source { if source {
if infos { if infos {
fmt.Println() o.Println()
} }
fmt.Println(video.VideoURL) o.Println(video.VideoURL)
} }
if !(infos || source) { if !(infos || source) {
fmt.Printf("%s\n\n", videoInformation) o.Printf("%s\n\n", videoInformation)
fileInfo, err := os.Stat(output) fileInfo, err := os.Stat(output)
if fileInfo != nil && fileInfo.IsDir() { if fileInfo != nil && fileInfo.IsDir() {
@ -123,16 +134,16 @@ func main() {
} }
} }
fmt.Printf(colorize(CYAN, "Downloading %s to '%s'..."), video.VivoURL, output) o.Printf(colorize(CYAN, "Downloading %s to '%s'..."), video.VivoURL, output)
if err := video.Download(file); err != nil { if err := video.Download(file); err != nil {
e.Fatalln(colorize(RED, err.Error())) e.Fatalln(colorize(RED, err.Error()))
} }
fmt.Printf(colorize(GREEN, " finished\n")) o.Printf(colorize(GREEN, " finished\n"))
} }
if i != flag.NArg()-1 && !clean { if i != flag.NArg()-1 && !clean {
fmt.Println("\n--------------------------------------------------") o.Println("\n--------------------------------------------------")
} }
} }
} }