//////////////////////////////////////////////////////////////////////////////////////////
// A multi-platform support c++11 library with focus on asynchronous socket I/O for any
// client application.
//////////////////////////////////////////////////////////////////////////////////////////
//
// detail/select_fd_set.hpp
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
// Copyright (c) 2012-2023 HALX99 (halx99 at live dot com)
#pragma once

#include <vector>
#include "yasio/detail/socket.hpp"

namespace yasio
{
YASIO__NS_INLINE
namespace inet
{
class select_fd_set {
public:
  select_fd_set() { reset(); }

  void reset()
  {
    FD_ZERO(&fd_set_[read_op]);
    FD_ZERO(&fd_set_[write_op]);
    FD_ZERO(&fd_set_[except_op]);
    max_descriptor_ = 0;
  }

  select_fd_set& operator=(select_fd_set& rhs)
  {
    ::memcpy(this->fd_set_, rhs.fd_set_, sizeof(rhs.fd_set_));
    max_descriptor_ = rhs.max_descriptor_;
    return *this;
  }

  int poll_io(timeval& waitd_tv) { return ::select(this->max_descriptor_, &(fd_set_[read_op]), &(fd_set_[write_op]), nullptr, &waitd_tv); }

  int is_set(socket_native_type fd, int events) const
  {
    int retval = 0;
    if (events & socket_event::read)
      retval |= FD_ISSET(fd, &fd_set_[read_op]);
    if (events & socket_event::write)
      retval |= FD_ISSET(fd, &fd_set_[write_op]);
    if (events & socket_event::error)
      retval |= FD_ISSET(fd, &fd_set_[except_op]);
    return retval;
  }

  void set(socket_native_type fd, int events)
  {
    if (yasio__testbits(events, socket_event::read))
      FD_SET(fd, &(fd_set_[read_op]));

    if (yasio__testbits(events, socket_event::write))
      FD_SET(fd, &(fd_set_[write_op]));

    if (yasio__testbits(events, socket_event::error))
      FD_SET(fd, &(fd_set_[except_op]));

    if (max_descriptor_ < static_cast<int>(fd) + 1)
      max_descriptor_ = static_cast<int>(fd) + 1;
  }

  void unset(socket_native_type fd, int events)
  {
    if (yasio__testbits(events, socket_event::read))
      FD_CLR(fd, &(fd_set_[read_op]));

    if (yasio__testbits(events, socket_event::write))
      FD_CLR(fd, &(fd_set_[write_op]));

    if (yasio__testbits(events, socket_event::error))
      FD_CLR(fd, &(fd_set_[except_op]));
  }

  int max_descriptor() const { return max_descriptor_; }

protected:
  enum
  {
    read_op,
    write_op,
    except_op,
    max_ops,
  };
  fd_set fd_set_[max_ops];
  int max_descriptor_ = 0;
};
} // namespace inet
} // namespace yasio