// telnet 命令退出 ctrl + ] 然后输入 quit退出  不会触发close事件
// ctrl + ] 然后 ctrl + c关闭连接 会触发close事件。
// 使用 telnet 127.0.0.1 3000 加入聊天室
var net = require('net');

var count = 0; // 用来追踪连接数
var users = {}; // 记录设置了昵称的用户

var server = net.createServer(function(conn) {
  conn.write(
    '\n > welcome to \033[92mnode-chat\033[39m!'
    + '\n > ' + count + ' other people are connected at this time.'
    + '\n > please write your name and press enter:'
  )

  count++;
  
  var nickname;
  // 接受用户输入
  conn.setEncoding('utf8');
  conn.on('data', function(data) {
    // 需要注意的是接受到的是一个Buffer 所以需要设置编码
    data = data.replace('\r\n', ''); // 删除回车符
    if(!nickname) { // 如果还没有设置昵称的情况下
      if(users[data]) {
        // 用户输入的昵称已经被注册
        conn.write('\033[90m> nickname already in use. try again:\033[39m ');
        return;
      } else {
        nickname = data;
        // 记录用户的连接
        users[nickname] = conn;
        // 广播给所有已经连接的用户，不排除自己
        broadcast('\033[90m > ' + nickname + ' joined the room\033[39m \n')
      }
    } else {
      // 如果设置了昵称 则之后的输入视为聊天消息 广播给除自己外的其他人
      broadcast('\033[96m > ' + nickname + ':\033[39m' + data + '\n', true)
    }
  })

  // 当客户端关闭连接时 计数器要递减
  conn.on('close', function() {
    count--;
    // 当有人断开连接时，需要清除users中对应的元素
    delete users[nickname];
    // 广播给其他人知道
    broadcast('\033[90m > ' + nickname + ' left the room\033[39m \n')
  })
  
  // 抽象广播方法
  // @params msg [String] 消息
  // @params exceptMyself [Boolean] 是否排除自己 
  function broadcast(msg, exceptMyself) {
    for(var n in users) {
      if(!exceptMyself || n != nickname) {
        // 为false时 不排除自己 在不排除自己时才向自己发送消息
        // 在排除自己时，向其他人发送消息
        users[n].write(msg);
      }
    }
  }
})

server.listen(3000, function() {
  console.log('\033[90m   server listening on: 3000\033[39m');
})