message parsing

lib/messages.go

@@ -0,0 +1,209 @@
+package lib
+
+import (
+	"fmt"
+	"math"
+	"regexp"
+	"strings"
+	"time"
+
+	"github.com/Cynosphere/comcord/state"
+	"github.com/bwmarrin/discordgo"
+	"github.com/mgutz/ansi"
+)
+
+var /*const*/ REGEX_CODEBLOCK = regexp.MustCompile(`(?i)\x60\x60\x60(?:([a-z0-9_+\-\.]+?)\n)?\n*([^\n].*?)\n*\x60\x60\x60`)
+
+type MessageOptions struct {
+  Content string
+  Name string
+  Channel string
+  Bot bool
+  Attachments []*discordgo.MessageAttachment
+  Stickers []*discordgo.Sticker
+  Reply *discordgo.Message
+  Timestamp time.Time
+  IsMention bool
+  IsDM bool
+  IsJoin bool
+  IsPin bool
+  IsDump bool
+  NoColor bool
+  InHistory bool
+}
+
+func FormatMessage(session *discordgo.Session, options MessageOptions) {
+
+  // TODO: timestamps for pin and join
+
+  // TODO: history lines
+
+  nameLength := len(options.Name) + 2
+  stateNameLength := state.GetNameLength()
+  if nameLength > stateNameLength {
+    state.SetNameLength(nameLength)
+  }
+
+  // TODO: replies
+
+  if options.IsDump {
+    if options.InHistory {
+      // TODO
+    } else {
+      wordCount := len(strings.Split(options.Content, " "))
+      lineCount := len(strings.Split(options.Content, "\n"))
+      wordsPlural := ""
+      linesPlural := ""
+
+  if wordCount > 1 {
+        wordsPlural = "s"
+      }
+      if lineCount > 1 {
+        linesPlural = "s"
+      }
+
+      str := fmt.Sprintf("<%s DUMPs in %d characters of %d word%s in %d line%s>", options.Name, len(options.Content), wordCount, wordsPlural, lineCount, linesPlural)
+
+      if options.NoColor {
+        fmt.Print(str)
+      }
+    }
+  } else {
+    // TODO: markdown
+
+    if options.IsDM {
+      name := fmt.Sprintf("*%s*", options.Name)
+      if !options.NoColor {
+        name = ansi.Color(name, "red+b")
+      }
+
+      fmt.Printf("%s %s\x07\n\r", name, options.Content)
+    } else if len(options.Content) > 1 &&
+    (strings.HasPrefix(options.Content, "*") && strings.HasSuffix(options.Content, "*") && !strings.HasPrefix(options.Content, "**") && !strings.HasSuffix(options.Content, "**")) ||
+    (strings.HasPrefix(options.Content, "_") && strings.HasSuffix(options.Content, "_") && !strings.HasPrefix(options.Content, "__") && !strings.HasSuffix(options.Content, "__")) {
+      str := fmt.Sprintf("<%s %s>", options.Name, options.Content[1:len(options.Content)-1])
+
+      if options.NoColor {
+        fmt.Print(str + "\n\r")
+      } else {
+        fmt.Print(ansi.Color(str, "green+b") + "\n\r")
+      }
+    } else if options.IsJoin {
+      // TODO
+    } else if options.IsPin {
+      // TODO
+    } else {
+      nameColor := "cyan+b"
+      if options.IsMention {
+        nameColor = "red+b"
+      } else if options.Bot {
+        nameColor = "yellow+b"
+      }
+
+      name := fmt.Sprintf("[%s]", options.Name)
+      if !options.NoColor {
+        name = ansi.Color(name, nameColor)
+      }
+
+      // FIXME: where is this off by 4 actually from
+      padding := strings.Repeat(" ", int(math.Abs(float64(stateNameLength) - float64(nameLength) - 4)))
+      str := fmt.Sprintf("%s%s %s", name, padding, options.Content)
+      if options.IsMention {
+        str = str + "\x07"
+      }
+      fmt.Print(str + "\n\r")
+    }
+  }
+
+  // TODO: attachments
+
+  // TODO: stickers
+
+  // TODO: links
+
+  // TODO: embeds
+
+  // TODO: lines output for history
+}
+
+func ProcessMessage(session *discordgo.Session, msg *discordgo.Message, options MessageOptions) {
+  channel, err := session.State.Channel(msg.ChannelID)
+  if err != nil {
+    return
+  }
+
+  guild, err := session.State.Guild(channel.GuildID)
+  if err != nil {
+    return
+  }
+
+  selfMember, err := session.State.Member(guild.ID, session.State.User.ID)
+  if err != nil {
+    return
+  }
+
+  hasMentionedRole := false
+  for _, role := range msg.MentionRoles {
+    for _, selfRole := range selfMember.Roles {
+      if role == selfRole {
+        hasMentionedRole = true
+        break;
+      }
+    }
+  }
+
+  isDirectlyMentioned := false
+  for _, user := range msg.Mentions {
+    if user.ID == session.State.User.ID {
+      isDirectlyMentioned = true
+      break;
+    }
+  }
+
+  isPing := msg.MentionEveryone || hasMentionedRole || isDirectlyMentioned
+  isDM := channel.Type == discordgo.ChannelTypeDM || channel.Type == discordgo.ChannelTypeGroupDM
+
+  currentChannel := state.GetCurrentChannel()
+  isCurrentChannel := currentChannel == msg.ChannelID
+
+  if !isCurrentChannel && !isDM && !isPing {
+    return
+  }
+
+  if isPing && !isCurrentChannel && !isDM {
+    str := fmt.Sprintf("**mentioned by %s in #%s in %s**", msg.Author.Username, channel.Name, guild.Name)
+    if options.NoColor {
+      fmt.Print(ansi.Color(str, "red+b"))
+    } else {
+      fmt.Print(str)
+    }
+    fmt.Print("\x07\n\r")
+    return
+  }
+
+  content, _ := msg.ContentWithMoreMentionsReplaced(session)
+  options.Content = content
+  options.Name = msg.Author.Username
+  options.Channel = msg.ChannelID
+  options.Bot = msg.Author.Bot
+  options.Attachments = msg.Attachments
+  options.Stickers = msg.StickerItems
+  options.Reply = msg.ReferencedMessage
+  options.IsMention = isPing
+  options.IsDM = isDM
+  options.IsJoin = msg.Type == discordgo.MessageTypeGuildMemberJoin
+  options.IsPin = msg.Type == discordgo.MessageTypeChannelPinnedMessage
+  options.IsDump = REGEX_CODEBLOCK.MatchString(content)
+
+  FormatMessage(session, options)
+}
+
+func ProcessQueue(session *discordgo.Session) {
+  queue := state.GetMessageQueue()
+
+  for _, msg := range queue {
+    ProcessMessage(session, msg, MessageOptions{NoColor: state.HasNoColor()})
+  }
+
+  state.EmptyMessageQueue()
+}