-
Notifications
You must be signed in to change notification settings - Fork 0
/
getter.go
94 lines (76 loc) · 1.58 KB
/
getter.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
package getter
import (
"crypto/rand"
"encoding/hex"
"fmt"
"os"
"path/filepath"
"strings"
"time"
"github.com/briandowns/spinner"
getter "github.com/hashicorp/go-getter"
)
func extractLastString(url string) string {
parts := strings.Split(url, "/")
return parts[len(parts)-1]
}
func Getter(src string) (string, error) {
RepoString := extractLastString(src)
spinner := newSpinner(fmt.Sprintf(" Extracting files from %s \n", RepoString))
spinner.Color("green", "bold")
spinner.Start()
defer spinner.Stop()
suffix, err := randomSuffix()
if err != nil {
return "", err
}
dst := filepath.Join(os.TempDir(), fmt.Sprintf("gcloc-extract-%s", suffix))
pwd, err := os.Getwd()
if err != nil {
return "", err
}
client := &getter.Client{
Src: src,
Dst: dst,
Pwd: pwd,
//Mode: getter.ClientModeAny,
Mode: getter.ClientModeDir,
}
if err := client.Get(); err != nil {
return "", err
}
symLink, err := isSymLink(dst)
if err != nil {
return "", err
}
if symLink {
origin, err := os.Readlink(dst)
if err != nil {
return "", err
}
return origin, nil
}
return dst, nil
}
func newSpinner(text string) *spinner.Spinner {
return spinner.New(
spinner.CharSets[35],
100*time.Millisecond,
spinner.WithSuffix(text),
)
}
func randomSuffix() (string, error) {
randBytes := make([]byte, 16)
_, err := rand.Read(randBytes)
if err != nil {
return "", err
}
return hex.EncodeToString(randBytes), nil
}
func isSymLink(path string) (bool, error) {
info, err := os.Lstat(path)
if err != nil {
return false, err
}
return info.Mode()&os.ModeSymlink != 0, nil
}