package main

import (
	"bufio"         
	"chatRoom/chatRoom/module" // 消息的編碼和解碼模塊
	"fmt"          
	"io"             
	"log"            
	"net"            
	"strconv"        
	"strings"       
	"sync"           
	"time"          
)

// 定義客戶端結(jié)構(gòu)體
type client struct {
	userChannel chan string // 用戶的消息通道
	conn        net.Conn    // 網(wǎng)絡(luò)連接
	name        string      // 用戶名
	addr        string      // 客戶端地址
}

// 定義在線用戶計(jì)數(shù)器
var count int

// 定義互斥鎖
var mu sync.Mutex

// 定義在線用戶列表
var onlineList map[string]*client

// 定義消息廣播通道
var message chan string

// 初始化函數(shù)
func init() {
	onlineList = make(map[string]*client) // 初始化在線用戶列表
	message = make(chan string, 1024)     // 初始化消息廣播通道
}

// 主函數(shù)
func main() {
	fmt.Println("端口監(jiān)聽(tīng)中...")
	listener, err := net.Listen("tcp", "127.0.0.1:8080")
	if err != nil {
		log.Fatal(err) // 如果監(jiān)聽(tīng)失敗，記錄錯(cuò)誤并退出程序
	}
	defer listener.Close()
	time.Sleep(time.Second)
	fmt.Println("端口監(jiān)聽(tīng)成功")

	// 啟動(dòng)管理消息廣播的協(xié)程
	go manger()

	// 主循環(huán)，接受客戶端連接
	for {
		conn, err := listener.Accept()
		if err != nil {
			continue
		}
		go handleConnection(conn)
	}
}

// 處理客戶端連接的函數(shù)
func handleConnection(conn net.Conn) {
	defer conn.Close()

	count++ // 增加在線用戶計(jì)數(shù)

	fmt.Println("有新用戶連接服務(wù)，當(dāng)前連接數(shù)：", count)

	// 添加新用戶
	addUser(conn)

	// 創(chuàng)建退出信號(hào)通道
	var quit = make(chan bool)

	// 啟動(dòng)寫(xiě)消息到客戶端的協(xié)程
	go writeMsgToClient(conn, quit)

	// 啟動(dòng)讀取客戶端消息的協(xié)程
	go readClient(conn, quit)

	// 監(jiān)聽(tīng)退出信號(hào)
	select {
	case <-quit:
		// 用戶下線處理
		connName := onlineList[conn.RemoteAddr().String()].name
		mu.Lock()
		close(onlineList[conn.RemoteAddr().String()].userChannel)
		mu.Unlock()
		mu.Lock()
		delete(onlineList, conn.RemoteAddr().String())
		mu.Unlock()
		count--
		message <- "< 系統(tǒng)消息 > [ " + connName + " ]" + "下線了 當(dāng)前在線人數(shù) " + strconv.Itoa(len(onlineList)) + " 人"
		fmt.Println("有用戶下線了，當(dāng)前連接數(shù)：", count)
		if count == 0 {
			fmt.Println("等待用戶連接中...")
		}

		return
	}
}

// 修改用戶名的方法
func (c *client) changeName(newUserName string) bool {
	mu.Lock()
	defer mu.Unlock()
	// 更新用戶名
	c.name = newUserName
	return true
}

// 管理消息廣播的函數(shù)
func manger() {
	fmt.Println("開(kāi)始監(jiān)聽(tīng) message通道")
	defer fmt.Println("結(jié)束監(jiān)聽(tīng) message通道")
	for msg := range message {
		mu.Lock()
		for _, v := range onlineList {
			v.userChannel <- msg
		}
		mu.Unlock()
	}
}

// 寫(xiě)消息到客戶端的協(xié)程
func writeMsgToClient(conn net.Conn, quit chan bool) {
	fmt.Println(onlineList[conn.RemoteAddr().String()].name, "的信息通道監(jiān)聽(tīng)成功")
	defer fmt.Println(onlineList[conn.RemoteAddr().String()].name, "的信息通道監(jiān)聽(tīng)結(jié)束")
	for msg := range onlineList[conn.RemoteAddr().String()].userChannel {
		king, err := module.Encode(msg + "\n")
		if err != nil {
			fmt.Println("發(fā)送消息失敗")
			continue
		}
		_, err = conn.Write(king)
		if err != nil {
			fmt.Println("發(fā)送消息失敗")
			quit <- true
		}
	}
	fmt.Println("函數(shù)writeMsgToClient函數(shù)結(jié)束")
}

// 添加新用戶
func addUser(conn net.Conn) client {
	fmt.Println("開(kāi)始使用添加新用戶" + conn.RemoteAddr().String())
	newUser := client{
		make(chan string), // 創(chuàng)建用戶消息通道
		conn,              // 網(wǎng)絡(luò)連接
		conn.RemoteAddr().String(), // 用戶名，初始化為客戶端地址
		conn.RemoteAddr().String(), // 客戶端地址
	}
	onlineList[conn.RemoteAddr().String()] = &newUser // 添加到在線用戶列表

	fmt.Println("addUser函數(shù)結(jié)束,用戶" + conn.RemoteAddr().String() + "添加成功")
	return newUser
}

// 讀取客戶端消息的協(xié)程
func readClient(conn net.Conn, quit chan bool) {
	fmt.Println("開(kāi)始讀取客戶端發(fā)送的信息")
	defer fmt.Println("客戶端發(fā)送信息讀取結(jié)束")

	userChannel := onlineList[conn.RemoteAddr().String()].userChannel

	reader := bufio.NewReader(conn)
	for {
		msg, err := module.Decode(reader)
		if err == io.EOF {
			quit <- true
		}
		if err != nil {
			fmt.Println("decode msg failed, err:", err)
			quit <- true
		}
		if len(msg) == 0 {
			continue
		}
		fmt.Println("收到client發(fā)來(lái)的數(shù)據(jù)：", msg)

		// 處理客戶端發(fā)送的不同類型的消息
		switch {
		case strings.HasPrefix(msg, "!@#$@!cd1changeName"):

			king := true
			oldName := onlineList[conn.RemoteAddr().String()].name
			newName := strings.TrimPrefix(msg, "!@#$@!cd1changeName")

			if strings.HasPrefix(msg, "!@#$@!cd1changeNameFirst") {
				newName = strings.TrimPrefix(msg, "!@#$@!cd1changeNameFirst")
			}

			if newName == "" {
				newName = conn.RemoteAddr().String()
			}

			for _, v := range onlineList {

				mapName := v.name
				if mapName == newName {

					king = false

					break
				}
			}

			if strings.HasPrefix(msg, "!@#$@!cd1changeNameFirst") && king == false {

				message <- "< 系統(tǒng)消息 > [ " + conn.RemoteAddr().String() + " ] [ " + oldName + " ] 上線了！"

				userChannel <- "< 系統(tǒng)消息 > [ " + onlineList[conn.RemoteAddr().String()].name + " ]" + "名字: " + newName + " 已存在，請(qǐng)更換一個(gè)名字嘗試"

				userChannel <- "< 系統(tǒng)消息 > 你當(dāng)前昵稱為: " + oldName + " ( 輸入cd2可進(jìn)行名字修改 )"

				continue
			}

			if king == false {

				userChannel <- "< 系統(tǒng)消息 > 昵稱修改失?。。。?

				userChannel <- "< 系統(tǒng)消息 > [ " + onlineList[conn.RemoteAddr().String()].name + " ]" + "名字: " + newName + " 已存在，請(qǐng)更換一個(gè)名字嘗試"

				userChannel <- "< 系統(tǒng)消息 > 你當(dāng)前昵稱為: " + oldName

				continue

			}

			isSuccess := onlineList[conn.RemoteAddr().String()].changeName(newName)
			if isSuccess {

				userChannel <- "!@#$@!cd1changeName" + newName

				if strings.HasPrefix(msg, "!@#$@!cd1changeNameFirst") {

					message <- "< 系統(tǒng)消息 > [ " + conn.RemoteAddr().String() + " ] [ " + newName + " ] 上線了！"

					time.Sleep(time.Millisecond * 50)

					userChannel <- "< 系統(tǒng)消息 > 你當(dāng)前的昵稱為：" + newName

					continue
				}

				userChannel <- "< 系統(tǒng)消息 > 昵稱修改成功 你當(dāng)前昵稱為: " + newName

			} else {

				userChannel <- "< 系統(tǒng)消息 > 昵稱修改失敗?。?！"

			}

			message <- "< 系統(tǒng)消息 > [ " + conn.RemoteAddr().String() + " ]" + " 舊昵稱為: " + oldName + " 新昵稱為: " + newName

		case strings.HasPrefix(msg, "!@#$@!cd4exit"):
			fmt.Println("[ " + onlineList[conn.RemoteAddr().String()].name + " ] " + "下線了")

			quit <- true

			return
		case strings.HasPrefix(msg, "!@#$@!menu"):

			userChannel <- "< 系統(tǒng)消息 > \n * ./cd1 或 ./menu       功能菜單\n * ./cd2 或 ./changeName 更改昵稱\n * ./cd3 或 ./online     在線用戶數(shù)量查詢\n * ../cd4 或 ./quit      退出聊天室"

		case strings.HasPrefix(msg, "!@#$@!cd3online"):

			fmt.Println("在線人數(shù)：", count)

			userChannel <- "< 系統(tǒng)消息 > 當(dāng)前在線人數(shù)：" + strconv.Itoa(count)

		default:

			message <- "[ " + onlineList[conn.RemoteAddr().String()].name + " ]" + ": " + msg

			fmt.Println("信息廣播成功")
		}
	}
}

package main

import (
	"bufio"         
	"chatRoom/chatRoom/module" // 消息的編碼和解碼模塊
	"fmt"            
	"io"            
	"net"            
	"os"            
	"strings"        
	"time"           
)

// 定義一個(gè)全局變量用于存儲(chǔ)用戶的昵稱
var name string

// 主函數(shù)
func main() {
	// 嘗試連接到服務(wù)器
	conn, err := net.Dial("tcp", "127.0.0.1:8080")
	if err != nil {
		fmt.Println("服務(wù)器連接失敗 err =", err)
		return
	}

	defer conn.Close()
	// 打印連接成功的消息
	fmt.Println("服務(wù)器連接成功")

	// 獲取用戶的昵稱
	name = getUserInput("請(qǐng)輸入你的昵稱：")
	// 構(gòu)建一條特殊的消息，用于通知服務(wù)器用戶昵稱
	data, err := module.Encode("!@#$@!cd1changeNameFirst" + name)
	if err != nil {
		
		fmt.Println("encode msg failed, err:", err)
		return
	}
	// 發(fā)送消息到服務(wù)器
	_, err = conn.Write(data)
	if err != nil {
		fmt.Println("發(fā)送數(shù)據(jù)失敗1 err =", err)
	}

	// 創(chuàng)建一個(gè)通道，用于接收退出信號(hào)
	var exit = make(chan bool)
	// 確保在函數(shù)退出時(shí)關(guān)閉通道
	defer close(exit)

	// 顯示歡迎信息和命令提示
	fmt.Println("--------------歡迎進(jìn)入多人聊天室系統(tǒng)----------------")
	fmt.Println("       * ./cd1 或 ./menu       功能菜單")
	fmt.Println("       * ./cd2 或 ./changeName 更改昵稱")
	fmt.Println("       * ./cd3 或 ./online     在線用戶數(shù)量查詢")
	fmt.Println("       * ./cd4 或 ./quit       退出聊天室")
	fmt.Println("---------------指令字母不區(qū)分大小寫(xiě)-----------------")

	// 啟動(dòng)一個(gè)協(xié)程用于讀取消息
	go readMsg(conn)

	// 啟動(dòng)一個(gè)協(xié)程用于處理用戶輸入
	go func() {
		for {
			// 獲取用戶輸入
			msg := getUserInput("")

			// 根據(jù)用戶輸入特殊消息處理命令
			if strings.EqualFold(msg, "./cd1") || strings.EqualFold(msg, "./menu") {
				msg = "!@#$@!menu"
			}
			if strings.EqualFold(msg, "./cd2") || strings.EqualFold(msg, "./changeName") {
				newMsg := getUserInput("請(qǐng)輸入新的昵稱：")
				msg = "!@#$@!cd1changeName" + newMsg
			}
			if strings.EqualFold(msg, "./cd3") || strings.EqualFold(msg, "./online") {
				msg = "!@#$@!cd3online"
			}
			if strings.EqualFold(msg, "./cd4") || strings.EqualFold(msg, "./quit") {
				msg = "!@#$@!cd4exit"
				// 編碼并發(fā)送退出消息
				data, err := module.Encode(msg)
				if err != nil {
					fmt.Println("消息數(shù)據(jù)失敗1, err:", err)
					return
				}
				_, err = conn.Write(data)
				if err != nil {
					// 如果發(fā)送失敗，打印錯(cuò)誤信息
					fmt.Println("發(fā)送數(shù)據(jù)失敗2 err =", err)
				}
				// 打印退出信息
				fmt.Println("正在退出...")
				// 發(fā)送退出信號(hào)
				exit <- true
				return
			}

			// 編碼并發(fā)送普通消息
			data, err := module.Encode(msg)
			if err != nil {
				// 如果消息編碼失敗，打印錯(cuò)誤信息并退出協(xié)程
				fmt.Println("發(fā)送數(shù)據(jù)失敗3, err:", err)
				return
			}
			_, err = conn.Write(data)
			if err != nil {
				// 如果發(fā)送失敗，打印錯(cuò)誤信息
				fmt.Println("發(fā)送數(shù)據(jù)失敗4 err =", err)
			}
		}
	}()

	// 主循環(huán)，監(jiān)聽(tīng)退出信號(hào)
	for {
		select {
		case <-exit:
			// 當(dāng)收到退出信號(hào)時(shí)，打印退出成功并退出程序
			fmt.Println("退出成功")
			return
		}
	}
}

// getUserInput 函數(shù)用于獲取用戶輸入
func getUserInput(prompt string) string {

	time.Sleep(time.Millisecond * 100)
	// 根據(jù)不同的提示信息顯示相應(yīng)的提示
	switch prompt {
	case "請(qǐng)輸入你的昵稱：":
		fmt.Print("請(qǐng)輸入你的昵稱：")
	case "請(qǐng)輸入新的昵稱：":
		fmt.Println("請(qǐng)輸入新的昵稱：")
	}
	// 創(chuàng)建一個(gè)標(biāo)準(zhǔn)輸入的緩沖讀取器
	reader := bufio.NewReader(os.Stdin)
	// 讀取一行輸入
	input, err := reader.ReadString('\n')
	if err != nil {
		// 如果讀取失敗，打印錯(cuò)誤信息并返回錯(cuò)誤信息
		fmt.Println("用戶輸入獲取失?。篹rr =", err)
		return "客戶端信息讀取錯(cuò)誤"
	}
	// 返回去掉空格的輸入字符串
	return strings.TrimSpace(input)
}

// readMsg 函數(shù)用于讀取消息
func readMsg(conn net.Conn) {

	defer conn.Close()
	// 創(chuàng)建一個(gè)緩沖讀取器來(lái)讀取連接中的數(shù)據(jù)
	reader := bufio.NewReader(conn)
	for {
		// 解碼消息
		msg, err := module.Decode(reader)
		if err == io.EOF {
			// 如果遇到EOF（文件結(jié)束），表示連接已斷開(kāi)
			fmt.Println("服務(wù)器連接已斷開(kāi) ")
			// 終止程序
			os.Exit(1)
		}
		if err != nil {
		
			fmt.Println("服務(wù)器斷開(kāi)連接 2 err =", err)
			return
		}
		if msg == "" {
			// 如果消息為空，則跳過(guò)本次循環(huán)
			continue
		}
		if strings.HasPrefix(msg, "!@#$@!cd1changeName") {
			// 如果消息是更改昵稱的通知
			msg1 := strings.TrimPrefix(msg, "!@#$@!cd1changeName")
			name = strings.TrimRight(msg1, "\n")
			// 更新昵稱
			continue
		}
		// 打印消息的時(shí)間戳和內(nèi)容
		fmt.Print("【 ", time.Now().Format("15:04"), " 】", msg)
	}
}

package module

import (
	"bufio"
	"bytes"
	"encoding/binary"
)

func Encode(message string) ([]byte, error) {
	// 讀取消息的長(zhǎng)度，轉(zhuǎn)換成int32類型（占4個(gè)字節(jié)）
	var length = int32(len(message))
	var pkg = new(bytes.Buffer)
	// 寫(xiě)入消息頭
	err := binary.Write(pkg, binary.LittleEndian, length)
	if err != nil {
		return nil, err
	}
	// 寫(xiě)入消息實(shí)體
	err = binary.Write(pkg, binary.LittleEndian, []byte(message))
	if err != nil {
		return nil, err
	}
	return pkg.Bytes(), nil
}

// Decode 解碼消息
func Decode(reader *bufio.Reader) (string, error) {
	// 讀取消息的長(zhǎng)度
	lengthByte, _ := reader.Peek(4) // 讀取前4個(gè)字節(jié)的數(shù)據(jù)
	lengthBuff := bytes.NewBuffer(lengthByte)
	var length int32
	err := binary.Read(lengthBuff, binary.LittleEndian, &length)
	if err != nil {
		return "", err
	}
	// Buffered返回緩沖中現(xiàn)有的可讀取的字節(jié)數(shù)。
	if int32(reader.Buffered()) < length+4 {
		return "", err
	}

	// 讀取真正的消息數(shù)據(jù)
	pack := make([]byte, int(4+length))
	_, err = reader.Read(pack)
	if err != nil {
		return "", err
	}
	return string(pack[4:]), nil
}