/*
 * Copyright (c) 2017-2019 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/transport/interfaces/socket_consumer.h>

namespace transport {

namespace interface {

ConsumerSocket::ConsumerSocket(int protocol)
    : ConsumerSocket(protocol, internal_io_service_) {}

ConsumerSocket::ConsumerSocket(int protocol, asio::io_service &io_service)
    : io_service_(io_service),
      portal_(std::make_shared<Portal>(io_service_)),
      async_downloader_(),
      interest_lifetime_(default_values::interest_lifetime),
      min_window_size_(default_values::min_window_size),
      max_window_size_(default_values::max_window_size),
      current_window_size_(-1),
      max_retransmissions_(
          default_values::transport_protocol_max_retransmissions),
      /****** RAAQM Parameters ******/
      minimum_drop_probability_(default_values::minimum_drop_probability),
      sample_number_(default_values::sample_number),
      gamma_(default_values::gamma_value),
      beta_(default_values::beta_value),
      drop_factor_(default_values::drop_factor),
      /****** END RAAQM Parameters ******/
      rate_estimation_alpha_(default_values::rate_alpha),
      rate_estimation_observer_(nullptr),
      rate_estimation_choice_(0),
      is_async_(false),
      verify_signature_(false),
      content_buffer_(nullptr),
      on_interest_output_(VOID_HANDLER),
      on_interest_timeout_(VOID_HANDLER),
      on_interest_satisfied_(VOID_HANDLER),
      on_content_object_input_(VOID_HANDLER),
      on_content_object_verification_(VOID_HANDLER),
      on_content_object_(VOID_HANDLER),
      on_manifest_(VOID_HANDLER),
      on_payload_retrieved_(VOID_HANDLER),
      virtual_download_(false),
      rtt_stats_(false),
      timer_(portal_->getIoService()),
      timer_interval_milliseconds_(0) {
  switch (protocol) {
    case TransportProtocolAlgorithms::VEGAS:
      transport_protocol_ = std::make_shared<VegasTransportProtocol>(this);
      break;
    case TransportProtocolAlgorithms::CBR:
      transport_protocol_ = std::make_shared<CbrTransportProtocol>(this);
      break;
    case TransportProtocolAlgorithms::RTC:
      transport_protocol_ = std::make_shared<RTCTransportProtocol>(this);
      break;
    case TransportProtocolAlgorithms::RAAQM:
    default:
      transport_protocol_ = std::make_shared<RaaqmTransportProtocol>(this);
      break;
  }
}

ConsumerSocket::~ConsumerSocket() {
  stop();

  async_downloader_.stop();

  transport_protocol_.reset();
  portal_.reset();
}

void ConsumerSocket::connect() { portal_->connect(); }

int ConsumerSocket::consume(const Name &name,
                            utils::SharableVector<uint8_t> &receive_buffer) {
  if (transport_protocol_->isRunning()) {
    return CONSUMER_BUSY;
  }

  content_buffer_ = receive_buffer.shared_from_this();

  network_name_ = name;
  network_name_.setSuffix(0);
  is_async_ = false;

  transport_protocol_->start(receive_buffer);

  return CONSUMER_READY;
}

int ConsumerSocket::asyncConsume(
    const Name &name,
    std::shared_ptr<utils::SharableVector<uint8_t>> receive_buffer) {
  // XXX Try to move the name here, instead of copying it!!
  if (!async_downloader_.stopped()) {
    async_downloader_.add([this, receive_buffer, name]() {
      network_name_ = std::move(name);
      network_name_.setSuffix(0);
      is_async_ = true;
      transport_protocol_->start(*receive_buffer);
    });
  }

  return CONSUMER_READY;
}

void ConsumerSocket::asyncSendInterest(Interest::Ptr &&interest,
                                       Portal::ConsumerCallback *callback) {
  if (!async_downloader_.stopped()) {
    // TODO Workaround, to be fixed!
    auto i = interest.release();
    async_downloader_.add([this, i, callback]() mutable {
      Interest::Ptr _interest(i);
      portal_->setConsumerCallback(callback);
      portal_->sendInterest(std::move(_interest));
      portal_->runEventsLoop();
    });
  }
}

void ConsumerSocket::stop() {
  if (transport_protocol_->isRunning()) {
    transport_protocol_->stop();
  }

  //is_running_ = false;
}

void ConsumerSocket::resume() {
  if(!transport_protocol_->isRunning()){
    transport_protocol_->resume();
  }
}

asio::io_service &ConsumerSocket::getIoService() {
  return portal_->getIoService();
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    double socket_option_value) {
  switch (socket_option_key) {
    case MIN_WINDOW_SIZE:
      min_window_size_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case MAX_WINDOW_SIZE:
      max_window_size_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case CURRENT_WINDOW_SIZE:
      current_window_size_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case GAMMA_VALUE:
      gamma_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case BETA_VALUE:
      beta_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case DROP_FACTOR:
      drop_factor_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case MINIMUM_DROP_PROBABILITY:
      minimum_drop_probability_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case RATE_ESTIMATION_ALPHA:
      if (socket_option_value >= 0 && socket_option_value < 1) {
        rate_estimation_alpha_ = socket_option_value;
      } else {
        rate_estimation_alpha_ = ALPHA;
      }
      return SOCKET_OPTION_SET;
    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    uint32_t socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::INPUT_BUFFER_SIZE:
      input_buffer_size_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case GeneralTransportOptions::OUTPUT_BUFFER_SIZE:
      output_buffer_size_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case GeneralTransportOptions::MAX_INTEREST_RETX:
      max_retransmissions_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case GeneralTransportOptions::INTEREST_LIFETIME:
      interest_lifetime_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case ConsumerCallbacksOptions::INTEREST_RETRANSMISSION:
      if (socket_option_value == VOID_HANDLER) {
        on_interest_retransmission_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::INTEREST_EXPIRED:
      if (socket_option_value == VOID_HANDLER) {
        on_interest_timeout_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::INTEREST_SATISFIED:
      if (socket_option_value == VOID_HANDLER) {
        on_interest_satisfied_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::INTEREST_OUTPUT:
      if (socket_option_value == VOID_HANDLER) {
        on_interest_output_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::CONTENT_OBJECT_INPUT:
      if (socket_option_value == VOID_HANDLER) {
        on_content_object_input_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::CONTENT_OBJECT_TO_VERIFY:
      if (socket_option_value == VOID_HANDLER) {
        on_content_object_verification_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case ConsumerCallbacksOptions::CONTENT_RETRIEVED:
      if (socket_option_value == VOID_HANDLER) {
        on_payload_retrieved_ = VOID_HANDLER;
        return SOCKET_OPTION_SET;
      }

    case RateEstimationOptions::RATE_ESTIMATION_BATCH_PARAMETER:
      if (socket_option_value > 0) {
        rate_estimation_batching_parameter_ = socket_option_value;
      } else {
        rate_estimation_batching_parameter_ = BATCH;
      }
      return SOCKET_OPTION_SET;

    case RateEstimationOptions::RATE_ESTIMATION_CHOICE:
      if (socket_option_value > 0) {
        rate_estimation_choice_ = socket_option_value;
      } else {
        rate_estimation_choice_ = RATE_CHOICE;
      }
      return SOCKET_OPTION_SET;

    case GeneralTransportOptions::TIMER_INTERVAL:
      timer_interval_milliseconds_ = socket_option_value;
      TRANSPORT_LOGD("Ok set %d", timer_interval_milliseconds_);
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    bool socket_option_value) {
  switch (socket_option_key) {
    case OtherOptions::VIRTUAL_DOWNLOAD:
      virtual_download_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case RaaqmTransportOptions::RTT_STATS:
      rtt_stats_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case GeneralTransportOptions::VERIFY_SIGNATURE:
      verify_signature_ = socket_option_value;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    Name socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::NETWORK_NAME:
      network_name_ = socket_option_value;
      return SOCKET_OPTION_SET;
    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    std::list<Prefix> socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ConsumerContentObjectCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_OBJECT_INPUT:
      on_content_object_input_ = socket_option_value;
      ;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ProducerContentObjectCallback socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(
    int socket_option_key,
    ConsumerContentObjectVerificationCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_OBJECT_TO_VERIFY:
      on_content_object_verification_ = socket_option_value;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ConsumerInterestCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::INTEREST_RETRANSMISSION:
      on_interest_retransmission_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case ConsumerCallbacksOptions::INTEREST_OUTPUT:
      on_interest_output_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case ConsumerCallbacksOptions::INTEREST_EXPIRED:
      on_interest_timeout_ = socket_option_value;
      return SOCKET_OPTION_SET;

    case ConsumerCallbacksOptions::INTEREST_SATISFIED:
      on_interest_satisfied_ = socket_option_value;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ProducerInterestCallback socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ConsumerContentCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_RETRIEVED:
      on_payload_retrieved_ = socket_option_value;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ConsumerManifestCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::MANIFEST_INPUT:
      on_manifest_ = socket_option_value;
      return SOCKET_OPTION_SET;

    default:
      return SOCKET_OPTION_NOT_SET;
  }
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, ProducerContentCallback socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    IcnObserver *socket_option_value) {
  if (socket_option_key == RateEstimationOptions::RATE_ESTIMATION_OBSERVER) {
    rate_estimation_observer_ = socket_option_value;
    return SOCKET_OPTION_SET;
  }

  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    HashAlgorithm socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    utils::CryptoSuite socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(
    int socket_option_key, const utils::Identity &socket_option_value) {
  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    const std::string &socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::CERTIFICATE:
      key_id_ = verifier_.addKeyFromCertificate(socket_option_value);

      if (key_id_ != nullptr) {
        return SOCKET_OPTION_SET;
      }

      break;

    case DataLinkOptions::OUTPUT_INTERFACE:
      output_interface_ = socket_option_value;
      portal_->setOutputInterface(output_interface_);
      return SOCKET_OPTION_SET;
  }

  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::setSocketOption(int socket_option_key,
                                    ConsumerTimerCallback socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::TIMER_EXPIRES:
      on_timer_expires_ = socket_option_value;
      return SOCKET_OPTION_SET;
  }

  return SOCKET_OPTION_NOT_SET;
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    double &socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::MIN_WINDOW_SIZE:
      socket_option_value = min_window_size_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::MAX_WINDOW_SIZE:
      socket_option_value = max_window_size_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::CURRENT_WINDOW_SIZE:
      socket_option_value = current_window_size_;
      return SOCKET_OPTION_GET;

      // RAAQM parameters

    case RaaqmTransportOptions::GAMMA_VALUE:
      socket_option_value = gamma_;
      return SOCKET_OPTION_GET;

    case RaaqmTransportOptions::BETA_VALUE:
      socket_option_value = beta_;
      return SOCKET_OPTION_GET;

    case RaaqmTransportOptions::DROP_FACTOR:
      socket_option_value = drop_factor_;
      return SOCKET_OPTION_GET;

    case RaaqmTransportOptions::MINIMUM_DROP_PROBABILITY:
      socket_option_value = minimum_drop_probability_;
      return SOCKET_OPTION_GET;

    case RateEstimationOptions::RATE_ESTIMATION_ALPHA:
      socket_option_value = rate_estimation_alpha_;
      return SOCKET_OPTION_GET;
    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    uint32_t &socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::INPUT_BUFFER_SIZE:
      socket_option_value = input_buffer_size_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::OUTPUT_BUFFER_SIZE:
      socket_option_value = output_buffer_size_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::MAX_INTEREST_RETX:
      socket_option_value = max_retransmissions_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::INTEREST_LIFETIME:
      socket_option_value = interest_lifetime_;
      return SOCKET_OPTION_GET;

    case RaaqmTransportOptions::SAMPLE_NUMBER:
      socket_option_value = sample_number_;
      return SOCKET_OPTION_GET;

    case RateEstimationOptions::RATE_ESTIMATION_BATCH_PARAMETER:
      socket_option_value = rate_estimation_batching_parameter_;
      return SOCKET_OPTION_GET;

    case RateEstimationOptions::RATE_ESTIMATION_CHOICE:
      socket_option_value = rate_estimation_choice_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    bool &socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::ASYNC_MODE:
      socket_option_value = is_async_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::RUNNING:
      socket_option_value = transport_protocol_->isRunning();
       return SOCKET_OPTION_GET;

    case OtherOptions::VIRTUAL_DOWNLOAD:
      socket_option_value = virtual_download_;
      return SOCKET_OPTION_GET;

    case RaaqmTransportOptions::RTT_STATS:
      socket_option_value = rtt_stats_;
      return SOCKET_OPTION_GET;

    case GeneralTransportOptions::VERIFY_SIGNATURE:
      socket_option_value = verify_signature_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    Name &socket_option_value) {
  switch (socket_option_key) {
    case GeneralTransportOptions::NETWORK_NAME:
      socket_option_value = network_name_;
      return SOCKET_OPTION_GET;
    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    std::list<Prefix> &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ConsumerContentObjectCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_OBJECT_INPUT:
      socket_option_value = on_content_object_input_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ProducerContentObjectCallback &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(
    int socket_option_key,
    ConsumerContentObjectVerificationCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_OBJECT_TO_VERIFY:
      socket_option_value = on_content_object_verification_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ConsumerInterestCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::INTEREST_RETRANSMISSION:
      socket_option_value = on_interest_retransmission_;
      return SOCKET_OPTION_GET;

    case ConsumerCallbacksOptions::INTEREST_OUTPUT:
      socket_option_value = on_interest_output_;
      return SOCKET_OPTION_GET;

    case ConsumerCallbacksOptions::INTEREST_EXPIRED:
      socket_option_value = on_interest_timeout_;
      return SOCKET_OPTION_GET;

    case ConsumerCallbacksOptions::INTEREST_SATISFIED:
      socket_option_value = on_interest_satisfied_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ProducerInterestCallback &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ConsumerContentCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::CONTENT_RETRIEVED:
      socket_option_value = on_payload_retrieved_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ConsumerManifestCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::MANIFEST_INPUT:
      socket_option_value = on_manifest_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, std::shared_ptr<Portal> &socket_option_value) {
  switch (socket_option_key) {
    case PORTAL:
      socket_option_value = portal_;
      return SOCKET_OPTION_GET;

    default:
      return SOCKET_OPTION_NOT_GET;
  }
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    IcnObserver **socket_option_value) {
  if (socket_option_key == RATE_ESTIMATION_OBSERVER) {
    *socket_option_value = (rate_estimation_observer_);
    return SOCKET_OPTION_GET;
  }

  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    HashAlgorithm &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    utils::CryptoSuite &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    utils::Identity &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ProducerContentCallback &socket_option_value) {
  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(int socket_option_key,
                                    std::string &socket_option_value) {
  switch (socket_option_key) {
    case DataLinkOptions::OUTPUT_INTERFACE:
      socket_option_value = output_interface_;
      return SOCKET_OPTION_GET;
  }

  return SOCKET_OPTION_NOT_GET;
}

int ConsumerSocket::getSocketOption(
    int socket_option_key, ConsumerTimerCallback &socket_option_value) {
  switch (socket_option_key) {
    case ConsumerCallbacksOptions::TIMER_EXPIRES:
      socket_option_value = on_timer_expires_;
      return SOCKET_OPTION_GET;
  }

  return SOCKET_OPTION_NOT_GET;
}

}  // namespace interface

}  // end namespace transport