70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
|
package b2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Upload B2 upload information
|
||
|
|
type Upload struct {
|
||
|
|
BucketID string `json:"bucketId"`
|
||
|
|
UploadURL string `json:"uploadUrl"`
|
||
|
|
AuthToken string `json:"authorizationToken"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// UploadFile uploads one file to B2
|
||
|
|
func (u *Upload) UploadFile(data io.Reader, fileName string, fileSize int64, contentType string, sha1 string, mtime *time.Time, info map[string]string) (*FileInfo, error) {
|
||
|
|
req, err := http.NewRequest("POST", u.UploadURL, data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// content length is necessary for buffers like os.File
|
||
|
|
req.ContentLength = fileSize
|
||
|
|
|
||
|
|
// use B2's autodetect content type if one is not passed
|
||
|
|
if contentType == "" {
|
||
|
|
contentType = "b2/x-auto"
|
||
|
|
}
|
||
|
|
|
||
|
|
// encode fileName via URL encoding per B2's documentation
|
||
|
|
fileEncoded, err := url.Parse(fileName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
fileName = fileEncoded.String()
|
||
|
|
|
||
|
|
req.Header.Add("Authorization", u.AuthToken)
|
||
|
|
req.Header.Add("X-Bz-File-Name", fileName)
|
||
|
|
req.Header.Add("Content-Type", contentType)
|
||
|
|
req.Header.Add("X-Bz-Content-Sha1", sha1)
|
||
|
|
|
||
|
|
// B2 requires time to be in UNIX milliseconds
|
||
|
|
if mtime != nil {
|
||
|
|
req.Header.Add("X-Bz-Info-src_last_modified_millis", fmt.Sprint(mtime.UnixNano()/1000000))
|
||
|
|
}
|
||
|
|
|
||
|
|
if info != nil {
|
||
|
|
for name, value := range info {
|
||
|
|
req.Header.Add("X-Bz-Info-"+name, value)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := http.DefaultClient.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
fileInfo := &FileInfo{}
|
||
|
|
err = readResp(resp, fileInfo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return fileInfo, nil
|
||
|
|
}
|