-
Notifications
You must be signed in to change notification settings - Fork 228
/
main.go
87 lines (77 loc) · 1.59 KB
/
main.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
package main
import (
"context"
"flag"
"log"
"os"
"os/signal"
"sync"
getter "github.com/hashicorp/go-getter"
)
func main() {
modeRaw := flag.String("mode", "any", "get mode (any, file, dir)")
progress := flag.Bool("progress", false, "display terminal progress")
flag.Parse()
args := flag.Args()
if len(args) < 2 {
log.Fatalf("Expected two args: URL and dst")
os.Exit(1)
}
// Get the mode
var mode getter.ClientMode
switch *modeRaw {
case "any":
mode = getter.ClientModeAny
case "file":
mode = getter.ClientModeFile
case "dir":
mode = getter.ClientModeDir
default:
log.Fatalf("Invalid client mode, must be 'any', 'file', or 'dir': %s", *modeRaw)
os.Exit(1)
}
// Get the pwd
pwd, err := os.Getwd()
if err != nil {
log.Fatalf("Error getting wd: %s", err)
}
opts := []getter.ClientOption{}
if *progress {
opts = append(opts, getter.WithProgress(defaultProgressBar))
}
ctx, cancel := context.WithCancel(context.Background())
// Build the client
client := &getter.Client{
Ctx: ctx,
Src: args[0],
Dst: args[1],
Pwd: pwd,
Mode: mode,
Options: opts,
}
wg := sync.WaitGroup{}
wg.Add(1)
errChan := make(chan error, 2)
go func() {
defer wg.Done()
defer cancel()
if err := client.Get(); err != nil {
errChan <- err
}
}()
c := make(chan os.Signal)
signal.Notify(c, os.Interrupt)
select {
case sig := <-c:
signal.Reset(os.Interrupt)
cancel()
wg.Wait()
log.Printf("signal %v", sig)
case <-ctx.Done():
wg.Wait()
log.Printf("success!")
case err := <-errChan:
wg.Wait()
log.Fatalf("Error downloading: %s", err)
}
}