forked from External/ergo
71 lines
1.2 KiB
Go
71 lines
1.2 KiB
Go
package irc
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
type Client struct {
|
|
conn net.Conn
|
|
send chan<- string
|
|
recv <-chan string
|
|
username string
|
|
realname string
|
|
nick string
|
|
registered bool
|
|
invisible bool
|
|
}
|
|
|
|
func NewClient(conn net.Conn) *Client {
|
|
client := new(Client)
|
|
client.conn = conn
|
|
client.send = StringWriteChan(conn)
|
|
client.recv = StringReadChan(conn)
|
|
return client
|
|
}
|
|
|
|
// Adapt `chan string` to a `chan Message`.
|
|
func (c *Client) Communicate(server chan<- *ClientMessage) {
|
|
for str := range c.recv {
|
|
m := ParseMessage(str)
|
|
if m != nil {
|
|
server <- &ClientMessage{c, m}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *Client) Nick() string {
|
|
if c.nick != "" {
|
|
return c.nick
|
|
}
|
|
return "<guest>"
|
|
}
|
|
|
|
func (c *Client) UModeString() string {
|
|
if c.invisible {
|
|
return "+i"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (c *Client) HasNick() bool {
|
|
return c.nick != ""
|
|
}
|
|
|
|
func (c *Client) HasUser() bool {
|
|
return c.username != ""
|
|
}
|
|
|
|
func (c *Client) Hostname() string {
|
|
addr := c.conn.RemoteAddr().String()
|
|
index := strings.LastIndex(addr, ":")
|
|
if index != -1 {
|
|
return addr[0:index]
|
|
}
|
|
return addr
|
|
}
|
|
|
|
func (c *Client) UserHost() string {
|
|
return fmt.Sprintf("%s!%s@%s", c.nick, c.username, c.Hostname())
|
|
}
|