/*
 * Copyright (c) 2021-2022 Cisco and/or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <hicn/apps/utils/logger.h>
#include <hicn/http-proxy/http_proxy.h>

using namespace transport;

int usage(char* program) {
  LoggerInfo() << "USAGE: " << program << "[-C|-S] [options] <http_prefix>\n"
               << "Server or Client: \n"
               << "  -P [FIRST_IPv6_WORD_HEX]\n"
               << "  -t [number of threads]\n"
               << "Client Options: \n"
               << "  -L [PROXY_LISTEN_PORT]\n"
               << "Server Options: \n"
               << "  -a [ORIGIN_IP_ADDRESS]\n"
               << "  -p [ORIGIN_PORT]\n"
               << "  -c [CACHE_SIZE]\n"
               << "  -m [MTU]"
               << "  -l [DEFAULT_CONTENT_LIFETIME] (seconds)\n"
               << "  -M (enable manifest)\n";
  LoggerInfo() << "Example Server:\n"
               << "  " << program
               << " -S -a example.com -p 80 -c 10000 -m 1300 -l 7200 -M -t 1 "
                  "http://httpserver\n"
               << "Example Client:\n"
               << "  " << program << " -C -L 9091 http://httpserver\n";
  return -1;
}

struct Params : HTTPProxy::ClientParams, HTTPProxy::ServerParams {
  void printParams() override {
    if (client) {
      HTTPProxy::ClientParams::printParams();
    } else if (server) {
      HTTPProxy::ServerParams::printParams();
    } else {
      throw std::runtime_error(
          "Proxy configured as client and server at the same time.");
    }

    LoggerInfo() << "\tN Threads: " << n_thread;
  }

  HTTPProxy* instantiateProxyAsValue() {
    if (client) {
      HTTPProxy::ClientParams* p = dynamic_cast<HTTPProxy::ClientParams*>(this);
      return new transport::HTTPProxy(*p, n_thread);
    } else if (server) {
      HTTPProxy::ServerParams* p = dynamic_cast<HTTPProxy::ServerParams*>(this);
      return new transport::HTTPProxy(*p, n_thread);
    } else {
      throw std::runtime_error(
          "Proxy configured as client and server at the same time.");
    }
  }

  bool client = false;
  bool server = false;
  std::uint16_t n_thread = 1;
};

int main(int argc, char** argv) {
  Params params;

  params.prefix = "http://hicn-http-proxy";
  params.origin_address = "127.0.0.1";
  params.origin_port = "80";
  params.cache_size = "50000";
  params.mtu = "1500";
  params.first_ipv6_word = "b001";
  params.content_lifetime = "7200;";  // seconds
  params.manifest = false;
  params.tcp_listen_port = 8080;

  int opt;
  while ((opt = getopt(argc, argv, "CSa:p:c:m:P:l:ML:t:")) != -1) {
    switch (opt) {
      case 'C':
        if (params.server) {
          LoggerErr() << "Cannot be both client and server (both -C anc -S "
                         "options specified.).";
          return usage(argv[0]);
        }
        params.client = true;
        break;
      case 'S':
        if (params.client) {
          LoggerErr() << "Cannot be both client and server (both -C anc -S "
                         "options specified.).";
          return usage(argv[0]);
        }
        params.server = true;
        break;
      case 'a':
        params.origin_address = optarg;
        break;
      case 'p':
        params.origin_port = optarg;
        break;
      case 'c':
        params.cache_size = optarg;
        break;
      case 'm':
        params.mtu = optarg;
        break;
      case 'P':
        params.first_ipv6_word = optarg;
        break;
      case 'l':
        params.content_lifetime = optarg;
        break;
      case 'L':
        params.tcp_listen_port = std::stoul(optarg);
        break;
      case 'M':
        params.manifest = true;
        break;
      case 't':
        params.n_thread = std::stoul(optarg);
        break;
      case 'h':
      default:
        return usage(argv[0]);
    }
  }

  if (argv[optind] == 0) {
    LoggerInfo() << "Using default prefix " << params.prefix;
  } else {
    params.prefix = argv[optind];
  }

  params.printParams();
  auto proxy = params.instantiateProxyAsValue();
  proxy->run();
  delete proxy;

  return 0;
}