@@ -66,3 +66,21 @@ func (s channelService) GuildChannels(g geeksbot.Guild) ([]geeksbot.Channel, err
|
||||
}
|
||||
return channels, nil
|
||||
}
|
||||
|
||||
func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
|
||||
channel, err := s.Channel(id)
|
||||
if err == sql.ErrNoRows {
|
||||
guild, err := GuildService.GetOrCreateGuild(guild_id)
|
||||
if err != nil {
|
||||
return geeksbot.Channel{}, err
|
||||
}
|
||||
channel, err = s.CreateChannel(geeksbot.Channel{
|
||||
ID: id,
|
||||
Guild: guild,
|
||||
Admin: false,
|
||||
Default: false,
|
||||
NewPatron: false,
|
||||
})
|
||||
}
|
||||
return channel, err
|
||||
}
|
||||
|
||||
@@ -99,3 +99,15 @@ func (s guildService) DeleteRole(r geeksbot.Role) error {
|
||||
_, err := s.db.Exec(queryString, r.ID)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s guildService) GetOrCreateGuild(id string) (geeksbot.Guild, error) {
|
||||
guild, err := s.Guild(id)
|
||||
if err == sql.ErrNoRows {
|
||||
guild = geeksbot.Guild{
|
||||
ID: id,
|
||||
Prefixes: []string{},
|
||||
}
|
||||
guild, err = s.CreateGuild(guild)
|
||||
}
|
||||
return guild, err
|
||||
}
|
||||
|
||||
@@ -36,3 +36,16 @@ func (s userService) UpdateUser(u geeksbot.User) (geeksbot.User, error) {
|
||||
_, err := s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin)
|
||||
return u, err
|
||||
}
|
||||
|
||||
func (s userService) GetOrCreateUser(id string) (geeksbot.User, error) {
|
||||
user, err := s.User(id)
|
||||
if err == sql.ErrNoRows {
|
||||
user, err = s.CreateUser(geeksbot.User{
|
||||
ID: id,
|
||||
IsActive: true,
|
||||
IsAdmin: false,
|
||||
IsStaff: false,
|
||||
})
|
||||
}
|
||||
return user, err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user