main.cpp

/*
 * Most comments are located in main.cpp and SRecv.cpp files so I don't repeat myself in every file.
*/
#include <cstdlib>
#include <iostream>
#include <string>
#include "echelonheaders.h"

// Port and bufSize declaration
int PORT = 7777;
int BUFFER_SIZE = 4096;



int parseInt(const std::string& value, const std::string& flag) {
  try {
    return std::stoi(value);
  }
  catch (...) {
    std::cerr << "Invalid value for " << flag << ": " << value << "\n";
    exit(1);
  }
}


double parseDouble(const std::string& value, const std::string& flag) {
  try {
    return std::stod(value);
  }
  catch (...) {
    std::cerr << "Invalid value for " << flag << ": " << value << "\n";
    exit(1);
  }
}



Config parseArgs(int argc, char** argv) {
  Config cfg;

  if (argc < 3) exit(1);

  cfg.mode = argv[1]; // ./ecf client <-
  cfg.action = argv[2]; // ./ecf client send <-

  int i=3; // initial argc argument 
  

  // Mode and action arguments parsing
  if(cfg.mode == "server" && cfg.action == "recv"){}

  else if(cfg.mode == "server" && cfg.action == "send") {

    if(i >= argc) {
      std::cerr << "Usage: \necf server send <file>\n";
      exit(1);
    }

    cfg.file = argv[i++]; // takes argv[3] as file: ./ecf client send video.mp4 <- , then increments I
  }

  else if(cfg.mode == "client" && cfg.action == "send") {

    if(i + 1 >= argc) {
      std::cerr << "Usage: \necf client send <file> <ip>\n";
      exit(1);
    }

    cfg.file = argv[i++];
    cfg.ip = argv[i++]; // argv[4] as ip: ./ecf client send video.mp4 dwpo.xyz <-
  }

  else if(cfg.mode == "client" && cfg.action == "recv") {

    if(i >= argc) {
      std::cerr << "Usage: \necf client recv <ip>\n";
      exit(1);
    }

    cfg.ip = argv[i++];
  }

  else {
    std::cerr << "Unknown mode/action\n";
    exit(1);
  }



  // Additional flags parsing
  for(; i < argc; i++) {
    std::string arg = argv[i];

    if(arg == "--always" || arg == "-a") {
      cfg.continuous = true;
    }

    else if (arg == "--passkey" && i+1 < argc) {
      cfg.passkey = argv[++i];
    }

    else if (arg == "--speed" && i+1 < argc) {
      cfg.speed = parseDouble(argv[++i], arg);
    }

    else if (arg == "--port" && i+1 < argc) {
      cfg.port = parseInt(argv[++i], arg);
    }

    else if (arg == "--buffer" && i+1 < argc) {
      cfg.bufSize = parseInt(argv[++i], arg);
    }

    else if (arg == "-n" && i+1 < argc) {
      cfg.attemptAmount = parseInt(argv[++i], arg);
    }

    else if (arg == "-i" && i+1 < argc) {
      cfg.attemptInterval = parseInt(argv[++i], arg);
    }

    else if (arg.find("--passkey=") == 0) {
      if(arg.substr(10).length() < 1) {
        std::cerr << "Invalid passkey.\n";
        exit(1);
      }
      else {
        cfg.passkey = arg.substr(10);
      }
    }

    else if (arg.find("--speed=") == 0) {
      cfg.speed = parseDouble(arg.substr(8), arg);
    } 

    else if (arg.find("--port=") == 0) {
      cfg.port = parseInt(arg.substr(7), arg);
    }

    else if (arg.find("--buffer=") == 0) {
      cfg.bufSize = parseInt(arg.substr(9), arg);
    }

    else if (arg.find("-n=") == 0) {
      cfg.attemptAmount = parseInt(arg.substr(3), arg);
    }

    else if (arg.find("-i=") == 0) {
      cfg.attemptInterval = parseInt(arg.substr(3), arg);
    }

    else {
      std::cout << "Unknown argument: " << arg << "\n";
      exit(1);
    }
  }

  return cfg;
}



int main(int argc, char* argv[]) {

  // --help

  if(argc < 3) {
    std::cout << "\n----server----\n";
    printServerHelp(argc, argv);
    std::cout << "\n----client----\n";
    printClientHelp(argc, argv);
    return 1;
  }

  Config cfg = parseArgs(argc, argv);

  PORT = cfg.port;
  BUFFER_SIZE = cfg.bufSize;
  

  if(cfg.port < 1 || cfg.port > 65535) {
    std::cerr << "Port must be between 1 and 65535\n";
    return 1;
  }

  if (cfg.bufSize < 256) {
    std::cerr << "Buffer too small\n";
    return 1;
  }


  if(cfg.mode == "server") {
    if(cfg.action == "recv") {
      if(cfg.continuous) {
        while(true) {
          serverRecv(cfg);
        }
      }

      return serverRecv(cfg);
    }

    else if(cfg.action == "send") {
      return serverSend(cfg);
    }
  }

  else if(cfg.mode == "client") {
    if(cfg.action == "send") {
      return clientSend(cfg);
    }
    
    else if(cfg.action == "recv") {
      if(cfg.continuous) {
        while(true) {
          clientRecv(cfg);
        }
      }
      return clientRecv(cfg);
    }
  }

  return 0;
}