#include "animations.hpp"

#include <format>
#include <ranges>

#include "log.hpp"
#include "strings.hpp"
#include "toast.hpp"
#include "vector_.hpp"

using namespace anm2ed::util;
using namespace anm2ed::resource;
using namespace anm2ed::types;

namespace anm2ed::imgui
{
  void Animations::update(Manager& manager, Settings& settings, Resources& resources, Clipboard& clipboard)
  {
    auto& document = *manager.get();
    auto& anm2 = document.anm2;
    auto& reference = document.reference;
    auto& selection = document.animation.selection;
    auto& mergeSelection = document.merge.selection;
    auto& mergeReference = document.merge.reference;
    auto& overlayIndex = document.overlayIndex;

    auto rename_format_get = [&](int index)
    { return std::format("###Document #{} Animation #{}", manager.selected, index); };

    auto rename = [&]()
    {
      if (!selection.empty()) renameQueued = *selection.begin();
    };

    auto add = [&]()
    {
      auto behavior = [&]()
      {
        anm2::Animation animation;
        animation.name = localize.get(TEXT_NEW_ANIMATION);
        if (anm2::Animation* referenceAnimation = document.animation_get())
        {
          for (auto [id, layerAnimation] : referenceAnimation->layerAnimations)
            animation.layerAnimations[id] = anm2::Item();
          animation.layerOrder = referenceAnimation->layerOrder;
          for (auto [id, nullAnimation] : referenceAnimation->nullAnimations)
            animation.nullAnimations[id] = anm2::Item();
        }
        animation.rootAnimation.frames.emplace_back(anm2::Frame());

        auto index = (int)anm2.animations.items.size();
        if (!selection.empty())
        {
          index = *selection.rbegin() + 1;
          index = std::min(index, (int)anm2.animations.items.size());
        }

        if (anm2.animations.items.empty()) anm2.animations.defaultAnimation = animation.name;

        anm2.animations.items.insert(anm2.animations.items.begin() + index, animation);
        selection = {index};
        reference = {index};
        newAnimationSelectedIndex = index;
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_ADD_ANIMATION), Document::ANIMATIONS, behavior());
    };

    auto remove = [&]()
    {
      auto behavior = [&]()
      {
        if (!selection.empty())
        {
          for (auto it = selection.rbegin(); it != selection.rend(); ++it)
          {
            auto i = *it;
            if (overlayIndex == i) overlayIndex = -1;
            if (reference.animationIndex == i) reference.animationIndex = -1;
            anm2.animations.items.erase(anm2.animations.items.begin() + i);
          }
          selection.clear();
        }
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_REMOVE_ANIMATIONS), Document::ANIMATIONS, behavior());
    };

    auto duplicate = [&]()
    {
      auto behavior = [&]()
      {
        auto duplicated = selection;
        auto end = std::ranges::max(duplicated);
        for (auto& id : duplicated)
        {
          anm2.animations.items.insert(anm2.animations.items.begin() + end, anm2.animations.items[id]);
          selection.insert(++end);
          selection.erase(id);
        }
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_REMOVE_ANIMATIONS), Document::ANIMATIONS, behavior());
    };

    auto merge = [&]()
    {
      auto behavior = [&]()
      {
        if (mergeSelection.contains(overlayIndex)) overlayIndex = -1;
        auto merged = anm2.animations_merge(mergeReference, mergeSelection, (merge::Type)settings.mergeType,
                                            settings.mergeIsDeleteAnimationsAfter);

        selection = {merged};
        reference = {merged};
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_MERGE_ANIMATIONS), Document::ANIMATIONS, behavior());
    };

    auto merge_popup_open = [&]()
    {
      mergePopup.open();
      mergeSelection.clear();
      mergeReference = *selection.begin();
    };

    auto merge_quick = [&]()
    {
      auto behavior = [&]()
      {
        int merged{};
        if (selection.contains(overlayIndex)) overlayIndex = -1;

        if (selection.size() > 1)
          merged = anm2.animations_merge(*selection.begin(), selection);
        else if (selection.size() == 1 && *selection.begin() != (int)anm2.animations.items.size() - 1)
        {
          auto start = *selection.begin();
          auto next = *selection.begin() + 1;
          std::set<int> animationSet{};
          animationSet.insert(start);
          animationSet.insert(next);

          merged = anm2.animations_merge(start, animationSet);
        }
        else
          return;

        selection = {merged};
        reference = {merged};
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_MERGE_ANIMATIONS), Document::ANIMATIONS, behavior());
    };

    auto default_set = [&]()
    {
      DOCUMENT_EDIT(document, localize.get(EDIT_DEFAULT_ANIMATION), Document::ANIMATIONS,
                    anm2.animations.defaultAnimation = anm2.animations.items[*selection.begin()].name);
    };

    auto copy = [&]()
    {
      if (selection.empty()) return;

      std::string clipboardText{};
      for (auto& i : selection)
        clipboardText += anm2.animations.items[i].to_string();
      clipboard.set(clipboardText);
    };

    auto cut = [&]()
    {
      copy();
      DOCUMENT_EDIT(document, localize.get(EDIT_CUT_ANIMATIONS), Document::ANIMATIONS, remove());
    };

    auto paste = [&]()
    {
      if (clipboard.is_empty()) return;

      auto behavior = [&]()
      {
        auto clipboardText = clipboard.get();
        auto start = selection.empty() ? anm2.animations.items.size() : *selection.rbegin() + 1;
        std::set<int> indices{};
        std::string errorString{};
        if (anm2.animations_deserialize(clipboardText, start, indices, &errorString))
        {
          if (!indices.empty())
          {
            auto index = *indices.rbegin();
            selection = {index};
            reference = {index};
            newAnimationSelectedIndex = index;
          }
        }
        else
        {
          toasts.push(
              std::vformat(localize.get(TOAST_DESERIALIZE_ANIMATIONS_FAILED), std::make_format_args(errorString)));
          logger.error(std::vformat(localize.get(TOAST_DESERIALIZE_ANIMATIONS_FAILED, anm2ed::ENGLISH),
                                    std::make_format_args(errorString)));
        }
      };

      DOCUMENT_EDIT(document, localize.get(EDIT_PASTE_ANIMATIONS), Document::ANIMATIONS, behavior());
    };

    if (ImGui::Begin(localize.get(LABEL_ANIMATIONS_WINDOW), &settings.windowIsAnimations))
    {
      if (ImGui::IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows) && ImGui::IsKeyPressed(ImGuiKey_Escape))
        reference = {};

      auto childSize = size_without_footer_get();

      if (ImGui::BeginChild("##Animations Child", childSize, ImGuiChildFlags_Borders))
      {

        selection.start(anm2.animations.items.size());

        for (auto [i, animation] : std::views::enumerate(anm2.animations.items))
        {
          ImGui::PushID((int)i);

          auto isDefault = anm2.animations.defaultAnimation == animation.name;
          auto isReferenced = reference.animationIndex == (int)i;
          auto isNewAnimation = newAnimationSelectedIndex == (int)i;

          auto font = isDefault && isReferenced ? font::BOLD_ITALICS
                      : isDefault               ? font::BOLD
                      : isReferenced            ? font::ITALICS
                                                : font::REGULAR;

          ImGui::PushFont(resources.fonts[font].get(), font::SIZE);
          ImGui::SetNextItemSelectionUserData((int)i);

          if (isNewAnimation || renameQueued == i)
          {
            renameState = RENAME_FORCE_EDIT;
            renameQueued = -1;
          }
          if (selectable_input_text(animation.name, rename_format_get(i), animation.name, selection.contains((int)i),
                                    ImGuiSelectableFlags_None, renameState))
          {
            reference = {(int)i};
            document.frames.clear();

            if (renameState == RENAME_BEGIN)
              document.snapshot(localize.get(SNAPSHOT_RENAME_ANIMATION));
            else if (renameState == RENAME_FINISHED)
            {
              if (anm2.animations.items.size() == 1) anm2.animations.defaultAnimation = animation.name;
              document.change(Document::ANIMATIONS);
            }
          }

          if (isNewAnimation)
          {
            isUpdateScroll = true;
            newAnimationSelectedIndex = -1;
          }
          ImGui::PopFont();

          if (isUpdateScroll && isReferenced)
          {
            ImGui::SetScrollHereY(0.5f);
            isUpdateScroll = false;
          }

          if (ImGui::BeginItemTooltip())
          {
            ImGui::PushFont(resources.fonts[font::BOLD].get(), font::SIZE);
            ImGui::TextUnformatted(animation.name.c_str());
            ImGui::PopFont();

            if (isDefault)
            {
              ImGui::PushFont(resources.fonts[font::ITALICS].get(), font::SIZE);
              ImGui::TextUnformatted(localize.get(BASIC_DEFAULT));
              ImGui::PopFont();
            }

            ImGui::TextUnformatted(
                std::vformat(localize.get(FORMAT_LENGTH), std::make_format_args(animation.frameNum)).c_str());
            auto loopLabel = animation.isLoop ? localize.get(BASIC_YES) : localize.get(BASIC_NO);
            ImGui::TextUnformatted(std::vformat(localize.get(FORMAT_LOOP), std::make_format_args(loopLabel)).c_str());

            ImGui::EndTooltip();
          }

          if (ImGui::BeginDragDropSource())
          {
            static std::vector<int> dragDropSelection{};
            dragDropSelection.assign(selection.begin(), selection.end());
            ImGui::SetDragDropPayload("Animation Drag Drop", dragDropSelection.data(),
                                      dragDropSelection.size() * sizeof(int));
            for (auto& i : dragDropSelection)
              ImGui::Text("%s", anm2.animations.items[(int)i].name.c_str());
            ImGui::EndDragDropSource();
          }

          if (ImGui::BeginDragDropTarget())
          {
            if (auto payload = ImGui::AcceptDragDropPayload("Animation Drag Drop"))
            {
              auto payloadIndices = (int*)(payload->Data);
              auto payloadCount = payload->DataSize / sizeof(int);
              std::vector<int> indices(payloadIndices, payloadIndices + payloadCount);
              std::sort(indices.begin(), indices.end());
              DOCUMENT_EDIT(document, localize.get(EDIT_MOVE_ANIMATIONS), Document::ANIMATIONS,
                            selection = vector::move_indices(anm2.animations.items, indices, i));
            }
            ImGui::EndDragDropTarget();
          }

          ImGui::PopID();
        }

        selection.finish();

        if (shortcut(manager.chords[SHORTCUT_RENAME], shortcut::FOCUSED)) rename();
        if (shortcut(manager.chords[SHORTCUT_MERGE], shortcut::FOCUSED)) merge_quick();
        if (shortcut(manager.chords[SHORTCUT_CUT], shortcut::FOCUSED)) cut();
        if (shortcut(manager.chords[SHORTCUT_COPY], shortcut::FOCUSED)) copy();
        if (shortcut(manager.chords[SHORTCUT_PASTE], shortcut::FOCUSED)) paste();

        if (ImGui::BeginPopupContextWindow("##Context Menu", ImGuiPopupFlags_MouseButtonRight))
        {
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_UNDO), settings.shortcutUndo.c_str(), false,
                              document.is_able_to_undo()))
            document.undo();

          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_REDO), settings.shortcutRedo.c_str(), false,
                              document.is_able_to_redo()))
            document.redo();

          ImGui::Separator();

          if (ImGui::MenuItem(localize.get(BASIC_RENAME), settings.shortcutRename.c_str(), false,
                              selection.size() == 1))
            rename();
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_ADD), settings.shortcutAdd.c_str())) add();
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_DUPLICATE), settings.shortcutDuplicate.c_str())) duplicate();
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_MERGE), settings.shortcutMerge.c_str())) merge_quick();
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_REMOVE), settings.shortcutRemove.c_str())) remove();
          if (ImGui::MenuItem(localize.get(SHORTCUT_STRING_DEFAULT), settings.shortcutDefault.c_str(), false,
                              selection.size() == 1))
            default_set();

          ImGui::Separator();

          if (ImGui::MenuItem(localize.get(BASIC_CUT), settings.shortcutCut.c_str(), false, !selection.empty())) cut();
          if (ImGui::MenuItem(localize.get(BASIC_COPY), settings.shortcutCopy.c_str(), false, !selection.empty()))
            copy();
          if (ImGui::MenuItem(localize.get(BASIC_PASTE), settings.shortcutPaste.c_str(), false, !clipboard.is_empty()))
            paste();
          ImGui::EndPopup();
        }
      }
      ImGui::EndChild();

      auto widgetSize = widget_size_with_row_get(5);

      shortcut(manager.chords[SHORTCUT_ADD]);
      if (ImGui::Button(localize.get(BASIC_ADD), widgetSize)) add();
      set_item_tooltip_shortcut(localize.get(TOOLTIP_ADD_ANIMATION), settings.shortcutAdd);

      ImGui::SameLine();

      ImGui::BeginDisabled(selection.empty());
      shortcut(manager.chords[SHORTCUT_DUPLICATE]);
      if (ImGui::Button(localize.get(BASIC_DUPLICATE), widgetSize)) duplicate();
      ImGui::EndDisabled();
      set_item_tooltip_shortcut(localize.get(TOOLTIP_DUPLICATE_ANIMATION), settings.shortcutDuplicate);

      ImGui::SameLine();

      ImGui::BeginDisabled(selection.size() != 1);
      if (ImGui::Button(localize.get(LABEL_MERGE), widgetSize)) merge_popup_open();
      ImGui::EndDisabled();
      set_item_tooltip_shortcut(localize.get(TOOLTIP_OPEN_MERGE_POPUP), settings.shortcutMerge);

      ImGui::SameLine();

      ImGui::BeginDisabled(selection.empty());
      shortcut(manager.chords[SHORTCUT_REMOVE]);
      if (ImGui::Button(localize.get(BASIC_REMOVE), widgetSize)) remove();
      ImGui::EndDisabled();
      set_item_tooltip_shortcut(localize.get(TOOLTIP_REMOVE_ANIMATION), settings.shortcutRemove);

      ImGui::SameLine();

      ImGui::BeginDisabled(selection.size() != 1);
      shortcut(manager.chords[SHORTCUT_DEFAULT]);
      if (ImGui::Button(localize.get(BASIC_DEFAULT), widgetSize)) default_set();
      ImGui::EndDisabled();
      set_item_tooltip_shortcut(localize.get(TOOLTIP_SET_DEFAULT_ANIMATION), settings.shortcutDefault);

      mergePopup.trigger();

      if (ImGui::BeginPopupModal(mergePopup.label(), &mergePopup.isOpen, ImGuiWindowFlags_NoResize))
      {
        auto close = [&]()
        {
          mergeSelection.clear();
          mergePopup.close();
        };

        auto& type = settings.mergeType;
        auto& isDeleteAnimationsAfter = settings.mergeIsDeleteAnimationsAfter;

        auto footerSize = footer_size_get();
        auto optionsSize = child_size_get(2);
        auto deleteAfterSize = child_size_get();
        auto animationsSize =
            ImVec2(0, ImGui::GetContentRegionAvail().y -
                          (optionsSize.y + deleteAfterSize.y + footerSize.y + ImGui::GetStyle().ItemSpacing.y * 3));

        if (ImGui::BeginChild(localize.get(LABEL_ANIMATIONS_CHILD), animationsSize, ImGuiChildFlags_Borders))
        {
          mergeSelection.start(anm2.animations.items.size());

          for (int i = 0; i < (int)anm2.animations.items.size(); i++)
          {
            if (i == mergeReference) continue;

            auto& animation = anm2.animations.items[i];

            ImGui::PushID(i);

            ImGui::SetNextItemSelectionUserData(i);
            ImGui::Selectable(animation.name.c_str(), mergeSelection.contains(i));

            ImGui::PopID();
          }

          mergeSelection.finish();
        }
        ImGui::EndChild();

        if (ImGui::BeginChild("##Merge Options", optionsSize, ImGuiChildFlags_Borders))
        {
          auto size = ImVec2(optionsSize.x * 0.5f, optionsSize.y - ImGui::GetStyle().WindowPadding.y * 2);

          if (ImGui::BeginChild("##Merge Options 1", size))
          {
            ImGui::RadioButton(localize.get(LABEL_APPEND_FRAMES), &type, merge::APPEND);
            ImGui::RadioButton(localize.get(LABEL_PREPEND_FRAMES), &type, merge::PREPEND);
          }
          ImGui::EndChild();

          ImGui::SameLine();

          if (ImGui::BeginChild("##Merge Options 2", size))
          {
            ImGui::RadioButton(localize.get(LABEL_REPLACE_FRAMES), &type, merge::REPLACE);
            ImGui::RadioButton(localize.get(LABEL_IGNORE_FRAMES), &type, merge::IGNORE);
          }
          ImGui::EndChild();
        }
        ImGui::EndChild();

        if (ImGui::BeginChild("##Merge Delete After", deleteAfterSize, ImGuiChildFlags_Borders))
          ImGui::Checkbox(localize.get(LABEL_DELETE_ANIMATIONS_AFTER), &isDeleteAnimationsAfter);
        ImGui::EndChild();

        auto widgetSize = widget_size_with_row_get(2);

        ImGui::BeginDisabled(mergeSelection.empty());
        if (ImGui::Button(localize.get(LABEL_MERGE), widgetSize))
        {
          merge();
          close();
        }
        ImGui::EndDisabled();

        ImGui::SameLine();
        if (ImGui::Button(localize.get(LABEL_CLOSE), widgetSize)) close();

        ImGui::EndPopup();
      }
    }
    ImGui::End();

    auto isNextAnimation = shortcut(manager.chords[SHORTCUT_NEXT_ANIMATION], shortcut::GLOBAL);
    auto isPreviousAnimation = shortcut(manager.chords[SHORTCUT_PREVIOUS_ANIMATION], shortcut::GLOBAL);

    if ((isPreviousAnimation || isNextAnimation) && !anm2.animations.items.empty())
    {
      if (isPreviousAnimation)
        reference.animationIndex = glm::clamp(--reference.animationIndex, 0, (int)anm2.animations.items.size() - 1);
      if (isNextAnimation)
        reference.animationIndex = glm::clamp(++reference.animationIndex, 0, (int)anm2.animations.items.size() - 1);

      selection = {reference.animationIndex};
      isUpdateScroll = true;
    }
  }
}