feat: add custom file name support for saved files and improve error messages

dao/db.go

@@ -3,12 +3,14 @@ package dao
 import (
 	"os"
 	"path/filepath"
+	"time"
 
 	"github.com/glebarez/sqlite"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
 	"github.com/krau/SaveAny-Bot/types"
 	"gorm.io/gorm"
+	glogger "gorm.io/gorm/logger"
 )
 
 var db *gorm.DB
@@ -19,7 +21,16 @@ func Init() {
 		os.Exit(1)
 	}
 	var err error
-	db, err = gorm.Open(sqlite.Open(config.Cfg.DB.Path), &gorm.Config{})
+	db, err = gorm.Open(sqlite.Open(config.Cfg.DB.Path), &gorm.Config{
+		Logger: glogger.New(logger.L, glogger.Config{
+			Colorful:                  true,
+			SlowThreshold:             time.Second * 5,
+			LogLevel:                  glogger.Error,
+			IgnoreRecordNotFoundError: true,
+			ParameterizedQueries:      true,
+		}),
+		PrepareStmt: true,
+	})
 	if err != nil {
 		logger.L.Fatal("Failed to open database: ", err)
 		os.Exit(1)

dao/file.go

@@ -2,8 +2,12 @@ package dao
 
 import "github.com/krau/SaveAny-Bot/types"
 
-func AddReceivedFile(receivedFile *types.ReceivedFile) error {
-	return db.Create(receivedFile).Error
+func SaveReceivedFile(receivedFile *types.ReceivedFile) error {
+	record, err := GetReceivedFileByChatAndMessageID(receivedFile.ChatID, receivedFile.MessageID)
+	if err == nil {
+		receivedFile.ID = record.ID
+	}
+	return db.Save(receivedFile).Error
 }
 
 func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*types.ReceivedFile, error) {
@@ -15,10 +19,6 @@ func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*types.Rece
 	return &receivedFile, nil
 }
 
-func UpdateReceivedFile(receivedFile *types.ReceivedFile) error {
-	return db.Save(receivedFile).Error
-}
-
 func DeleteReceivedFile(receivedFile *types.ReceivedFile) error {
 	return db.Delete(receivedFile).Error
 }