feat: add timestamp vars

target · Apr 8, 2021 · 509927c · 509927c
1 parent cef183b
commit 509927c
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 37 deletions.
diff --git a/remote/discord/helper.go b/remote/discord/helper.go
@@ -35,6 +35,8 @@ func populateMessage(message models.Message, msgType models.MessageType, channel
 	message.Vars["_channel.id"] = channel
 	message.Vars["_channel.name"] = message.ChannelName
 
+	message.Vars["_source.timestamp"] = timeStamp
+
 	// Populate message user sender
 	// These will be accessible on rules via ${_user.email}, etc
 	if user != nil { // nil user implies a message from an api/bot (i.e. not an actual user)

diff --git a/remote/discord/remote.go b/remote/discord/remote.go
@@ -88,6 +88,7 @@ func (c *Client) Read(inputMsgs chan<- models.Message, rules map[string]models.R
 	}
 
 	bot.Name = botuser.Username
+	bot.ID = botuser.ID
 
 	foundGuild := false
 
@@ -177,6 +178,11 @@ func handleDiscordMessage(bot *models.Bot, inputMsgs chan<- models.Message) inte
 			return
 		}
 
+		// ignore messages from self
+		if m.Author.ID == bot.ID {
+			return
+		}
+
 		// Process message
 		message := models.NewMessage()
 		switch m.Type {

diff --git a/remote/slack/helper.go b/remote/slack/helper.go
@@ -166,6 +166,7 @@ func handleCallBack(api *slack.Client, event slackevents.EventsAPIInnerEvent, bo
 
 			link, err := api.GetPermalink(&slack.PermalinkParameters{Channel: channel, Ts: timestamp})
 			if err != nil {
+				bot.Log.Errorf("unable to retrieve link to message: %s", err.Error())
 				link = ""
 			}
 
@@ -481,6 +482,10 @@ func populateMessage(message models.Message, msgType models.MessageType, channel
 		// make link to trigger message available
 		message.Vars["_source.link"] = message.SourceLink
 
+		// make timestamp information available
+		message.Vars["_source.timestamp"] = timeStamp
+		message.Vars["_source.thread_timestamp"] = threadTimestamp
+
 		// Populate message with user information (i.e. who sent the message)
 		// These will be accessible on rules via ${_user.email}, ${_user.id}, etc.
 		if user != nil { // nil user implies a message from an api/bot (i.e. not an actual user)
@@ -616,6 +621,7 @@ func readFromRTM(rtm *slack.RTM, inputMsgs chan<- models.Message, bot *models.Bo
 
 				link, err := rtm.GetPermalink(&slack.PermalinkParameters{Channel: channel, Ts: timestamp})
 				if err != nil {
+					bot.Log.Errorf("unable to retrieve link to message: %s", err.Error())
 					link = ""
 				}
 

diff --git a/remote/telegram/remote.go b/remote/telegram/remote.go
@@ -69,47 +69,52 @@ func (c *Client) Read(inputMsgs chan<- models.Message, rules map[string]models.R
 		}
 
 		// check if we should respond to other bot messages
-		if m.From.IsBot && !bot.RespondToBots {
+		if m.From != nil && m.From.IsBot && !bot.RespondToBots {
 			continue
 		}
 
-		// only process messages not from out bot
-		if m.From.ID != botuser.ID {
-			msg, mentioned := processMessageText(m.Text, bot.Name)
-
-			// support slash commands
-			if len(m.Command()) > 0 {
-				fullCmd := fmt.Sprintf("%s %s", m.Command(), m.CommandArguments())
-				mentioned = true
-				msg = strings.TrimSpace(fullCmd)
-			}
-
-			message := models.NewMessage()
-			message.Timestamp = strconv.FormatInt(m.Time().Unix(), 10)
-			message.Type = mapMessageType(*m)
-			message.Input = msg
-			message.Output = ""
-			message.ID = strconv.Itoa(m.MessageID)
-			message.Service = models.MsgServiceChat
-			message.BotMentioned = mentioned
-			message.ChannelID = strconv.FormatInt(m.Chat.ID, 10)
-
-			// populate message with metadata
-			if m.From != nil {
-				message.Vars["_user.name"] = m.From.UserName
-				message.Vars["_user.firstname"] = m.From.FirstName
-				message.Vars["_user.lastname"] = m.From.LastName
-				message.Vars["_user.id"] = strconv.Itoa(m.From.ID)
-				message.Vars["_user.realnamenormalized"] = fmt.Sprintf("%s %s", m.From.FirstName, m.From.LastName)
-				message.Vars["_user.displayname"] = m.From.UserName
-				message.Vars["_user.displaynamenormalized"] = m.From.UserName
-			}
-
-			message.Vars["_channel.id"] = message.ChannelID
-			message.Vars["_channel.name"] = m.Chat.Title
-
-			inputMsgs <- message
+		// only process messages not from our bot
+		// Note: it looks like "From" is not being populated currently
+		if m.From != nil && m.From.ID == botuser.ID {
+			continue
 		}
+
+		msg, mentioned := processMessageText(m.Text, bot.Name)
+
+		// support slash commands
+		if len(m.Command()) > 0 {
+			fullCmd := fmt.Sprintf("%s %s", m.Command(), m.CommandArguments())
+			mentioned = true
+			msg = strings.TrimSpace(fullCmd)
+		}
+
+		message := models.NewMessage()
+		message.Timestamp = strconv.FormatInt(m.Time().Unix(), 10)
+		message.Type = mapMessageType(*m)
+		message.Input = msg
+		message.Output = ""
+		message.ID = strconv.Itoa(m.MessageID)
+		message.Service = models.MsgServiceChat
+		message.BotMentioned = mentioned
+		message.ChannelID = strconv.FormatInt(m.Chat.ID, 10)
+
+		// populate message with metadata
+		if m.From != nil {
+			message.Vars["_user.name"] = m.From.UserName
+			message.Vars["_user.firstname"] = m.From.FirstName
+			message.Vars["_user.lastname"] = m.From.LastName
+			message.Vars["_user.id"] = strconv.Itoa(m.From.ID)
+			message.Vars["_user.realnamenormalized"] = fmt.Sprintf("%s %s", m.From.FirstName, m.From.LastName)
+			message.Vars["_user.displayname"] = m.From.UserName
+			message.Vars["_user.displaynamenormalized"] = m.From.UserName
+		}
+
+		message.Vars["_channel.id"] = message.ChannelID
+		message.Vars["_channel.name"] = m.Chat.Title
+
+		message.Vars["_source.timestamp"] = strconv.Itoa(m.Date)
+
+		inputMsgs <- message
 	}
 }