mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
package imguploader
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
"golang.org/x/oauth2/google"
|
|
)
|
|
|
|
type GCSUploader struct {
|
|
keyFile string
|
|
bucket string
|
|
log log.Logger
|
|
}
|
|
|
|
func NewGCSUploader(keyFile, bucket string) *GCSUploader {
|
|
return &GCSUploader{
|
|
keyFile: keyFile,
|
|
bucket: bucket,
|
|
log: log.New("gcsuploader"),
|
|
}
|
|
}
|
|
|
|
func (u *GCSUploader) Upload(ctx context.Context, imageDiskPath string) (string, error) {
|
|
key := util.GetRandomString(20) + ".png"
|
|
|
|
log.Debug("Opening key file ", u.keyFile)
|
|
|
|
data, err := ioutil.ReadFile(u.keyFile)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
log.Debug("Creating JWT conf")
|
|
|
|
conf, err := google.JWTConfigFromJSON(data, "https://www.googleapis.com/auth/devstorage.read_write")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
log.Debug("Creating HTTP client")
|
|
|
|
client := conf.Client(ctx)
|
|
|
|
err = u.uploadFile(client, imageDiskPath, key)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return fmt.Sprintf("https://storage.googleapis.com/%s/%s", u.bucket, key), nil
|
|
}
|
|
|
|
func (u *GCSUploader) uploadFile(client *http.Client, imageDiskPath, key string) error {
|
|
log.Debug("Opening image file ", imageDiskPath)
|
|
|
|
fileReader, err := os.Open(imageDiskPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
reqUrl := fmt.Sprintf(
|
|
"https://www.googleapis.com/upload/storage/v1/b/%s/o?uploadType=media&name=%s&predefinedAcl=publicRead",
|
|
u.bucket,
|
|
key,
|
|
)
|
|
|
|
log.Debug("Request URL: ", reqUrl)
|
|
|
|
req, err := http.NewRequest("POST", reqUrl, fileReader)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req.Header.Add("Content-Type", "image/png")
|
|
|
|
log.Debug("Sending POST request to GCS")
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Debug("GCS API response header", resp.Header)
|
|
|
|
if resp.StatusCode != 200 {
|
|
return errors.New(fmt.Sprintf("GCS response status code %d", resp.StatusCode))
|
|
}
|
|
|
|
return nil
|
|
}
|