3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-12-22 18:52:41 +01:00
ergo/irc/socket.go
2014-02-16 22:24:49 -08:00

102 lines
1.7 KiB
Go

package irc
import (
"bufio"
"log"
"net"
"strings"
)
type Socket struct {
conn net.Conn
reader *bufio.Reader
receive chan string
send chan string
writer *bufio.Writer
}
func NewSocket(conn net.Conn) *Socket {
socket := &Socket{
conn: conn,
reader: bufio.NewReader(conn),
receive: make(chan string),
send: make(chan string),
writer: bufio.NewWriter(conn),
}
go socket.readLines()
go socket.writeLines()
return socket
}
func (socket *Socket) String() string {
return socket.conn.RemoteAddr().String()
}
func (socket *Socket) Close() {
socket.conn.Close()
}
func (socket *Socket) Read() <-chan string {
return socket.receive
}
func (socket *Socket) Write(lines []string) {
for _, line := range lines {
socket.send <- line
}
return
}
func (socket *Socket) readLines() {
for {
line, err := socket.reader.ReadString('\n')
if err != nil {
if DEBUG_NET {
log.Printf("%s → error: %s", socket, err)
}
break
}
line = strings.TrimSpace(line)
if len(line) == 0 {
continue
}
if DEBUG_NET {
log.Printf("%s → %s", socket, line)
}
socket.receive <- line
}
close(socket.receive)
}
func (socket *Socket) writeLines() {
for line := range socket.send {
if DEBUG_NET {
log.Printf("%s ← %s", socket, line)
}
if _, err := socket.writer.WriteString(line); socket.maybeLogWriteError(err) {
break
}
if _, err := socket.writer.WriteString(CRLF); socket.maybeLogWriteError(err) {
break
}
if err := socket.writer.Flush(); socket.maybeLogWriteError(err) {
break
}
}
}
func (socket *Socket) maybeLogWriteError(err error) bool {
if err != nil {
if DEBUG_NET {
log.Printf("%s ← error: %s", socket, err)
}
return true
}
return false
}