/*
 * WAMR bridge for QuickBEAM NIFs.
 * Thin C wrapper around WAMR's embedding API.
 */

#ifndef _WAMR_BRIDGE_H_
#define _WAMR_BRIDGE_H_

#include <stdint.h>
#include <stdbool.h>
#include "wamr/include/wasm_export.h"

typedef struct WamrModule WamrModule;
typedef struct WamrInstance WamrInstance;

typedef struct WamrNativeModule {
    const char *module_name;
    NativeSymbol *symbols;
    uint32_t symbol_count;
} WamrNativeModule;

/* Initialize the WAMR runtime. Call once at NIF load. */
bool wamr_bridge_init(void);

/* Destroy the WAMR runtime. Call at NIF unload. */
void wamr_bridge_destroy(void);

/* Compile a WASM binary into a module. Returns NULL on error. */
WamrModule *wamr_bridge_compile(const uint8_t *bytes, uint32_t len,
                                char *err_buf, uint32_t err_buf_size);

/* Free a compiled module. */
void wamr_bridge_free_module(WamrModule *mod);

/* Validate a WASM binary without full compilation. */
bool wamr_bridge_validate(const uint8_t *bytes, uint32_t len);

/* Instantiate a compiled module. Returns NULL on error. */
WamrInstance *wamr_bridge_start(WamrModule *mod,
                                uint32_t stack_size,
                                uint32_t heap_size,
                                char *err_buf, uint32_t err_buf_size);

WamrInstance *wamr_bridge_start_with_native_modules(
    WamrModule *mod,
    uint32_t stack_size,
    uint32_t heap_size,
    const WamrNativeModule *modules,
    uint32_t module_count,
    char *err_buf, uint32_t err_buf_size);

void wamr_bridge_unregister_native_modules(const WamrNativeModule *modules,
                                           uint32_t module_count);

/* Destroy an instance. */
void wamr_bridge_stop(WamrInstance *inst);

/* Get the signature of an exported function by name. */
bool wamr_bridge_function_signature(WamrInstance *inst,
                                     const char *func_name,
                                     uint32_t *param_count,
                                     wasm_valkind_t *param_types,
                                     uint32_t *result_count,
                                     wasm_valkind_t *result_types,
                                     char *err_buf, uint32_t err_buf_size);

/* Call an exported function by name with typed values. */
bool wamr_bridge_call_typed(WamrInstance *inst,
                             const char *func_name,
                             const wasm_val_t *params,
                             uint32_t param_count,
                             wasm_val_t *results,
                             uint32_t result_count,
                             char *err_buf, uint32_t err_buf_size);

void wamr_bridge_set_instruction_limit(WamrInstance *inst, int instruction_count);

/* Get the number of exports. */
int32_t wamr_bridge_export_count(WamrModule *mod);

/* Get export info by index. Returns false if index out of range. */
bool wamr_bridge_export_info(WamrModule *mod, int32_t index,
                              const char **name, int32_t *kind);

/* Get the number of imports. */
int32_t wamr_bridge_import_count(WamrModule *mod);

/* Get import info by index. Returns false if index out of range. */
bool wamr_bridge_import_info(WamrModule *mod, int32_t index,
                              const char **module_name, const char **name,
                              int32_t *kind);

/* Memory operations on an instance. */
uint32_t wamr_bridge_memory_size(WamrInstance *inst);
int32_t wamr_bridge_memory_grow(WamrInstance *inst, uint32_t delta);
bool wamr_bridge_read_memory(WamrInstance *inst, uint32_t offset,
                              uint8_t *buf, uint32_t len);
bool wamr_bridge_write_memory(WamrInstance *inst, uint32_t offset,
                               const uint8_t *buf, uint32_t len);

bool wamr_bridge_read_global(WamrInstance *inst, const char *name,
                              wasm_val_t *value,
                              char *err_buf, uint32_t err_buf_size);

bool wamr_bridge_write_global(WamrInstance *inst, const char *name,
                               const wasm_val_t *value,
                               char *err_buf, uint32_t err_buf_size);

#endif /* _WAMR_BRIDGE_H_ */