#include <iostream>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <functional>
#include <fstream>
#include <chrono>
#include <thread>
#include <ctime>
#include <iomanip>
#include <csignal>
#include <loguru.hpp>

const char * HOST = "192.168.30.89";
const uint32_t PORT = 9999;
const size_t MAX_EVENT_SIZE = 33484; // in bytes
const size_t MIN_EVENT_SIZE = 204; // in bytes
const size_t MAX_CHANNEL_SIZE = 4184;
const size_t MIN_CHANNEL_SIZE = 24;
const uint8_t EVENT_MARKER = 0xA5;
const uint8_t CHANNEL_MARKER = 0x5A;

int sock = 0;
bool running = true;
bool bad_data = true;

int open_socket();
void signalHandler( int signum );
void setup_logger();
size_t process_channel(char const * ptr, size_t len);
size_t process_channel(char const * ptr, size_t len){
   if ((len < MIN_CHANNEL_SIZE) || (len > MAX_CHANNEL_SIZE)) {
      LOG_F(WARNING, "channel length must be between %u and %u.",
            MIN_CHANNEL_SIZE, MAX_CHANNEL_SIZE);
      return 0;
   }

   size_t bytes_processed = 0;
   size_t bytes_left = len;
   bool valid_channel = false;
   uint32_t * wordData = (uint32_t *)ptr;

   while ((bytes_left > 0) && !valid_channel) {
      if ((ptr[0] & 0xFF) != CHANNEL_MARKER){
         LOG_F(WARNING, "bad_data, expecting %02X, received %02X",
               static_cast<uint32_t>(CHANNEL_MARKER), static_cast<uint32_t>(ptr[0]));
         bytes_left -= 4;
      } else {
         valid_channel = true;

         size_t channel_size = ((wordData[0] & 0xFFFFFF00) >> 8) * sizeof(uint32_t);
         if ((channel_size > MAX_CHANNEL_SIZE) || (channel_size < MIN_CHANNEL_SIZE)) {
            bad_data = true;
            LOG_F(WARNING, "Invalid channel size: %u", channel_size);
         }

         LOG_F(INFO, "channel size %u", channel_size);
         bytes_left -= (channel_size - 4);
      }
   }

   // return number of bytes cosumed of the channel
   return len - bytes_left;
}


int main(int argc, char * argv[])
{
   setup_logger();
   loguru::init(argc, argv);
   signal(SIGINT, signalHandler);
   int ret = open_socket();
   if (ret != 0)
      return -1;

   char char_buffer[MAX_EVENT_SIZE];
   uint32_t * wordData = (uint32_t*) char_buffer;
   ssize_t nread;

   fd_set select_fds;                // fd's used by select
   struct timeval timeout;           // Time value for time out
   //Timeout set for 0 sec + 200 millisec
   timeout.tv_sec = 0;         
   timeout.tv_usec = 200*1000;
   std::chrono::duration<int, std::milli> polling_period(2);

   while (running) {
      // Setup the descriptor set for select()
      FD_ZERO(&select_fds);            
      FD_SET(sock, &select_fds);           
      if ( select(sock + 1, &select_fds, NULL, NULL, &timeout) == 0 ){
         std::this_thread::sleep_for(polling_period);
      }
      else{
         // read in first 4 bytes
         nread = recv(sock, char_buffer, 4, 0);
         LOG_F(INFO, "read in %u bytes: 0x%08X", nread, wordData[0]);
         // if not A5 marker, go to next iteration (read in 4 more bytes)
         if ((char_buffer[0] & 0xFF) != 0xA5){
            bad_data = true;
            continue;
         }
         else {
            bad_data = false;
            size_t frameSize = 4 * (wordData[0] >> 8);
            nread = recv(sock, char_buffer, frameSize - 4, 0);
            // nread + 4 must be between MIN_EVENT_SIZE and MAX_EVENT_SIZE, and equal to frameSize.
            if ((nread != frameSize - 4) || (frameSize < MIN_EVENT_SIZE)|| (frameSize > MAX_EVENT_SIZE)){
               bad_data = true;
               continue;
            }
            else {
               // ok, this seems to be good data
               LOG_F(INFO, "next %u bytes (expecting %u): %08X %08X %08X %08X",
                     nread, frameSize - 4, wordData[0], wordData[1], wordData[2], wordData[3]);
               LOG_F(INFO, "%08X %08X %08X %08X", wordData[4], wordData[5], wordData[6], wordData[7]);
               uint8_t board_id = (wordData[0] >> 24) & 0xFF;
               uint8_t version = wordData[0] & 0xFF;
               uint16_t spill_id = (wordData[1] & 0x00FFFF00) >> 8;
               uint8_t channel_count = (wordData[1]) & 0x000000FF;
               LOG_F(INFO, "board %u, version %u, spill %u, channel_count %u",
                     static_cast<uint32_t>(board_id),
                     static_cast<uint32_t>(version),
                     static_cast<uint32_t>(spill_id),
                     static_cast<uint32_t>(channel_count));

               if ((board_id > 4) || (board_id < 1) || (version != 1) || (channel_count > 8) || (channel_count < 1)) {
                  bad_data = true;
                  continue;
               }
               else {
                  size_t bytes_left = frameSize - 3;
                  size_t bytes_processed = 0;
                  while (bytes_left >= MIN_CHANNEL_SIZE) {
                     bytes_processed = process_channel(char_buffer + 2 * 4, bytes_left);
                     bytes_left -= bytes_processed;
                  }
               }

            }
         }
      }
   }

   close(sock);
   return 0;
}

void signalHandler( int signum ) {
   RAW_LOG_F(WARNING, "Interrupt signal (%d) received, exiting...", signum);
   running = false;
}

int open_socket(){
   struct sockaddr_in serv_addr;
   if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
      std::cerr << "Cannot create socket" << std::endl;
      return -1;
   }

   memset(&serv_addr, '0', sizeof(serv_addr));
   serv_addr.sin_family = AF_INET;
   serv_addr.sin_port = htons(PORT);

   if(inet_pton(AF_INET, HOST, &serv_addr.sin_addr)<=0) {
      std::cerr << "Invalid address or Address not supported" << std::endl;
      return -1;
   }

   if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
      std::cerr << "Cannot connect to " << HOST << ":" << PORT << std::endl;
      return -1;
   }

   return 0;
}

void setup_logger(){
   loguru::g_stderr_verbosity = loguru::Verbosity_INFO;
   // loguru::g_stderr_verbosity = loguru::Verbosity_WARNING;
   loguru::g_preamble_date    = true; // The date field
	loguru::g_preamble_time    = true; // The time of the current day
	loguru::g_preamble_uptime  = true; // The time since init call
	loguru::g_preamble_thread  = false; // The logging thread
	loguru::g_preamble_file    = false; // The file from which the log originates from
	loguru::g_preamble_verbose = false; // The verbosity field
	loguru::g_preamble_pipe    = false; // The pipe symbol right before the message
}