#include <assert.h>
#include <string>
#include <napi.h>
#include "../lib/levenstein.cpp"
#include "../lib/fibonacci.cpp"
#include "../lib/fermat.cpp"
#include "../lib/regression.cpp"
#include "../lib/sha256.cpp"

using namespace Napi;

std::vector<double> convertToVector (const Array& a) {
  auto length = a.Length();
  std::vector<double> result;
  for (size_t i = 0; i < length; i++) {
    result.emplace_back(a[i].As<Number>().DoubleValue());
  }
  return result;
}

std::vector<double> convertToVector (const Float64Array& a) {
  size_t s = a.ElementLength();
  auto data = a.Data();
  return std::vector<double>(data, data + s);
}

Value Levenstein(const CallbackInfo& info) {
  assert(info[0].IsString());
  std::string word = info[0].As<String>().Utf8Value();
  std::string reference = info[1].As<String>().Utf8Value();
  auto result = levenstein(word, reference);
  return Number::New(info.Env(), result);
}

Value Fibonacci(const CallbackInfo& info) {
  assert(info[0].IsNumber());
  unsigned int n = info[0].As<Number>().Uint32Value();
  auto result = fibonacci(n);
  return Number::New(info.Env(), result);
}

Value Fermat(const CallbackInfo& info) {
  assert(info[0].IsNumber());
  unsigned int n = info[0].As<Number>().Uint32Value();
  unsigned int k = info[1].As<Number>().Uint32Value();
  auto result = isPrime(n, k);
  return Boolean::New(info.Env(), result);
}

Value Regression(const CallbackInfo& info) {
  auto y = info[0].IsTypedArray() ? convertToVector(info[0].As<Float64Array>()) :
    convertToVector(info[0].As<Array>());
  auto x = info[1].IsTypedArray() ? convertToVector(info[1].As<Float64Array>()) :
    convertToVector(info[1].As<Array>());
  auto line = regression(y, x);
  auto Env = info.Env();
  auto result = Object::New(Env);
  result.Set("slope", Number::New(Env, line.slope));
  result.Set("intercept", Number::New(Env, line.intercept));
  result.Set("r2", Number::New(Env, line.r2));
  return result;
}

Value Sha256(const CallbackInfo& info) {
  assert(info[0].IsString());
  auto input = info[0].As<String>().Utf8Value();
  return String::New(info.Env(), SHA256(input));
}

Object Init(Env env, Object exports) {
  exports.Set("levenstein", Function::New(env, Levenstein));
  exports.Set("fibonacci", Function::New(env, Fibonacci));
  exports.Set("fermat", Function::New(env, Fermat));
  exports.Set("regression", Function::New(env, Regression));
  exports.Set("sha256", Function::New(env, Sha256));
  return exports;
}

NODE_API_MODULE(addon, Init)