mirror of
https://github.com/ergochat/ergo.git
synced 2024-12-22 18:52:41 +01:00
82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
// Copyright (c) 2012-2014 Jeremy Latt
|
|
// Copyright (c) 2016- Daniel Oaks <daniel@danieloaks.net>
|
|
// released under the MIT license
|
|
|
|
package irc
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
// Socket represents an IRC socket.
|
|
type Socket struct {
|
|
Closed bool
|
|
conn net.Conn
|
|
reader *bufio.Reader
|
|
}
|
|
|
|
// NewSocket returns a new Socket.
|
|
func NewSocket(conn net.Conn) Socket {
|
|
return Socket{
|
|
conn: conn,
|
|
reader: bufio.NewReader(conn),
|
|
}
|
|
}
|
|
|
|
// Close stops a Socket from being able to send/receive any more data.
|
|
func (socket *Socket) Close() {
|
|
if socket.Closed {
|
|
return
|
|
}
|
|
socket.Closed = true
|
|
socket.conn.Close()
|
|
}
|
|
|
|
// Read returns a single IRC line from a Socket.
|
|
func (socket *Socket) Read() (string, error) {
|
|
if socket.Closed {
|
|
return "", io.EOF
|
|
}
|
|
|
|
lineBytes, err := socket.reader.ReadBytes('\n')
|
|
|
|
// convert bytes to string
|
|
line := string(lineBytes[:])
|
|
|
|
// read last message properly (such as ERROR/QUIT/etc), just fail next reads/writes
|
|
if err == io.EOF {
|
|
socket.Close()
|
|
}
|
|
|
|
if err == io.EOF && strings.TrimSpace(line) != "" {
|
|
// don't do anything
|
|
} else if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return strings.TrimRight(line, "\r\n"), nil
|
|
}
|
|
|
|
// Write sends the given string out of Socket.
|
|
func (socket *Socket) Write(data string) error {
|
|
if socket.Closed {
|
|
return io.EOF
|
|
}
|
|
|
|
// write data
|
|
_, err := socket.conn.Write([]byte(data))
|
|
if err != nil {
|
|
socket.Close()
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// WriteLine writes the given line out of Socket.
|
|
func (socket *Socket) WriteLine(line string) error {
|
|
return socket.Write(line + "\r\n")
|
|
}
|