feat: implement multi-user credits redeem and storage isolation

user_webdav.go

@@ -0,0 +1,106 @@
+package main
+
+import (
+	"path"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/studio-b12/gowebdav"
+)
+
+func cleanWebDAVRoot(root string) string {
+	root = strings.TrimSpace(strings.ReplaceAll(root, "\\", "/"))
+	root = path.Clean("/" + root)
+	if root == "/" {
+		return ""
+	}
+	return strings.Trim(root, "/")
+}
+
+func (a *App) getUserWebDAV(c *gin.Context) {
+	user := c.MustGet("user").(*User)
+	var cfg WebDAVConfig
+	if err := a.db.Where("user_id = ?", user.ID).First(&cfg).Error; err != nil {
+		c.JSON(200, gin.H{"enabled": false, "url": "", "username": "", "rootPath": "", "hasPassword": false})
+		return
+	}
+	c.JSON(200, gin.H{"enabled": cfg.Enabled, "url": cfg.URL, "username": cfg.Username, "rootPath": cfg.RootPath, "hasPassword": cfg.Password != ""})
+}
+
+func (a *App) putUserWebDAV(c *gin.Context) {
+	user := c.MustGet("user").(*User)
+	var in struct {
+		Enabled  bool   `json:"enabled"`
+		URL      string `json:"url"`
+		Username string `json:"username"`
+		Password string `json:"password"`
+		RootPath string `json:"rootPath"`
+	}
+	if c.BindJSON(&in) != nil {
+		c.Status(400)
+		return
+	}
+	root := cleanWebDAVRoot(in.RootPath)
+	var cfg WebDAVConfig
+	err := a.db.Where("user_id = ?", user.ID).First(&cfg).Error
+	if err != nil {
+		cfg = WebDAVConfig{UserID: user.ID, Enabled: in.Enabled, URL: strings.TrimRight(strings.TrimSpace(in.URL), "/"), Username: strings.TrimSpace(in.Username), Password: in.Password, RootPath: root}
+	} else {
+		cfg.Enabled = in.Enabled
+		cfg.URL = strings.TrimRight(strings.TrimSpace(in.URL), "/")
+		cfg.Username = strings.TrimSpace(in.Username)
+		cfg.RootPath = root
+		if in.Password != "" {
+			cfg.Password = in.Password
+		}
+	}
+	if err := a.db.Save(&cfg).Error; err != nil {
+		c.JSON(500, gin.H{"error": err.Error()})
+		return
+	}
+	c.JSON(200, gin.H{"ok": true})
+}
+
+func (a *App) testUserWebDAV(c *gin.Context) {
+	user := c.MustGet("user").(*User)
+	var in struct{ URL, Username, Password, RootPath string }
+	if c.BindJSON(&in) != nil {
+		c.Status(400)
+		return
+	}
+	if in.URL == "" {
+		var cfg WebDAVConfig
+		if a.db.Where("user_id = ?", user.ID).First(&cfg).Error == nil {
+			in.URL, in.Username, in.Password, in.RootPath = cfg.URL, cfg.Username, cfg.Password, cfg.RootPath
+		}
+	}
+	client := gowebdav.NewClient(in.URL, in.Username, in.Password)
+	client.SetTimeout(20 * time.Second)
+	if err := client.Connect(); err != nil {
+		c.JSON(502, gin.H{"ok": false, "error": err.Error()})
+		return
+	}
+	c.JSON(200, gin.H{"ok": true, "rootPath": cleanWebDAVRoot(in.RootPath)})
+}
+
+func (a *App) storageForTask(t *Task) (RemoteStorage, string, bool) {
+	var user User
+	if a.db.First(&user, t.UserID).Error != nil {
+		return nil, "", false
+	}
+	if user.Role == RoleAdmin {
+		if a.cfg.WebDAVURL == "" {
+			return nil, "", false
+		}
+		return a.storage, a.cfg.WebDAVPath, a.storage != nil
+	}
+	if !a.hasPermission(user.ID, PermWebDAV) {
+		return nil, "", false
+	}
+	var cfg WebDAVConfig
+	if a.db.Where("user_id = ? AND enabled = ?", user.ID, true).First(&cfg).Error != nil || cfg.URL == "" {
+		return nil, "", false
+	}
+	return NewWebDAVStorage(Config{WebDAVURL: cfg.URL, WebDAVUser: cfg.Username, WebDAVPass: cfg.Password}), cfg.RootPath, true
+}