/*******************************************************************************
 * Copyright 2013-2023 Aerospike, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/

#include "client.h"
#include "command.h"
#include "async.h"
#include "conversions.h"
#include "policy.h"
#include "log.h"
#include "string.h"

extern "C" {
#include <aerospike/aerospike.h>
#include <aerospike/aerospike_info.h>
#include <aerospike/as_node.h>
}

#define INFO_REQUEST_LEN 50

using namespace v8;

typedef struct node_info_s {
	char *info;
	char node[AS_NODE_NAME_SIZE];
} node_info;

class InfoForeachCommand : public AerospikeCommand {
  public:
	InfoForeachCommand(AerospikeClient *client, Local<Function> callback_)
		: AerospikeCommand("InfoForeach", client, callback_)
	{
		results = as_vector_create(sizeof(node_info), 4);
	}

	~InfoForeachCommand()
	{
		if (policy != NULL)
			cf_free(policy);
		if (request != NULL && strlen(request) > 0)
			cf_free(request);
		as_vector_destroy(results);
	}

	as_policy_info *policy = NULL;
	char *request = NULL;
	as_vector *results;
};

bool aerospike_info_callback(const as_error *error, const as_node *node,
							 const char *info_req, char *response, void *udata)
{
	InfoForeachCommand *cmd = reinterpret_cast<InfoForeachCommand *>(udata);
	LogInfo *log = cmd->log;
	node_info result;

	if (strlen(node->name) > 0) {
		as_v8_debug(log, "Response from node %s", node->name);
		if (as_strlcpy(result.node, node->name, AS_NODE_NAME_SIZE) >
			AS_NODE_NAME_SIZE) {
			as_v8_info(log, "Node name exceeds max. length (%d)",
					   AS_NODE_NAME_SIZE);
		}
	}
	else {
		result.node[0] = '\0';
		as_v8_debug(log, "No host name from cluster");
	}

	if (response != NULL) {
		as_v8_debug(log, "Response is %s", response);
		result.info = strdup(response);
	}
	else {
		result.info = NULL;
		as_v8_debug(log, "No response from cluster");
	}

	as_vector_append(cmd->results, (void *)&result);

	return true;
}

static void *prepare(const Nan::FunctionCallbackInfo<Value> &info)
{
	Nan::HandleScope scope;
	AerospikeClient *client =
		Nan::ObjectWrap::Unwrap<AerospikeClient>(info.This());
	InfoForeachCommand *cmd =
		new InfoForeachCommand(client, info[2].As<Function>());
	LogInfo *log = client->log;

	if (info[0]->IsString()) {
		cmd->request = (char *)malloc(INFO_REQUEST_LEN);
		if (as_strlcpy(cmd->request, *Nan::Utf8String(info[0].As<String>()),
					   INFO_REQUEST_LEN) > INFO_REQUEST_LEN) {
			return CmdSetError(cmd, AEROSPIKE_ERR_PARAM,
							   "Info request exceeds max. length (%d)",
							   INFO_REQUEST_LEN);
		}
	}
	else {
		cmd->request = (char *)"";
	}

	if (info[1]->IsObject()) {
		cmd->policy = (as_policy_info *)cf_malloc(sizeof(as_policy_info));
		if (infopolicy_from_jsobject(cmd->policy, info[1].As<Object>(), log) !=
			AS_NODE_PARAM_OK) {
			return CmdSetError(cmd, AEROSPIKE_ERR_PARAM,
							   "Policy parameter is invalid");
		}
	}

	return cmd;
}

static void execute(uv_work_t *req)
{
	InfoForeachCommand *cmd = reinterpret_cast<InfoForeachCommand *>(req->data);
	LogInfo *log = cmd->log;

	if (!cmd->CanExecute()) {
		return;
	}

	as_v8_debug(log, "Sending info command \"%s\" to all cluster hosts",
				cmd->request);
	aerospike_info_foreach(cmd->as, &cmd->err, cmd->policy, cmd->request,
						   aerospike_info_callback, (void *)cmd);
}

static void respond(uv_work_t *req, int status)
{
	Nan::HandleScope scope;
	InfoForeachCommand *cmd = reinterpret_cast<InfoForeachCommand *>(req->data);
	LogInfo *log = cmd->log;

	if (cmd->IsError()) {
		cmd->ErrorCallback();
	}
	else {
		as_vector *results = cmd->results;
		Local<Array> v8Results = Nan::New<Array>(results->size);
		as_v8_debug(log, "num of responses %d", results->size);
		for (uint32_t i = 0; i < results->size; i++) {
			node_info *result = (node_info *)as_vector_get(results, i);
			const char *info = result->info;
			const char *node = result->node;

			Local<Object> v8Result = Nan::New<Object>();
			Local<Object> v8Node = Nan::New<Object>();

			if (node != NULL && strlen(node) > 0) {
				as_v8_debug(log, "Node name: %s", node);
				Nan::Set(v8Node, Nan::New("node_id").ToLocalChecked(),
						 Nan::New(node).ToLocalChecked());
			}

			Nan::Set(v8Result, Nan::New("host").ToLocalChecked(), v8Node);

			if (info != NULL && strlen(info) > 0) {
				as_v8_debug(log, "Info response: %s", info);
				Nan::Set(v8Result, Nan::New("info").ToLocalChecked(),
						 Nan::New(info).ToLocalChecked());
				cf_free((void *)info);
			}

			Nan::Set(v8Results, i, v8Result);
		}

		Local<Value> argv[] = {Nan::Null(), v8Results};
		cmd->Callback(2, argv);
	}

	delete cmd;
	delete req;
}

NAN_METHOD(AerospikeClient::InfoForeach)
{
	TYPE_CHECK_OPT(info[0], IsString, "Request must be a string");
	TYPE_CHECK_OPT(info[1], IsObject, "Policy must be an object");
	TYPE_CHECK_REQ(info[2], IsFunction, "Callback must be a function");

	async_invoke(info, prepare, execute, respond);
}