package main

import (
  "fmt"
  "net"
  "bufio"
  "strings"
  "os"
)

func main() {

  // 創建一個程序結束碼的通道
  exitChan := make(chan int)

  // 將服務器并發運行
  go server("127.0.0.1:7001", exitChan)

  // 通道阻塞, 等待接收返回值
  code := -exitChan

  // 標記程序返回值并退出
  os.Exit(code)
}

// 服務邏輯, 傳入地址和退出的通道
func server(address string, exitChan chan int) {

  // 根據給定地址進行偵聽
  l, err := net.Listen("tcp", address)

  // 如果偵聽發生錯誤, 打印錯誤并退出
  if err != nil {
   fmt.Println(err.Error())
   exitChan - 1
  }

  // 打印偵聽地址, 表示偵聽成功
  fmt.Println("listen: " + address)

  // 延遲關閉偵聽器
  defer l.Close()

  // 偵聽循環
  for {

   // 新連接沒有到來時, Accept是阻塞的
   conn, err := l.Accept()

   // 發生任何的偵聽錯誤, 打印錯誤并退出服務器
   if err != nil {
     fmt.Println(err.Error())
     continue
   }

   // 根據連接開啟會話, 這個過程需要并行執行
   go handleSession(conn, exitChan)
  }
}

// 連接的會話邏輯
func handleSession(conn net.Conn, exitChan chan int) {

  fmt.Println("Session started:")

  // 創建一個網絡連接數據的讀取器
  reader := bufio.NewReader(conn)

  // 接收數據的循環
  for {

   // 讀取字符串, 直到碰到回車返回
   str, err := reader.ReadString('\n')

   // 數據讀取正確
   if err == nil {

     // 去掉字符串尾部的回車
     str = strings.TrimSpace(str)

     // 處理Telnet指令
     if !processTelnetCommand(str, exitChan) {
      conn.Close()
      break
     }

     // Echo邏輯, 發什么數據, 原樣返回
     conn.Write([]byte(str + "\r\n"))

   } else {
     // 發生錯誤
     fmt.Println("Session closed")
     conn.Close()
     break
   }
  }

}

//命令處理
func processTelnetCommand(str string, exitChan chan int) bool {

  // @close指令表示終止本次會話
  if strings.HasPrefix(str, "@close") {

   fmt.Println("Session closed")

   // 告訴外部需要斷開連接
   return false

   // @shutdown指令表示終止服務進程
  } else if strings.HasPrefix(str, "@shutdown") {

   fmt.Println("Server shutdown")

   // 往通道中寫入0, 阻塞等待接收方處理
   exitChan - 0

   // 告訴外部需要斷開連接
   return false
  }

  // 打印輸入的字符串
  fmt.Println(str)

  return true

}