#include "rsa_jwk.h"

Handle<JwkRsa> JwkRsa::From(Handle<ScopedEVP_PKEY> pkey, int &key_type) {
	LOG_FUNC();

	LOG_INFO("Check key_type");
	if (!(key_type == NODESSL_KT_PRIVATE || key_type == NODESSL_KT_PUBLIC)) {
		THROW_ERROR("Wrong value of key_type");
	}

	LOG_INFO("Check pkey");
	if (pkey == nullptr) {
		THROW_ERROR("Key value is nullptr");
	}
	if (EVP_PKEY_base_id(pkey->Get())!= EVP_PKEY_RSA) {
		THROW_ERROR("Key is not RSA type");
	}

	LOG_INFO("Create JWK Object");
	Handle<JwkRsa> jwk(new JwkRsa());

	RSA *rsa = EVP_PKEY_get1_RSA(pkey->Get());

	LOG_INFO("Convert RSA to JWK");
	jwk->type = key_type;
	LOG_INFO("Get RSA public key");

#if NODE_MODULE_VERSION < 60 || OPENSSL_1_0_2

    jwk->n = BN_dup(rsa->n);
    jwk->e = BN_dup(rsa->e);
    
    if (key_type == NODESSL_KT_PRIVATE) {
        LOG_INFO("Get RSA private key");
        jwk->d = BN_dup(rsa->d);
        jwk->p = BN_dup(rsa->p);
        jwk->q = BN_dup(rsa->q);
        jwk->dp = BN_dup(rsa->dmp1);
        jwk->dq = BN_dup(rsa->dmq1);
        jwk->qi = BN_dup(rsa->iqmp);
    }
    
#else
    
    // NODE v10
    
    const BIGNUM *n, *e, *d, *p, *q, *dmp1, *dmq1, *iqmp;
    RSA_get0_key(rsa, &n, &e, &d);
    RSA_get0_factors(rsa, &p, &q);
    RSA_get0_crt_params(rsa, &dmp1, &dmq1, &iqmp);
    jwk->n = BN_dup(n);
    jwk->e = BN_dup(e);
    if (key_type == NODESSL_KT_PRIVATE) {
        LOG_INFO("Get RSA private key");
        jwk->d = BN_dup(d);
        jwk->p = BN_dup(p);
        jwk->q = BN_dup(q);
        jwk->dp = BN_dup(dmp1);
        jwk->dq = BN_dup(dmq1);
        jwk->qi = BN_dup(iqmp);
    }
    
#endif

	return jwk;
}

Handle<ScopedEVP_PKEY> JwkRsa::To(int &key_type) {
	LOG_FUNC();

	LOG_INFO("Check key_type");
	if (!(key_type == NODESSL_KT_PRIVATE || key_type == NODESSL_KT_PUBLIC)) {
		THROW_ERROR("Wrong value of key_type");
	}

	if (strcmp(this->kty, "RSA") != 0) {
		THROW_ERROR("JWK key is not RSA");
	}

	RSA* rsa_key = RSA_new();

	LOG_INFO("set public key");

#if NODE_MODULE_VERSION < 60 || OPENSSL_1_0_2
    
    rsa_key->n = BN_dup(this->n.Get());
    rsa_key->e = BN_dup(this->e.Get());
    
    if (key_type == NODESSL_KT_PRIVATE) {
        LOG_INFO("set private key");
        rsa_key->d = BN_dup(this->d.Get());
        rsa_key->p = BN_dup(this->p.Get());
        rsa_key->q = BN_dup(this->q.Get());
        rsa_key->dmp1 = BN_dup(this->dp.Get());
        rsa_key->dmq1 = BN_dup(this->dq.Get());
        rsa_key->iqmp = BN_dup(this->qi.Get());
    }

#else
    
    // NODE v10
    
    RSA_set0_key(rsa_key, BN_dup(this->n.Get()), BN_dup(this->e.Get()), NULL);

	if (key_type == NODESSL_KT_PRIVATE) {
		LOG_INFO("set private key");
		RSA_set0_key(rsa_key, NULL, NULL, BN_dup(this->d.Get()));
		RSA_set0_factors(rsa_key, BN_dup(this->p.Get()), BN_dup(this->q.Get()));
		RSA_set0_crt_params(rsa_key, BN_dup(this->dp.Get()), BN_dup(this->dq.Get()), BN_dup(this->qi.Get()));
	}

#endif

	LOG_INFO("set key");
	ScopedEVP_PKEY pkey(EVP_PKEY_new());
	EVP_PKEY_assign_RSA(pkey.Get(), rsa_key);

	return Handle<ScopedEVP_PKEY>(new ScopedEVP_PKEY(pkey));
}