From 1165a7fbf2be2590d9376da1c83022a977ebcf5b Mon Sep 17 00:00:00 2001
From: tj <mail@jager.no>
Date: Thu, 23 Sep 2010 18:54:34 +0200
Subject: [PATCH] Fix compile + bit more robust reconnect code

---
 Makefile        |  2 +-
 irc.go          | 21 ++++++++++++++++-----
 irc_callback.go |  2 +-
 irc_struct.go   | 23 ++++++++++++-----------
 4 files changed, 30 insertions(+), 18 deletions(-)

diff --git a/Makefile b/Makefile
index d266c78..393ab31 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,4 @@
-include $(GOROOT)/src/Make.$(GOARCH)
+include $(GOROOT)/src/Make.inc
 
 TARG=irc
 GOFILES=irc.go irc_struct.go irc_callback.go
diff --git a/irc.go b/irc.go
index 88ab7d7..14517db 100644
--- a/irc.go
+++ b/irc.go
@@ -17,14 +17,16 @@ const (
 	VERSION = "GolangBOT v1.0"
 )
 
+var error bool
+
 
 func reader(irc *IRCConnection) {
 	br := bufio.NewReader(irc.socket)
-	for {
+	for !error {
 		msg, err := br.ReadString('\n')
 		if err != nil {
 			irc.Error <- err
-			return
+			break
 		}
 		irc.lastMessage = time.Seconds()
 		msg = msg[0 : len(msg)-2] //Remove \r\n
@@ -53,18 +55,20 @@ func reader(irc *IRCConnection) {
 		}
 		irc.RunCallbacks(event)
 	}
+	irc.syncreader <- true
 }
 
 func writer(irc *IRCConnection) {
-	for {
+	for !error {
 		b := []byte(<-irc.pwrite)
 		_, err := irc.socket.Write(b)
 		if err != nil {
 			fmt.Printf("%s\n", err)
 			irc.Error <- err
-			return
+			break
 		}
 	}
+	irc.syncwriter <- true
 }
 
 //Pings the server if we have not recived any messages for 5 minutes
@@ -101,6 +105,8 @@ func (irc *IRCConnection) SendRaw(message string) {
 }
 
 func (i *IRCConnection) Reconnect() os.Error {
+	<-i.syncreader
+	<-i.syncwriter
 	for {
 		fmt.Printf("Reconnecting to %s\n", i.server)
 		var err os.Error
@@ -110,6 +116,7 @@ func (i *IRCConnection) Reconnect() os.Error {
 		}
 		fmt.Printf("Error: %s\n", err)
 	}
+	error = false
 	fmt.Printf("Connected to %s (%s)\n", i.server, i.socket.RemoteAddr())
 	go reader(i)
 	go writer(i)
@@ -120,7 +127,9 @@ func (i *IRCConnection) Reconnect() os.Error {
 
 func (i *IRCConnection) Loop() {
 	for {
-		<-i.Error
+		e := <-i.Error
+		fmt.Printf("Error: %s\n", e)
+		error = true
 		i.Reconnect()
 	}
 }
@@ -137,6 +146,8 @@ func (i *IRCConnection) Connect(server string) os.Error {
 	i.pread = make(chan string, 100)
 	i.pwrite = make(chan string, 100)
 	i.Error = make(chan os.Error, 10)
+	i.syncreader = make(chan bool)
+	i.syncwriter = make(chan bool)
 	go reader(i)
 	go writer(i)
 	go pinger(i)
diff --git a/irc_callback.go b/irc_callback.go
index d8d7d00..83dcbd5 100644
--- a/irc_callback.go
+++ b/irc_callback.go
@@ -79,7 +79,7 @@ func (irc *IRCConnection) setupCallbacks() {
 	})
 
 	irc.AddCallback("CTCP_TIME", func(e *IRCEvent) {
-		ltime := time.LocalTime();
+		ltime := time.LocalTime()
 		irc.SendRaw(fmt.Sprintf("NOTICE %s :\x01TIME %s\x01", e.Nick, ltime.String()))
 	})
 
diff --git a/irc_struct.go b/irc_struct.go
index bd99452..4ca8105 100644
--- a/irc_struct.go
+++ b/irc_struct.go
@@ -10,19 +10,20 @@ import (
 )
 
 type IRCConnection struct {
-	socket        net.Conn
-	pread, pwrite chan string
-	Error         chan os.Error
-	nick          string
-	user          string
-	registered    bool
-	server        string
+	socket                 net.Conn
+	pread, pwrite          chan string
+	Error                  chan os.Error
+	syncreader, syncwriter chan bool
+	nick                   string
+	user                   string
+	registered             bool
+	server                 string
 
 	events map[string][]func(*IRCEvent)
-	
-	lastMessage int64;
-	ticker <-chan int64;
-	ticker2 <-chan int64;
+
+	lastMessage int64
+	ticker      <-chan int64
+	ticker2     <-chan int64
 }
 
 type IRCEvent struct {