mirror-nourybot/cmd/bot/channel.go

package main

import (
	"fmt"

	"github.com/gempir/go-twitch-irc/v3"
	"github.com/lyx0/nourybot/internal/data"
	"github.com/lyx0/nourybot/pkg/commands/decapi"
	"github.com/lyx0/nourybot/pkg/common"
)

// AddChannel takes in a channel name, then calls GetIdByLogin for the
// channels ID and inserts both the name and id value into the database.
// If there is no error thrown the TwitchClient joins the channel afterwards.
func (app *Application) AddChannel(login string, message twitch.PrivateMessage) {
	userId, err := decapi.GetIdByLogin(login)
	if err != nil {
		app.Logger.Error(err)
		return
	}

	// Initialize a new channel struct holding the values that will be
	// passed into the app.Models.Channels.Insert() method.
	channel := &data.Channel{
		Login:    login,
		TwitchID: userId,
	}

	err = app.Models.Channels.Insert(channel)
	if err != nil {
		reply := fmt.Sprintf("Something went wrong FeelsBadMan %s", err)
		common.Send(message.Channel, reply, app.TwitchClient)
		return
	} else {
		app.TwitchClient.Join(login)
		reply := fmt.Sprintf("Added channel %s", login)
		common.Send(message.Channel, reply, app.TwitchClient)
		return
	}
}

// GetAllChannels() queries the database and lists all channels.
// Only used for debug/information purposes.
func (app *Application) GetAllChannels() {
	channel, err := app.Models.Channels.GetAll()
	if err != nil {
		app.Logger.Error(err)
		return
	}
	app.Logger.Infow("All channels:",
		"channel", channel)
}

// DeleteChannel queries the database for a channel name and if it exists
// deletes the channel and makes the bot depart said channel.
func (app *Application) DeleteChannel(login string, message twitch.PrivateMessage) {
	err := app.Models.Channels.Delete(login)
	if err != nil {
		common.Send(message.Channel, "Something went wrong FeelsBadMan", app.TwitchClient)
		app.Logger.Error(err)
		return
	}

	app.TwitchClient.Depart(login)

	reply := fmt.Sprintf("Deleted channel %s", login)
	common.Send(message.Channel, reply, app.TwitchClient)
}

// InitialJoin is called on startup and queries the database for a list of
// channels which the TwitchClient then joins.
func (app *Application) InitialJoin() {
	// GetJoinable returns a slice of channel names.
	channel, err := app.Models.Channels.GetJoinable()
	if err != nil {
		app.Logger.Error(err)
		return
	}

	// Iterate over the slice of channels and join each.
	for _, v := range channel {
		app.TwitchClient.Join(v)
		app.Logger.Infow("Joining channel",
			"channel", v)
	}
}