refactor(webdav): use write stream to upload file

bot/handlers.go

@@ -23,7 +23,7 @@ import (
 )
 
 func RegisterHandlers(dispatcher dispatcher.Dispatcher) {
-	dispatcher.AddHandler(handlers.NewAnyUpdate(checkPermission))
+	dispatcher.AddHandler(handlers.NewMessage(filters.Message.All, checkPermission))
 	dispatcher.AddHandler(handlers.NewCommand("start", start))
 	dispatcher.AddHandler(handlers.NewCommand("help", help))
 	dispatcher.AddHandler(handlers.NewCommand("silent", silent))
@@ -325,6 +325,15 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
 }
 
 func AddToQueue(ctx *ext.Context, update *ext.Update) error {
+	if !slice.Contain(config.Cfg.Telegram.Admins, update.CallbackQuery.UserID) {
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "你没有权限",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
 	args := strings.Split(string(update.CallbackQuery.Data), " ")
 	messageID, _ := strconv.Atoi(args[1])
 	logger.L.Tracef("Got add to queue: chatID: %d, messageID: %d, storage: %s", update.EffectiveChat().GetID(), messageID, args[2])

storage/webdav/webdav.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"os"
+	"path"
 	"path/filepath"
 	"strings"
 	"time"
@@ -32,17 +33,19 @@ func (w *Webdav) Init() {
 }
 
 func (w *Webdav) Save(ctx context.Context, filePath, storagePath string) error {
-	storagePath = basePath + "/" + storagePath
+	storagePath = path.Join(basePath, storagePath)
 	if err := Client.MkdirAll(filepath.Dir(storagePath), os.ModePerm); err != nil {
 		logger.L.Errorf("Failed to create directory %s: %v", filepath.Dir(storagePath), err)
 		return errors.New("webdav: failed to create directory")
 	}
-	fileBytes, err := os.ReadFile(filePath)
+	file, err := os.Open(filePath)
 	if err != nil {
-		logger.L.Errorf("Failed to read file %s: %v", filePath, err)
+		logger.L.Errorf("Failed to open file %s: %v", filePath, err)
 		return err
 	}
-	if err := Client.Write(storagePath, fileBytes, os.ModePerm); err != nil {
+	defer file.Close()
+
+	if err := Client.WriteStream(storagePath, file, os.ModePerm); err != nil {
 		logger.L.Errorf("Failed to write file %s: %v", storagePath, err)
 		return errors.New("webdav: failed to write file")
 	}