#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include "../include/arboricx.h"

static uint8_t *read_file(const char *path, size_t *out_len) {
    FILE *f = fopen(path, "rb");
    if (!f) return NULL;
    fseek(f, 0, SEEK_END);
    *out_len = ftell(f);
    fseek(f, 0, SEEK_SET);
    uint8_t *buf = malloc(*out_len);
    fread(buf, 1, *out_len, f);
    fclose(f);
    return buf;
}

int main() {
    clock_t t0 = clock();
    arb_ctx_t *ctx = arboricx_init();
    clock_t t1 = clock();
    if (!ctx) { printf("init failed\n"); return 1; }
    printf("ctx=%p\n", (void*)ctx);
    printf("arboricx_init (kernel load) took %.3f ms\n", (double)(t1 - t0) * 1000.0 / CLOCKS_PER_SEC);

    size_t bundle_len;
    uint8_t *bundle = read_file("../../test/fixtures/append.arboricx", &bundle_len);
    if (!bundle) { printf("bundle not found\n"); return 1; }
    printf("bundle size=%zu\n", bundle_len);

    uint32_t bundle_tree = arb_of_bytes(ctx, bundle, bundle_len);
    printf("bundle_tree=%u\n", bundle_tree);

    uint32_t tag = arb_of_number(ctx, 1);
    printf("tag=%u\n", tag);

    uint32_t arg1 = arb_of_string(ctx, "Hello, ");
    uint32_t arg2 = arb_of_string(ctx, "world!");
    printf("arg1=%u arg2=%u\n", arg1, arg2);

    uint32_t list_tail = arb_fork(ctx, arg2, arb_leaf(ctx));
    uint32_t args_list = arb_fork(ctx, arg1, list_tail);
    printf("args_list=%u\n", args_list);

    uint32_t app0 = arb_app(ctx, arb_kernel_root(ctx), tag);
    uint32_t app1 = arb_app(ctx, app0, bundle_tree);
    uint32_t app2 = arb_app(ctx, app1, args_list);
    printf("app2=%u\n", app2);

    printf("reducing...\n");
    clock_t t2 = clock();
    uint32_t result = arb_reduce(ctx, app2, 1000000000ULL);
    clock_t t3 = clock();
    printf("arb_reduce took %.3f ms, result=%u\n", (double)(t3 - t2) * 1000.0 / CLOCKS_PER_SEC, result);

    int ok;
    uint32_t value, rest;
    if (!arb_unwrap_result(ctx, result, &ok, &value, &rest)) {
        printf("unwrap_result failed\n");
        return 1;
    }
    printf("ok=%d value=%u\n", ok, value);

    uint64_t htag;
    uint32_t payload;
    if (!arb_unwrap_host_value(ctx, value, &htag, &payload)) {
        printf("unwrap_host_value failed\n");
        return 1;
    }
    printf("htag=%lu payload=%u\n", htag, payload);

    uint8_t *str_ptr;
    size_t str_len;
    if (!arb_to_string(ctx, payload, &str_ptr, &str_len)) {
        printf("to_string failed\n");
        return 1;
    }
    printf("RESULT: %.*s\n", (int)str_len, str_ptr);
    arboricx_free_buf(ctx, str_ptr, str_len);

    free(bundle);
    arboricx_free(ctx);
    printf("done\n");
    return 0;
}