/*
 *  Copyright 2019 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */
#include "test/time_controller/real_time_controller.h"

#include <memory>
#include <string>
#include <utility>

#include "absl/strings/string_view.h"
#include "api/field_trials_view.h"
#include "api/task_queue/default_task_queue_factory.h"
#include "api/task_queue/task_queue_factory.h"
#include "api/units/time_delta.h"
#include "rtc_base/null_socket_server.h"
#include "rtc_base/socket_server.h"
#include "rtc_base/thread.h"
#include "system_wrappers/include/clock.h"

namespace webrtc {
namespace {
class MainThread : public Thread {
 public:
  MainThread()
      : Thread(std::make_unique<NullSocketServer>(), false),
        current_setter_(this) {
    DoInit();
  }
  ~MainThread() override {
    Stop();
    DoDestroy();
  }

 private:
  CurrentThreadSetter current_setter_;
};
}  // namespace
RealTimeController::RealTimeController(const FieldTrialsView* field_trials)
    : task_queue_factory_(CreateDefaultTaskQueueFactory(field_trials)),
      main_thread_(std::make_unique<MainThread>()) {
  main_thread_->SetName("Main", this);
}

Clock* RealTimeController::GetClock() {
  return Clock::GetRealTimeClock();
}

TaskQueueFactory* RealTimeController::GetTaskQueueFactory() {
  return task_queue_factory_.get();
}

std::unique_ptr<Thread> RealTimeController::CreateThread(
    const std::string& name,
    std::unique_ptr<SocketServer> socket_server) {
  if (!socket_server)
    socket_server = std::make_unique<NullSocketServer>();
  auto res = std::make_unique<Thread>(std::move(socket_server));
  res->SetName(name, nullptr);
  res->Start();
  return res;
}

std::unique_ptr<Thread> RealTimeController::CreateThreadWithSocketServer(
    absl::string_view name,
    SocketServer* socket_server) {
  std::unique_ptr<Thread> res;
  if (!socket_server) {
    res = std::make_unique<Thread>(std::make_unique<NullSocketServer>());
  } else {
    res = std::make_unique<Thread>(socket_server);
  }
  res->SetName(name, nullptr);
  res->Start();
  return res;
}

Thread* RealTimeController::GetMainThread() {
  return main_thread_.get();
}

void RealTimeController::AdvanceTime(TimeDelta duration) {
  main_thread_->ProcessMessages(duration.ms());
}

}  // namespace webrtc