rename package

pkg/thumbnail/imgsource/webdav.go

@@ -0,0 +1,66 @@
+package imgsource
+
+import (
+	"context"
+	"fmt"
+	"image"
+	"net/http"
+	"net/url"
+	"path"
+
+	"github.com/owncloud/ocis-thumbnails/pkg/config"
+)
+
+// NewWebDavSource creates a new webdav instance.
+func NewWebDavSource(cfg config.WebDavSource) WebDav {
+	return WebDav{
+		baseURL: cfg.BaseURL,
+	}
+}
+
+// WebDav implements the Source interface for webdav services
+type WebDav struct {
+	baseURL string
+}
+
+const (
+	// WebDavAuth is the parameter name for the autorization token
+	WebDavAuth = "Authorization"
+)
+
+// Get downloads the file from a webdav service
+func (s WebDav) Get(ctx context.Context, file string) (image.Image, error) {
+	u, _ := url.Parse(s.baseURL)
+	u.Path = path.Join(u.Path, file)
+	req, err := http.NewRequest(http.MethodGet, u.String(), nil)
+	if err != nil {
+		return nil, fmt.Errorf("could not get the image \"%s\" error: %s", file, err.Error())
+	}
+
+	auth, ok := authorization(ctx)
+	if !ok {
+		return nil, fmt.Errorf("could not get image \"%s\" error: authorization is missing", file)
+	}
+	req.Header.Add("Authorization", auth)
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("could not get the image \"%s\" error: %s", file, err.Error())
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("could not get the image \"%s\". Request returned with statuscode %d ", file, resp.StatusCode)
+	}
+
+	img, _, err := image.Decode(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("could not decode the image \"%s\". error: %s", file, err.Error())
+	}
+	return img, nil
+}
+
+func authorization(ctx context.Context) (string, bool) {
+	auth, ok := ctx.Value(WebDavAuth).(string)
+	return auth, ok
+}