// Copyright (C) 2001 - 2021 Rob Caelers & Raymond Penners
// All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

#ifdef HAVE_CONFIG_H
#  include "config.h"
#endif

#include <algorithm>
#include <cstdint>
#include <string>
#include <spdlog/spdlog.h>

#include "ui/windows/WindowsHarpoonLocker.hh"

#include "input-monitor/Harpoon.hh"

#include "debug.hh"

namespace
{
  auto get_window_title(HWND window) -> std::string
  {
    int length = GetWindowTextLengthA(window);
    std::string text(static_cast<size_t>(length) + 1, '\0');
    int copied = GetWindowTextA(window, text.data(), static_cast<int>(text.size()));
    text.resize(static_cast<size_t>(std::max(copied, 0)));
    return text;
  }
} // namespace

bool
WindowsHarpoonLocker::can_lock()
{
  return true;
}

void
WindowsHarpoonLocker::block_input(bool block)
{
  if (block)
    {
      Harpoon::block_input();
    }
  else
    {
      Harpoon::unblock_input();
    }

  UINT uPreviousState = 0;
  SystemParametersInfo(SPI_SETSCREENSAVERRUNNING, block, &uPreviousState, 0);
}

void
WindowsHarpoonLocker::prepare_lock()
{
  TRACE_ENTRY();
  active_window = GetForegroundWindow();

  if (active_window != nullptr)
    {
      HWND window = static_cast<HWND>(active_window);
      std::string text = get_window_title(window);
      TRACE_MSG("Save active window: {}", text);
      spdlog::info("Save active window: {} {}", text, reinterpret_cast<intptr_t>(window));
    }
}

void
WindowsHarpoonLocker::lock()
{
  TRACE_ENTRY();
  block_input(TRUE);
}

void
WindowsHarpoonLocker::unlock()
{
  TRACE_ENTRY();
  block_input(FALSE);
  if (active_window != nullptr)
    {
      HWND window = static_cast<HWND>(active_window);
      std::string text = get_window_title(window);
      TRACE_MSG("Restore active window: {}", text);
      spdlog::info("Restore active window: {} {}", text, reinterpret_cast<intptr_t>(window));
      SetForegroundWindow(window);
      active_window = nullptr;
    }
}
