#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>

@interface ClientModel : NSObject
@property(nonatomic,assign)int clientSocket;
@property(nonatomic,copy)NSString *clientName;
@end

#import "ViewController.h"

#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#import "ClientModel.h"
static int const kMaxConnectCount = 5;

@interface ViewController()
@property (weak) IBOutlet NSTextField *textField;
//@property (nonatomic,assign)int client_socket; //客戶端socket
@property (unsafe_unretained) IBOutlet NSTextView *textView;
 @property (nonatomic,strong)NSMutableArray *clientArray;
 @property (nonatomic,strong)NSMutableArray *clientNameArray;
@end

@implementation ViewController

- (NSMutableArray *)clientArray {
 if (!_clientArray) {
  _clientArray = [NSMutableArray array];
 }
 return _clientArray;
}
- (NSMutableArray *)clientNameArray {
 if (!_clientNameArray) {
  _clientNameArray = [NSMutableArray array];
 }
 return _clientNameArray;
}

- (void)viewDidLoad {
 [super viewDidLoad];
 // Do any additional setup after loading the view.
 //創建socket
 int server_socket = socket(AF_INET, SOCK_STREAM, 0);
 if (server_socket == -1) {
  NSLog(@"創建失敗");
  [self showLogsWithString:@"socket創建失敗"];

 }else{
  //綁定地址和端口
  struct sockaddr_in server_addr;
  server_addr.sin_len = sizeof(struct sockaddr_in);
  server_addr.sin_family = AF_INET;
  server_addr.sin_port = htons(1234);
  server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
  bzero(&(server_addr.sin_zero), 8);

  int bind_result = bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr));
  if (bind_result == -1) {
   NSLog(@"綁定端口失敗");
   [self showLogsWithString:@"綁定端口失敗"];

  }else{
   if (listen(server_socket, kMaxConnectCount)==-1) {
    NSLog(@"監聽失敗");
    [self showLogsWithString:@"監聽失敗"];

   }else{
    for (int i = 0; i < kMaxConnectCount; i++) {
     //接受客戶端的鏈接
     [self acceptClientWithServerSocket:server_socket];
    }
   }
  }
 }
}


- (void)setRepresentedObject:(id)representedObject {
 [super setRepresentedObject:representedObject];

 // Update the view, if already loaded.
}

//創建線程接受客戶端
-(void)acceptClientWithServerSocket:(int)server_socket{
 struct sockaddr_in client_address;
 socklen_t address_len;
 dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
 dispatch_async(queue, ^{
  //創建新的socket
  while (1) {
   int client_socket = accept(server_socket, (struct sockaddr*)&client_address,&address_len );
   if (client_socket == -1) {
    [self showLogsWithString:@"接受客戶端鏈接失敗"];
    NSLog(@"接受客戶端鏈接失敗");
   }else{
    NSString *acceptInfo = [NSString stringWithFormat:@"客戶端 in,socket:%d",client_socket];
    [self showLogsWithString:acceptInfo];

    //接受客戶端數據
    [self recvFromClinetWithSocket:client_socket];
   }
  }
 });
}

//接受客戶端數據
- (void)recvFromClinetWithSocket:(int)client_socket{
 while (1) {
  //接受客戶端傳來的數據
  char buf[1024] = {0};
  long iReturn = recv(client_socket, buf, 1024, 0);
  if (iReturn>0) {
   NSLog(@"客戶端來消息了");
   NSString *str = [NSString stringWithCString:buf encoding:NSUTF8StringEncoding];
   [self showLogsWithString:[NSString stringWithFormat:@"客戶端來消息了:%@",str]];
   [self checkRecvStr:str andClientSocket:client_socket];
  }else if (iReturn == -1){
   NSLog(@"讀取消息失敗");
   [self showLogsWithString:@"讀取消息失敗"];
   break;
  }else if (iReturn == 0){
   NSLog(@"客戶端走了");
   [self showLogsWithString:[NSString stringWithFormat:@"客戶端 out socket:%d",client_socket]];
   NSMutableArray *array = [NSMutableArray arrayWithArray:self.clientArray];
   for (ClientModel *model in array) {
    if (model.clientSocket == client_socket) {
     [self.clientNameArray removeObject:model.clientName];
     [self.clientArray removeObject:model];
    }
   }

   close(client_socket);

   break;
  }
 }
}

 //檢查接受到的字符串
- (void)checkRecvStr:(NSString*)str andClientSocket:(int)socket{
 if ([str hasPrefix:@"name:"]) {
  NSString *name = [str substringFromIndex:5];

  ClientModel *model = [[ClientModel alloc] init];
  model.clientSocket = socket;
  model.clientName = name;


  if (self.clientArray.count > 0) {
   int flag = 999;
   //用戶名不能相同
   int i = 0;

   for (ClientModel *client in self.clientArray) {

    //改名
    if (client.clientSocket == socket) {
     NSString *oldName = self.clientNameArray[i];
     self.clientNameArray[i] = name;
     self.clientArray[i] = model;

     for (ClientModel *oldclient in self.clientArray) {
      [self sendMsg:[NSString stringWithFormat:@"%@ 改名 %@",oldName,name] toClient:oldclient.clientSocket];
      [self showLogsWithString:[NSString stringWithFormat:@"%@ 改名 %@",oldName,name]];
      NSString *list = [self.clientNameArray componentsJoinedByString:@","];
      //向客戶端推送當前在線列表
      [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:oldclient.clientSocket];
     }

     flag = 2;

    }else{
     if ([client.clientName isEqualToString:model.clientName]) {
      //用戶名已存在
      flag = 1;
      break;
     }
    }
    i++;

   }
   if (flag != 1 & flag != 2) {
    [self.clientArray addObject:model];
    [self.clientNameArray addObject:model.clientName];
    //向客戶端推送當前在線列表
    for (ClientModel *client in self.clientArray) {
     [self sendMsg:[NSString stringWithFormat:@"%@,上線了",name] toClient:client.clientSocket];
     NSString *list = [self.clientNameArray componentsJoinedByString:@","];
     //向客戶端推送當前在線列表
     [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:client.clientSocket];
    }

    //給當前客戶端發送一條歡迎信息
    NSString *msg = [NSString stringWithFormat:@"Welcome %@ !",name];
    [self sendMsg:msg toClient:socket];
    [self showLogsWithString:msg];

   }else if (flag == 1){
    [self sendMsg:@"注冊用戶名失敗，用戶名已經存在，請重新設置用戶名" toClient:socket];
    [self showLogsWithString:[NSString stringWithFormat:@"socket %d 注冊用戶名失敗，設置的用戶名已經存在",socket]];

    for (ClientModel *model in self.clientArray) {

     [name isEqualToString:model.clientName];
    }



   }
  }else{
   [self.clientArray addObject:model];
   [self.clientNameArray addObject:model.clientName];
   //向客戶端推送當前在線列表
   //給當前客戶端發送一條歡迎信息
   NSString *msg = [NSString stringWithFormat:@"Welcome %@ !",name];
   [self sendMsg:msg toClient:socket];
   [self showLogsWithString:msg];

   NSString *list = [self.clientNameArray componentsJoinedByString:@","];
   //向客戶端推送當前在線列表
   [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:socket];

  }

 }
 //給某人發消息
 else if ([str hasPrefix:@"to:"]){
  NSRange nameRange = [str rangeOfString:@"*"];
  NSString *name = [str substringWithRange:NSMakeRange(3, nameRange.location-3)];
  NSString *content = [str substringFromIndex:nameRange.location+1];
  NSString *fromClientName;
  //找出發送者
  for (ClientModel *model in self.clientArray) {
   if (socket == model.clientSocket) {
    fromClientName = model.clientName;
    break;
   }
  }

  //給目標發送信息
  for (ClientModel *model in self.clientArray) {
   if ([name isEqualToString:model.clientName]) {
    NSString *msg = [NSString stringWithFormat:@"%@ to you\n%@",fromClientName,content];
    [self sendMsg:msg toClient:model.clientSocket];

    [self showLogsWithString:[NSString stringWithFormat:@"%@ 發送給 %@ 內容是：%@",fromClientName,name,content]];
    break;

   }
  }

 }
}

//給客戶端發送信息
- (void)sendMsg:(NSString*)msg toClient:(int)socket{
 char *buf[1024] = {0};
 const char *p1 = (char*)buf;
 p1 = [msg cStringUsingEncoding:NSUTF8StringEncoding];
 send(socket, p1, 1024, 0);
}

 //在界面上顯示日志
- (void)showLogsWithString:(NSString*)str {
 dispatch_async(dispatch_get_main_queue(), ^{
  NSString *newStr = [NSString stringWithFormat:@"\n%@",str];
  self.textView.string = [self.textView.string stringByAppendingString:newStr];
 });
}

@end

#import "ViewController.h"
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
@interface ViewController ()<UITableViewDelegate,UITableViewDataSource>
//服務器socket
@property (nonatomic,assign)int server_socket;

//UI
@property (weak, nonatomic) IBOutlet UITextField *userNameField;
@property (weak, nonatomic) IBOutlet UITextView *chatView;
@property (weak, nonatomic) IBOutlet UITextField *msgField;
@property (weak, nonatomic) IBOutlet UILabel *toName;
@property (weak, nonatomic) IBOutlet UIView *onlineUserView;
@property (nonatomic,strong)UITableView *onlineTable;

//user列表
@property (nonatomic,strong)NSMutableArray *userArray;

@end

@implementation ViewController
- (NSMutableArray *)userArray {
 if (!_userArray) {
  _userArray = [NSMutableArray array];
 }
 return _userArray;
}
- (void)viewDidLoad {
 [super viewDidLoad];
 [self.userNameField becomeFirstResponder];
 self.userNameField.text = @"";
 self.msgField.text = @"";
 //添加table用戶列表
 self.onlineTable = [[UITableView alloc] initWithFrame:self.onlineUserView.frame style:UITableViewStylePlain];
 self.onlineTable.delegate = self;
 self.onlineTable.dataSource = self;
 [self.view addSubview:self.onlineTable];

 int server_socket = socket(AF_INET, SOCK_STREAM, 0);
 if (server_socket == -1) {
  NSLog(@"創建失敗");
 }else{
  //綁定地址和端口
  struct sockaddr_in server_addr;
  server_addr.sin_len = sizeof(struct sockaddr_in);
  server_addr.sin_family = AF_INET;
  server_addr.sin_port = htons(1234);
  server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
  bzero(&(server_addr.sin_zero), 8);

  //接受客戶端的鏈接
  dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
  dispatch_async(queue, ^{
   //創建新的socket
   int aResult = connect(server_socket, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_in));
   if (aResult == -1) {
    NSLog(@"鏈接失敗");
   }else{
    self.server_socket = server_socket;
    [self acceptFromServer];
   }
  });
 }
}

//從服務端接受消息
- (void)acceptFromServer{
 while (1) {
  //接受服務器傳來的數據
  char buf[1024];
  long iReturn = recv(self.server_socket, buf, 1024, 0);
  if (iReturn>0) {
   NSString *str = [NSString stringWithCString:buf encoding:NSUTF8StringEncoding];

   //篩選前綴
   if ([str hasPrefix:@"list:"]) {
    NSString *arrayStr = [str substringFromIndex:5];
    NSArray *list = [arrayStr componentsSeparatedByString:@","];
    self.userArray = [NSMutableArray arrayWithArray:list];
    dispatch_async(dispatch_get_main_queue(), ^{
     [self.onlineTable reloadData];
    });
    NSLog(@"當前在線用戶列表：%@",arrayStr);
   }else{
    //回到主線程 界面上顯示內容
    [self showLogsWithString:str];
   }

  }else if (iReturn == -1){
   NSLog(@"接受失敗-1");
   break;
  }
 }
}

//在界面上顯示日志
- (void)showLogsWithString:(NSString*)str {
 dispatch_async(dispatch_get_main_queue(), ^{
  NSString *newStr = [NSString stringWithFormat:@"\n%@",str];
  self.chatView.text = [self.chatView.text stringByAppendingString:newStr];
 });
}

- (void)didReceiveMemoryWarning {
 [super didReceiveMemoryWarning];
 // Dispose of any resources that can be recreated.
}

//設置用戶名
- (IBAction)clickSetUserName:(id)sender {
 NSString *msg = [NSString stringWithFormat:@"name:%@",self.userNameField.text] ;
 [self sendMsg:msg];
// [self showLogsWithString:msg];
 [self.msgField becomeFirstResponder];
}

//發送信息
- (IBAction)clickSendMsg:(id)sender {
 if ([self.msgField.text isEqualToString:@""] || ![self.userArray containsObject:self.userNameField.text] || [self.toName.text isEqualToString:self.userNameField.text]) {
  [self showLogsWithString:@"請設置用戶名、檢查發送對象、消息不能為空"];
  return;
 }
 NSString *msg = [NSString stringWithFormat:@"to:%@*%@",self.toName.text,self.msgField.text];
 [self sendMsg:msg];
 NSString *displayMsg = [NSString stringWithFormat:@"to:%@\n%@",self.toName.text,self.msgField.text];
 [self showLogsWithString:displayMsg];
 self.msgField.text = @"";

}

//給客戶端發送信息
- (void)sendMsg:(NSString*)msg {
 char *buf[1024] = {0};
 const char *p1 = (char*)buf;
 p1 = [msg cStringUsingEncoding:NSUTF8StringEncoding];
 send(self.server_socket, p1, 1024, 0);
}

#pragma mark - TableViewDelegate & dataSource
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
 return self.userArray.count;
}
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{
 static NSString *cellId = @"onlinetableviewcellid";
 UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:cellId];
 if (cell == nil) {
  cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:cellId];
 }else{
  NSLog(@"cell重用了");
 }
 cell.textLabel.text = self.userArray[indexPath.row];
 return cell;
}

//點擊cell
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath{
 self.toName.text = self.userArray[indexPath.row];
 [self.msgField becomeFirstResponder];
}
@end