// one sever to more client chat room
//This is chat sever
package main

import (
    "fmt"
    "net"
    "strings"
)

var ConnMap map[string]net.Conn = make(map[string]net.Conn)  //声明一个集合

//ConnMap := make(map[string]net.Conn)

func main() {
    listen_socket, err := net.Listen("tcp", "127.0.0.1:8000")  //打开监听接口
    if err != nil {
        fmt.Println("server start error")
    }

    defer listen_socket.Close()
    fmt.Println("server is wating ....")

    for {
        conn, err := listen_socket.Accept()  //收到来自客户端发来的消息
        if err != nil {
            fmt.Println("conn fail ...")
        }
        fmt.Println(conn.RemoteAddr(), "connect successed")

        go handle(conn)  //创建线程
    }
}

func handle(conn net.Conn) {
    for {
        data := make([]byte, 255)  //创建字节流 （此处同 一对一 通信）
        msg_read, err := conn.Read(data)  //声明并将从客户端读取的消息赋给msg_read 和err
        if msg_read == 0 || err != nil {
            continue
        }

        //解析协议
        msg_str := strings.Split(string(data[0:msg_read]), "|")  //将从客户端收到的字节流分段保存到msg_str这个数组中

        switch msg_str[0] {
        case "nick":  //加入聊天室
            fmt.Println(conn.RemoteAddr(), "-->", msg_str[1])  //nick占在数组下标0上，客户端上写的昵称占在数组下标1上
            for k, v := range ConnMap {  //遍历集合中存储的客户端消息
                if k != msg_str[1] {
                    v.Write([]byte("[" + msg_str[1] + "]: join..."))
                }
            }
            ConnMap[msg_str[1]] = conn
        case "say":   //转发消息
            for k, v := range ConnMap {  //k指客户端昵称   v指客户端连接服务器端后的地址
                if k != msg_str[1] {  //判断是不是给自己发，如果不是
                    fmt.Println("Send "+msg_str[2]+" to ", k)  //服务器端将消息转发给集合中的每一个客户端
                    v.Write([]byte("[" + msg_str[1] + "]: " + msg_str[2]))  //给除了自己的每一个客户端发送自己之前要发送的消息
                }
            }
        case "quit":  //退出
            for k, v := range ConnMap {  //遍历集合中的客户端昵称
                if k != msg_str[1] {  //如果昵称不是自己
                    v.Write([]byte("[" + msg_str[1] + "]: quit"))  //给除了自己的其他客户端昵称发送退出的消息，并使Write方法阻塞
                }
            }
            delete(ConnMap, msg_str[1])  //退出聊天室
        }
    }
}

// one sever to more client chat room
//This is chat client
package main

import (
    "fmt"
    "net"
)

var nick string = ""  //声明聊天室的昵称

func main() {
    conn, err := net.Dial("tcp", "127.0.0.1:8000")  //打开监听端口
    if err != nil {
        fmt.Println("conn fail...")
    }
    defer conn.Close()
    fmt.Println("client connect server successed \n")

    //给自己取一个聊天室的昵称
    fmt.Printf("Make a nickname:")
    fmt.Scanf("%s", &nick)  //输入昵称
    fmt.Println("hello : ", nick)  //客户端输出
    conn.Write([]byte("nick|" + nick))  //将信息发送给服务器端

    go Handle(conn)  //创建线程

    var msg string
    for {
        msg = ""  //声明一个空的消息
        fmt.Scan(&msg)  //输入消息
        conn.Write([]byte("say|" + nick + "|" + msg))  //三段字节流 say | 昵称 | 发送的消息
        if msg == "quit" {  //如果消息为quit
            conn.Write([]byte("quit|" + nick))  //将quit字节流发送给服务器端
            break  //程序结束运行
        }
    }
}

func Handle(conn net.Conn) {

    for {

        data := make([]byte, 255)  //创建一个字节流
        msg_read, err := conn.Read(data)  //将读取的字节流赋值给msg_read和err
        if msg_read == 0 || err != nil {  //如果字节流为0或者有错误
            break
        }

        fmt.Println(string(data[0:msg_read]))  //把字节流转换成字符串
    }
}