#ifndef IRC_H
#define IRC_H
#include <boost/asio.hpp>
#include <boost/asio/ssl.hpp>

#include <boost/format.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/optional.hpp>
#include <boost/signals2/mutex.hpp>
#include <chrono>
#include <iomanip> // put_time
#include <string>
// #include <vector>
#include <algorithm>
#include <ctime> // time_t
#include <fstream>
#include <set>

void string_toupper(std::string &str);

std::vector<std::string> split_limit(std::string &text, int max = -1);
std::vector<std::string> irc_split(std::string &text);
std::string parse_nick(std::string &name);
void remove_channel_modes(std::string &nick);

class message_stamp {
public:
  message_stamp() { time(&stamp); }
  std::time_t stamp;
  std::vector<std::string> buffer;
};

// using error_code = boost::system::error_code;

class ircClient {
  using error_code = boost::system::error_code;

public:
  ircClient(boost::asio::io_context &io_context);

  // async startup
  void begin(void);

  // thread-safe write to IRC
  void write(std::string output);

  // configuration
  std::string hostname;
  std::string port;
  std::string nick;
  std::string username = "bzbz";
  std::string realname;
  std::string autojoin;

  // filename to use for logfile
  std::string debug_output;
  std::ofstream debug_file;

protected:
  boost::signals2::mutex talkto_lock;
  std::string _talkto;

public:
  std::string talkto(void) {
    talkto_lock.lock();
    std::string ret{_talkto};
    talkto_lock.unlock();
    return ret;
  };

  void talkto(std::string talkvalue) {
    talkto_lock.lock();
    _talkto = talkvalue;
    talkto_lock.unlock();
  };

  // channels / users
  boost::signals2::mutex channels_lock;
  std::map<std::string, std::set<std::string>> channels;

  /*
    std::set<std::string> channels;
    std::set<std::string> users;
      std::vector<std::string> motd;
  */

  // thread-safe buffer access
  /*
  void buffer_append(std::vector<std::string> &data);
  int buffer_size(void);
  std::vector<std::string> buffer_pop(void);
  boost::optional<std::vector<std::string>> buffer_maybe_pop(void);
  */
  void message(std::string msg);
  std::atomic<bool> shutdown;

  // thread-safe messages access
  void message_append(message_stamp &msg);
  boost::optional<message_stamp> message_pop(void);

private:
  boost::signals2::mutex lock;
  std::vector<std::vector<std::string>> buffer;
  std::vector<message_stamp> messages;

  bool registered;
  std::string original_nick;
  int nick_retry;

  bool logging;
  std::ofstream &log(void);

  // async callbacks
  void on_resolve(error_code error,
                  boost::asio::ip::tcp::resolver::results_type results);
  void on_connect(error_code error,
                  boost::asio::ip::tcp::endpoint const &endpoint);

  void on_handshake(error_code error);
  void on_write(error_code error, std::size_t bytes_transferred);
  void read_until(error_code error, std::size_t bytes);
  void on_shutdown(error_code error);
  // end async callback

  void receive(std::string &text);

  std::string registration(void);

  // initialization order matters for socket, ssl_context!
  //
  // socket up here causes segmentation fault!
  // boost::asio::ssl::stream<ip::tcp::socket> socket;

  boost::asio::ip::tcp::resolver resolver;
  boost::asio::ssl::context ssl_context;
  boost::asio::streambuf response;
  boost::asio::ssl::stream<boost::asio::ip::tcp::socket> socket;

  boost::asio::io_context &context;
};

#endif