#pragma once

#include "EventLoop.h"
#include "Acceptor.h"
#include "InetAddress.h"
#include "noncopyable.h"
#include "EventLoopThreadPool.h"
#include "Callbacks.h"
#include "TcpConnection.h"
#include "Buffer.h"

#include <functional>
#include <string>
#include <memory>
#include <atomic>
#include <unordered_map>

//对外部服务器进行编程应用的类
//相当于封装了TCP所有细节然后可以暴露给第三方程序可以直接调用API进行网络IO的作用
class TcpServer : noncopyable{
  public:
    using ThreadInitCallback = std::function<void(EventLoop*)>;

    enum Option{ //确定是否对端口可重用
      kNoReusePort,
      kReusePort,
    };

    TcpServer(EventLoop* loop, const InetAddress& listenAddr, const std::string& nameArg, Option option = kNoReusePort);
    ~TcpServer();

    void setThreadInitcallback(const ThreadInitCallback& cb){ threadInitCallback_ = cb; }
    void setConnectionCallback(const ConnectionCallback& cb){ connectionCallback_ = cb; }
    void setMessageCallback(const MessageCallback& cb) { messageCallback_ = cb; }
    void setWriteCompleteCallback(const WriteCompleteCallback& cb){ writeCompleteCallback_ = cb; }


    //设置底层subLoop的个数
    void setThreadNum(int numThreads);
    
    //开启服务器监听
    void start();

  private:
    void newConneciton(int sockfd, const InetAddress& peerAddr); //新连接到来
    void removeConnection(const TcpConnectionPtr& conn); //删除连接
    void removeConnectionInLoop(const TcpConnectionPtr& conn);

    using ConnectionMap = std::unordered_map<std::string, TcpConnectionPtr>;

    EventLoop* loop_; //baseLoop 用户定义的loop
    const std::string ipPort_;
    const std::string name_;
    std::unique_ptr<Acceptor> acceptor_; //运行在mainLoop，任务就是监听新连接事件
    std::shared_ptr<EventLoopThreadPool> threadPool_; //one loop per thread
    
    ConnectionCallback connectionCallback_; //有新连接时的回调
    MessageCallback messageCallback_; //有读写消息时的回调
    WriteCompleteCallback writeCompleteCallback_; //消息发送完成时的回调

    ThreadInitCallback threadInitCallback_; //loop线程初始化时的回调

    std::atomic_int started_;

    int nextConnId_;
    ConnectionMap connections_; //保存所有的连接
};