Add timestamp for messages coming from user connections

3 files changed, 7 insertions, 3 deletions

diff --git a/server/message.go b/server/message.go
index 11644dc..cd5add6 100644
--- a/server/message.go
+++ b/server/message.go
@@ -30,7 +30,11 @@ func (s *Server) SendBroadcast(msg string) {
 	}
 }
 
-func (s *Server) handleMessage(sender net.Conn, msg core.Message) error {
+func (s *Server) handleMessage(sender net.Conn, connType core.ConnType, msg core.Message) error {
+	if connType == core.ConnTypeUser {
+		msg.Timestamp = time.Now()
+	}
+
 	log.Println("message:", msg.Source, "->", msg.Target, msg.Content)
 
 	addr, err := core.ReadAddr(msg.Target)
diff --git a/server/remote.go b/server/remote.go
index 5d86da2..6f9bd66 100644
--- a/server/remote.go
+++ b/server/remote.go
@@ -126,7 +126,7 @@ func (s *Server) readRemoteInput(conn net.Conn) error {
 func (s *Server) handleRemotePayload(sender net.Conn, payload any) error {
 	switch v := payload.(type) {
 	case core.Message:
-		return s.handleMessage(sender, v)
+		return s.handleMessage(sender, core.ConnTypeServer, v)
 	default:
 		return core.ErrUnexpectedPayloadType
 	}
diff --git a/server/user.go b/server/user.go
index 4f78b6a..002ed89 100644
--- a/server/user.go
+++ b/server/user.go
@@ -115,7 +115,7 @@ func (s *Server) readUserInput(user *User, conn net.Conn) error {
 func (s *Server) handleUserPayload(user *User, sender net.Conn, payload any) error {
 	switch v := payload.(type) {
 	case core.Message:
-		return s.handleMessage(sender, v)
+		return s.handleMessage(sender, core.ConnTypeUser, v)
 	case core.Usermode:
 		return s.handleUsermode(user, sender, v)
 	default: