HugeFrog24/go-telegram-bot
1
0
Fork 0
mirror of https://github.com/HugeFrog24/go-telegram-bot.git synced 2026-03-02 08:24:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated owner management

Browse Source
This commit is contained in:
HugeFrog24
2024-10-22 14:54:39 +02:00
parent aca2008404
commit 972973633a
5 changed files with 145 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
bot.go
View File

@@ -63,6 +63,10 @@ func NewBot(db *gorm.DB, config BotConfig, clock Clock) (*Bot, error) {
}
if err := db.Create(&owner).Error; err != nil {
// If unique constraint is violated, another owner already exists
if strings.Contains(err.Error(), "unique index") {
return nil, fmt.Errorf("an owner already exists for this bot")
}
return nil, fmt.Errorf("failed to create owner user: %w", err)
}
} else if err != nil {
@@ -123,6 +127,10 @@ func (b *Bot) getOrCreateUser(userID int64, username string, isOwner bool) (User
}
if err := b.db.Create(&user).Error; err != nil {
// Handle unique constraint for owner
if isOwner && strings.Contains(err.Error(), "unique index") {
return User{}, fmt.Errorf("an owner already exists for this bot")
}
return User{}, err
}
} else {
@@ -139,11 +147,11 @@ func (b *Bot) getOrCreateUser(userID int64, username string, isOwner bool) (User
return User{}, err
}
// Promote to owner
user.Role, err = b.getRoleByName("owner")
role, err := b.getRoleByName("owner")
if err != nil {
return User{}, err
}
user.RoleID = user.Role.ID
user.RoleID = role.ID
user.IsOwner = true
if err := b.db.Save(&user).Error; err != nil {
return User{}, err