vegapokerbot/internal/db/repository/integration.go

package repository

import (
	"gitea.neur0tx.site/Neur0toxine/vegapokerbot/internal/db/model"
	"gitea.neur0tx.site/Neur0toxine/vegapokerbot/internal/db/util"
	"gorm.io/gorm"
)

type Integration struct {
	db *gorm.DB
}

func NewIntegration(db *gorm.DB) *Integration {
	return &Integration{db: db}
}

func (r *Integration) LoadForChatAndType(chatID uint64, typ model.IntegrationType) (*model.Integration, error) {
	var integration model.Integration
	if err := r.db.Model(model.Integration{ChatID: chatID, Type: typ}).First(&integration).Error; err != nil {
		return nil, util.HandleRecordNotFound(err)
	}
	return &integration, nil
}

func (r *Integration) Save(integration *model.Integration) error {
	if integration.ID == 0 {
		return r.db.Create(integration).Error
	}
	return r.db.Model(integration).Save(integration).Error
}

func (r *Integration) DeleteForChat(id uint64) error {
	return r.db.Where("chat_id = ?", id).Delete(&model.Integration{}).Error
}