Add option to enable processing on post update

server/config.go

@@ -12,6 +12,7 @@ import (
 // Config from config.json
 type Config struct {
 	EnableAdminCommand bool
+	EnableOnUpdate     bool
 	Links              []Link
 }
 
@@ -74,6 +75,7 @@ func (conf Config) ToConfig() map[string]interface{} {
 	}
 	return map[string]interface{}{
 		"EnableAdminCommand": conf.EnableAdminCommand,
+		"EnableOnUpdate":     conf.EnableOnUpdate,
 		"Links":              links,
 	}
 }

server/plugin.go

@@ -20,9 +20,7 @@ type Plugin struct {
 	confLock sync.RWMutex
 }
 
-// MessageWillBePosted is invoked when a message is posted by a user before it is committed
-// to the database.
-func (p *Plugin) MessageWillBePosted(c *plugin.Context, post *model.Post) (*model.Post, string) {
+func (p *Plugin) ProcessPost(c *plugin.Context, post *model.Post) (*model.Post, string) {
 	conf := p.getConfig()
 	postText := post.Message
 	offset := 0
@@ -98,3 +96,20 @@ func (p *Plugin) MessageWillBePosted(c *plugin.Context, post *model.Post) (*mode
 
 	return post, ""
 }
+
+// MessageWillBePosted is invoked when a message is posted by a user before it is committed
+// to the database.
+func (p *Plugin) MessageWillBePosted(c *plugin.Context, post *model.Post) (*model.Post, string) {
+	return p.ProcessPost(c, post)
+}
+
+// MessageWillBeUpdated is invoked when a message is updated by a user before it is committed
+// to the database.
+func (p *Plugin) MessageWillBeUpdated(c *plugin.Context, post *model.Post, _ *model.Post) (*model.Post, string) {
+	conf := p.getConfig()
+	if conf.EnableOnUpdate {
+		return p.ProcessPost(c, post)
+	} else {
+		return post, ""
+	}
+}