117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package irc
|
|
|
|
import "github.com/sirupsen/logrus"
|
|
import "github.com/spf13/viper"
|
|
import "net"
|
|
|
|
type ircd struct {
|
|
Running bool
|
|
serverName string
|
|
log *logrus.Logger
|
|
netName string
|
|
ircPort uint16
|
|
httpPort uint16
|
|
ircListener net.Listener
|
|
ircClients map[*ircClient]bool
|
|
newClients chan *ircClient
|
|
deadClients chan *ircClient
|
|
messageQueue chan *ircMessage
|
|
c *viper.Viper
|
|
}
|
|
|
|
// FIXME pull these from the config
|
|
const (
|
|
CONN_HOST = "localhost"
|
|
CONN_PORT = "6667"
|
|
CONN_TYPE = "tcp"
|
|
)
|
|
|
|
func New(config *viper.Viper) *ircd {
|
|
s := new(ircd)
|
|
s.Running = true
|
|
s.ircClients = make(map[*ircClient]bool)
|
|
s.c = config
|
|
s.serverName = s.c.GetString("myhostname")
|
|
return s
|
|
}
|
|
|
|
func (s *ircd) SetLogger(logger *logrus.Logger) {
|
|
s.log = logger
|
|
}
|
|
|
|
func (s *ircd) SetServerName(name string) {
|
|
s.serverName = name
|
|
}
|
|
|
|
func (s *ircd) Start() {
|
|
s.log.Infof(
|
|
"sircd version %s (%s) built %s by %s starting.",
|
|
s.c.GetString("version"),
|
|
s.c.GetString("buildarch"),
|
|
s.c.GetString("buildtime"),
|
|
s.c.GetString("builduser"),
|
|
)
|
|
s.newClients = make(chan *ircClient, 128)
|
|
s.deadClients = make(chan *ircClient, 128)
|
|
s.messageQueue = make(chan *ircMessage, 128)
|
|
|
|
var err error
|
|
s.ircListener, err = net.Listen(CONN_TYPE, CONN_HOST+":"+CONN_PORT)
|
|
if err != nil {
|
|
s.log.Fatalln("Error listening:", err.Error())
|
|
}
|
|
s.log.Println("Listening for irc proto on " + CONN_HOST + ":" + CONN_PORT)
|
|
|
|
go func() {
|
|
conn, err := s.ircListener.Accept()
|
|
if err != nil {
|
|
s.log.Panicf("Error accepting: ", err.Error())
|
|
}
|
|
s.newClients <- newIrcClient(conn, s.log, s.messageQueue, s)
|
|
}()
|
|
|
|
// FIXME have this do 'go s.handlemessages()' and process an input channel
|
|
// of messages from all clients in its own goroutine, and then call
|
|
// directly into the main loop here and only do client i/o in this
|
|
// goroutine
|
|
go s.processMessages()
|
|
s.mainLoop()
|
|
}
|
|
|
|
func (s *ircd) processMessages() {
|
|
for {
|
|
message := <-s.messageQueue
|
|
s.processIRCMessage(message)
|
|
}
|
|
}
|
|
|
|
func (s *ircd) mainLoop() {
|
|
for {
|
|
select {
|
|
case client := <-s.newClients:
|
|
s.ircClients[client] = true
|
|
s.log.Println("new irc client: %s", client.Id)
|
|
go func() {
|
|
buf := make([]byte, 1024*1024)
|
|
for {
|
|
nbyte, err := client.conn.Read(buf)
|
|
if err != nil {
|
|
//this will surface it in the deadClients channel
|
|
//reader below
|
|
client.Kill()
|
|
break
|
|
} else {
|
|
fragment := make([]byte, nbyte)
|
|
copy(fragment, buf[:nbyte])
|
|
// this will populate the message channel
|
|
client.AppendInputBuffer(fragment)
|
|
}
|
|
}
|
|
}()
|
|
case deadClient := <-s.deadClients:
|
|
delete(s.ircClients, deadClient)
|
|
deadClient.CleanupAndClose()
|
|
}
|
|
}
|
|
}
|