sircd/irc/server.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()
		}
	}
}