#include "pgsodium.h"

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

PG_FUNCTION_INFO_V1(pgsodium_crypto_generichash);
Datum pgsodium_crypto_generichash(PG_FUNCTION_ARGS) {
    bytea *data;
    bytea *result;
    bytea *keyarg;
    unsigned char *key = NULL;
    size_t keylen = 0;
    size_t result_size;
    data = PG_GETARG_BYTEA_P(0);
    if (!PG_ARGISNULL(1)) {
        keyarg = PG_GETARG_BYTEA_P(1);
        key = PGSODIUM_UCHARDATA(keyarg);
        keylen = VARSIZE_ANY_EXHDR(keyarg);
        ERRORIF(keylen < crypto_generichash_KEYBYTES_MIN ||
                    keylen > crypto_generichash_KEYBYTES_MAX,
                "%s: invalid key");
    }
    result_size = VARHDRSZ + crypto_generichash_BYTES;
    result = _pgsodium_zalloc_bytea(result_size);
    crypto_generichash(PGSODIUM_UCHARDATA(result),
                       crypto_generichash_BYTES,
                       PGSODIUM_UCHARDATA(data),
                       VARSIZE_ANY_EXHDR(data),
                       key,
                       keylen);
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_crypto_shorthash_keygen);
Datum pgsodium_crypto_shorthash_keygen(PG_FUNCTION_ARGS) {
    size_t result_size = VARHDRSZ + crypto_shorthash_KEYBYTES;
    bytea *result = _pgsodium_zalloc_bytea(result_size);
    crypto_secretbox_keygen(PGSODIUM_UCHARDATA(result));
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_crypto_shorthash);
Datum pgsodium_crypto_shorthash(PG_FUNCTION_ARGS) {
    bytea *data;
    bytea *result;
    bytea *key;
    int result_size = VARHDRSZ + crypto_shorthash_BYTES;
    data = PG_GETARG_BYTEA_P(0);
    key = PG_GETARG_BYTEA_P(1);
    ERRORIF(VARSIZE_ANY_EXHDR(key) != crypto_shorthash_KEYBYTES, "%s: invalid key");
    result = _pgsodium_zalloc_bytea(result_size);
    crypto_shorthash(PGSODIUM_UCHARDATA(result),
                     PGSODIUM_UCHARDATA(data),
                     VARSIZE_ANY_EXHDR(data),
                     PGSODIUM_UCHARDATA(key));
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_crypto_generichash_by_id);
Datum pgsodium_crypto_generichash_by_id(PG_FUNCTION_ARGS) {
    bytea *data;
    bytea *result;
    bytea *keyarg;
    bytea *context;
    unsigned char *key = NULL;
    size_t keylen = 0;
    size_t result_size;
    data = PG_GETARG_BYTEA_P(0);
    if (!PG_ARGISNULL(1)) {
        unsigned long long key_id = PG_GETARG_INT64(1);
        context = PG_GETARG_BYTEA_P(2);
        keyarg = pgsodium_derive_helper(
            key_id, crypto_generichash_KEYBYTES, context);
        key = PGSODIUM_UCHARDATA(keyarg);
        keylen = VARSIZE_ANY_EXHDR(keyarg);
        ERRORIF(keylen < crypto_generichash_KEYBYTES_MIN ||
                    keylen > crypto_generichash_KEYBYTES_MAX,
                "%s: invalid key");
    }
    result_size = VARHDRSZ + crypto_generichash_BYTES;
    result = _pgsodium_zalloc_bytea(result_size);
    crypto_generichash(PGSODIUM_UCHARDATA(result),
                       crypto_generichash_BYTES,
                       PGSODIUM_UCHARDATA(data),
                       VARSIZE_ANY_EXHDR(data),
                       key,
                       keylen);
    PG_RETURN_BYTEA_P(result);
}

PG_FUNCTION_INFO_V1(pgsodium_crypto_shorthash_by_id);
Datum pgsodium_crypto_shorthash_by_id(PG_FUNCTION_ARGS) {
    bytea *data;
    bytea *result;
    bytea *key;
    bytea *context;
    uint64_t key_id;
    int result_size = VARHDRSZ + crypto_shorthash_BYTES;
    data = PG_GETARG_BYTEA_P(0);
    key_id = PG_GETARG_INT64(1);
    context = PG_GETARG_BYTEA_P(2);
    key = pgsodium_derive_helper(key_id, crypto_shorthash_KEYBYTES, context);
    result = _pgsodium_zalloc_bytea(result_size);
    crypto_shorthash(PGSODIUM_UCHARDATA(result),
                     PGSODIUM_UCHARDATA(data),
                     VARSIZE_ANY_EXHDR(data),
                     PGSODIUM_UCHARDATA(key));
    PG_RETURN_BYTEA_P(result);
}