forked from hashicorp/go-getter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
55 lines (47 loc) · 942 Bytes
/
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
package main
import (
"flag"
"log"
"os"
"github.com/hashicorp/go-getter"
)
func main() {
modeRaw := flag.String("mode", "any", "get mode (any, file, dir)")
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)
os.Exit(1)
}
// Build the client
client := &getter.Client{
Src: args[0],
Dst: args[1],
Pwd: pwd,
Mode: mode,
}
if err := client.Get(); err != nil {
log.Fatalf("Error downloading: %s", err)
os.Exit(1)
}
log.Println("Success!")
}