109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package profile
|
|
|
|
import (
|
|
"github.com/omnex/control-plane/api/internal/auth"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// GormStore : Store adossé à PostgreSQL via GORM.
|
|
type GormStore struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewGormStore(db *gorm.DB) *GormStore {
|
|
return &GormStore{db: db}
|
|
}
|
|
|
|
type Store interface {
|
|
UpdateUsername(id, newUsername string) (auth.User, error)
|
|
UpdatePassword(id, passwordHahs string) (auth.User, error)
|
|
GetTelegram(id string) (auth.User, error)
|
|
SetTelegram(id, telegram string) (auth.User, error)
|
|
GetAlertSettings(id string) (auth.User, error)
|
|
SetAlertSettings(id string, discordWebhookURL, telegramBotToken, telegramChatID string) (auth.User, error)
|
|
}
|
|
|
|
func (s *GormStore) UpdateUsername(id, newUsername string) (auth.User, error) {
|
|
result := s.db.Model(&auth.User{}).
|
|
Where("id = ?", id).
|
|
Update("username", newUsername)
|
|
|
|
if result.Error != nil {
|
|
return auth.User{}, result.Error
|
|
}
|
|
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (s *GormStore) UpdatePassword(id, passwordHash string) (auth.User, error) {
|
|
result := s.db.Model(&auth.User{}).
|
|
Where("id = ?", id).
|
|
Update("password_hash", passwordHash)
|
|
if result.Error != nil {
|
|
return auth.User{}, result.Error
|
|
}
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (s *GormStore) GetTelegram(id string) (auth.User, error) {
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (s *GormStore) SetTelegram(id, telegram string) (auth.User, error) {
|
|
result := s.db.Model(&auth.User{}).Where("id = ?", id).Update("telegram", telegram)
|
|
if result.Error != nil {
|
|
return auth.User{}, result.Error
|
|
}
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (s *GormStore) GetAlertSettings(id string) (auth.User, error) {
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
// SetAlertSettings enregistre les réglages d'alerte de cet admin. Une chaîne
|
|
// vide efface le champ correspondant (désactive ce canal pour cet admin).
|
|
func (s *GormStore) SetAlertSettings(id string, discordWebhookURL, telegramBotToken, telegramChatID string) (auth.User, error) {
|
|
updates := map[string]interface{}{
|
|
"alert_discord_webhook_url": nullIfEmpty(discordWebhookURL),
|
|
"alert_telegram_bot_token": nullIfEmpty(telegramBotToken),
|
|
"alert_telegram_chat_id": nullIfEmpty(telegramChatID),
|
|
}
|
|
if err := s.db.Model(&auth.User{}).Where("id = ?", id).Updates(updates).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
var user auth.User
|
|
if err := s.db.First(&user, "id = ?", id).Error; err != nil {
|
|
return auth.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func nullIfEmpty(s string) *string {
|
|
if s == "" {
|
|
return nil
|
|
}
|
|
return &s
|
|
}
|