From bab544261a86644e560d78594a38c0ec1b90f911 Mon Sep 17 00:00:00 2001
From: Jordan Evans <evans.jordan.m@gmail.com>
Date: Thu, 27 Feb 2014 12:00:15 -0800
Subject: [PATCH] properly quit irc after sending failure/success

---
 pkg/plugin/notify/irc.go | 36 ++++++++++++++++++++----------------
 1 file changed, 20 insertions(+), 16 deletions(-)

diff --git a/pkg/plugin/notify/irc.go b/pkg/plugin/notify/irc.go
index a98104252..93127e074 100644
--- a/pkg/plugin/notify/irc.go
+++ b/pkg/plugin/notify/irc.go
@@ -20,8 +20,8 @@ type IRC struct {
 	Success bool   `yaml:"on_success,omitempty"`
 	Failure bool   `yaml:"on_failure,omitempty"`
     SSL bool `yaml:"ssl,omitempty"`
-    ClientStarted bool
-    Client *irc.Conn
+	ClientStarted bool
+	Client *irc.Conn
 }
 
 func (i *IRC) Connect() {
@@ -30,20 +30,15 @@ func (i *IRC) Connect() {
     connected := make(chan bool)
     c.AddHandler(irc.CONNECTED,
         func(conn *irc.Conn, line *irc.Line) { 
-            conn.Join(i.Channel)
+	    conn.Join(i.Channel)
             connected <- true})
     c.Connect(i.Server)
     <-connected
-    i.Client = c
     i.ClientStarted = true
+    i.Client = c
 }
 
 func (i *IRC) Send(context *Context) error {
-       
-        if !i.ClientStarted {
-            i.Connect()
-        }
-
 	switch {
 	case context.Commit.Status == "Started" && i.Started:
 		return i.sendStarted(context)
@@ -57,24 +52,33 @@ func (i *IRC) Send(context *Context) error {
 
 func (i *IRC) sendStarted(context *Context) error {
 	msg := fmt.Sprintf(ircStartedMessage, context.Repo.Name, context.Commit.HashShort(), context.Commit.Author)
-	 i.send(i.Channel, msg)
-        return nil
+	i.send(i.Channel, msg)
+    return nil
 }
 
 func (i *IRC) sendFailure(context *Context) error {
 	msg := fmt.Sprintf(ircFailureMessage, context.Repo.Name, context.Commit.HashShort(), context.Commit.Author)
-	 i.send(i.Channel, msg)
-        return nil
+	i.send(i.Channel, msg)
+	if i.ClientStarted {
+		i.Client.Quit()
+	}
+    return nil
 }
 
 func (i *IRC) sendSuccess(context *Context) error {
 	msg := fmt.Sprintf(ircSuccessMessage, context.Repo.Name, context.Commit.HashShort(), context.Commit.Author)
-	 i.send(i.Channel, msg)
-        return nil
+	i.send(i.Channel, msg)
+	if i.ClientStarted {
+	    i.Client.Quit()
+	}
+    return nil
 }
 
 
 func (i *IRC) send(channel string, message string) error {
-    i.Client.Notice(channel, message)
+	if !i.ClientStarted {
+		i.Connect()
+	}
+	i.Client.Notice(channel, message)
     return nil
 }