139 lines
3.3 KiB
Go
139 lines
3.3 KiB
Go
package gowebdav
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func (c *Client) req(method, path string, body io.Reader, intercept func(*http.Request)) (req *http.Response, err error) {
|
|
// Tee the body, because if authorization fails we will need to read from it again.
|
|
var ba bytes.Buffer
|
|
bb := io.TeeReader(body, &ba)
|
|
|
|
r, err := http.NewRequest(method, PathEscape(Join(c.root, path)), &ba)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c.auth.Authorize(c, method, path)
|
|
|
|
for k, vals := range c.headers {
|
|
for _, v := range vals {
|
|
r.Header.Add(k, v)
|
|
}
|
|
}
|
|
|
|
if intercept != nil {
|
|
intercept(r)
|
|
}
|
|
|
|
rs, err := c.c.Do(r)
|
|
|
|
if rs.StatusCode == 401 && c.auth.Type() == "NoAuth" {
|
|
if strings.Index(rs.Header.Get("Www-Authenticate"), "Digest") > -1 {
|
|
c.auth = &DigestAuth{c.auth.User(), c.auth.Pass(), digestParts(rs)}
|
|
} else if strings.Index(rs.Header.Get("Www-Authenticate"), "Basic") > -1 {
|
|
c.auth = &BasicAuth{c.auth.User(), c.auth.Pass()}
|
|
} else {
|
|
return rs, newPathError("Authorize", c.root, rs.StatusCode)
|
|
}
|
|
return c.req(method, path, bb, intercept)
|
|
} else if rs.StatusCode == 401 {
|
|
return rs, newPathError("Authorize", c.root, rs.StatusCode)
|
|
}
|
|
|
|
return rs, err
|
|
}
|
|
|
|
func (c *Client) mkcol(path string) int {
|
|
rs, err := c.req("MKCOL", path, nil, nil)
|
|
defer rs.Body.Close()
|
|
if err != nil {
|
|
return 400
|
|
}
|
|
|
|
if rs.StatusCode == 201 || rs.StatusCode == 405 {
|
|
return 201
|
|
}
|
|
|
|
return rs.StatusCode
|
|
}
|
|
|
|
func (c *Client) options(path string) (*http.Response, error) {
|
|
return c.req("OPTIONS", path, nil, func(rq *http.Request) {
|
|
rq.Header.Add("Depth", "0")
|
|
})
|
|
}
|
|
|
|
func (c *Client) propfind(path string, self bool, body string, resp interface{}, parse func(resp interface{}) error) error {
|
|
rs, err := c.req("PROPFIND", path, strings.NewReader(body), func(rq *http.Request) {
|
|
if self {
|
|
rq.Header.Add("Depth", "0")
|
|
} else {
|
|
rq.Header.Add("Depth", "1")
|
|
}
|
|
rq.Header.Add("Content-Type", "text/xml;charset=UTF-8")
|
|
rq.Header.Add("Accept", "application/xml,text/xml")
|
|
rq.Header.Add("Accept-Charset", "utf-8")
|
|
// TODO add support for 'gzip,deflate;q=0.8,q=0.7'
|
|
rq.Header.Add("Accept-Encoding", "")
|
|
})
|
|
defer rs.Body.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if rs.StatusCode != 207 {
|
|
return fmt.Errorf("%s - %s %s", rs.Status, "PROPFIND", path)
|
|
}
|
|
|
|
return parseXML(rs.Body, resp, parse)
|
|
}
|
|
|
|
func (c *Client) doCopyMove(method string, oldpath string, newpath string, overwrite bool) (int, io.ReadCloser) {
|
|
rs, err := c.req(method, oldpath, nil, func(rq *http.Request) {
|
|
rq.Header.Add("Destination", Join(c.root, newpath))
|
|
if overwrite {
|
|
rq.Header.Add("Overwrite", "T")
|
|
} else {
|
|
rq.Header.Add("Overwrite", "F")
|
|
}
|
|
})
|
|
if err != nil {
|
|
return 400, nil
|
|
}
|
|
return rs.StatusCode, rs.Body
|
|
}
|
|
|
|
func (c *Client) copymove(method string, oldpath string, newpath string, overwrite bool) error {
|
|
s, data := c.doCopyMove(method, oldpath, newpath, overwrite)
|
|
defer data.Close()
|
|
|
|
switch s {
|
|
case 201, 204:
|
|
return nil
|
|
|
|
case 207:
|
|
// TODO handle multistat errors, worst case ...
|
|
log(fmt.Sprintf(" TODO handle %s - %s multistatus result %s", method, oldpath, String(data)))
|
|
|
|
case 409:
|
|
// TODO create dst path
|
|
}
|
|
|
|
return newPathError(method, oldpath, s)
|
|
}
|
|
|
|
func (c *Client) put(path string, stream io.Reader) int {
|
|
rs, err := c.req("PUT", path, stream, nil)
|
|
defer rs.Body.Close()
|
|
if err != nil {
|
|
return 400
|
|
}
|
|
|
|
return rs.StatusCode
|
|
}
|