package src //from yadisk import ( "encoding/json" "io" "net/url" "strconv" ) // UploadResponse struct is returned by the API for upload request. type UploadResponse struct { HRef string `json:"href"` Method string `json:"method"` Templated bool `json:"templated"` } // Upload will put specified data to Yandex.Disk. func (c *Client) Upload(data io.Reader, remotePath string, overwrite bool, contentType string) error { ur, err := c.UploadRequest(remotePath, overwrite) if err != nil { return err } if err := c.PerformUpload(ur.HRef, data, contentType); err != nil { return err } return nil } // UploadRequest will make an upload request and return a URL to upload data to. func (c *Client) UploadRequest(remotePath string, overwrite bool) (ur *UploadResponse, err error) { values := url.Values{} values.Add("path", remotePath) values.Add("overwrite", strconv.FormatBool(overwrite)) req, err := c.scopedRequest("GET", "/v1/disk/resources/upload?"+values.Encode(), nil) if err != nil { return nil, err } resp, err := c.HTTPClient.Do(req) if err != nil { return nil, err } if err := CheckAPIError(resp); err != nil { return nil, err } defer CheckClose(resp.Body, &err) ur, err = ParseUploadResponse(resp.Body) if err != nil { return nil, err } return ur, nil } // ParseUploadResponse tries to read and parse UploadResponse struct. func ParseUploadResponse(data io.Reader) (*UploadResponse, error) { dec := json.NewDecoder(data) var ur UploadResponse if err := dec.Decode(&ur); err == io.EOF { // ok } else if err != nil { return nil, err } // TODO: check if there is any trash data after JSON and crash if there is. return &ur, nil }