基本功能实现

internal/storage/webdav/webdav.go

@@ -0,0 +1,74 @@
+package webdav
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"path"
+	"strings"
+
+	"BingDailyImage/internal/storage"
+
+	"github.com/studio-b12/gowebdav"
+)
+
+type WebDAVStorage struct {
+	client          *gowebdav.Client
+	publicURLPrefix string
+}
+
+func NewWebDAVStorage(url, username, password, publicURLPrefix string) (*WebDAVStorage, error) {
+	client := gowebdav.NewClient(url, username, password)
+	if err := client.Connect(); err != nil {
+		// 有些 webdav 不支持 Connect，我们可以忽略错误或者做简单的探测
+	}
+	return &WebDAVStorage{
+		client:          client,
+		publicURLPrefix: publicURLPrefix,
+	}, nil
+}
+
+func (w *WebDAVStorage) Put(ctx context.Context, key string, r io.Reader, contentType string) (storage.StoredObject, error) {
+	// 确保目录存在
+	dir := path.Dir(key)
+	if dir != "." && dir != "/" {
+		if err := w.client.MkdirAll(dir, 0755); err != nil {
+			return storage.StoredObject{}, err
+		}
+	}
+
+	err := w.client.WriteStream(key, r, 0644)
+	if err != nil {
+		return storage.StoredObject{}, err
+	}
+
+	publicURL := ""
+	if w.publicURLPrefix != "" {
+		publicURL = fmt.Sprintf("%s/%s", strings.TrimSuffix(w.publicURLPrefix, "/"), key)
+	}
+
+	return storage.StoredObject{
+		Key:         key,
+		ContentType: contentType,
+		PublicURL:   publicURL,
+	}, nil
+}
+
+func (w *WebDAVStorage) Get(ctx context.Context, key string) (io.ReadCloser, string, error) {
+	reader, err := w.client.ReadStream(key)
+	if err != nil {
+		return nil, "", err
+	}
+	return reader, "", nil
+}
+
+func (w *WebDAVStorage) Delete(ctx context.Context, key string) error {
+	return w.client.Remove(key)
+}
+
+func (w *WebDAVStorage) PublicURL(key string) (string, bool) {
+	if w.publicURLPrefix != "" {
+		return fmt.Sprintf("%s/%s", strings.TrimSuffix(w.publicURLPrefix, "/"), key), true
+	}
+	return "", false
+}