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
|
package b2
import (
"io"
"net/http"
"strconv"
"strings"
"time"
)
// DownloadFileByID gets file contents by file ID. The ReadCloser must be
// closed by the caller once done reading.
//
// Note: the (*FileInfo).CustomMetadata values returned by this function are
// all represented as strings, because they are delivered by HTTP headers.
func (c *Client) DownloadFileByID(id string) (io.ReadCloser, *FileInfo, error) {
downloadURL := c.loginInfo.Load().(*LoginInfo).DownloadURL
res, err := c.hc.Get(downloadURL + apiPath + "b2_download_file_by_id?fileId=" + id)
if e, ok := UnwrapError(err); ok && e.Status == http.StatusUnauthorized {
if err = c.login(res); err == nil {
res, err = c.hc.Get(downloadURL + apiPath + "b2_download_file_by_id?fileId=" + id)
}
}
if err != nil {
debugf("download %s: %s", id, err)
return nil, nil, err
}
debugf("download %s (%s)", id, res.Header.Get("X-Bz-Content-Sha1"))
fi, err := parseFileInfoHeaders(res.Header)
return res.Body, fi, err
}
// DownloadFileByName gets file contents by file and bucket name.
// The ReadCloser must be closed by the caller once done reading.
//
// Note: the (*FileInfo).CustomMetadata values returned by this function are
// all represented as strings, because they are delivered by HTTP headers.
func (c *Client) DownloadFileByName(bucket, file string) (io.ReadCloser, *FileInfo, error) {
downloadURL := c.loginInfo.Load().(*LoginInfo).DownloadURL
res, err := c.hc.Get(downloadURL + "/file/" + bucket + "/" + file)
if e, ok := UnwrapError(err); ok && e.Status == http.StatusUnauthorized {
if err = c.login(res); err == nil {
res, err = c.hc.Get(downloadURL + "/file/" + bucket + "/" + file)
}
}
if err != nil {
debugf("download %s: %s", file, err)
return nil, nil, err
}
debugf("download %s (%s)", file, res.Header.Get("X-Bz-Content-Sha1"))
fi, err := parseFileInfoHeaders(res.Header)
return res.Body, fi, err
}
func parseFileInfoHeaders(h http.Header) (*FileInfo, error) {
fi := &FileInfo{
ID: h.Get("X-Bz-File-Id"),
Name: h.Get("X-Bz-File-Name"),
ContentType: h.Get("Content-Type"),
ContentSHA1: h.Get("X-Bz-Content-Sha1"),
Action: "upload",
}
timestamp, err := strconv.ParseInt(h.Get("X-Bz-Upload-Timestamp"), 10, 64)
if err != nil {
return nil, err
}
fi.UploadTimestamp = time.Unix(timestamp/1e3, timestamp%1e3*1e6)
fi.ContentLength, err = strconv.Atoi(h.Get("Content-Length"))
if err != nil {
return nil, err
}
fi.CustomMetadata = make(map[string]interface{})
for name := range h {
if !strings.HasPrefix(name, "X-Bz-Info-") {
continue
}
fi.CustomMetadata[name[len("X-Bz-Info-"):]] = h.Get(name)
}
return fi, nil
}
|
