module DuckDB::MemoryHelper

static VALUE rbduckdb_memory_helper_write_bigint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    int64_t *data;
    idx_t idx;
    int64_t val;

    (void)self;

    /* Convert Ruby values to C types */
    data = (int64_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    val = (int64_t)NUM2LL(value);

    /* Write the value */
    data[idx] = val;

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_boolean(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    bool *data;
    idx_t idx;

    (void)self;

    data = (bool *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = RTEST(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_date(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    duckdb_date *data;
    idx_t idx;

    data = (duckdb_date *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);

    data[idx] = rbduckdb_to_duckdb_date_from_value(
        rb_funcall(value, rb_intern("year"), 0),
        rb_funcall(value, rb_intern("month"), 0),
        rb_funcall(value, rb_intern("day"), 0)
    );

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_double(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    double *data;
    idx_t idx;
    double val;

    (void)self;

    data = (double *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    val = NUM2DBL(value);

    data[idx] = val;

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_float(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    float *data;
    idx_t idx;

    (void)self;

    data = (float *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (float)NUM2DBL(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_integer(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    int32_t *data;
    idx_t idx;
    int32_t val;

    (void)self;

    data = (int32_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    val = (int32_t)NUM2INT(value);

    data[idx] = val;

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_smallint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    int16_t *data;
    idx_t idx;

    (void)self;

    data = (int16_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (int16_t)NUM2INT(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_timestamp(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    duckdb_timestamp *data;
    idx_t idx;

    (void)self;

    if (!rb_obj_is_kind_of(value, rb_cTime)) {
        rb_raise(rb_eTypeError, "Expected Time object for TIMESTAMP");
    }

    data = (duckdb_timestamp *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);

    VALUE local_time = rb_funcall(value, rb_intern("getlocal"), 0);
    data[idx] = rbduckdb_to_duckdb_timestamp_from_value(
        rb_funcall(local_time, rb_intern("year"), 0),
        rb_funcall(local_time, rb_intern("month"), 0),
        rb_funcall(local_time, rb_intern("day"), 0),
        rb_funcall(local_time, rb_intern("hour"), 0),
        rb_funcall(local_time, rb_intern("min"), 0),
        rb_funcall(local_time, rb_intern("sec"), 0),
        rb_funcall(local_time, rb_intern("usec"), 0)
    );

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_timestamp_tz(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    duckdb_time_tz *data;
    idx_t idx;
    int64_t secs;
    int32_t usecs;

    (void)self;

    if (!rb_obj_is_kind_of(value, rb_cTime)) {
        rb_raise(rb_eTypeError, "Expected Time object for TIMESTAMP_TZ");
    }

    data = (duckdb_time_tz *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);

    secs = NUM2LL(rb_funcall(value, rb_intern("to_i"), 0));
    usecs = NUM2INT(rb_funcall(value, rb_intern("usec"), 0));
    data[idx].bits = (uint64_t)(secs * 1000000LL + usecs);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_tinyint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    int8_t *data;
    idx_t idx;

    (void)self;

    data = (int8_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (int8_t)NUM2INT(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_ubigint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    uint64_t *data;
    idx_t idx;

    (void)self;

    data = (uint64_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = NUM2ULL(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_uinteger(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    uint32_t *data;
    idx_t idx;

    (void)self;

    data = (uint32_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (uint32_t)NUM2ULL(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_usmallint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    uint16_t *data;
    idx_t idx;

    (void)self;

    data = (uint16_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (uint16_t)NUM2UINT(value);

    return Qnil;
}

static VALUE rbduckdb_memory_helper_write_utinyint(VALUE self, VALUE ptr, VALUE index, VALUE value) {
    uint8_t *data;
    idx_t idx;

    (void)self;

    data = (uint8_t *)NUM2ULL(ptr);
    idx = (idx_t)NUM2ULL(index);
    data[idx] = (uint8_t)NUM2UINT(value);

    return Qnil;
}