mirror of
https://github.com/lyx0/nourybot.git
synced 2024-11-13 19:49:55 +01:00
39 lines
1 KiB
Go
39 lines
1 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/lyx0/nourybot/cmd/bot"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// PartChannel takes in a target channel and querys the database for
|
|
// its name. If the channel is found it deletes the channel and returns a
|
|
// success message. If the channel couldn't be found it will return an error.
|
|
func PartChannel(target, channelName string, nb *bot.Bot) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
collection := nb.MongoClient.Database("nourybot").Collection("channels")
|
|
|
|
// Channel
|
|
// {{"name": string}, {"connect": bool}}
|
|
chnl := Channel{channelName, true}
|
|
|
|
res, insertErr := collection.DeleteOne(ctx, chnl)
|
|
|
|
// res.DeletedCount is 0 if trying to delete something that wasn't there.
|
|
if insertErr != nil || res.DeletedCount != 1 {
|
|
nb.Send(target, fmt.Sprintf("Error trying to part %s", channelName))
|
|
log.Error(insertErr)
|
|
return
|
|
}
|
|
|
|
nb.TwitchClient.Depart(channelName)
|
|
nb.Send(target, fmt.Sprintf("Parted %s", channelName))
|
|
|
|
// log.Info(res)
|
|
}
|