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

}