#include <algorithm>
#include <cstdlib>
#include <string>

#include "layout.h"
#include "localization.h"
#include "rd_log.h"
#include "render.h"

namespace crossdesk {

std::string CleanMarkdown(const std::string& markdown) {
  std::string result = markdown;

  // remove # title mark
  size_t pos = 0;
  while (pos < result.length()) {
    if (result[pos] == '\n' || pos == 0) {
      size_t line_start = (result[pos] == '\n') ? pos + 1 : pos;
      if (line_start < result.length() && result[line_start] == '#') {
        size_t hash_end = line_start;
        while (hash_end < result.length() &&
               (result[hash_end] == '#' || result[hash_end] == ' ')) {
          hash_end++;
        }

        result.erase(line_start, hash_end - line_start);
        pos = line_start;
        continue;
      }
    }
    pos++;
  }

  // remove ** bold mark
  pos = 0;
  while ((pos = result.find("**", pos)) != std::string::npos) {
    result.erase(pos, 2);
  }

  // remove all spaces
  result.erase(std::remove(result.begin(), result.end(), ' '), result.end());

  // replace . with 、
  pos = 0;
  while ((pos = result.find('.', pos)) != std::string::npos) {
    result.replace(pos, 1, "、");
    pos += 1;  // Move to next position after the replacement
  }

  return result;
}

int Render::UpdateNotificationWindow() {
  if (show_update_notification_window_ && update_available_) {
    const ImGuiViewport* viewport = ImGui::GetMainViewport();

    float update_notification_window_width = title_bar_button_width_ * 10.0f;
    float update_notification_window_height = title_bar_button_width_ * 8.0f;

    // #ifdef __APPLE__
    //     float font_scale = 0.3f;
    // #else
    //     float font_scale = 0.5f;
    // #endif

    ImGui::SetNextWindowPos(ImVec2((viewport->WorkSize.x - viewport->WorkPos.x -
                                    update_notification_window_width) /
                                       2,
                                   (viewport->WorkSize.y - viewport->WorkPos.y -
                                    update_notification_window_height) /
                                       2),
                            ImGuiCond_FirstUseEver);

    ImGui::SetNextWindowSize(ImVec2(update_notification_window_width,
                                    update_notification_window_height));

    ImGui::PushStyleColor(ImGuiCol_WindowBg, ImVec4(1.0f, 1.0f, 1.0f, 1.0f));
    ImGui::PushStyleVar(ImGuiStyleVar_WindowBorderSize, 1.0f);
    ImGui::PushStyleVar(ImGuiStyleVar_WindowRounding, 5.0f);
    ImGui::PushStyleVar(ImGuiStyleVar_FrameRounding, 3.0f);
    ImGui::Begin(
        localization::notification[localization_language_index_].c_str(),
        nullptr,
        ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoCollapse |
            ImGuiWindowFlags_NoSavedSettings | ImGuiWindowFlags_NoTitleBar);

    ImGui::SetCursorPosY(ImGui::GetCursorPosY() +
                         update_notification_window_height * 0.05f);

    // title: new version available
    ImGui::SetCursorPosX(update_notification_window_width * 0.1f);
    ImGui::SetWindowFontScale(0.55f);
    std::string title =
        localization::new_version_available[localization_language_index_] +
        ": v" + latest_version_;
    ImGui::Text("%s", title.c_str());
    ImGui::SetWindowFontScale(0.1f);

    ImGui::Spacing();

    // website link
    std::string download_text =
        localization::access_website[localization_language_index_] +
        "https://crossdesk.cn";
    ImGui::SetWindowFontScale(0.5f);
    Hyperlink(download_text, "https://crossdesk.cn",
              update_notification_window_width);
    ImGui::SetWindowFontScale(1.0f);

    ImGui::Spacing();

    float scrollable_height =
        update_notification_window_height - UPDATE_NOTIFICATION_RESERVED_HEIGHT;

    if (main_windows_system_chinese_font_ != nullptr) {
      ImGui::PushFont(main_windows_system_chinese_font_);
    }
    // scrollable content area
    ImGui::SetCursorPosX(update_notification_window_width * 0.05f);
    ImGui::BeginChild(
        "ScrollableContent",
        ImVec2(update_notification_window_width * 0.9f, scrollable_height),
        ImGuiChildFlags_Border, ImGuiWindowFlags_None);
    ImGui::SetWindowFontScale(0.5f);
    // set text wrap position to current available width (accounts for
    // scrollbar)
    float wrap_pos = ImGui::GetContentRegionAvail().x;
    ImGui::PushTextWrapPos(wrap_pos);

    // release name
    if (latest_version_info_.contains("releaseName") &&
        latest_version_info_["releaseName"].is_string() &&
        !latest_version_info_["releaseName"].empty()) {
      ImGui::SetCursorPosX(update_notification_window_width * 0.05f);
      std::string release_name =
          latest_version_info_["releaseName"].get<std::string>();
      ImGui::TextWrapped("%s", release_name.c_str());
      ImGui::Spacing();
    }

    // release notes
    if (!release_notes_.empty()) {
      ImGui::SetCursorPosX(update_notification_window_width * 0.05f);
      std::string cleaned_notes = CleanMarkdown(release_notes_);
      ImGui::TextWrapped("%s", cleaned_notes.c_str());
      ImGui::Spacing();
    }

    // release date
    if (latest_version_info_.contains("releaseDate") &&
        latest_version_info_["releaseDate"].is_string() &&
        !latest_version_info_["releaseDate"].empty()) {
      ImGui::SetCursorPosX(update_notification_window_width * 0.05f);
      std::string date_label =
          localization::release_date[localization_language_index_];
      std::string release_date = latest_version_info_["releaseDate"];
      std::string date_text = date_label + release_date;
      ImGui::Text("%s", date_text.c_str());
      ImGui::Spacing();
    }

    // pop text wrap position
    ImGui::PopTextWrapPos();

    ImGui::EndChild();

    // pop system font
    if (main_windows_system_chinese_font_ != nullptr) {
      ImGui::PopFont();
    }

    ImGui::Spacing();

    if (ConfigCenter::LANGUAGE::CHINESE == localization_language_) {
      ImGui::SetCursorPosX(update_notification_window_width * 0.407f);
    } else {
      ImGui::SetCursorPosX(update_notification_window_width * 0.367f);
    }

    ImGui::SetWindowFontScale(0.5f);
    // update button
    if (ImGui::Button(
            localization::update[localization_language_index_].c_str())) {
      // open download page
      std::string url = "https://crossdesk.cn";
#if defined(_WIN32)
      std::string cmd = "start " + url;
#elif defined(__APPLE__)
      std::string cmd = "open " + url;
#else
      std::string cmd = "xdg-open " + url;
#endif
      system(cmd.c_str());
      show_update_notification_window_ = false;
    }

    ImGui::SameLine();

    if (ImGui::Button(
            localization::cancel[localization_language_index_].c_str())) {
      show_update_notification_window_ = false;
    }

    ImGui::SetWindowFontScale(1.0f);

    ImGui::End();
    ImGui::PopStyleVar(3);
    ImGui::PopStyleColor();
  }
  return 0;
}

}  // namespace crossdesk