package main
import (
  "net"
  "fmt"
  "strings"
  "time"
)
//定義的此結(jié)構(gòu)體為全局map的value值，包括每一個用戶的姓名，ip地址和私人管道
type client struct {
  name string
  addr string
  C  chan string
}
/*這個函數(shù)是將私人管道中的內(nèi)容發(fā)送給用戶，配合全局管道Message使用可以實現(xiàn)廣播的功能，
單獨使用可以實現(xiàn)私聊的功能*/
func writemsg2client(clinet client,conn net.Conn) {
  for m := range clinet.C {
    conn.Write([]byte(m + "\n"))
  }
}
//這只是一個封裝好用來統(tǒng)一(發(fā)送信息格式)的小函數(shù)，不用在意
func makemsg(name string, addr string, s string) string {
  return "[" + addr + "]" + name + s
}
//每一個進入聊天室的用戶都將啟動一個handleconn的go程來處理事件
func handleconn(conn net.Conn) {
  defer conn.Close()
/*用戶連接進來以后要初始化全局map，把自己的信息加入到字典里，相當于進到聊天室里之前要登
記一下個人信息，注意姓名初始為ip地址。*/
  addr := conn.RemoteAddr().String()
  fmt.Printf("用戶%s進入了房間\n", addr)
  client := client{addr, addr, make(chan string)}
  //在這里啟動子go程，功能上面已經(jīng)提及
  go writemsg2client(client,conn)
  onlinemap[addr] = client
  //登錄進來一切準備就緒后就給所有人廣播上線信息啦
  Message <- makemsg(client.name, addr, "login")
  //下面這三個變量服務(wù)于下面一些小功能
  var haschat=make(chan bool)
  var ifquit=make(chan bool)
  var flag bool
  //從這單獨開啟一個go程來讀取用戶輸入的信息
  go func() {
    buf:=make([]byte,4096)
    for {
      n,_:=conn.Read(buf)
      if n==0 {
        fmt.Printf("%s離開了房間\n",client.name)
        ifquit<-true
        return
      }
      //改名功能的實現(xiàn)
      if string(buf[:7])=="Rename|" {
        client.name=strings.Split(string(buf[:n-1]),"|")[1]
        onlinemap[addr]=client
        conn.Write([]byte("rename success\n"))
      }else if string(buf[:n-1])=="/who"{
      //查詢在線用戶信息的功能
        for _,s:=range onlinemap{
          conn.Write([]byte(s.name+"online\n"))
        }
      }else if string(buf[:2])=="m|"&&strings.Count(string(buf[:n]),"|")==2 {
/*私聊功能的實現(xiàn)，其實私聊功能就是跳過了往全局Message里傳輸信息，
改為直接向私人管道里傳輸信息*/
        flag=false
        slice:=strings.Split(string(buf[:n-1]),"|")
        for _,a:=range onlinemap{
        //遍歷所有在線用戶，向指定的用戶管道中發(fā)送信息
          if a.name==slice[1]{
            flag=true
            a.C<-makemsg(client.name,addr,slice[2])
            conn.Write([]byte("send success"))
          }
        }
        if flag {
          conn.Write([]byte("no such man or not online"))
        }
      } else {
        Message<-makemsg(client.name,addr,string(buf[:n-1]))
      }
      haschat<-true
    }
  }()
  for {
    select {
    case <-haschat:
    //超時強踢
    case <-time.After(time.Minute*3):
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    case <-ifquit:
    //退出處理
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    }
  }
}
//這個函數(shù)用來將全局Message中的內(nèi)容全部塞到私人管道C里，實現(xiàn)上下線廣播和群聊的功能
func Manager() {
  for {
    msg := <-Message
    for _, s := range onlinemap {
      s.C <- msg
    }
  }
}
var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)
//主函數(shù)
func main() {
  listener, _ := net.Listen("tcp", "127.0.0.1:6666")
  defer listener.Close()
  //提前開啟全局Message的go程，防止被阻塞
  go Manager()
  for {
    conn, err := listener.Accept()
    if err != nil {
      fmt.Println("accept err", err)
      continue
    }
    //每一個連接進來的用戶都會被分配進入一個子go程，用來處理上面我們提到的各種功能
    go handleconn(conn)
  }
}