/
godl.go
109 lines (95 loc) · 2.22 KB
/
godl.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package main
import (
"net/http"
"os"
"os/signal"
"runtime"
"syscall"
"github.com/chzyer/flagx"
"gopkg.in/logex.v1"
)
type Config struct {
Proxy []string `flag:"p;usage=proxy"`
Server string `flag:"s;usage=godl will enter server mode if specified listen addr with -s"`
Overwrite bool `flag:"f;usage=overwritten if file is exists, false mean resume the progress from the meta file"`
MaxSpeed int64 `flag:"max;usage=max speed"`
ConnSize int `flag:"n;def=5;usage=specified the max connections connected"`
BlockBit uint `flag:"b;def=20;usage=block size represented by bit"`
Meta bool `flag:"usage=print meta"`
Progress bool `flag:"np;def=true;usage=show progress"`
Debug bool `flag:"v;usage=turn on debug mode"`
Url2 string `flag:"u;usage=url, same as specified at arg"`
Url string `flag:"[0];usage=url"`
Headers []string `flag:"H"`
obj *flagx.Object
}
func NewConfig() *Config {
var c Config
c.obj = flagx.Parse(&c)
if c.Url == "" && c.Url2 != "" {
c.Url = c.Url2
}
logex.ShowCode = c.Debug
return &c
}
func singleDn(c *Config, cwd string) {
if c.Url == "" {
c.obj.Usage()
return
}
tcfg := &TaskConfig{
Clean: c.Overwrite,
MaxSpeed: c.MaxSpeed,
Progress: c.Progress,
Proxy: c.Proxy,
ShowRealSp: c.Debug,
Headers: c.Headers,
}
task, err := NewDnTaskAuto(c.Url, cwd, c.BlockBit, tcfg)
if err != nil {
logex.Fatal(err)
}
if c.Meta {
task.Meta.header = nil
for i := range task.Meta.Blocks {
if task.Meta.Blocks[i] == nil {
task.Meta.Blocks = task.Meta.Blocks[:i]
break
}
}
logex.Pretty(task.Meta)
return
}
closeSignal := make(chan os.Signal, 1)
go func() {
task.Schedule(c.ConnSize)
closeSignal <- os.Interrupt
}()
signal.Notify(closeSignal,
os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGHUP)
<-closeSignal
task.Close()
if task.Meta.IsFinish() {
task.Meta.Remove()
} else {
task.Meta.Sync()
}
}
func main() {
runtime.GOMAXPROCS(4)
c := NewConfig()
cwd, err := os.Getwd()
if err != nil {
logex.Fatal(err)
}
if c.Server != "" {
mux := http.NewServeMux()
bindHandler(mux)
err := http.ListenAndServe(c.Server, mux)
if err != nil {
logex.Fatal(err)
}
return
}
singleDn(c, cwd)
}