Introduce basic chat functionality

I have yet to develop multi-user chats

chat-tests/cmd/serverFunc.go

@@ -6,7 +6,6 @@ import (
 	"log"
 	"net"
 	"strings"
-	"time"
 )
 
 var (
@@ -22,11 +21,6 @@ type User struct {
 	IP       string
 }
 
-type Message struct {
-	Contents string
-	Date     time.Time
-}
-
 func (u User) CreateUser(usr string, ip string) User {
 	u.Username = usr
 	u.IP = ip
@@ -49,27 +43,63 @@ func Server() {
 	}
 }
 
-func getUsername(conn net.Conn) string {
-	conn.Write([]byte("What's your name?\n\r"))
-	conn.Write([]byte("Choice: "))
+func getUsername(conn net.Conn) (s string, err error) {
+	conn.Write([]byte("What's your name?\nChoice: "))
 	name, err := bufio.NewReader(conn).ReadString('\n')
 	if err != nil {
-		log.Fatalf("Error occurred getting username: %v\n", err)
+		return "", err
 	}
 	trimmedName := strings.TrimRight(name, "\n")
-	return trimmedName
+	return trimmedName, nil
 }
 
-func getUserInput() {
+func getUserInput(conn net.Conn) (s string, err error) {
+	message, err := bufio.NewReader(conn).ReadString('\n')
+	if err != nil {
+		return "", err
+	}
 
+	return message, nil
 }
 
 func handleConn(conn net.Conn) {
 	defer conn.Close()
+
+	//////////////////////////////////
+	// Get user information
+	//////////////////////////////////
+	userName, err := getUsername(conn)
+	if err != nil {
+		log.Printf("Error occurred getting username: %v\n", err)
+		return
+	}
+	userIP := getIP(conn)
+	//////////////////////////////////
+
 	newUserTemplate := new(User)
-	newUser := newUserTemplate.CreateUser(getUsername(conn), getIP(conn))
-	fmt.Printf("Received connection from %v at %v\n", newUser.Username, newUser.IP)
-	fmt.Fprintln(conn, "Hello buddy!")
+	newUser := newUserTemplate.CreateUser(userName, userIP)
+	joinMessage := fmt.Sprintf("%v has joined the chat!", newUser.Username)
+	fmt.Println(joinMessage)
+	conn.Write([]byte(joinMessage))
+
+	//////////////////////////////////
+	for {
+		message, err := getUserInput(conn)
+		if err != nil {
+			quitMessage := fmt.Sprintf("%v has disconnected!\n", newUser.Username)
+			fmt.Print(quitMessage)
+			if _, err := conn.Write([]byte(quitMessage)); err != nil {
+				log.Printf("Error happened sending disconnect message: %v", err)
+			}
+			return
+		}
+		finalMessage := fmt.Sprintf("[%v] %v: %v", newUser.IP, newUser.Username, strings.TrimRight(message, "\n"))
+		fmt.Printf("%v\n", finalMessage)
+		if _, err := conn.Write([]byte(finalMessage)); err != nil {
+			log.Printf("Error happened sending message: %v", err)
+		}
+	}
+	//////////////////////////////////
 }
 
 func getIP(conn net.Conn) (IP string) {