-
Notifications
You must be signed in to change notification settings - Fork 6
/
main.go
89 lines (72 loc) · 1.9 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
88
89
package main
import (
"context"
"encoding/base64"
"io"
"io/ioutil"
"net/http"
"os"
"time"
"github.com/direktiv/direktiv-apps/pkg/direktivapps"
"google.golang.org/api/option"
"cloud.google.com/go/storage"
)
type GoogleUploadBlob struct {
Bucket string `json:"bucket"`
Object string `json:"upload-name"`
Data string `json:"data"`
ServiceAccountKey string `json:"serviceAccountKey"`
}
const credFile = "/creds"
const code = "com.google-upload.error"
func main() {
direktivapps.StartServer(GoogleUpload)
}
func GoogleUpload(w http.ResponseWriter, r *http.Request) {
obj := new(GoogleUploadBlob)
_, err := direktivapps.Unmarshal(obj, r)
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
ctx := context.Background()
err = ioutil.WriteFile(credFile, []byte(obj.ServiceAccountKey), 0777)
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
client, err := storage.NewClient(ctx, option.WithCredentialsFile(credFile))
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
defer client.Close()
decoded, err := base64.StdEncoding.DecodeString(obj.Data)
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
err = ioutil.WriteFile(obj.Object, decoded, 0700)
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
f, err := os.Open(obj.Object)
if err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
defer f.Close()
ctx, cancel := context.WithTimeout(ctx, time.Second*50)
defer cancel()
wc := client.Bucket(obj.Bucket).Object(obj.Object).NewWriter(ctx)
if _, err = io.Copy(wc, f); err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
if err := wc.Close(); err != nil {
direktivapps.RespondWithError(w, code, err.Error())
return
}
direktivapps.Respond(w, []byte{})
}