From 17b8b86d68badfe27c29c7448548f9eb97dd26e9 Mon Sep 17 00:00:00 2001
From: Wim <wim@42.be>
Date: Sat, 15 Apr 2017 23:46:01 +0200
Subject: [PATCH] Add support for edited messages (gitter)

---
 bridge/gitter/gitter.go  | 47 ++++++++++++++++++++++++++++++----------
 changelog.md             | 11 +++++++++-
 matterbridge.toml.sample |  9 ++++++++
 3 files changed, 54 insertions(+), 13 deletions(-)

diff --git a/bridge/gitter/gitter.go b/bridge/gitter/gitter.go
index af71d892..a962f923 100644
--- a/bridge/gitter/gitter.go
+++ b/bridge/gitter/gitter.go
@@ -2,10 +2,12 @@ package bgitter
 
 import (
 	"fmt"
+	"strings"
+	"time"
+
 	"github.com/42wim/matterbridge/bridge/config"
 	log "github.com/Sirupsen/logrus"
 	"github.com/sromku/go-gitter"
-	"strings"
 )
 
 type Bgitter struct {
@@ -71,27 +73,48 @@ func (b *Bgitter) JoinChannel(channel string) error {
 	}
 	users, _ := b.c.GetUsersInRoom(roomID)
 	b.Users = append(b.Users, users...)
+
+	// we need to use (experimental) faye for edits.
+	// streaming API doesn't show edits.
+	if !b.Config.EditDisable {
+		faye := b.c.Faye(roomID)
+		go faye.Listen()
+		go func(stream *gitter.Faye, room string) {
+			for event := range stream.Event {
+				b.handleEvent(event, room)
+			}
+		}(faye, room.Name)
+		return nil
+	}
 	stream := b.c.Stream(roomID)
 	go b.c.Listen(stream)
 
 	go func(stream *gitter.Stream, room string) {
 		for event := range stream.Event {
-			switch ev := event.Data.(type) {
-			case *gitter.MessageReceived:
-				// check for ZWSP to see if it's not an echo
-				if !strings.HasSuffix(ev.Message.Text, "​") {
-					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account)
-					b.Remote <- config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room,
-						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username)}
-				}
-			case *gitter.GitterConnectionClosed:
-				flog.Errorf("connection with gitter closed for room %s", room)
-			}
+			b.handleEvent(event, room)
 		}
 	}(stream, room.Name)
 	return nil
 }
 
+func (b *Bgitter) handleEvent(event gitter.Event, room string) {
+	flog.Debugf("event: %#v %#v", event.Data, event.Data.(*gitter.MessageReceived).Message.EditedAt.Sub(time.Now()).String())
+	switch ev := event.Data.(type) {
+	case *gitter.MessageReceived:
+		// check for ZWSP to see if it's not an echo
+		if !strings.HasSuffix(ev.Message.Text, "​") {
+			if !ev.Message.EditedAt.IsZero() && !b.Config.EditDisable {
+				ev.Message.Text = ev.Message.Text + b.Config.EditSuffix
+			}
+			flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account)
+			b.Remote <- config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room,
+				Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username)}
+		}
+	case *gitter.GitterConnectionClosed:
+		flog.Errorf("connection with gitter closed for room %s", room)
+	}
+}
+
 func (b *Bgitter) Send(msg config.Message) error {
 	flog.Debugf("Receiving %#v", msg)
 	roomID := b.getRoomID(msg.Channel)
diff --git a/changelog.md b/changelog.md
index f208f44b..fd640095 100644
--- a/changelog.md
+++ b/changelog.md
@@ -1,8 +1,17 @@
 # v0.12.0
+## Changes
+* gitter: The default support for edited messages enables experimental FAYE support for gitter.
+  If you need stable code add EditDisable=true to your gitter configuration.
+
 ## New features
 * general: add support for edited messages. 
   Add new keyword EditDisable (false/true), default false. Which means by default edited messages will be sent to other bridges.
-  Add new keyword EditSuffix , default "". You can change this eg to "(edited)", this will be appended to every edit message.
+  Add new keyword EditSuffix , default "". You can change this eg to " (edited)", this will be appended to every edit message.
+
+* discord: Add support for edited messages. See EditDisable/EditSuffix
+* mattermost: Add support for edited messages. See EditDisable/EditSuffix
+* telegram: Add support for edited messages. See EditDisable/EditSuffix
+* gitter: Add support for edited messages. See EditDisable/EditSuffix
 
 ## Enhancements
 * discord: Strip custom emoji metadata (discord). Closes #148
diff --git a/matterbridge.toml.sample b/matterbridge.toml.sample
index 33a01fc5..a54da452 100644
--- a/matterbridge.toml.sample
+++ b/matterbridge.toml.sample
@@ -282,6 +282,15 @@ ShowJoinPart=false
 #REQUIRED
 Token="Yourtokenhere"
 
+#Disable sending of edits to other bridges
+#WARNING: the default uses FAYE which is experimental. If you need stability set EditDisable=true
+#OPTIONAL (default false)
+EditDisable=true
+
+#Message to be appended to every edited message
+#OPTIONAL (default empty)
+EditSuffix=" (edited)"
+
 #Nicks you want to ignore. 
 #Messages from those users will not be sent to other bridges.
 #OPTIONAL