#include "plugin_effects.h"
#include "effect_receivers.h"
#include "effect_draw_budget.h"
#include "effect_asset.h"
#include "Extension/Decals/native_decals.h"
#include "Extension/Skater/no_bail.h"
#include "Engine/Game/Abi/native_data.h"
#include "Engine/Game/Build/addresses.h"
#include "Engine/Game/Build/20260929/effects.h"
#include "Engine/Game/Build/20260929/engine.h"
#include "Engine/Game/Build/20260929/replay_activity.h"
#include "Engine/Game/Build/supported_build.h"
#include <Windows.h>
#include <algorithm>
#include <atomic>
#include <cstring>
#include <map>
#include <mutex>

namespace dingosdk::effects {
namespace {
namespace fx = addr::effects;
namespace decal = decals::native;
struct Entry {
    std::string owner;
    std::uintptr_t particle{};
    decal::Handle decal;
    ReceiverId receiver{};
    Matrix local{}, drawn{};
    std::uint64_t expires{};
    bool stopped{};
};
struct State {
    std::mutex mutex;
    std::map<std::uint64_t, Entry> entries;
    std::map<std::string, std::pair<std::uintptr_t, std::uint64_t>, std::less<>> assets;
    Receivers receivers;
    DrawBudget budget;
    std::uintptr_t base{}, service{}, manager{}, world{}, entity{};
    std::uint64_t serial{}, budget_at{}, epoch{1};
    DWORD thread{};
    std::atomic<bool> suspended{true};
    bool checked{}, valid{}, allowed{};
};
State& state() { static auto* s = new State; return *s; }
template<class F> F function(const State& s, std::uintptr_t rva) { return reinterpret_cast<F>(s.base + rva); }
std::uintptr_t pointer(std::uintptr_t address) { std::uintptr_t value{}; memory::peek(address, value); return value; }
std::uintptr_t service(const State& s) {
    const auto p = pointer(s.base + fx::service), vtable = pointer(p);
    if (!p || !vtable) return 0;
    for (const auto slot : fx::service_methods) {
        const auto method = pointer(vtable + slot);
        if (method < s.base || method >= s.base + supported_build::game_image_size) return 0;
    }
    return p;
}
bool same_services(const State& s) { return service(s) == s.service && decal::manager(s.base) == s.manager; }
void destroy(State& s, Entry& e) {
    if (e.particle) {
        function<void(*)(std::uintptr_t*, bool, float)>(s, fx::stop.rva)(&e.particle, true, 0);
        function<void(*)(std::uintptr_t*)>(s, fx::release.rva)(&e.particle);
    }
    if (e.decal.id) decal::release(s.base, e.decal);
}
void reset(State& s, bool same_world = true) {
    if (s.allowed || !s.entries.empty()) ++s.epoch;
    if (same_world && s.valid && s.thread == GetCurrentThreadId() && same_services(s))
        for (auto& [id, e] : s.entries) destroy(s, e);
    s.entries.clear(); s.assets.clear();
    if (s.valid && s.thread == GetCurrentThreadId()) (void)s.budget.update(s.base, 0);
    s.allowed = false;
}
bool usable(const State& s) { return s.valid && s.allowed && !s.suspended && s.thread == GetCurrentThreadId() && same_services(s); }
Entry* owned(State& s, std::string_view plugin, std::uint64_t id) {
    const auto it = s.entries.find(id);
    return usable(s) && it != s.entries.end() && it->second.owner == plugin ? &it->second : nullptr;
}
std::uintptr_t asset(State& s, const std::string& name) {
    if (!s.assets.contains(name) && s.assets.size() >= 512) return 0;
    auto& cached = s.assets[name];
    const auto now = GetTickCount64();
    if (cached.first || now < cached.second) return cached.first;
    cached.second = now + 2000;
    if (const auto find = game::native_data().find_asset)
        for (std::uint16_t domain = 0; domain < fx::domain_count; ++domain)
            if (pointer(s.base + addr::engine::domain_owners + sizeof(std::uintptr_t) * domain))
                if (const auto found = find(domain, name.c_str())) { cached.first = found; break; }
    return cached.first;
}
bool replaying(std::uintptr_t base) {
    namespace r = addr::replay_activity;
    const auto client = pointer(base + r::client), manager = pointer(client + r::manager);
    const auto lease = pointer(manager + r::lease);
    return lease && pointer(lease) == base + r::lease_vtable && pointer(lease + r::lease_manager) == manager;
}
}
void tick(std::uintptr_t base, std::uintptr_t client, bool ready, bool noclip, bool editor) noexcept {
    try {
        auto& s = state(); std::lock_guard lock(s.mutex);
        if (!s.thread) s.thread = GetCurrentThreadId();
        if (s.thread != GetCurrentThreadId()) return;
        if (!s.checked) {
            s.base = base; s.checked = true; s.valid = base != 0;
            for (const auto& f : {fx::descriptor_init, fx::descriptor_destroy, fx::create, fx::move, fx::stop, fx::release}) {
                std::array<unsigned char, 32> bytes{};
                s.valid = memory::read(base + f.rva, bytes) && bytes == f.bytes && s.valid;
            }
            s.valid = s.valid && decal::contracts_match(base) && decal::opacity_contracts_match(base) &&
                decal::surface_contracts_match(base) && s.receivers.initialize(base);
        }
        if (!s.valid || base != s.base) return;
        const auto current = service(s), manager = decal::manager(base);
        NoBailSkater skater;
        const bool playing = ready && !s.suspended && !noclip && !editor && !replaying(base) && no_bail_skater(skater);
        const auto world = pointer(client + fx::client_world);
        // The transition hook releases references before teardown. If a world
        // was replaced without that hook, never submit its old IDs to a new one.
        if (!playing || current != s.service || manager != s.manager || world != s.world || skater.entity != s.entity)
            reset(s, world == s.world);
        s.service = current; s.manager = manager; s.world = world; s.entity = skater.entity;
        s.allowed = playing && world && current && manager;
        if (!s.allowed) return;
        s.receivers.begin(world);
        const auto now = GetTickCount64();
        for (auto it = s.entries.begin(); it != s.entries.end();) {
            auto& e = it->second;
            Matrix pose{};
            if (now >= e.expires || (e.receiver[0] && !s.receivers.pose(e.receiver, pose))) {
                destroy(s, e); it = s.entries.erase(it); continue;
            }
            if (e.receiver[0]) {
                const auto transform = surface::world(pose, e.local);
                if (surface::changed(transform, e.drawn)) {
                    decal::set_transform(base, manager, e.decal.id, transform); e.drawn = transform;
                }
            }
            ++it;
        }
        if (now >= s.budget_at) {
            const auto count = std::count_if(s.entries.begin(), s.entries.end(), [](const auto& e) { return e.second.decal.id != 0; });
            (void)s.budget.update(base, count ? 4096 : 0); s.budget_at = now + 1000;
        }
    } catch (...) {}
}
void before_level_transition(unsigned next) noexcept {
    auto& s = state();
    if (next == 3 || next == 14 || next == 22 || next == 24 || next == 25) {
        s.suspended = true;
        try { std::lock_guard lock(s.mutex); reset(s); } catch (...) {}
    } else if (next == 13 || next == 21) s.suspended = false;
}
std::uint64_t create(std::string_view plugin, const plugins::EffectRequest& r) {
    auto& s = state(); std::lock_guard lock(s.mutex);
    if (!usable(s) || !plugins::private_effect_asset(r.asset, r.decal) || !plugins::valid_effect_matrix(r.transform) ||
        !(r.lift >= 0 && r.lift <= 1)) return 0;
    const auto total = std::count_if(s.entries.begin(), s.entries.end(), [&](const auto& e) {
        return (e.second.decal.id != 0) == r.decal;
    });
    if (total >= (r.decal ? 4096 : 64)) return 0;
    const auto mine = std::count_if(s.entries.begin(), s.entries.end(), [&](const auto& e) {
        return e.second.owner == plugin && (e.second.decal.id != 0) == r.decal;
    });
    if (mine >= (r.decal ? 2048 : 24)) return 0;
    const auto found = asset(s, r.asset);
    if (!asset_type_matches(found, r.decal)) return 0;
    Entry e; e.owner = plugin; e.expires = GetTickCount64() + 600000;
    e.drawn = r.transform;
    if (r.decal) {
        if (r.attach) {
            Receiver receiver;
            if (!s.receivers.bind(r.transform, r.lift, receiver)) return 0;
            e.receiver = receiver.id; e.local = surface::local(receiver.pose, r.transform);
        }
        e.decal = decal::create(s.base, s.manager, found, r.transform, decal::ReceiverMask::all);
        if (!e.decal.id) return 0;
        if (mine == 0) { (void)s.budget.update(s.base, 4096); s.budget_at = GetTickCount64() + 1000; }
    } else {
        alignas(16) std::array<std::byte, fx::descriptor_size> descriptor{};
        alignas(16) const auto transform = r.transform;
        function<void(*)(void*, std::uintptr_t, const float*)>(s, fx::descriptor_init.rva)(descriptor.data(), found, transform.data());
        std::memcpy(descriptor.data() + fx::descriptor_flags, &fx::descriptor_flags_value, sizeof(fx::descriptor_flags_value));
        function<void(*)(std::uintptr_t, std::uintptr_t*, void*)>(s, fx::create.rva)(0, &e.particle, descriptor.data());
        function<void(*)(void*)>(s, fx::descriptor_destroy.rva)(descriptor.data());
        if (!e.particle) return 0;
    }
    const auto id = ++s.serial;
    s.entries.emplace(id, std::move(e)); return id;
}
void move(std::string_view plugin, std::uint64_t id, const Matrix& matrix) {
    if (!plugins::valid_effect_matrix(matrix)) return;
    auto& s = state(); std::lock_guard lock(s.mutex);
    if (auto* e = owned(s, plugin, id)) {
        if (e->particle) { alignas(16) const auto m = matrix; function<void(*)(std::uintptr_t*, const float*)>(s, fx::move.rva)(&e->particle, m.data()); }
        else if (!e->receiver[0]) decal::set_transform(s.base, s.manager, e->decal.id, matrix);
    }
}
void stop(std::string_view plugin, std::uint64_t id) {
    auto& s = state(); std::lock_guard lock(s.mutex);
    if (auto* e = owned(s, plugin, id); e && e->particle && !e->stopped) {
        function<void(*)(std::uintptr_t*, bool, float)>(s, fx::stop.rva)(&e->particle, false, 0); e->stopped = true;
    }
}
void opacity(std::string_view plugin, std::uint64_t id, float value) {
    if (!(value >= 0 && value <= 1)) return;
    auto& s = state(); std::lock_guard lock(s.mutex);
    if (auto* e = owned(s, plugin, id); e && e->decal.id) decal::set_opacity(s.base, s.manager, e->decal.id, value);
}
void release(std::string_view plugin, std::uint64_t id) {
    auto& s = state(); std::lock_guard lock(s.mutex);
    const auto it = s.entries.find(id);
    if (it == s.entries.end() || it->second.owner != plugin) return;
    if (s.valid && s.thread == GetCurrentThreadId() && same_services(s)) destroy(s, it->second);
    s.entries.erase(it);
}
std::pair<bool, std::uint64_t> status() {
    auto& s = state(); std::lock_guard lock(s.mutex); return {usable(s), s.epoch};
}
void clear(std::string_view plugin) {
    auto& s = state(); std::lock_guard lock(s.mutex);
    for (auto it = s.entries.begin(); it != s.entries.end();) {
        if (it->second.owner != plugin) { ++it; continue; }
        if (s.valid && s.thread == GetCurrentThreadId() && same_services(s)) destroy(s, it->second);
        it = s.entries.erase(it);
    }
    if (s.valid && s.thread == GetCurrentThreadId() && s.entries.empty()) (void)s.budget.update(s.base, 0);
}
}