forked from RecursiveForest/goty
/
goty.go
87 lines (79 loc) · 1.73 KB
/
goty.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package goty
import (
"bufio"
"fmt"
"net"
"os"
"strings"
)
type IRCConn struct {
Sock *net.TCPConn
Read, Write chan string
Closed chan bool
}
func Dial(server, nick string) (*IRCConn, error) {
read := make(chan string, 1000)
write := make(chan string, 1000)
closed := make(chan bool)
con := &IRCConn{nil, read, write, closed}
err := con.Connect(server, nick)
return con, err
}
func (con *IRCConn) Connect(server, nick string) error {
if raddr, err := net.ResolveTCPAddr("tcp", server); err != nil {
return err
} else {
if c, err := net.DialTCP("tcp", nil, raddr); err != nil {
return err
} else {
con.Sock = c
r := bufio.NewReader(con.Sock)
w := bufio.NewWriter(con.Sock)
go func() {
L:
for {
select {
case <-con.Closed:
fmt.Fprintf(os.Stderr, "goty: read closed\n")
break L
default:
if str, err := r.ReadString(byte('\n')); err != nil {
fmt.Fprintf(os.Stderr, "goty: read: %s\n", err)
break L
} else {
if strings.HasPrefix(str, "PING") {
con.Write <- "PONG" + str[4:len(str)-2]
} else {
con.Read <- str[0 : len(str)-2]
}
}
}
}
fmt.Println("done with send goroutine")
}()
go func() {
for {
str, ok := <-con.Write
if ok == false {
fmt.Fprintf(os.Stderr, "goty: write closed\n")
break
}
if _, err := w.WriteString(str + "\r\n"); err != nil {
fmt.Fprintf(os.Stderr, "goty: write: %s\n", err)
break
}
w.Flush()
}
}()
con.Write <- "NICK " + nick
con.Write <- "USER bot * * :..."
}
}
return nil
}
func (con *IRCConn) Close() error {
con.Closed <- true
close(con.Write)
close(con.Read)
return con.Sock.Close()
}