-
Notifications
You must be signed in to change notification settings - Fork 0
/
uploader.go
105 lines (89 loc) · 2.5 KB
/
uploader.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
package upload
import (
"bytes"
"fmt"
"os/exec"
"regexp"
log "github.com/sirupsen/logrus"
)
func isValidURL(toTest string) bool {
re := regexp.MustCompile(`^(?:https?:\/\/)?(?:[^@\/\n]+@)?(?:www\.)?([^:\/\n]+)`)
log.Debugf("Pattern: %v\n", re.String())
if re.MatchString(toTest) {
return true
}
return false
}
//uploadExeCMD exec bash script to upload file
//upload.sh -a 'username:password' file_host_name /path/to/file
func uploadExeCMD(f, path, filehost, username, password, minSpeed string) (l string, err error) {
/**************************************************
*
* @f file
* @filehost sitename
* @path where the script is located path/to/app/
* @username username
* @password password
*
**************************************************/
uploadPath := fmt.Sprintf("%supload.sh", path)
log.Debugf("Uploader Path: %s", uploadPath)
//Check if ffmpeg is installed
path, err = exec.LookPath(uploadPath)
if err != nil {
log.Fatalf("Bash script for upload is required")
}
log.Debugf("Bash script for upload is available at %s\n", path)
var login string
if password == "" {
login = username
} else {
login = fmt.Sprintf("%s:%s", username, password)
}
//cmd := exec.Command(parts[0], parts[1])
cmd := exec.Command(uploadPath, "--min-rate", minSpeed, "-a", login, filehost, f)
var out bytes.Buffer
var stderr bytes.Buffer
cmd.Stdout = &out
cmd.Stderr = &stderr
err = cmd.Run()
if err != nil {
log.WithFields(log.Fields{
"err": err.Error(),
"cmd": cmd,
"output": stderr.String(),
"handler": "UploaderExeCMD",
}).Errorf("EXEC[Uploader] -Error in EXEC")
return l, err
}
l = out.String()
log.Debugf("Result: " + out.String())
if ok := isValidURL(l); !ok {
log.WithFields(log.Fields{
"err": out.String(),
"cmd": cmd,
"output": stderr.String(),
"handler": "UploaderExeCMD",
}).Errorf("EXEC[Uploader] - Not a valid URL in response")
return l, err
}
return l, nil
}
// ToFileHost The main function to call
/**************************************************
*
* @f file to upload, full path
* @filehost name of filehost to upload
* @path where the script is located path/to/app/
* @username username
* @password password
* @minSpeed min speed of upload in kb/s EX: 1500
*
**************************************************/
func ToFileHost(f, path, filehost, username, password, minSpeed string) (l string, err error) {
l, err = uploadExeCMD(f, path, filehost, username, password, minSpeed)
if err != nil {
return l, err
}
return l, nil
}