From 1e18fc3c5a4bf9ad22d84ecf71c5be395f514938 Mon Sep 17 00:00:00 2001 From: Harry Jeffery Date: Sat, 24 Aug 2013 07:48:28 +0100 Subject: [PATCH] Separated code more --- Makefile | 2 +- client.go | 249 ++++++++++++++++++++++++ rosella.go | 560 +---------------------------------------------------- server.go | 308 +++++++++++++++++++++++++++++ 4 files changed, 564 insertions(+), 555 deletions(-) create mode 100644 client.go create mode 100644 server.go diff --git a/Makefile b/Makefile index 3fdb6cd..d438b68 100644 --- a/Makefile +++ b/Makefile @@ -1,5 +1,5 @@ rosella: main.go rosella.go - go build -o ircd main.go rosella.go + go build -o ircd main.go rosella.go server.go client.go clean: rm ircd diff --git a/client.go b/client.go new file mode 100644 index 0000000..5ac1baa --- /dev/null +++ b/client.go @@ -0,0 +1,249 @@ +package main + +import ( + "bytes" + "fmt" + "io" + "log" + "sort" + "strings" + "time" +) + +func (c *Client) setNick(nick string) { + if c.nick != "" { + delete(c.server.clientMap, c.nick) + for _, channel := range c.channelMap { + delete(channel.clientMap, c.nick) + } + } + + //Set up new nick + oldNick := c.nick + c.nick = nick + c.server.clientMap[c.nick] = c + + clients := make([]string, 0, 100) + + for _, channel := range c.channelMap { + channel.clientMap[c.nick] = c + + //Collect list of client nicks who can see us + for client := range channel.clientMap { + clients = append(clients, client) + } + } + + //By sorting the nicks and skipping duplicates we send each client one message + sort.Strings(clients) + prevNick := "" + for _, nick := range clients { + if nick == prevNick { + continue + } + prevNick = nick + + client, exists := c.server.clientMap[nick] + if exists { + client.reply(rplNickChange, oldNick, c.nick) + } + } +} + +func (c *Client) joinChannel(channelName string) { + channel, exists := c.server.channelMap[channelName] + if exists == false { + channel = &Channel{name: channelName, + topic: "", + clientMap: make(map[string]*Client)} + c.server.channelMap[channelName] = channel + } + + channel.clientMap[c.nick] = c + c.channelMap[channelName] = channel + + for _, client := range channel.clientMap { + client.reply(rplJoin, c.nick, channelName) + } + + if channel.topic != "" { + c.reply(rplTopic, channelName, channel.topic) + } else { + c.reply(rplNoTopic, channelName) + } + + nicks := make([]string, 0, 100) + for nick := range channel.clientMap { + nicks = append(nicks, nick) + } + + c.reply(rplNames, channelName, strings.Join(nicks, " ")) +} + +func (c *Client) partChannel(channelName string) { + channel, exists := c.server.channelMap[channelName] + if exists == false { + return + } + + //Notify clients of the part + for _, client := range channel.clientMap { + client.reply(rplPart, c.nick, channelName) + } + + delete(channel.clientMap, c.nick) + delete(c.channelMap, channelName) +} + +func (c *Client) disconnect() { + c.connected = false + c.signalChan <- signalStop +} + +//Send a reply to a user with the code specified +func (c *Client) reply(code int, args ...string) { + if c.connected == false { + return + } + + switch code { + case rplWelcome: + c.outputChan <- fmt.Sprintf(":%s 001 %s :Welcome to %s", c.server.name, c.nick, c.server.name) + case rplJoin: + c.outputChan <- fmt.Sprintf(":%s JOIN %s", args[0], args[1]) + case rplPart: + c.outputChan <- fmt.Sprintf(":%s PART %s", args[0], args[1]) + case rplTopic: + c.outputChan <- fmt.Sprintf(":%s 332 %s %s :%s", c.server.name, c.nick, args[0], args[1]) + case rplNoTopic: + c.outputChan <- fmt.Sprintf(":%s 331 %s %s :No topic is set", c.server.name, c.nick, args[0]) + case rplNames: + //TODO: break long lists up into multiple messages + c.outputChan <- fmt.Sprintf(":%s 353 %s = %s :%s", c.server.name, c.nick, args[0], args[1]) + c.outputChan <- fmt.Sprintf(":%s 366 %s", c.server.name, c.nick) + case rplNickChange: + c.outputChan <- fmt.Sprintf(":%s NICK %s", args[0], args[1]) + case rplKill: + c.outputChan <- fmt.Sprintf(":%s KILL %s A %s", c.server.name, c.nick, args[0]) + case rplMsg: + c.outputChan <- fmt.Sprintf(":%s PRIVMSG %s %s", args[0], args[1], args[2]) + case rplList: + c.outputChan <- fmt.Sprintf(":%s 321 %s", c.server.name, c.nick) + for _, listItem := range args { + c.outputChan <- fmt.Sprintf(":%s 322 %s %s", c.server.name, c.nick, listItem) + } + c.outputChan <- fmt.Sprintf(":%s 323 %s", c.server.name, c.nick) + case rplOper: + c.outputChan <- fmt.Sprintf(":%s 381 %s :You are now an operator", c.server.name, c.nick) + case errMoreArgs: + c.outputChan <- fmt.Sprintf(":%s 461 %s :Not enough params", c.server.name, c.nick) + case errNoNick: + c.outputChan <- fmt.Sprintf(":%s 431 %s :No nickname given", c.server.name, c.nick) + case errInvalidNick: + c.outputChan <- fmt.Sprintf(":%s 432 %s %s :Erronenous nickname", c.server.name, c.nick, args[0]) + case errNickInUse: + c.outputChan <- fmt.Sprintf(":%s 433 %s %s :Nick already in use", c.server.name, c.nick, args[0]) + case errAlreadyReg: + c.outputChan <- fmt.Sprintf(":%s 462 :You need a valid nick first", c.server.name) + case errNoSuchNick: + c.outputChan <- fmt.Sprintf(":%s 401 %s %s :No such nick/channel", c.server.name, c.nick, args[0]) + case errUnknownCommand: + c.outputChan <- fmt.Sprintf(":%s 421 %s %s :Unknown command", c.server.name, c.nick, args[0]) + case errNotReg: + c.outputChan <- fmt.Sprintf(":%s 451 :You have not registered", c.server.name) + case errPassword: + c.outputChan <- fmt.Sprintf(":%s 464 %s :Error, password incorrect", c.server.name, c.nick) + case errNoPriv: + c.outputChan <- fmt.Sprintf(":%s 481 %s :Permission denied", c.server.name, c.nick) + } +} + +func (c *Client) clientThread() { + defer c.connection.Close() + + readSignalChan := make(chan int, 3) + writeSignalChan := make(chan int, 3) + writeChan := make(chan string, 100) + + go c.readThread(readSignalChan) + go c.writeThread(writeSignalChan, writeChan) + + defer func() { + //Part from all channels + for channelName := range c.channelMap { + c.partChannel(channelName) + } + + delete(c.server.clientMap, c.nick) + }() + + for { + select { + case signal := <-c.signalChan: + if signal == signalStop { + readSignalChan <- signalStop + writeSignalChan <- signalStop + return + } + case line := <-c.outputChan: + select { + case writeChan <- line: + //It worked + default: + log.Printf("Dropped a line for client: %q", c.nick) + //Do nothing, dropping the line + } + } + } + +} + +func (c *Client) readThread(signalChan chan int) { + for { + select { + case signal := <-signalChan: + if signal == signalStop { + return + } + default: + c.connection.SetReadDeadline(time.Now().Add(time.Second * 3)) + buf := make([]byte, 512) + ln, err := c.connection.Read(buf) + if err != nil { + if err == io.EOF { + c.disconnect() + return + } + continue + } + + rawLines := buf[:ln] + lines := bytes.Split(rawLines, []byte("\r\n")) + for _, line := range lines { + if len(line) > 0 { + c.server.eventChan <- Event{client: c, input: string(line)} + } + } + } + } +} + +func (c *Client) writeThread(signalChan chan int, outputChan chan string) { + for { + select { + case signal := <-signalChan: + if signal == signalStop { + return + } + case output := <-outputChan: + line := []byte(fmt.Sprintf("%s\r\n", output)) + + c.connection.SetWriteDeadline(time.Now().Add(time.Second * 30)) + _, err := c.connection.Write(line) + if err != nil { + c.disconnect() + return + } + } + } +} diff --git a/rosella.go b/rosella.go index 460ca02..215f067 100644 --- a/rosella.go +++ b/rosella.go @@ -1,17 +1,6 @@ package main -import ( - "bytes" - "crypto/sha1" - "fmt" - "io" - "log" - "net" - "regexp" - "sort" - "strings" - "time" -) +import "net" type Server struct { eventChan chan Event @@ -34,17 +23,17 @@ type Client struct { channelMap map[string]*Channel } +type Event struct { + client *Client + input string +} + type Channel struct { name string topic string clientMap map[string]*Client } -type Event struct { - client *Client - input string -} - const ( signalStop int = iota ) @@ -72,540 +61,3 @@ const ( errPassword errNoPriv ) - -var ( - nickRegexp = regexp.MustCompile(`^[a-zA-Z\[\]_^{|}][a-zA-Z0-9\[\]_^{|}]*$`) - channelRegexp = regexp.MustCompile(`^#[a-z0-9_\-]+$`) -) - -func NewServer() *Server { - return &Server{eventChan: make(chan Event), - name: "rosella", - clientMap: make(map[string]*Client), - channelMap: make(map[string]*Channel), - operatorMap: make(map[string]string)} -} - -func (s *Server) Run() { - go func() { - for { - event := <-s.eventChan - defer func() { - if err := recover(); err != nil { - log.Printf("Recovered from panic()") - log.Printf("%s sent %q", event.client.nick, event.input) - log.Println(err) - } - }() - - s.handleEvent(<-s.eventChan) - } - }() -} - -func (s *Server) HandleConnection(conn net.Conn) { - - client := &Client{server: s, - connection: conn, - outputChan: make(chan string), - signalChan: make(chan int, 3), - channelMap: make(map[string]*Channel), - connected: true} - - go client.clientThread() -} - -func (s *Server) handleEvent(e Event) { - fields := strings.Fields(e.input) - - if len(fields) < 1 { - return - } - - if strings.HasPrefix(fields[0], ":") { - fields = fields[1:] - } - - command := strings.ToUpper(fields[0]) - args := fields[1:] - - switch { - case command == "NICK": - if len(args) < 1 { - e.client.reply(errNoNick) - return - } - - newNick := args[0] - - //Check newNick is of valid formatting (regex) - if nickRegexp.MatchString(newNick) == false { - e.client.reply(errInvalidNick, newNick) - return - } - - if _, exists := s.clientMap[newNick]; exists { - e.client.reply(errNickInUse, newNick) - return - } - - //Protect the server name from being used - if newNick == s.name { - e.client.reply(errNickInUse, newNick) - return - } - - e.client.setNick(newNick) - - case command == "USER": - if e.client.nick == "" { - e.client.reply(rplKill, "Your nickname is already being used") - e.client.disconnect() - } else { - e.client.reply(rplWelcome) - e.client.registered = true - } - - case command == "JOIN": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) < 1 { - e.client.reply(errMoreArgs) - return - } - - if args[0] == "0" { - //Quit all channels - for channel := range e.client.channelMap { - e.client.partChannel(channel) - } - return - } - - channels := strings.Split(args[0], ",") - for _, channel := range channels { - //Join the channel if it's valid - if channelRegexp.Match([]byte(channel)) { - e.client.joinChannel(channel) - } - } - - case command == "PART": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) < 1 { - e.client.reply(errMoreArgs) - return - } - - channels := strings.Split(args[0], ",") - for _, channel := range channels { - //Part the channel if it's valid - if channelRegexp.Match([]byte(channel)) { - e.client.partChannel(channel) - } - } - - case command == "PRIVMSG": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) < 2 { - e.client.reply(errMoreArgs) - return - } - - message := strings.Join(args[1:], " ") - - channel, chanExists := s.channelMap[args[0]] - client, clientExists := s.clientMap[args[0]] - - if chanExists { - for _, c := range channel.clientMap { - if c != e.client { - c.reply(rplMsg, e.client.nick, args[0], message) - } - } - } else if clientExists { - client.reply(rplMsg, e.client.nick, client.nick, message) - } else { - e.client.reply(errNoSuchNick, args[0]) - } - - case command == "QUIT": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - e.client.disconnect() - - case command == "TOPIC": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) < 1 { - e.client.reply(errMoreArgs) - return - } - - channel, exists := s.channelMap[args[0]] - if exists == false { - e.client.reply(errNoSuchNick, args[0]) - return - } - - channelName := args[0] - - if len(args) == 1 { - e.client.reply(rplTopic, channelName, channel.topic) - return - } - - if args[1] == ":" { - channel.topic = "" - for _, client := range channel.clientMap { - client.reply(rplNoTopic, channelName) - } - } else { - topic := strings.Join(args[1:], " ") - topic = strings.TrimPrefix(topic, ":") - channel.topic = topic - - for _, client := range channel.clientMap { - client.reply(rplTopic, channelName, channel.topic) - } - } - - case command == "LIST": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) == 0 { - chanList := make([]string, 0, len(s.channelMap)) - - for channelName, channel := range s.channelMap { - listItem := fmt.Sprintf("%s %d :%s", channelName, len(channel.clientMap), channel.topic) - chanList = append(chanList, listItem) - } - - e.client.reply(rplList, chanList...) - - } else { - channels := strings.Split(args[0], ",") - chanList := make([]string, 0, len(channels)) - - for _, channelName := range channels { - if channel, exists := s.channelMap[channelName]; exists { - listItem := fmt.Sprintf("%s %d :%s", channelName, len(channel.clientMap), channel.topic) - chanList = append(chanList, listItem) - } - } - - e.client.reply(rplList, chanList...) - } - case command == "OPER": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if len(args) < 2 { - e.client.reply(errMoreArgs) - return - } - - username := args[0] - password := args[1] - - if hashedPassword, exists := s.operatorMap[username]; exists { - h := sha1.New() - io.WriteString(h, password) - pass := fmt.Sprintf("%x", h.Sum(nil)) - if hashedPassword == pass { - e.client.operator = true - e.client.reply(rplOper) - return - } - } - e.client.reply(errPassword) - - case command == "KILL": - if e.client.registered == false { - e.client.reply(errNotReg) - return - } - - if e.client.operator == false { - e.client.reply(errNoPriv) - return - } - - if len(args) < 1 { - e.client.reply(errMoreArgs) - return - } - - nick := args[0] - - if client, exists := s.clientMap[nick]; exists { - client.reply(rplKill, "An operator has disconnected you.") - client.disconnect() - } else { - e.client.reply(errNoSuchNick, nick) - } - - default: - e.client.reply(errUnknownCommand, command) - } -} - -func (c *Client) joinChannel(channelName string) { - channel, exists := c.server.channelMap[channelName] - if exists == false { - channel = &Channel{name: channelName, - topic: "", - clientMap: make(map[string]*Client)} - c.server.channelMap[channelName] = channel - } - - channel.clientMap[c.nick] = c - c.channelMap[channelName] = channel - - for _, client := range channel.clientMap { - client.reply(rplJoin, c.nick, channelName) - } - - if channel.topic != "" { - c.reply(rplTopic, channelName, channel.topic) - } else { - c.reply(rplNoTopic, channelName) - } - - nicks := make([]string, 0, 100) - for nick := range channel.clientMap { - nicks = append(nicks, nick) - } - - c.reply(rplNames, channelName, strings.Join(nicks, " ")) -} - -func (c *Client) partChannel(channelName string) { - channel, exists := c.server.channelMap[channelName] - if exists == false { - return - } - - //Notify clients of the part - for _, client := range channel.clientMap { - client.reply(rplPart, c.nick, channelName) - } - - delete(channel.clientMap, c.nick) - delete(c.channelMap, channelName) -} - -func (c *Client) clientThread() { - defer c.connection.Close() - - readSignalChan := make(chan int, 3) - writeSignalChan := make(chan int, 3) - writeChan := make(chan string, 100) - - go c.readThread(readSignalChan) - go c.writeThread(writeSignalChan, writeChan) - - defer func() { - //Part from all channels - for channelName := range c.channelMap { - c.partChannel(channelName) - } - - delete(c.server.clientMap, c.nick) - }() - - for { - select { - case signal := <-c.signalChan: - if signal == signalStop { - readSignalChan <- signalStop - writeSignalChan <- signalStop - return - } - case line := <-c.outputChan: - select { - case writeChan <- line: - //It worked - default: - log.Printf("Dropped a line for client: %q", c.nick) - //Do nothing, dropping the line - } - } - } - -} - -func (c *Client) readThread(signalChan chan int) { - for { - select { - case signal := <-signalChan: - if signal == signalStop { - return - } - default: - c.connection.SetReadDeadline(time.Now().Add(time.Second * 3)) - buf := make([]byte, 512) - ln, err := c.connection.Read(buf) - if err != nil { - if err == io.EOF { - c.disconnect() - return - } - continue - } - - rawLines := buf[:ln] - lines := bytes.Split(rawLines, []byte("\r\n")) - for _, line := range lines { - if len(line) > 0 { - c.server.eventChan <- Event{client: c, input: string(line)} - } - } - } - } -} - -func (c *Client) writeThread(signalChan chan int, outputChan chan string) { - for { - select { - case signal := <-signalChan: - if signal == signalStop { - return - } - case output := <-outputChan: - line := []byte(fmt.Sprintf("%s\r\n", output)) - - c.connection.SetWriteDeadline(time.Now().Add(time.Second * 30)) - _, err := c.connection.Write(line) - if err != nil { - c.disconnect() - return - } - } - } -} - -func (c *Client) disconnect() { - c.connected = false - c.signalChan <- signalStop -} - -//Send a reply to a user with the code specified -func (c *Client) reply(code int, args ...string) { - if c.connected == false { - return - } - - switch code { - case rplWelcome: - c.outputChan <- fmt.Sprintf(":%s 001 %s :Welcome to %s", c.server.name, c.nick, c.server.name) - case rplJoin: - c.outputChan <- fmt.Sprintf(":%s JOIN %s", args[0], args[1]) - case rplPart: - c.outputChan <- fmt.Sprintf(":%s PART %s", args[0], args[1]) - case rplTopic: - c.outputChan <- fmt.Sprintf(":%s 332 %s %s :%s", c.server.name, c.nick, args[0], args[1]) - case rplNoTopic: - c.outputChan <- fmt.Sprintf(":%s 331 %s %s :No topic is set", c.server.name, c.nick, args[0]) - case rplNames: - //TODO: break long lists up into multiple messages - c.outputChan <- fmt.Sprintf(":%s 353 %s = %s :%s", c.server.name, c.nick, args[0], args[1]) - c.outputChan <- fmt.Sprintf(":%s 366 %s", c.server.name, c.nick) - case rplNickChange: - c.outputChan <- fmt.Sprintf(":%s NICK %s", args[0], args[1]) - case rplKill: - c.outputChan <- fmt.Sprintf(":%s KILL %s A %s", c.server.name, c.nick, args[0]) - case rplMsg: - c.outputChan <- fmt.Sprintf(":%s PRIVMSG %s %s", args[0], args[1], args[2]) - case rplList: - c.outputChan <- fmt.Sprintf(":%s 321 %s", c.server.name, c.nick) - for _, listItem := range args { - c.outputChan <- fmt.Sprintf(":%s 322 %s %s", c.server.name, c.nick, listItem) - } - c.outputChan <- fmt.Sprintf(":%s 323 %s", c.server.name, c.nick) - case rplOper: - c.outputChan <- fmt.Sprintf(":%s 381 %s :You are now an operator", c.server.name, c.nick) - case errMoreArgs: - c.outputChan <- fmt.Sprintf(":%s 461 %s :Not enough params", c.server.name, c.nick) - case errNoNick: - c.outputChan <- fmt.Sprintf(":%s 431 %s :No nickname given", c.server.name, c.nick) - case errInvalidNick: - c.outputChan <- fmt.Sprintf(":%s 432 %s %s :Erronenous nickname", c.server.name, c.nick, args[0]) - case errNickInUse: - c.outputChan <- fmt.Sprintf(":%s 433 %s %s :Nick already in use", c.server.name, c.nick, args[0]) - case errAlreadyReg: - c.outputChan <- fmt.Sprintf(":%s 462 :You need a valid nick first", c.server.name) - case errNoSuchNick: - c.outputChan <- fmt.Sprintf(":%s 401 %s %s :No such nick/channel", c.server.name, c.nick, args[0]) - case errUnknownCommand: - c.outputChan <- fmt.Sprintf(":%s 421 %s %s :Unknown command", c.server.name, c.nick, args[0]) - case errNotReg: - c.outputChan <- fmt.Sprintf(":%s 451 :You have not registered", c.server.name) - case errPassword: - c.outputChan <- fmt.Sprintf(":%s 464 %s :Error, password incorrect", c.server.name, c.nick) - case errNoPriv: - c.outputChan <- fmt.Sprintf(":%s 481 %s :Permission denied", c.server.name, c.nick) - } -} - -func (c *Client) setNick(nick string) { - if c.nick != "" { - delete(c.server.clientMap, c.nick) - for _, channel := range c.channelMap { - delete(channel.clientMap, c.nick) - } - } - - //Set up new nick - oldNick := c.nick - c.nick = nick - c.server.clientMap[c.nick] = c - - clients := make([]string, 0, 100) - - for _, channel := range c.channelMap { - channel.clientMap[c.nick] = c - - //Collect list of client nicks who can see us - for client := range channel.clientMap { - clients = append(clients, client) - } - } - - //By sorting the nicks and skipping duplicates we send each client one message - sort.Strings(clients) - prevNick := "" - for _, nick := range clients { - if nick == prevNick { - continue - } - prevNick = nick - - client, exists := c.server.clientMap[nick] - if exists { - client.reply(rplNickChange, oldNick, c.nick) - } - } -} diff --git a/server.go b/server.go new file mode 100644 index 0000000..811c3ea --- /dev/null +++ b/server.go @@ -0,0 +1,308 @@ +package main + +import ( + "crypto/sha1" + "fmt" + "io" + "log" + "net" + "regexp" + "strings" +) + +var ( + nickRegexp = regexp.MustCompile(`^[a-zA-Z\[\]_^{|}][a-zA-Z0-9\[\]_^{|}]*$`) + channelRegexp = regexp.MustCompile(`^#[a-z0-9_\-]+$`) +) + +func NewServer() *Server { + return &Server{eventChan: make(chan Event), + name: "rosella", + clientMap: make(map[string]*Client), + channelMap: make(map[string]*Channel), + operatorMap: make(map[string]string)} +} + +func (s *Server) Run() { + go func() { + for event := range s.eventChan { + defer func() { + if err := recover(); err != nil { + log.Printf("Recovered from panic()") + log.Printf("%s sent %q", event.client.nick, event.input) + log.Println(err) + } + }() + + s.handleEvent(<-s.eventChan) + } + }() +} + +func (s *Server) HandleConnection(conn net.Conn) { + client := &Client{server: s, + connection: conn, + outputChan: make(chan string), + signalChan: make(chan int, 3), + channelMap: make(map[string]*Channel), + connected: true} + + go client.clientThread() +} + +func (s *Server) handleEvent(e Event) { + fields := strings.Fields(e.input) + + if len(fields) < 1 { + return + } + + if strings.HasPrefix(fields[0], ":") { + fields = fields[1:] + } + + command := strings.ToUpper(fields[0]) + args := fields[1:] + + switch { + case command == "NICK": + if len(args) < 1 { + e.client.reply(errNoNick) + return + } + + newNick := args[0] + + //Check newNick is of valid formatting (regex) + if nickRegexp.MatchString(newNick) == false { + e.client.reply(errInvalidNick, newNick) + return + } + + if _, exists := s.clientMap[newNick]; exists { + e.client.reply(errNickInUse, newNick) + return + } + + //Protect the server name from being used + if newNick == s.name { + e.client.reply(errNickInUse, newNick) + return + } + + e.client.setNick(newNick) + + case command == "USER": + if e.client.nick == "" { + e.client.reply(rplKill, "Your nickname is already being used") + e.client.disconnect() + } else { + e.client.reply(rplWelcome) + e.client.registered = true + } + + case command == "JOIN": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) < 1 { + e.client.reply(errMoreArgs) + return + } + + if args[0] == "0" { + //Quit all channels + for channel := range e.client.channelMap { + e.client.partChannel(channel) + } + return + } + + channels := strings.Split(args[0], ",") + for _, channel := range channels { + //Join the channel if it's valid + if channelRegexp.Match([]byte(channel)) { + e.client.joinChannel(channel) + } + } + + case command == "PART": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) < 1 { + e.client.reply(errMoreArgs) + return + } + + channels := strings.Split(args[0], ",") + for _, channel := range channels { + //Part the channel if it's valid + if channelRegexp.Match([]byte(channel)) { + e.client.partChannel(channel) + } + } + + case command == "PRIVMSG": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) < 2 { + e.client.reply(errMoreArgs) + return + } + + message := strings.Join(args[1:], " ") + + channel, chanExists := s.channelMap[args[0]] + client, clientExists := s.clientMap[args[0]] + + if chanExists { + for _, c := range channel.clientMap { + if c != e.client { + c.reply(rplMsg, e.client.nick, args[0], message) + } + } + } else if clientExists { + client.reply(rplMsg, e.client.nick, client.nick, message) + } else { + e.client.reply(errNoSuchNick, args[0]) + } + + case command == "QUIT": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + e.client.disconnect() + + case command == "TOPIC": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) < 1 { + e.client.reply(errMoreArgs) + return + } + + channel, exists := s.channelMap[args[0]] + if exists == false { + e.client.reply(errNoSuchNick, args[0]) + return + } + + channelName := args[0] + + if len(args) == 1 { + e.client.reply(rplTopic, channelName, channel.topic) + return + } + + if args[1] == ":" { + channel.topic = "" + for _, client := range channel.clientMap { + client.reply(rplNoTopic, channelName) + } + } else { + topic := strings.Join(args[1:], " ") + topic = strings.TrimPrefix(topic, ":") + channel.topic = topic + + for _, client := range channel.clientMap { + client.reply(rplTopic, channelName, channel.topic) + } + } + + case command == "LIST": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) == 0 { + chanList := make([]string, 0, len(s.channelMap)) + + for channelName, channel := range s.channelMap { + listItem := fmt.Sprintf("%s %d :%s", channelName, len(channel.clientMap), channel.topic) + chanList = append(chanList, listItem) + } + + e.client.reply(rplList, chanList...) + + } else { + channels := strings.Split(args[0], ",") + chanList := make([]string, 0, len(channels)) + + for _, channelName := range channels { + if channel, exists := s.channelMap[channelName]; exists { + listItem := fmt.Sprintf("%s %d :%s", channelName, len(channel.clientMap), channel.topic) + chanList = append(chanList, listItem) + } + } + + e.client.reply(rplList, chanList...) + } + case command == "OPER": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if len(args) < 2 { + e.client.reply(errMoreArgs) + return + } + + username := args[0] + password := args[1] + + if hashedPassword, exists := s.operatorMap[username]; exists { + h := sha1.New() + io.WriteString(h, password) + pass := fmt.Sprintf("%x", h.Sum(nil)) + if hashedPassword == pass { + e.client.operator = true + e.client.reply(rplOper) + return + } + } + e.client.reply(errPassword) + + case command == "KILL": + if e.client.registered == false { + e.client.reply(errNotReg) + return + } + + if e.client.operator == false { + e.client.reply(errNoPriv) + return + } + + if len(args) < 1 { + e.client.reply(errMoreArgs) + return + } + + nick := args[0] + + if client, exists := s.clientMap[nick]; exists { + client.reply(rplKill, "An operator has disconnected you.") + client.disconnect() + } else { + e.client.reply(errNoSuchNick, nick) + } + + default: + e.client.reply(errUnknownCommand, command) + } +}