Add support for outgoing webhook token

1 files changed, 10 insertions, 1 deletions

diff --git a/matterhook/matterhook.go b/matterhook/matterhook.go
index 8e87cc7b..ed062499 100644
--- a/matterhook/matterhook.go
+++ b/matterhook/matterhook.go
@@ -45,8 +45,10 @@ type Client struct {
 	Config
 }
 
+// Config for client.
 type Config struct {
-	Port int
+	Port  int    // Port to listen on.
+	Token string // Only allow this token from Mattermost. (Allow everything when empty)
 }
 
 // New Mattermost client.
@@ -96,6 +98,13 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		http.NotFound(w, r)
 		return
 	}
+	if c.Token != "" {
+		if msg.Token != c.Token {
+			log.Println("invalid token " + msg.Token + " from " + r.RemoteAddr)
+			http.NotFound(w, r)
+			return
+		}
+	}
 	c.In <- msg
 }