Repo Reorg

cmd/goff/main.go

@@ -0,0 +1,147 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/goff"
+	"github.com/dustinpianalto/goff/internal/events"
+	"github.com/dustinpianalto/goff/internal/exts"
+	"github.com/dustinpianalto/goff/internal/postgres"
+	"github.com/dustinpianalto/goff/pkg/email"
+
+	//"github.com/MikeModder/anpan"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+var (
+	Token string
+)
+
+//func init() {
+//	flag.StringVar(&Token, "t", "", "Bot Token")
+//	flag.Parse()
+//}
+
+func main() {
+	Token = os.Getenv("DISCORDGO_TOKEN")
+	dg, err := discordgo.New("Bot " + Token)
+	if err != nil {
+		fmt.Println("There was an error when creating the Discord Session, ", err)
+		return
+	}
+	dg.State.MaxMessageCount = 100
+	dg.StateEnabled = true
+
+	dg.Identify = discordgo.Identify{
+		Intents: discordgo.MakeIntent(discordgo.IntentsAll),
+	}
+
+	postgres.ConnectDatabase(os.Getenv("DATABASE_URL"))
+	postgres.InitializeDatabase()
+	//utils.LoadTestData()
+
+	us := &postgres.UserService{DB: postgres.DB}
+	gs := &postgres.GuildService{DB: postgres.DB}
+
+	//prefixes := []string{
+	//	"Go.",
+	//}
+	owners := []string{
+		"351794468870946827",
+	}
+
+	manager := disgoman.CommandManager{
+		Prefixes:         getPrefixes,
+		Owners:           owners,
+		StatusManager:    disgoman.GetDefaultStatusManager(),
+		ErrorChannel:     make(chan disgoman.CommandError, 10),
+		Commands:         make(map[string]*disgoman.Command),
+		IgnoreBots:       true,
+		CheckPermissions: false,
+	}
+
+	goffManager := goff.CommandManager{
+		UserService:    us,
+		GuildService:   gs,
+		CommandManager: manager,
+	}
+
+	// Add Command Handlers
+	exts.AddCommandHandlers(&goffManager)
+
+	//if _, ok := handler.Commands["help"]; !ok {
+	//	handler.AddDefaultHelpCommand()
+	//}
+
+	dg.AddHandler(goffManager.OnMessage)
+	dg.AddHandler(goffManager.StatusManager.OnReady)
+	dg.AddHandler(events.OnMessageUpdate)
+	dg.AddHandler(events.OnMessageDelete)
+	dg.AddHandler(events.OnGuildMemberAddLogging)
+	dg.AddHandler(events.OnGuildMemberRemoveLogging)
+
+	err = dg.Open()
+	if err != nil {
+		fmt.Println("There was an error opening the connection, ", err)
+		return
+	}
+
+	// Start the Error handler in a goroutine
+	go ErrorHandler(goffManager.ErrorChannel)
+
+	// Start the Logging handler in a goroutine
+	go events.LoggingHandler(events.LoggingChannel)
+
+	// Start the task handler in a goroutine
+	go events.ProcessTasks(dg, 1)
+
+	go email.RecieveEmail(dg)
+
+	fmt.Println("The Bot is now running.")
+	sc := make(chan os.Signal, 1)
+	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
+	<-sc
+
+	fmt.Println("Shutting Down...")
+	err = dg.Close()
+	if err != nil {
+		fmt.Println(err)
+	}
+}
+
+func getPrefixes(guildID string) []string {
+	queryString := "Select prefix from prefixes p, x_guilds_prefixes xgp where xgp.guild_id = $1 and xgp.prefix_id = p.id"
+	rows, err := postgres.DB.Query(queryString, guildID)
+	if err != nil {
+		log.Println(err)
+		return []string{"Go.", "go."}
+	}
+	var prefixes []string
+	for rows.Next() {
+		var prefix string
+		err = rows.Scan(&prefix)
+		if err != nil {
+			log.Println(err)
+			return []string{"Go.", "go."}
+		}
+		prefixes = append(prefixes, prefix)
+	}
+	return prefixes
+}
+
+func ErrorHandler(ErrorChan chan disgoman.CommandError) {
+	for ce := range ErrorChan {
+		msg := ce.Message
+		if msg == "" {
+			msg = ce.Error.Error()
+		}
+		_, _ = ce.Context.Send(msg)
+		fmt.Println(ce.Error)
+	}
+}