mirror of https://github.com/ergochat/ergo.git
96 lines
1.6 KiB
Go
96 lines
1.6 KiB
Go
package irc
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"net"
|
|
)
|
|
|
|
const (
|
|
R = '→'
|
|
W = '←'
|
|
EOF = ""
|
|
)
|
|
|
|
type Socket struct {
|
|
conn net.Conn
|
|
writer *bufio.Writer
|
|
}
|
|
|
|
func NewSocket(conn net.Conn, commands chan<- Command) *Socket {
|
|
socket := &Socket{
|
|
conn: conn,
|
|
writer: bufio.NewWriter(conn),
|
|
}
|
|
|
|
go socket.readLines(commands)
|
|
|
|
return socket
|
|
}
|
|
|
|
func (socket *Socket) String() string {
|
|
return socket.conn.RemoteAddr().String()
|
|
}
|
|
|
|
func (socket *Socket) Close() {
|
|
socket.conn.Close()
|
|
Log.debug.Printf("%s closed", socket)
|
|
}
|
|
|
|
func (socket *Socket) readLines(commands chan<- Command) {
|
|
commands <- NewProxyCommand(AddrLookupHostname(socket.conn.RemoteAddr()))
|
|
|
|
scanner := bufio.NewScanner(socket.conn)
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
if len(line) == 0 {
|
|
continue
|
|
}
|
|
Log.debug.Printf("%s → %s", socket, line)
|
|
|
|
msg, err := ParseCommand(line)
|
|
if err != nil {
|
|
// TODO error messaging to client
|
|
continue
|
|
}
|
|
commands <- msg
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
Log.debug.Printf("%s error: %s", socket, err)
|
|
}
|
|
|
|
commands <- NewQuitCommand("connection closed")
|
|
|
|
close(commands)
|
|
}
|
|
|
|
func (socket *Socket) Write(line string) (err error) {
|
|
if _, err = socket.writer.WriteString(line); socket.isError(err, W) {
|
|
return
|
|
}
|
|
|
|
if _, err = socket.writer.WriteString(CRLF); socket.isError(err, W) {
|
|
return
|
|
}
|
|
|
|
go socket.flush()
|
|
|
|
Log.debug.Printf("%s ← %s", socket, line)
|
|
return
|
|
}
|
|
|
|
func (socket *Socket) flush() {
|
|
socket.isError(socket.writer.Flush(), W)
|
|
}
|
|
|
|
func (socket *Socket) isError(err error, dir rune) bool {
|
|
if err != nil {
|
|
if err != io.EOF {
|
|
Log.debug.Printf("%s %c error: %s", socket, dir, err)
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|