From 3ccc1bb4e88ac90f3f1e3cb42c151ff97522f28a Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Thu, 6 Feb 2020 12:18:37 +0100
Subject: [PATCH] Handle PING

---
 server.go | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/server.go b/server.go
index 24cf3d4..d96d18e 100644
--- a/server.go
+++ b/server.go
@@ -63,6 +63,23 @@ func (c *conn) WriteMessage(msg *irc.Message) error {
 	return c.irc.WriteMessage(msg)
 }
 
+func (c *conn) handleMessage(msg *irc.Message) error {
+	switch msg.Command {
+	case "PING":
+		// TODO: handle params
+		return c.WriteMessage(&irc.Message{
+			Command: "PONG",
+			Params: []string{c.srv.Hostname},
+		})
+	default:
+		if c.registered {
+			return c.handleMessageRegistered(msg)
+		} else {
+			return c.handleMessageUnregistered(msg)
+		}
+	}
+}
+
 func (c *conn) handleMessageUnregistered(msg *irc.Message) error {
 	switch msg.Command {
 	case "NICK":
@@ -133,7 +150,7 @@ func (c *conn) register() error {
 	return nil
 }
 
-func (c *conn) handleMessage(msg *irc.Message) error {
+func (c *conn) handleMessageRegistered(msg *irc.Message) error {
 	switch msg.Command {
 	case "NICK", "USER":
 		return ircError{&irc.Message{
@@ -170,11 +187,7 @@ func (s *Server) handleConn(netConn net.Conn) error {
 		}
 		log.Println(msg)
 
-		if c.registered {
-			err = c.handleMessage(msg)
-		} else {
-			err = c.handleMessageUnregistered(msg)
-		}
+		err = c.handleMessage(msg)
 		if ircErr, ok := err.(ircError); ok {
 			ircErr.Message.Prefix = s.prefix()
 			if err := c.WriteMessage(ircErr.Message); err != nil {