gowebdav/utils.go

114 lines
2.2 KiB
Go
Raw Normal View History

2014-10-23 10:39:55 +02:00
package gowebdav
import (
"bytes"
2014-10-23 11:29:26 +02:00
"encoding/xml"
2014-10-23 10:39:55 +02:00
"io"
2018-05-23 13:41:06 +02:00
"net/url"
2014-10-23 10:39:55 +02:00
"strconv"
"strings"
"time"
)
2021-09-17 13:23:09 +02:00
// PathEscape escapes all segments of a given path
2018-05-23 13:41:06 +02:00
func PathEscape(path string) string {
s := strings.Split(path, "/")
for i, e := range s {
s[i] = url.PathEscape(e)
}
return strings.Join(s, "/")
}
2017-10-05 16:22:10 +02:00
// FixSlash appends a trailing / to our string
func FixSlash(s string) string {
if !strings.HasSuffix(s, "/") {
s += "/"
}
return s
}
2017-10-05 16:22:10 +02:00
// FixSlashes appends and prepends a / if they are missing
2014-10-23 14:26:08 +02:00
func FixSlashes(s string) string {
if !strings.HasPrefix(s, "/") {
2014-10-23 14:26:08 +02:00
s = "/" + s
}
2014-10-23 14:26:08 +02:00
return FixSlash(s)
}
2017-10-05 16:22:10 +02:00
// Join joins two paths
2014-10-23 10:39:55 +02:00
func Join(path0 string, path1 string) string {
return strings.TrimSuffix(path0, "/") + "/" + strings.TrimPrefix(path1, "/")
}
2017-10-05 16:22:10 +02:00
// String pulls a string out of our io.Reader
2014-10-23 10:39:55 +02:00
func String(r io.Reader) string {
buf := new(bytes.Buffer)
2018-05-16 12:35:57 +02:00
// TODO - make String return an error as well
2017-10-05 16:22:10 +02:00
_, _ = buf.ReadFrom(r)
2014-10-23 10:39:55 +02:00
return buf.String()
}
func parseUint(s *string) uint {
if n, e := strconv.ParseUint(*s, 10, 32); e == nil {
return uint(n)
}
return 0
}
func parseInt64(s *string) int64 {
if n, e := strconv.ParseInt(*s, 10, 64); e == nil {
return n
}
return 0
}
2014-10-23 10:39:55 +02:00
func parseModified(s *string) time.Time {
if t, e := time.Parse(time.RFC1123, *s); e == nil {
return t
}
return time.Unix(0, 0)
}
2014-10-23 11:29:26 +02:00
2014-10-24 12:39:35 +02:00
func parseXML(data io.Reader, resp interface{}, parse func(resp interface{}) error) error {
2014-10-23 11:29:26 +02:00
decoder := xml.NewDecoder(data)
for t, _ := decoder.Token(); t != nil; t, _ = decoder.Token() {
switch se := t.(type) {
case xml.StartElement:
if se.Name.Local == "response" {
if e := decoder.DecodeElement(resp, &se); e == nil {
2014-10-24 12:39:35 +02:00
if err := parse(resp); err != nil {
return err
}
2014-10-23 11:29:26 +02:00
}
}
}
}
2014-10-24 12:39:35 +02:00
return nil
2014-10-23 11:29:26 +02:00
}
// limitedReadCloser wraps a io.ReadCloser and limits the number of bytes that can be read from it.
type limitedReadCloser struct {
rc io.ReadCloser
remaining int
}
func (l *limitedReadCloser) Read(buf []byte) (int, error) {
if l.remaining <= 0 {
return 0, io.EOF
}
if len(buf) > l.remaining {
buf = buf[0:l.remaining]
}
n, err := l.rc.Read(buf)
l.remaining -= n
return n, err
}
func (l *limitedReadCloser) Close() error {
return l.rc.Close()
}