forked from rapidloop/mybot
-
Notifications
You must be signed in to change notification settings - Fork 16
/
utils.go
90 lines (80 loc) · 1.95 KB
/
utils.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
package main
import (
"io/ioutil"
"log"
"math/rand"
"net/http"
"os/exec"
"path/filepath"
"time"
)
var filenames = []string{
"have-a-good-day-have-it.jpg",
"pro-smite-player.jpg",
"esports-legend.jpg",
"awkward-dancer.jpg",
"Malakhor.jpg",
"seeya-very-much.jpg",
"presumably.jpg",
"that-sounded-weird-but-im-sticking-to-it.jpeg",
}
func init() {
rand.Seed(time.Now().UnixNano())
}
var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
func RandStringRunes(n int) string {
b := make([]rune, n)
for i := range b {
b[i] = letterRunes[rand.Intn(len(letterRunes))]
}
return string(b)
}
func randomFile() string {
return RandStringRunes(6) + "-" + filenames[rand.Intn(len(filenames))]
}
func SaveTempFile(b []byte) string {
file, err := ioutil.TempFile("", "slack_image")
if err != nil {
log.Fatalf("error saving file: %s", err)
}
if _, err = file.Write(b); err != nil {
log.Fatalf("error writing file: %s", err)
}
if err = file.Close(); err != nil {
log.Fatalf("error closing file: %s", err)
}
return file.Name()
}
func SaveFile(b []byte) string {
file := randomFile()
path := filepath.Join(base_path, file)
ioutil.WriteFile(path, b, 0644)
return base_url + file
}
func GetFile(file File) []byte {
client := &http.Client{
Timeout: time.Second * 20,
}
request, err := http.NewRequest(http.MethodGet, file.URLPrivateDownload, nil)
if err != nil {
log.Fatalf("error creating request: %s", err)
}
request.Header.Add("Authorization", "Bearer "+token)
response, err := client.Do(request)
if err != nil {
log.Fatalf("error downloading file\n%v\n%v", file, err)
}
defer response.Body.Close()
body, err := ioutil.ReadAll(response.Body)
if err != nil {
log.Fatalf("error downloading file\n%v\n%v", file, err)
}
return body
}
func Chrisify(file string) []byte {
out, err := exec.Command(chrisify, "--haar", haar, file).Output()
if err != nil {
log.Fatalf("couldn't chrisify: %s", err)
}
return out
}