modules.go

package modules

import (
	"context"
	"database/sql"
	"fmt"

	"github.com/bwmarrin/discordgo"
	"github.com/galexrt/fivenet/gen/go/proto/resources/users"
	"github.com/galexrt/fivenet/pkg/config"
	"github.com/galexrt/fivenet/pkg/mstlystcdata"
	"github.com/galexrt/fivenet/query/fivenet/table"
	"go.uber.org/zap"
)

var (
	tOauth2Accs = table.FivenetOauth2Accounts
	tAccs       = table.FivenetAccounts
	tUsers      = table.Users.AS("users")
)

var Modules = map[string]NewModuleFunc{}

type NewModuleFunc func(*BaseModule) (Module, error)

type Module interface {
	Run(settings *users.DiscordSyncSettings) ([]*discordgo.MessageEmbed, error)
}

func GetModule(name string, base *BaseModule) (Module, error) {
	fn, ok := Modules[name]
	if !ok {
		return nil, fmt.Errorf("no module found by name %s", name)
	}

	// "Wrap" logger with module name
	base.logger = base.logger.Named(name)

	return fn(base)
}

type BaseModule struct {
	ctx     context.Context
	logger  *zap.Logger
	db      *sql.DB
	discord *discordgo.Session
	guild   *discordgo.Guild
	job     string
	cfg     *config.DiscordBot

	enricher *mstlystcdata.Enricher
}

func NewBaseModule(ctx context.Context, logger *zap.Logger, db *sql.DB, discord *discordgo.Session, guild *discordgo.Guild, job string, cfg *config.DiscordBot, enricher *mstlystcdata.Enricher) *BaseModule {
	return &BaseModule{
		ctx:     ctx,
		logger:  logger,
		db:      db,
		discord: discord,
		guild:   guild,
		job:     job,
		cfg:     cfg,

		enricher: enricher,
	}
}