/*
  Copyright <2018-2023> <scott.e.graves@protonmail.com>

  Permission is hereby granted, free of charge, to any person obtaining a copy
  of this software and associated documentation files (the "Software"), to deal
  in the Software without restriction, including without limitation the rights
  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  copies of the Software, and to permit persons to whom the Software is
  furnished to do so, subject to the following conditions:

  The above copyright notice and this permission notice shall be included in all
  copies or substantial portions of the Software.

  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  SOFTWARE.
*/
#ifdef _WIN32

#include "platform/win32_platform.hpp"

#include "providers/i_provider.hpp"
#include "utils/error_utils.hpp"

namespace repertory {
auto lock_data::get_mount_state(const provider_type & /*pt*/, json &mount_state)
    -> bool {
  const auto ret = get_mount_state(mount_state);
  if (ret) {
    const auto mount_id =
        app_config::get_provider_display_name(pt_) + unique_id_;
    mount_state = mount_state[mount_id].empty()
                      ? json({{"Active", false}, {"Location", ""}, {"PID", -1}})
                      : mount_state[mount_id];
  }

  return ret;
}

auto lock_data::get_mount_state(json &mount_state) -> bool {
  HKEY key;
  auto ret = !::RegCreateKeyEx(
      HKEY_CURRENT_USER,
      ("SOFTWARE\\" + std::string(REPERTORY_DATA_NAME) + "\\Mounts").c_str(), 0,
      nullptr, 0, KEY_ALL_ACCESS, nullptr, &key, nullptr);
  if (ret) {
    DWORD i = 0u;
    DWORD data_size = 0u;
    std::string name;
    name.resize(32767u);
    auto name_size = static_cast<DWORD>(name.size());
    while (ret &&
           (::RegEnumValue(key, i, &name[0], &name_size, nullptr, nullptr,
                           nullptr, &data_size) == ERROR_SUCCESS)) {
      std::string data;
      data.resize(data_size);
      name_size++;
      if ((ret = !::RegEnumValue(key, i++, &name[0], &name_size, nullptr,
                                 nullptr, reinterpret_cast<LPBYTE>(&data[0]),
                                 &data_size))) {
        mount_state[name.c_str()] = json::parse(data);
        name_size = static_cast<DWORD>(name.size());
        data_size = 0u;
      }
    }
    ::RegCloseKey(key);
  }
  return ret;
}

auto lock_data::grab_lock(std::uint8_t retry_count) -> lock_result {
  auto ret = lock_result::success;
  if (mutex_handle_ == INVALID_HANDLE_VALUE) {
    ret = lock_result::failure;
  } else {
    for (auto i = 0;
         (i <= retry_count) && ((mutex_state_ = ::WaitForSingleObject(
                                     mutex_handle_, 100)) == WAIT_TIMEOUT);
         i++) {
    }

    switch (mutex_state_) {
    case WAIT_OBJECT_0: {
      ret = lock_result::success;
      auto should_reset = true;
      json mount_state;
      if (get_mount_state(pt_, mount_state)) {
        if (mount_state["Active"].get<bool>() &&
            mount_state["Location"] == "elevating") {
          should_reset = false;
        }
      }

      if (should_reset) {
        if (not set_mount_state(false, "", -1)) {
          utils::error::raise_error(__FUNCTION__, "failed to set mount state");
        }
      }
    } break;

    case WAIT_TIMEOUT:
      ret = lock_result::locked;
      break;

    default:
      ret = lock_result::failure;
      break;
    }
  }

  return ret;
}

void lock_data::release() {
  if (mutex_handle_ != INVALID_HANDLE_VALUE) {
    if ((mutex_state_ == WAIT_OBJECT_0) || (mutex_state_ == WAIT_ABANDONED)) {
      ::ReleaseMutex(mutex_handle_);
    }
    ::CloseHandle(mutex_handle_);
    mutex_handle_ = INVALID_HANDLE_VALUE;
  }
}

auto lock_data::set_mount_state(bool active, const std::string &mount_location,
                                const std::int64_t &pid) -> bool {
  auto ret = false;
  if (mutex_handle_ != INVALID_HANDLE_VALUE) {
    const auto mount_id =
        app_config::get_provider_display_name(pt_) + unique_id_;
    json mount_state;
    [[maybe_unused]] auto success = get_mount_state(mount_state);
    if ((mount_state.find(mount_id) == mount_state.end()) ||
        (mount_state[mount_id].find("Active") == mount_state[mount_id].end()) ||
        (mount_state[mount_id]["Active"].get<bool>() != active) ||
        (active && ((mount_state[mount_id].find("Location") ==
                     mount_state[mount_id].end()) ||
                    (mount_state[mount_id]["Location"].get<std::string>() !=
                     mount_location)))) {
      HKEY key;
      if ((ret = !::RegCreateKeyEx(
               HKEY_CURRENT_USER,
               ("SOFTWARE\\" + std::string(REPERTORY_DATA_NAME) + "\\Mounts")
                   .c_str(),
               0, nullptr, 0, KEY_ALL_ACCESS, nullptr, &key, nullptr))) {
        const auto str = json({{"Active", active},
                               {"Location", active ? mount_location : ""},
                               {"PID", active ? pid : -1}})
                             .dump(0);
        ret = !::RegSetValueEx(key, &mount_id[0], 0, REG_SZ,
                               reinterpret_cast<const BYTE *>(&str[0]),
                               static_cast<DWORD>(str.size()));
        ::RegCloseKey(key);
      }
    } else {
      ret = true;
    }
  }

  return ret;
}

auto create_meta_attributes(
    std::uint64_t accessed_date, std::uint32_t attributes,
    std::uint64_t changed_date, std::uint64_t creation_date, bool directory,
    std::uint32_t gid, const std::string &key, std::uint32_t mode,
    std::uint64_t modified_date, std::uint32_t osx_backup,
    std::uint32_t osx_flags, std::uint64_t size, const std::string &source_path,
    std::uint32_t uid, std::uint64_t written_date) -> api_meta_map {
  return {
      {META_ACCESSED, std::to_string(accessed_date)},
      {META_ATTRIBUTES, std::to_string(attributes)},
      {META_BACKUP, std::to_string(osx_backup)},
      {META_CHANGED, std::to_string(changed_date)},
      {META_CREATION, std::to_string(creation_date)},
      {META_DIRECTORY, utils::string::from_bool(directory)},
      {META_GID, std::to_string(gid)},
      {META_KEY, key},
      {META_MODE, std::to_string(mode)},
      {META_MODIFIED, std::to_string(modified_date)},
      {META_OSXFLAGS, std::to_string(osx_flags)},
      {META_PINNED, "0"},
      {META_SIZE, std::to_string(size)},
      {META_SOURCE, source_path},
      {META_UID, std::to_string(uid)},
      {META_WRITTEN, std::to_string(written_date)},
  };
}

auto provider_meta_handler(i_provider &provider, bool directory,
                           const api_file &file) -> api_error {
  const auto meta = create_meta_attributes(
      file.accessed_date,
      directory ? FILE_ATTRIBUTE_DIRECTORY
                : FILE_ATTRIBUTE_NORMAL | FILE_ATTRIBUTE_ARCHIVE,
      file.changed_date, file.creation_date, directory, 0u, file.key,
      directory ? S_IFDIR : S_IFREG, file.modified_date, 0u, 0u, file.file_size,
      file.source_path, 0u, file.modified_date);
  auto res = provider.set_item_meta(file.api_path, meta);
  if (res == api_error::success) {
    event_system::instance().raise<filesystem_item_added>(
        file.api_path, file.api_parent, directory);
  }

  return res;
}
} // namespace repertory

#endif //_WIN32
