3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-12-22 18:52:41 +01:00
ergo/irc/client.go
2014-02-14 19:35:25 -08:00

249 lines
4.3 KiB
Go

package irc
import (
"fmt"
"log"
"net"
"sync"
"time"
)
type Client struct {
atime time.Time
away bool
awayMessage string
channels ChannelSet
ctime time.Time
destroyed bool
hostname string
idleTimer *time.Timer
invisible bool
loginTimer *time.Timer
mutex *sync.Mutex
nick string
operator bool
phase Phase
quitTimer *time.Timer
realname string
replies chan Reply
server *Server
socket *Socket
username string
}
func NewClient(server *Server, conn net.Conn) *Client {
now := time.Now()
client := &Client{
atime: now,
channels: make(ChannelSet),
ctime: now,
hostname: AddrLookupHostname(conn.RemoteAddr()),
phase: server.InitPhase(),
replies: make(chan Reply),
server: server,
socket: NewSocket(conn),
mutex: &sync.Mutex{},
}
client.loginTimer = time.AfterFunc(LOGIN_TIMEOUT, client.Destroy)
go client.readCommands()
go client.writeReplies()
return client
}
func (client *Client) Touch() {
if client.IsDestroyed() {
return
}
client.atime = time.Now()
if client.quitTimer != nil {
client.quitTimer.Stop()
}
if client.idleTimer == nil {
client.idleTimer = time.AfterFunc(IDLE_TIMEOUT, client.Idle)
} else {
client.idleTimer.Reset(IDLE_TIMEOUT)
}
}
func (client *Client) Idle() {
if client.quitTimer == nil {
client.quitTimer = time.AfterFunc(QUIT_TIMEOUT, client.ConnectionTimeout)
} else {
client.quitTimer.Reset(QUIT_TIMEOUT)
}
client.Reply(RplPing(client.server, client))
}
func (client *Client) ConnectionTimeout() {
msg := &QuitCommand{
message: "connection timeout",
}
msg.SetClient(client)
client.server.Command(msg)
}
func (client *Client) ConnectionClosed() {
if client.IsDestroyed() {
return
}
msg := &QuitCommand{
message: "connection closed",
}
msg.SetClient(client)
client.server.Command(msg)
}
func (c *Client) readCommands() {
for line := range c.socket.Read() {
m, err := ParseCommand(line)
if err != nil {
switch err {
case NotEnoughArgsError:
c.Reply(ErrNeedMoreParams(c.server, line))
default:
c.Reply(ErrUnknownCommand(c.server, line))
}
continue
}
m.SetClient(c)
c.server.Command(m)
}
if c.phase == Normal {
c.ConnectionClosed()
} else {
c.Destroy()
}
}
func (client *Client) writeReplies() {
for reply := range client.replies {
if client.IsDestroyed() {
if DEBUG_CLIENT {
log.Printf("%s ← %s dropped", client, reply)
}
continue
}
if DEBUG_CLIENT {
log.Printf("%s ← %s", client, reply)
}
if client.socket.Write(reply.Format(client)) != nil {
break
}
}
}
func (client *Client) IsDestroyed() bool {
client.mutex.Lock()
defer client.mutex.Unlock()
return client.destroyed
}
func (client *Client) Destroy() {
if client.IsDestroyed() {
return
}
if DEBUG_CLIENT {
log.Printf("%s destroying", client)
}
client.mutex.Lock()
client.destroyed = true
client.socket.Close()
if client.idleTimer != nil {
client.idleTimer.Stop()
}
if client.quitTimer != nil {
client.quitTimer.Stop()
}
client.channels = make(ChannelSet) // clear channel list
client.server.clients.Remove(client)
client.mutex.Unlock()
if DEBUG_CLIENT {
log.Printf("%s destroyed", client)
}
}
func (client *Client) Reply(reply Reply) {
if client.replies == nil {
if DEBUG_CLIENT {
log.Printf("%s dropped %s", client, reply)
}
return
}
client.replies <- reply
}
func (client *Client) HasNick() bool {
return client.nick != ""
}
func (client *Client) HasUsername() bool {
return client.username != ""
}
func (client *Client) InterestedClients() ClientSet {
clients := make(ClientSet)
for channel := range client.channels {
for member := range channel.members {
clients.Add(member)
}
}
return clients
}
// <mode>
func (c *Client) ModeString() (str string) {
if c.invisible {
str += Invisible.String()
}
if c.operator {
str += Operator.String()
}
if len(str) > 0 {
str = "+" + str
}
return
}
func (c *Client) UserHost() string {
username := "*"
if c.HasUsername() {
username = c.username
}
return fmt.Sprintf("%s!%s@%s", c.Nick(), username, c.hostname)
}
func (c *Client) Nick() string {
if c.HasNick() {
return c.nick
}
return "*"
}
func (c *Client) Id() string {
return c.UserHost()
}
func (c *Client) String() string {
return c.UserHost()
}