From 43738dbc89c888967181309002dc46b2c64990a0 Mon Sep 17 00:00:00 2001
From: Wim <wim@42.be>
Date: Sat, 24 Oct 2015 18:44:45 +0200
Subject: Refactor and IconURL support

---
 config.go                |  1 +
 matterbridge.conf.sample |  1 +
 matterbridge.go          | 25 +++++++++++++------------
 3 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/config.go b/config.go
index 62d0ee17..efd6eaf8 100644
--- a/config.go
+++ b/config.go
@@ -20,6 +20,7 @@ type Config struct {
 		Port         int
 		ShowJoinPart bool
 		Token        string
+		IconURL      string
 	}
 }
 
diff --git a/matterbridge.conf.sample b/matterbridge.conf.sample
index 7cbdce1f..231a3c63 100644
--- a/matterbridge.conf.sample
+++ b/matterbridge.conf.sample
@@ -11,3 +11,4 @@ url="http://yourdomain/hooks/yourhookkey"
 port=9999
 showjoinpart=true
 #token=yourtokenfrommattermost
+IconURL="http://youricon.png"
diff --git a/matterbridge.go b/matterbridge.go
index 219f062e..94d38954 100644
--- a/matterbridge.go
+++ b/matterbridge.go
@@ -45,30 +45,31 @@ func (b *Bridge) createIRC(name string) *irc.Connection {
 }
 
 func (b *Bridge) handlePrivMsg(event *irc.Event) {
-	matterMessage := matterhook.OMessage{}
+	msg := ""
 	if event.Code == "CTCP_ACTION" {
-		matterMessage.Text = event.Nick + " "
+		msg = event.Nick + " "
 	}
-	matterMessage.Text += event.Message()
-	matterMessage.UserName = "irc-" + event.Nick
-	b.m.Send(matterMessage)
+	msg += event.Message()
+	b.Send("irc-"+event.Nick, msg)
 }
 
 func (b *Bridge) handleJoinPart(event *irc.Event) {
-	matterMessage := matterhook.OMessage{}
-	matterMessage.Text = "irc-" + event.Nick + " " + strings.ToLower(event.Code) + "s " + event.Message()
-	matterMessage.UserName = b.Config.IRC.Nick
-	b.m.Send(matterMessage)
+	b.Send(b.Config.IRC.Nick, "irc-"+event.Nick+" "+strings.ToLower(event.Code)+"s "+event.Message())
 }
 
 func (b *Bridge) handleOther(event *irc.Event) {
-	matterMessage := matterhook.OMessage{}
 	switch event.Code {
 	case "353":
-		matterMessage.UserName = b.Config.IRC.Nick
-		matterMessage.Text = event.Message() + " currently on IRC"
+		b.Send(b.Config.IRC.Nick, event.Message()+" currently on IRC")
 	}
+}
+
+func (b *Bridge) Send(nick string, message string) error {
+	matterMessage := matterhook.OMessage{IconURL: b.Config.Mattermost.IconURL}
+	matterMessage.UserName = nick
+	matterMessage.Text = message
 	b.m.Send(matterMessage)
+	return nil
 }
 
 func (b *Bridge) handleMatter() {
-- 
cgit v1.2.3