36 lines
870 B
Go
36 lines
870 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"mime/multipart"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MediaStorage interface {
|
||
|
|
Upload(fileName string, file multipart.File) (location string, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// BunnyCDN
|
||
|
|
type BunnyCdnStorage struct {
|
||
|
|
// Access Key
|
||
|
|
key string
|
||
|
|
// Storage zone name
|
||
|
|
storageZoneName string
|
||
|
|
// Base location
|
||
|
|
baseLocation string
|
||
|
|
}
|
||
|
|
|
||
|
|
var bunnyCdnStorageUrl = "https://storage.bunnycdn.com"
|
||
|
|
|
||
|
|
func (b BunnyCdnStorage) Upload(fileName string, file multipart.File) (location string, err error) {
|
||
|
|
client := &http.Client{}
|
||
|
|
req, _ := http.NewRequest("PUT", bunnyCdnStorageUrl+"/"+url.PathEscape(b.storageZoneName)+"/"+url.PathEscape("/"+fileName), file)
|
||
|
|
req.Header.Add("AccessKey", b.key)
|
||
|
|
resp, err := client.Do(req)
|
||
|
|
if err != nil || resp.StatusCode != 201 {
|
||
|
|
return "", errors.New("failed to upload file to BunnyCDN")
|
||
|
|
}
|
||
|
|
return b.baseLocation + fileName, nil
|
||
|
|
}
|