#include <hyprland/src/Compositor.hpp>
#include <hyprland/src/plugins/PluginAPI.hpp>

#include "globals.hpp"

namespace selection_hook {
inline CFunctionHook* g_LastSelectionHook = nullptr;

void hook_updateDecos(void* thisptr, PHLWINDOW window) {
	bool explicitly_selected = g_Hy3Layout->shouldRenderSelected(window);

	auto lastWindow = g_pCompositor->m_pLastWindow;
	if (explicitly_selected) {
		g_pCompositor->m_pLastWindow = window;
	}

	((void (*)(void*, PHLWINDOW)) g_LastSelectionHook->m_pOriginal)(thisptr, window);

	if (explicitly_selected) {
		g_pCompositor->m_pLastWindow = lastWindow;
	}
}

void init() {
	static const auto decoUpdateCandidates =
	    HyprlandAPI::findFunctionsByName(PHANDLE, "updateWindowAnimatedDecorationValues");

	if (decoUpdateCandidates.size() != 1) {
		g_LastSelectionHook = nullptr;

		hy3_log(
		    ERR,
		    "expected one matching function to hook for"
		    "\"updateWindowAnimatedDecorationValues\", found {}",
		    decoUpdateCandidates.size()
		);

		HyprlandAPI::addNotificationV2(
		    PHANDLE,
		    {
		        {"text",
		         "Failed to load function hooks: "
		         "\"updateWindowAnimatedDecorationValues\""},
		        {"time", (uint64_t) 10000},
		        {"color", CColor(1.0, 0.0, 0.0, 1.0)},
		        {"icon", ICON_ERROR},
		    }
		);
		return;
	}

	g_LastSelectionHook = HyprlandAPI::createFunctionHook(
	    PHANDLE,
	    decoUpdateCandidates[0].address,
	    (void*) &hook_updateDecos
	);
}

void enable() {
	if (g_LastSelectionHook != nullptr) {
		g_LastSelectionHook->hook();
	}
}

void disable() {
	if (g_LastSelectionHook != nullptr) {
		g_LastSelectionHook->unhook();
	}
}
} // namespace selection_hook