#include "cru/win/native/native_window.hpp"

#include "cru/common/format.hpp"
#include "cru/common/logger.hpp"
#include "cru/win/graph/direct/graph_factory.hpp"
#include "cru/win/native/cursor.hpp"
#include "cru/win/native/exception.hpp"
#include "cru/win/native/ui_application.hpp"
#include "cru/win/native/window_class.hpp"
#include "cru/win/native/window_render_target.hpp"
#include "dpi_util.hpp"
#include "window_d2d_painter.hpp"
#include "window_manager.hpp"

#include <cassert>
#include <windowsx.h>

namespace cru::platform::native::win {
inline Point PiToDip(const POINT& pi_point) {
  return Point(PixelToDipX(pi_point.x), PixelToDipY(pi_point.y));
}

WinNativeWindow::WinNativeWindow(WinUiApplication* application,
                                 std::shared_ptr<WindowClass> window_class,
                                 DWORD window_style, WinNativeWindow* parent) {
  assert(application);  // application can't be null.
  assert(parent == nullptr ||
         parent->IsValid());  // Parent window is not valid.

  application_ = application;
  parent_window_ = parent;

  const auto window_manager = application->GetWindowManager();

  hwnd_ = CreateWindowExW(
      0, window_manager->GetGeneralWindowClass()->GetName(), L"", window_style,
      CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
      parent == nullptr ? nullptr : parent->GetWindowHandle(), nullptr,
      application->GetInstanceHandle(), nullptr);

  if (hwnd_ == nullptr)
    throw Win32Error(::GetLastError(), "Failed to create window.");

  window_manager->RegisterWindow(hwnd_, this);

  window_render_target_.reset(new WindowRenderTarget(
      graph::win::direct::DirectGraphFactory::GetInstance(), hwnd_));
}

WinNativeWindow::~WinNativeWindow() {
  if (IsValid()) {
    SetDeleteThisOnDestroy(false);  // avoid double delete.
    Close();
  }
}

bool WinNativeWindow::IsValid() { return hwnd_ != nullptr; }

void WinNativeWindow::SetDeleteThisOnDestroy(bool value) {
  delete_this_on_destroy_ = value;
}

void WinNativeWindow::Close() {
  if (IsValid()) DestroyWindow(hwnd_);
}

bool WinNativeWindow::IsVisible() {
  if (IsValid()) return ::IsWindowVisible(hwnd_);
  return false;
}

void WinNativeWindow::SetVisible(bool is_visible) {
  if (!IsValid()) return;
  is_visible ? ShowWindow(hwnd_, SW_SHOWNORMAL) : ShowWindow(hwnd_, SW_HIDE);
}
Size WinNativeWindow::GetClientSize() {
  if (!IsValid()) return Size{};

  const auto pixel_rect = GetClientRectPixel();
  return Size(PixelToDipX(pixel_rect.right), PixelToDipY(pixel_rect.bottom));
}

void WinNativeWindow::SetClientSize(const Size& size) {
  if (IsValid()) {
    const auto window_style =
        static_cast<DWORD>(GetWindowLongPtr(hwnd_, GWL_STYLE));
    const auto window_ex_style =
        static_cast<DWORD>(GetWindowLongPtr(hwnd_, GWL_EXSTYLE));

    RECT rect;
    rect.left = 0;
    rect.top = 0;
    rect.right = DipToPixelX(size.width);
    rect.bottom = DipToPixelY(size.height);
    if (!AdjustWindowRectEx(&rect, window_style, FALSE, window_ex_style))
      throw Win32Error(::GetLastError(),
                       "Failed to invoke AdjustWindowRectEx.");

    if (!SetWindowPos(hwnd_, nullptr, 0, 0, rect.right - rect.left,
                      rect.bottom - rect.top, SWP_NOZORDER | SWP_NOMOVE))
      throw Win32Error(::GetLastError(), "Failed to invoke SetWindowPos.");
  }
}

Rect WinNativeWindow::GetWindowRect() {
  if (!IsValid()) return Rect{};

  RECT rect;
  if (!::GetWindowRect(hwnd_, &rect))
    throw Win32Error(::GetLastError(), "Failed to invoke GetWindowRect.");

  return Rect::FromVertices(PixelToDipX(rect.left), PixelToDipY(rect.top),
                            PixelToDipX(rect.right), PixelToDipY(rect.bottom));
}

void WinNativeWindow::SetWindowRect(const Rect& rect) {
  if (IsValid()) {
    if (!SetWindowPos(hwnd_, nullptr, DipToPixelX(rect.left),
                      DipToPixelY(rect.top), DipToPixelX(rect.GetRight()),
                      DipToPixelY(rect.GetBottom()), SWP_NOZORDER))
      throw Win32Error(::GetLastError(), "Failed to invoke SetWindowPos.");
  }
}

Point WinNativeWindow::GetMousePosition() {
  if (IsValid()) {
    POINT p;
    if (!::GetCursorPos(&p))
      throw Win32Error(::GetLastError(), "Failed to get cursor position.");
    if (!::ScreenToClient(hwnd_, &p))
      throw Win32Error(::GetLastError(), "Failed to call ScreenToClient.");
    return PiToDip(p);
  }
  return Point{};
}

bool WinNativeWindow::CaptureMouse() {
  if (IsValid()) {
    ::SetCapture(hwnd_);
    return true;
  }
  return false;
}

bool WinNativeWindow::ReleaseMouse() {
  if (IsValid()) {
    const auto result = ::ReleaseCapture();
    return result != 0;
  }
  return false;
}

void WinNativeWindow::Repaint() {
  if (IsValid()) {
    if (!::InvalidateRect(hwnd_, nullptr, FALSE))
      throw Win32Error(::GetLastError(), "Failed to invalidate window.");
    if (!::UpdateWindow(hwnd_))
      throw Win32Error(::GetLastError(), "Failed to update window.");
  }
}

graph::Painter* WinNativeWindow::BeginPaint() {
  return new WindowD2DPainter(this);
}

void WinNativeWindow::SetCursor(std::shared_ptr<Cursor> cursor) {
  if (!IsValid()) return;
  assert(cursor);
  WinCursor* c = static_cast<WinCursor*>(cursor.get());

  auto outputError = [] {
    log::Debug(L"Failed to set cursor. Last error code: {}.", ::GetLastError());
  };

  if (!::SetClassLongPtrW(hwnd_, GCLP_HCURSOR,
                          reinterpret_cast<LONG_PTR>(c->GetHandle()))) {
    outputError();
    return;
  }

  ::POINT point;
  if (!::GetCursorPos(&point)) {
    outputError();
    return;
  }

  ::RECT rect;
  if (!::GetClientRect(hwnd_, &rect)) {
    outputError();
    return;
  }

  ::POINT lefttop{rect.left, rect.top};
  ::POINT rightbottom{rect.right, rect.bottom};
  if (!::ClientToScreen(hwnd_, &lefttop)) {
    outputError();
    return;
  }

  if (!::ClientToScreen(hwnd_, &rightbottom)) {
    outputError();
    return;
  }

  if (point.x >= lefttop.x && point.y >= lefttop.y &&
      point.x <= rightbottom.x && point.y <= rightbottom.y) {
    ::SetCursor(c->GetHandle());
  }
}

bool WinNativeWindow::HandleNativeWindowMessage(HWND hwnd, UINT msg,
                                                WPARAM w_param, LPARAM l_param,
                                                LRESULT* result) {
  WindowNativeMessageEventArgs args{
      WindowNativeMessage{hwnd, msg, w_param, l_param}};
  native_message_event_.Raise(args);
  if (args.IsHandled()) {
    *result = args.GetResult();
    return true;
  }

  switch (msg) {
    case WM_PAINT:
      OnPaintInternal();
      *result = 0;
      return true;
    case WM_ERASEBKGND:
      *result = 1;
      return true;
    case WM_SETFOCUS:
      OnSetFocusInternal();
      *result = 0;
      return true;
    case WM_KILLFOCUS:
      OnKillFocusInternal();
      *result = 0;
      return true;
    case WM_MOUSEMOVE: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseMoveInternal(point);
      *result = 0;
      return true;
    }
    case WM_LBUTTONDOWN: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseDownInternal(platform::native::MouseButton::Left, point);
      *result = 0;
      return true;
    }
    case WM_LBUTTONUP: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseUpInternal(platform::native::MouseButton::Left, point);
      *result = 0;
      return true;
    }
    case WM_RBUTTONDOWN: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseDownInternal(platform::native::MouseButton::Right, point);
      *result = 0;
      return true;
    }
    case WM_RBUTTONUP: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseUpInternal(platform::native::MouseButton::Right, point);
      *result = 0;
      return true;
    }
    case WM_MBUTTONDOWN: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseDownInternal(platform::native::MouseButton::Middle, point);
      *result = 0;
      return true;
    }
    case WM_MBUTTONUP: {
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      OnMouseUpInternal(platform::native::MouseButton::Middle, point);
      *result = 0;
      return true;
    }
    case WM_MOUSEWHEEL:
      POINT point;
      point.x = GET_X_LPARAM(l_param);
      point.y = GET_Y_LPARAM(l_param);
      ScreenToClient(hwnd, &point);
      OnMouseWheelInternal(GET_WHEEL_DELTA_WPARAM(w_param), point);
      *result = 0;
      return true;
    case WM_KEYDOWN:
      OnKeyDownInternal(static_cast<int>(w_param));
      *result = 0;
      return true;
    case WM_KEYUP:
      OnKeyUpInternal(static_cast<int>(w_param));
      *result = 0;
      return true;
    case WM_CHAR:
      OnCharInternal(static_cast<wchar_t>(w_param));
      *result = 0;
      return true;
    case WM_SIZE:
      OnResizeInternal(LOWORD(l_param), HIWORD(l_param));
      *result = 0;
      return true;
    case WM_ACTIVATE:
      if (w_param == WA_ACTIVE || w_param == WA_CLICKACTIVE)
        OnActivatedInternal();
      else if (w_param == WA_INACTIVE)
        OnDeactivatedInternal();
      *result = 0;
      return true;
    case WM_DESTROY:
      OnDestroyInternal();
      *result = 0;
      return true;
    default:
      return false;
  }
}

RECT WinNativeWindow::GetClientRectPixel() {
  RECT rect;
  if (!GetClientRect(hwnd_, &rect))
    throw Win32Error(::GetLastError(), "Failed to invoke GetClientRect.");
  return rect;
}

void WinNativeWindow::OnDestroyInternal() {
  application_->GetWindowManager()->UnregisterWindow(hwnd_);
  hwnd_ = nullptr;
  destroy_event_.Raise(nullptr);
  if (delete_this_on_destroy_)
    application_->InvokeLater([this] { delete this; });
}

void WinNativeWindow::OnPaintInternal() {
  paint_event_.Raise(nullptr);
  ValidateRect(hwnd_, nullptr);
}

void WinNativeWindow::OnResizeInternal(const int new_width,
                                       const int new_height) {
  if (!(new_width == 0 && new_height == 0)) {
    window_render_target_->ResizeBuffer(new_width, new_height);
    resize_event_.Raise(Size{PixelToDipX(new_width), PixelToDipY(new_height)});
  }
}

void WinNativeWindow::OnSetFocusInternal() {
  has_focus_ = true;
  focus_event_.Raise(true);
}

void WinNativeWindow::OnKillFocusInternal() {
  has_focus_ = false;
  focus_event_.Raise(false);
}

void WinNativeWindow::OnMouseMoveInternal(const POINT point) {
  // when mouse was previous outside the window
  if (!is_mouse_in_) {
    // invoke TrackMouseEvent to have WM_MOUSELEAVE sent.
    TRACKMOUSEEVENT tme;
    tme.cbSize = sizeof tme;
    tme.dwFlags = TME_LEAVE;
    tme.hwndTrack = hwnd_;

    TrackMouseEvent(&tme);

    is_mouse_in_ = true;
    mouse_enter_leave_event_.Raise(true);
  }

  mouse_move_event_.Raise(PiToDip(point));
}

void WinNativeWindow::OnMouseLeaveInternal() {
  is_mouse_in_ = false;
  mouse_enter_leave_event_.Raise(false);
}

void WinNativeWindow::OnMouseDownInternal(platform::native::MouseButton button,
                                          POINT point) {
  const auto dip_point = PiToDip(point);
  mouse_down_event_.Raise({button, dip_point});
}

void WinNativeWindow::OnMouseUpInternal(platform::native::MouseButton button,
                                        POINT point) {
  const auto dip_point = PiToDip(point);
  mouse_up_event_.Raise({button, dip_point});
}

void WinNativeWindow::OnMouseWheelInternal(short delta, POINT point) {}

void WinNativeWindow::OnKeyDownInternal(int virtual_code) {
  key_down_event_.Raise(virtual_code);
}

void WinNativeWindow::OnKeyUpInternal(int virtual_code) {
  key_up_event_.Raise(virtual_code);
}

void WinNativeWindow::OnCharInternal(wchar_t c) {}

void WinNativeWindow::OnActivatedInternal() {}

void WinNativeWindow::OnDeactivatedInternal() {}
}  // namespace cru::platform::native::win