32 lines
888 B
Go
32 lines
888 B
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 {
|
|
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
|
|
}
|