#pragma once
#include <jsi/jsi.h>
#include <ReactCommon/CallInvoker.h>
#include <functional>
#include <memory>
#include <string>

using namespace facebook;

namespace rnllama_jsi {

    using PromiseResultGenerator = std::function<jsi::Value(jsi::Runtime&)>;
    using PromiseTask = std::function<PromiseResultGenerator()>;
    using JsiFunctionPtr = std::shared_ptr<jsi::Function>;

    jsi::Value createPromiseTask(
        jsi::Runtime& runtime,
        std::shared_ptr<react::CallInvoker> callInvoker,
        PromiseTask task,
        int contextId = -1,
        bool trackTask = true
    );

    JsiFunctionPtr makeJsiFunction(
        jsi::Runtime& runtime,
        const jsi::Value& value,
        std::shared_ptr<react::CallInvoker> callInvoker
    );

    jsi::Value createJsiError(
        jsi::Runtime& runtime,
        const std::string& message
    );

    // Schedule an async callback on the JS thread with TaskManager tracking.
    // This ensures releaseContext will wait for all pending callbacks before deletion.
    // The callback receives a bool indicating if it should proceed (false if shutting down).
    void invokeAsyncTracked(
        std::shared_ptr<react::CallInvoker> callInvoker,
        int contextId,
        std::function<void(bool shouldProceed)> callback
    );

    // Safe console.log wrapper for JSI context
    inline void consoleLog(jsi::Runtime& runtime, const std::string& message) {
        auto console = runtime.global().getPropertyAsObject(runtime, "console");
        auto log = console.getPropertyAsFunction(runtime, "log");
        log.call(runtime, jsi::String::createFromUtf8(runtime, message));
    }

    inline void consoleWarn(jsi::Runtime& runtime, const std::string& message) {
        auto console = runtime.global().getPropertyAsObject(runtime, "console");
        auto warn = console.getPropertyAsFunction(runtime, "warn");
        warn.call(runtime, jsi::String::createFromUtf8(runtime, message));
    }

    inline void consoleError(jsi::Runtime& runtime, const std::string& message) {
        auto console = runtime.global().getPropertyAsObject(runtime, "console");
        auto error = console.getPropertyAsFunction(runtime, "error");
        error.call(runtime, jsi::String::createFromUtf8(runtime, message));
    }
}