Repo Reorg

2020-09-18 18:39:51 -08:00
parent a4b3fc5a21
commit fc0f1b0ed3
33 changed files with 489 additions and 352 deletions
@@ -0,0 +1,36 @@
+package events
+
+import (
+	"fmt"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustinpianalto/goff/internal/postgres"
+)
+
+var LoggingChannel = make(chan *LogEvent, 10)
+
+type LogEvent struct {
+	// Embed with log message
+	Embed discordgo.MessageEmbed
+	// Guild to log event in
+	GuildID string
+	// Discordgo Session. Needed for sending messages
+	Session *discordgo.Session
+}
+
+func LoggingHandler(lc chan *LogEvent) {
+	for event := range lc {
+		var channelID string
+		row := postgres.DB.QueryRow("SELECT logging_channel FROM guilds where id=$1", event.GuildID)
+		err := row.Scan(&channelID)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		if channelID == "" {
+			return
+		}
+
+		_, _ = event.Session.ChannelMessageSendEmbed(channelID, &event.Embed)
+	}
+}
@@ -0,0 +1,167 @@
+package events
+
+import (
+	"fmt"
+	"log"
+	"strconv"
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustinpianalto/goff/internal/discord_utils"
+	"github.com/dustinpianalto/goff/internal/postgres"
+)
+
+func OnGuildMemberAddLogging(s *discordgo.Session, member *discordgo.GuildMemberAdd) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in OnGuildMemberAddLogging", r)
+		}
+	}()
+	var channelID string
+	row := postgres.DB.QueryRow("SELECT logging_channel FROM guilds where id=$1", member.GuildID)
+	err := row.Scan(&channelID)
+	if err != nil || channelID == "" {
+		return
+	}
+	guild, err := s.State.Guild(member.GuildID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+
+	var title string
+	if member.User.Bot {
+		title = "Bot Joined"
+	} else {
+		title = "Member Joined"
+	}
+
+	thumb := &discordgo.MessageEmbedThumbnail{
+		URL: member.User.AvatarURL(""),
+	}
+
+	int64ID, _ := strconv.ParseInt(member.User.ID, 10, 64)
+	snow := discord_utils.ParseSnowflake(int64ID)
+
+	field := &discordgo.MessageEmbedField{
+		Name:   "User was created:",
+		Value:  discord_utils.ParseDateString(snow.CreationTime),
+		Inline: false,
+	}
+
+	joinTime, _ := member.JoinedAt.Parse()
+
+	embed := &discordgo.MessageEmbed{
+		Title:       title,
+		Description: fmt.Sprintf("%v (%v) Has Joined the Server", member.User.Mention(), member.User.ID),
+		Color:       0x0cc56a,
+		Thumbnail:   thumb,
+		Footer: &discordgo.MessageEmbedFooter{
+			Text:    fmt.Sprintf("Current Member Count: %v", guild.MemberCount),
+			IconURL: guild.IconURL(),
+		},
+		Timestamp: joinTime.Format(time.RFC3339),
+		Fields:    []*discordgo.MessageEmbedField{field},
+	}
+	s.ChannelMessageSendEmbed(channelID, embed)
+}
+
+func OnGuildMemberRemoveLogging(s *discordgo.Session, member *discordgo.GuildMemberRemove) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in OnGuildMemberAddLogging", r)
+		}
+	}()
+	timeNow := time.Now()
+	var channelID string
+	row := postgres.DB.QueryRow("SELECT logging_channel FROM guilds where id=$1", member.GuildID)
+	err := row.Scan(&channelID)
+	if err != nil || channelID == "" {
+		return
+	}
+	guild, err := s.State.Guild(member.GuildID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+
+	var title string
+	if member.User.Bot {
+		title = "Bot Left"
+	} else {
+		title = "Member Left"
+	}
+
+	thumb := &discordgo.MessageEmbedThumbnail{
+		URL: member.User.AvatarURL(""),
+	}
+
+	desc := ""
+	al, err := s.GuildAuditLog(member.GuildID, "", "", 20, 1)
+	if err != nil {
+		log.Println(err)
+	} else {
+		for _, log := range al.AuditLogEntries {
+			if log.TargetID == member.User.ID {
+				int64ID, _ := strconv.ParseInt(log.ID, 10, 64)
+				logSnow := discord_utils.ParseSnowflake(int64ID)
+				if timeNow.Sub(logSnow.CreationTime).Seconds() <= 10 {
+					user, err := s.User(log.UserID)
+					if err == nil {
+						desc = fmt.Sprintf("%v (%v) was Kicked by: %v\nReason: %v", member.User.String(), member.User.ID, user.String(), log.Reason)
+					} else {
+						desc = fmt.Sprintf("%v (%v) was Kicked by: %v\nReason: %v", member.User.String(), member.User.ID, log.UserID, log.Reason)
+					}
+					break
+				}
+			}
+		}
+	}
+	if desc == "" {
+		desc = fmt.Sprintf("%v (%v) Has Left the Server", member.User.String(), member.User.ID)
+	}
+
+	embed := &discordgo.MessageEmbed{
+		Title:       title,
+		Description: desc,
+		Color:       0xff9431,
+		Thumbnail:   thumb,
+		Footer: &discordgo.MessageEmbedFooter{
+			Text:    fmt.Sprintf("Current Member Count: %v", guild.MemberCount),
+			IconURL: guild.IconURL(),
+		},
+		Timestamp: timeNow.Format(time.RFC3339),
+	}
+	s.ChannelMessageSendEmbed(channelID, embed)
+}
+
+func AddMemberToDatabase(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in AddMemberToDatabase", r)
+		}
+	}()
+	queryString := `INSERT INTO users (id, banned, logging, is_active, is_staff, is_admin) 
+						VALUES ($1, false, false, true, false, false)`
+	_, err := postgres.DB.Exec(queryString, m.User.ID)
+	if err != nil {
+		log.Println(fmt.Errorf("error inserting %s into database: %w", m.User.ID, err))
+		return
+	}
+	log.Printf("New User: %s\n", m.User.ID)
+}
+
+func MarkMemberInactive(s *discordgo.Session, m *discordgo.GuildMemberRemove) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in MarkMemberInactive")
+		}
+	}()
+	queryString := `UPDATE users SET is_active = false WHERE id = $1`
+	_, err := postgres.DB.Exec(queryString, m.User.ID)
+	if err != nil {
+		log.Println(fmt.Errorf("error marking %s as inactive: %w", m.User.ID, err))
+		return
+	}
+	log.Println("User left: %s")
+}
@@ -0,0 +1,97 @@
+package events
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustinpianalto/goff/internal/postgres"
+)
+
+func OnMessageUpdate(session *discordgo.Session, m *discordgo.MessageUpdate) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in OnMessageUpdate", r)
+		}
+	}()
+	msg := m.BeforeUpdate
+	if msg.Author.Bot {
+		return
+	}
+	var channelID string
+	row := postgres.DB.QueryRow("SELECT logging_channel FROM guilds where id=$1", msg.GuildID)
+	err := row.Scan(&channelID)
+	if err != nil || channelID == "" {
+		return
+	}
+	channel, err := session.State.Channel(msg.ChannelID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	embed := &discordgo.MessageEmbed{
+		Title:       fmt.Sprintf("Message Edited: %v", msg.ID),
+		Description: fmt.Sprintf("**Before:** %v\n**After:** %v\nIn Channel: %v", msg.Content, m.Content, channel.Mention()),
+		Color:       session.State.UserColor(msg.Author.ID, channelID),
+		Footer: &discordgo.MessageEmbedFooter{
+			Text:    fmt.Sprintf("Author: %v", msg.Author.String()),
+			IconURL: msg.Author.AvatarURL(""),
+		},
+	}
+	session.ChannelMessageSendEmbed(channelID, embed)
+}
+
+func OnMessageDelete(session *discordgo.Session, m *discordgo.MessageDelete) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Println("Recovered from panic in OnMessageDelete", r)
+		}
+	}()
+	msg := m.BeforeDelete
+	if msg == nil {
+		log.Printf("Message Deleted but the original message was not in my cache so we are ignoring it.\nMessage ID: %v\nGuild ID: %v\nChannel ID: %v\n", m.ID, m.GuildID, m.ChannelID)
+		return
+	}
+	if msg.Author.Bot {
+		return
+	}
+	var channelID string
+	row := postgres.DB.QueryRow("SELECT logging_channel FROM guilds where id=$1", msg.GuildID)
+	err := row.Scan(&channelID)
+	if err != nil || channelID == "" {
+		return
+	}
+	channel, err := session.State.Channel(msg.ChannelID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	desc := ""
+	al, err := session.GuildAuditLog(msg.GuildID, "", "", 72, 1)
+	if err != nil {
+		log.Println(err)
+	} else {
+		for _, log := range al.AuditLogEntries {
+			if log.TargetID == msg.Author.ID && log.Options.ChannelID == msg.ChannelID {
+				user, err := session.User(log.UserID)
+				if err == nil {
+					desc = fmt.Sprintf("**Content:** %v\nIn Channel: %v\nDeleted By: %v", msg.Content, channel.Mention(), user.Mention())
+				}
+				break
+			}
+		}
+	}
+	if desc == "" {
+		desc = fmt.Sprintf("**Content:** %v\nIn Channel: %v", msg.Content, channel.Mention())
+	}
+	embed := &discordgo.MessageEmbed{
+		Title:       fmt.Sprintf("Message Deleted: %v", msg.ID),
+		Description: desc,
+		Color:       session.State.UserColor(msg.Author.ID, channelID),
+		Footer: &discordgo.MessageEmbedFooter{
+			Text:    fmt.Sprintf("Author: %v", msg.Author.String()),
+			IconURL: msg.Author.AvatarURL(""),
+		},
+	}
+	session.ChannelMessageSendEmbed(channelID, embed)
+}
@@ -0,0 +1,134 @@
+package events
+
+import (
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustinpianalto/goff/internal/postgres"
+)
+
+type Task struct {
+	ID           int64
+	Type         string
+	Content      string
+	GuildID      string
+	ChannelID    string
+	UserID       string
+	CreationTime time.Time
+	TriggerTime  time.Time
+}
+
+func processTask(task *Task, s *discordgo.Session) {
+	query := "SELECT completed, processing from tasks where id = $1"
+	res, err := postgres.DB.Query(query, task.ID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	var completed bool
+	var processing bool
+	res.Next()
+	err = res.Scan(&completed, &processing)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	if completed || processing {
+		return
+	}
+	closeQuery := "Update tasks set completed = true where id = $1"
+	processQuery := "UPDATE tasks SET processing = true WHERE id = $1"
+	defer postgres.DB.Exec(closeQuery, task.ID)
+	_, err = postgres.DB.Exec(processQuery, task.ID)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	log.Println(fmt.Sprintf("Processing task %v", task.ID))
+	guild, err := s.Guild(task.GuildID)
+	if err != nil {
+		log.Print(fmt.Sprintf("Can't find guild with ID %v. Canceling task %v.", task.GuildID, task.ID))
+		return
+	}
+	channel, err := s.Channel(task.ChannelID)
+	if err != nil {
+		log.Print(fmt.Sprintf("Can't find channel with ID %v. Canceling task %v.", task.ChannelID, task.ID))
+		return
+	}
+	if channel.GuildID != guild.ID {
+		log.Print(fmt.Sprintf("The channel %v is not in guild %v. Canceling task %v.", channel.Name, guild.Name, task.ID))
+		return
+	}
+	member, err := s.GuildMember(guild.ID, task.UserID)
+	if err != nil {
+		log.Print(fmt.Sprintf("Can't find user with ID %v in guild %v. Canceling task %v.", task.UserID, guild.Name, task.ID))
+		return
+	}
+	if task.Type == "Reminder" {
+		color := s.State.UserColor(member.User.ID, channel.ID)
+		e := discordgo.MessageEmbed{
+			Title:       "REMINDER",
+			Description: task.Content,
+			Timestamp:   task.CreationTime.Format(time.RFC3339),
+			Color:       color,
+			Footer: &discordgo.MessageEmbedFooter{
+				Text: "Created: ",
+			},
+		}
+		msg := discordgo.MessageSend{
+			Content: member.Mention(),
+			Embed:   &e,
+		}
+		_, err = s.ChannelMessageSendComplex(channel.ID, &msg)
+		if err != nil {
+			log.Println(err)
+		}
+	}
+	processQuery = "UPDATE tasks SET processing = false WHERE id = $1"
+	_, err = postgres.DB.Exec(processQuery, task.ID)
+	if err != nil {
+		log.Println(err)
+	}
+
+}
+
+func getTasksToRun() []Task {
+	query := "SELECT id, type, content, guild_id, channel_id, user_id, creation_time, trigger_time " +
+		"from tasks where completed is false and processing is false and trigger_time < $1"
+	res, err := postgres.DB.Query(query, time.Now())
+	if err != nil {
+		log.Println(err)
+	}
+	var tasks []Task
+	for res.Next() {
+		var t Task
+		err = res.Scan(&t.ID, &t.Type, &t.Content, &t.GuildID, &t.ChannelID, &t.UserID, &t.CreationTime, &t.TriggerTime)
+		if err != nil {
+			log.Println(err)
+		}
+		for _, task := range tasks {
+			if task.ID == t.ID {
+				continue
+			}
+		}
+		tasks = append(tasks, t)
+	}
+
+	return tasks
+}
+
+func ProcessTasks(s *discordgo.Session, interval int) {
+	for {
+		time.Sleep(time.Duration(interval * 1e9))
+
+		tasks := getTasksToRun()
+
+		if len(tasks) > 0 {
+			for _, t := range tasks {
+				go processTask(&t, s)
+			}
+		}
+	}
+}