// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "remoting/host/win/rdp_desktop_session.h"

#include <memory>

#include "base/strings/utf_string_conversions.h"
#include "remoting/base/auto_thread_task_runner.h"
#include "remoting/host/base/screen_resolution.h"
#include "remoting/host/win/chromoting_module.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"

namespace remoting {

RdpDesktopSession::RdpDesktopSession() {}

RdpDesktopSession::~RdpDesktopSession() {}

STDMETHODIMP RdpDesktopSession::Connect(
    long width,
    long height,
    long dpi_x,
    long dpi_y,
    BSTR terminal_id,
    DWORD port_number,
    IRdpDesktopSessionEventHandler* event_handler) {
  event_handler_ = event_handler;

  scoped_refptr<AutoThreadTaskRunner> task_runner =
      ChromotingModule::task_runner();
  DCHECK(task_runner->BelongsToCurrentThread());

  client_ = std::make_unique<RdpClient>(
      task_runner, task_runner,
      ScreenResolution(webrtc::DesktopSize(width, height),
                       webrtc::DesktopVector(dpi_x, dpi_y)),
      base::WideToUTF8(terminal_id), port_number, this);
  return S_OK;
}

STDMETHODIMP RdpDesktopSession::Disconnect() {
  client_.reset();
  event_handler_ = nullptr;
  return S_OK;
}

STDMETHODIMP RdpDesktopSession::ChangeResolution(long width,
                                                 long height,
                                                 long dpi_x,
                                                 long dpi_y) {
  if (client_) {
    client_->ChangeResolution(
        ScreenResolution(webrtc::DesktopSize(width, height),
                         webrtc::DesktopVector(dpi_x, dpi_y)));
  }
  return S_OK;
}

STDMETHODIMP RdpDesktopSession::InjectSas() {
  if (client_) {
    client_->InjectSas();
  }
  return S_OK;
}

void RdpDesktopSession::OnRdpConnected() {
  HRESULT result = event_handler_->OnRdpConnected();
  CHECK(SUCCEEDED(result)) << "OnRdpConnected() failed: 0x" << std::hex
                           << result << std::dec << ".";
}

void RdpDesktopSession::OnRdpClosed() {
  HRESULT result = event_handler_->OnRdpClosed();
  CHECK(SUCCEEDED(result)) << "OnRdpClosed() failed: 0x" << std::hex << result
                           << std::dec << ".";
}

}  // namespace remoting