ci: build web

cmd/cli/flags.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+)
+
+type args struct {
+	url         string
+	connections *int
+	dir         *string
+}
+
+func parse() *args {
+	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
+	if err != nil {
+		panic(err)
+	}
+	var args args
+	args.connections = flag.Int("C", 16, "Concurrent connections.")
+	args.dir = flag.String("D", dir, "Store directory.")
+	flag.Parse()
+	t := flag.Args()
+	if len(t) > 0 {
+		args.url = t[0]
+	} else {
+		gPrintln("missing url parameter")
+		gPrintln("try 'gopeed -h' for more information")
+		os.Exit(1)
+	}
+	return &args
+}
+
+func gPrint(msg string) {
+	fmt.Print("gopeed: " + msg)
+}
+
+func gPrintln(msg string) {
+	gPrint(msg + "\n")
+}

cmd/cli/main.go

@@ -0,0 +1,80 @@
+package main
+
+import (
+	"fmt"
+	"github.com/monkeyWie/gopeed/pkg/base"
+	"github.com/monkeyWie/gopeed/pkg/download"
+	"github.com/monkeyWie/gopeed/pkg/util"
+	"strings"
+	"sync"
+)
+
+const progressWidth = 20
+
+func main() {
+	args := parse()
+
+	var wg sync.WaitGroup
+	_, err := download.Boot().
+		URL(args.url).
+		Listener(func(event *download.Event) {
+			if event.Key == download.EventKeyProgress {
+				printProgress(event.Task, "downloading...")
+			}
+			if event.Key == download.EventKeyFinally {
+				var title string
+				if event.Err != nil {
+					title = "fail"
+				} else {
+					title = "complete"
+				}
+				printProgress(event.Task, title)
+				fmt.Println()
+				if event.Err != nil {
+					fmt.Printf("reason: %s", event.Err.Error())
+				} else {
+					fmt.Printf("saving path: %s", *args.dir)
+				}
+				wg.Done()
+			}
+		}).
+		Create(&base.Options{
+			Path:        *args.dir,
+			Connections: *args.connections,
+		})
+	if err != nil {
+		panic(err)
+	}
+	wg.Add(1)
+	wg.Wait()
+}
+
+var (
+	lastLineLen = 0
+	sb          = new(strings.Builder)
+)
+
+func printProgress(task *download.Task, title string) {
+	rate := float64(task.Progress.Downloaded) / float64(task.Res.Size)
+	completeWidth := int(progressWidth * rate)
+	speed := util.ByteFmt(task.Progress.Speed)
+	totalSize := util.ByteFmt(task.Res.Size)
+	sb.WriteString(fmt.Sprintf("\r%s [", title))
+	for i := 0; i < progressWidth; i++ {
+		if i < completeWidth {
+			sb.WriteString("■")
+		} else {
+			sb.WriteString("□")
+		}
+	}
+	sb.WriteString(fmt.Sprintf("] %.1f%%    %s/s    %s", rate*100, speed, totalSize))
+	if lastLineLen != 0 {
+		paddingLen := lastLineLen - sb.Len()
+		if paddingLen > 0 {
+			sb.WriteString(strings.Repeat(" ", paddingLen))
+		}
+	}
+	lastLineLen = sb.Len()
+	fmt.Print(sb.String())
+	sb.Reset()
+}