#include "helpers.h"
#include "random.h"
#include "sodium.h"

using namespace facebook;

namespace react_native_nacl {
  void install_random(jsi::Runtime& jsiRuntime) {
    auto getRandomBytes = jsi::Function::createFromHostFunction(
      jsiRuntime,
      jsi::PropNameID::forAscii(jsiRuntime, "getRandomBytes"),
      1,
      [](jsi::Runtime& jsiRuntime, const jsi::Value& thisValue, const jsi::Value* arguments, size_t count) -> jsi::Value {
        int64_t size;
        if (arguments[0].isNumber()) {
          size = arguments[0].asNumber();
        } else if (arguments[0].isBigInt()) {
          size = arguments[0].asBigInt(jsiRuntime).asInt64(jsiRuntime);
        } else {
          throw jsi::JSError(jsiRuntime, "[react-native-nacl-jsi] getRandomBytes size must be a number or a BigInt");
        }

        jsi::ArrayBuffer arrayBuffer = getArrayBuffer(jsiRuntime, (double)size);
        uint8_t* data = arrayBuffer.data(jsiRuntime);
        randombytes_buf(data, size);

        return arrayBuffer;
      }
    );
    jsiRuntime.global().setProperty(jsiRuntime, "getRandomBytes", std::move(getRandomBytes));
  }
}