SaveAny-Bot/database/user.go

package database

import "context"

func CreateUser(ctx context.Context, chatID int64) error {
	if _, err := GetUserByChatID(ctx, chatID); err == nil {
		return nil
	}
	return db.Create(&User{ChatID: chatID}).Error
}

func GetAllUsers(ctx context.Context) ([]User, error) {
	var users []User
	err := db.Preload("Dirs").
		WithContext(ctx).
		Preload("Rules").
		Find(&users).Error
	return users, err
}

func GetUserByChatID(ctx context.Context, chatID int64) (*User, error) {
	var user User
	err := db.
		Preload("Dirs").
		WithContext(ctx).
		Preload("Rules").
		Where("chat_id = ?", chatID).First(&user).Error
	return &user, err
}

func UpdateUser(ctx context.Context, user *User) error {
	if _, err := GetUserByChatID(ctx, user.ChatID); err != nil {
		return err
	}
	return db.WithContext(ctx).Save(user).Error
}

func DeleteUser(ctx context.Context, user *User) error {
	return db.WithContext(ctx).Unscoped().Select("Dirs", "Rules").Delete(user).Error
}