From 09713d40ba8b78671964efe83cdd1da124d44114 Mon Sep 17 00:00:00 2001
From: Patrick Connolly <patrick.c.connolly@gmail.com>
Date: Thu, 15 Nov 2018 04:00:21 +0800
Subject: Fix file caching issue (slack). #572 (#575)

---
 bridge/slack/handlers.go | 29 ++++++++++++++++++++++-------
 bridge/slack/slack.go    |  8 ++------
 2 files changed, 24 insertions(+), 13 deletions(-)

(limited to 'bridge')

diff --git a/bridge/slack/handlers.go b/bridge/slack/handlers.go
index c4c68961..17249bda 100644
--- a/bridge/slack/handlers.go
+++ b/bridge/slack/handlers.go
@@ -132,7 +132,16 @@ func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool {
 		ev.SubMessage.Edited == nil {
 		return true
 	}
-	return false
+	return b.filesCached(ev.Files)
+}
+
+func (b *Bslack) filesCached(files []slack.File) bool {
+	for i := range files {
+		if !b.fileCached(&files[i]) {
+			return false
+		}
+	}
+	return true
 }
 
 // handleMessageEvent handles the message events. Together with any called sub-methods,
@@ -248,10 +257,9 @@ func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message,
 
 // handleDownloadFile handles file download
 func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) error {
-	if b.fileIsAvailable(file) {
+	if b.fileCached(file) {
 		return nil
 	}
-
 	// Check that the file is neither too large nor blacklisted.
 	if err := helper.HandleDownloadSize(b.Log, rmsg, file.Name, int64(file.Size), b.General); err != nil {
 		b.Log.WithError(err).Infof("Skipping download of incoming file.")
@@ -273,11 +281,18 @@ func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) erro
 	return nil
 }
 
-func (b *Bslack) fileIsAvailable(file *slack.File) bool {
-	// Only download a file if it is not in the cache or if it has been entered more than a minute ago.
-	if ts, ok := b.cache.Get("file" + file.ID); ok && time.Since(ts.(time.Time)) > time.Minute {
+// fileCached implements Matterbridge's caching logic for files
+// shared via Slack.
+//
+// We consider that a file was cached if its ID was added in the last minute or
+// it's name was registered in the last 10 seconds. This ensures that an
+// identically named file but with different content will be uploaded correctly
+// (the assumption is that such name collisions will not occur within the given
+// timeframes).
+func (b *Bslack) fileCached(file *slack.File) bool {
+	if ts, ok := b.cache.Get("file" + file.ID); ok && time.Since(ts.(time.Time)) < time.Minute {
 		return true
-	} else if ts, ok = b.cache.Get("filename" + file.Name); ok && time.Since(ts.(time.Time)) > 10*time.Second {
+	} else if ts, ok = b.cache.Get("filename" + file.Name); ok && time.Since(ts.(time.Time)) < 10*time.Second {
 		return true
 	}
 	return false
diff --git a/bridge/slack/slack.go b/bridge/slack/slack.go
index 8030eb9c..1f51de95 100644
--- a/bridge/slack/slack.go
+++ b/bridge/slack/slack.go
@@ -384,9 +384,7 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID string) {
 		// we can't match on the file ID yet, so we have to match on the filename too.
 		ts := time.Now()
 		b.Log.Debugf("Adding file %s to cache at %s with timestamp", fi.Name, ts.String())
-		if !b.cache.Add("filename"+fi.Name, ts) {
-			b.Log.Warnf("Failed to add file %s to cache at %s with timestamp", fi.Name, ts.String())
-		}
+		b.cache.Add("filename"+fi.Name, ts)
 		res, err := b.sc.UploadFile(slack.FileUploadParameters{
 			Reader:         bytes.NewReader(*fi.Data),
 			Filename:       fi.Name,
@@ -399,9 +397,7 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID string) {
 		}
 		if res.ID != "" {
 			b.Log.Debugf("Adding file ID %s to cache with timestamp %s", res.ID, ts.String())
-			if !b.cache.Add("file"+res.ID, ts) {
-				b.Log.Warnf("Failed to add file ID %s to cache with timestamp %s", res.ID, ts.String())
-			}
+			b.cache.Add("file"+res.ID, ts)
 		}
 	}
 }
-- 
cgit v1.2.3