#include "pgsodium.h"

PG_FUNCTION_INFO_V1(pgsodium_randombytes_random);
Datum pgsodium_randombytes_random(PG_FUNCTION_ARGS) {
    PG_RETURN_UINT32(randombytes_random());
}

PG_FUNCTION_INFO_V1(pgsodium_randombytes_uniform);
Datum pgsodium_randombytes_uniform(PG_FUNCTION_ARGS) {
    uint32_t upper_bound = PG_GETARG_UINT32(0);
    PG_RETURN_UINT32(randombytes_uniform(upper_bound));
}

PG_FUNCTION_INFO_V1(pgsodium_randombytes_buf);
Datum pgsodium_randombytes_buf(PG_FUNCTION_ARGS) {
    size_t size = PG_GETARG_UINT32(0);
    size_t result_size = VARHDRSZ + size;
    bytea *result = _pgsodium_zalloc_bytea(result_size);
    randombytes_buf(VARDATA(result), size);
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_randombytes_new_seed);
Datum pgsodium_randombytes_new_seed(PG_FUNCTION_ARGS) {
    size_t result_size = VARHDRSZ + randombytes_SEEDBYTES;
    bytea *result = _pgsodium_zalloc_bytea(result_size);
    randombytes_buf(VARDATA(result), randombytes_SEEDBYTES);
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_randombytes_buf_deterministic);
Datum pgsodium_randombytes_buf_deterministic(PG_FUNCTION_ARGS) {
    size_t size = PG_GETARG_UINT32(0);
    bytea *seed = PG_GETARG_BYTEA_P(1);
    size_t result_size = VARHDRSZ + size;
    bytea *result = _pgsodium_zalloc_bytea(result_size);
    randombytes_buf_deterministic(
        VARDATA(result), size, PGSODIUM_UCHARDATA(seed));
    PG_RETURN_BYTEA_P(result);
}