GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-06-08 08:59:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: complete core features

Browse Source
This commit is contained in:
krau
2024-11-09 09:07:00 +08:00
parent fbdfc04ad8
commit 20e06fbf46
14 changed files with 380 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dao/db.go
View File

@@ -7,7 +7,7 @@ import (
"github.com/glebarez/sqlite"
"github.com/krau/SaveAny-Bot/config"
"github.com/krau/SaveAny-Bot/logger"
"github.com/krau/SaveAny-Bot/model"
"github.com/krau/SaveAny-Bot/types"
"gorm.io/gorm"
)
@@ -25,7 +25,7 @@ func Init() {
os.Exit(1)
}
logger.L.Debug("Database connected")
db.AutoMigrate(&model.ReceivedFile{}, &model.User{})
db.AutoMigrate(&types.ReceivedFile{}, &types.User{})
for _, admin := range config.Cfg.Telegram.Admins {
CreateUser(int64(admin))

12
dao/file.go
View File

@@ -1,13 +1,13 @@
package dao
import "github.com/krau/SaveAny-Bot/model"
import "github.com/krau/SaveAny-Bot/types"
func AddReceivedFile(receivedFile *model.ReceivedFile) error {
func AddReceivedFile(receivedFile *types.ReceivedFile) error {
return db.Create(receivedFile).Error
}
func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.ReceivedFile, error) {
var receivedFile model.ReceivedFile
func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*types.ReceivedFile, error) {
var receivedFile types.ReceivedFile
err := db.Where("chat_id = ? AND message_id = ?", chatID, messageID).First(&receivedFile).Error
if err != nil {
return nil, err
@@ -15,10 +15,10 @@ func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.Rece
return &receivedFile, nil
}
func UpdateReceivedFile(receivedFile *model.ReceivedFile) error {
func UpdateReceivedFile(receivedFile *types.ReceivedFile) error {
return db.Save(receivedFile).Error
}
func DeleteReceivedFile(receivedFile *model.ReceivedFile) error {
func DeleteReceivedFile(receivedFile *types.ReceivedFile) error {
return db.Delete(receivedFile).Error
}

10
dao/user.go
View File

@@ -1,22 +1,22 @@
package dao
import (
"github.com/krau/SaveAny-Bot/model"
"github.com/krau/SaveAny-Bot/types"
)
func CreateUser(userID int64) error {
if _, err := GetUserByUserID(userID); err == nil {
return nil
}
return db.Create(&model.User{UserID: userID}).Error
return db.Create(&types.User{UserID: userID}).Error
}
func GetUserByUserID(userID int64) (*model.User, error) {
var user model.User
func GetUserByUserID(userID int64) (*types.User, error) {
var user types.User
err := db.Where("user_id = ?", userID).First(&user).Error
return &user, err
}
func UpdateUser(user *model.User) error {
func UpdateUser(user *types.User) error {
return db.Save(user).Error
}