/**
* @license Apache-2.0
*
* Copyright (c) 2020 The Stdlib Authors.
*
* 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 "stdlib/strided/napi/unary.h"
#include "stdlib/strided/base/function_object.h"
#include "stdlib/strided/napi/addon_arguments.h"
#include <node_api.h>
#include <stdint.h>
#include <assert.h>

/**
* Invokes a strided array interface which applies a unary callback to an input strided array based on provided JavaScript arguments.
*
* ## Notes
*
* -   This function expects that the callback `info` argument provides access to the following JavaScript arguments:
*
*     -   `N`: number of indexed elements
*     -   `X`: input array
*     -   `strideX`: `X` stride length
*     -   `Y`: destination array
*     -   `strideY`: `Y` stride length
*
* @param env    environment under which the function is invoked
* @param info   callback data
* @param obj    strided array function object
*/
void stdlib_strided_napi_unary( napi_env env, napi_callback_info info, const struct StridedFunctionObject *obj ) {
	napi_status status;

	// Total number of input arguments:
	int64_t nargs = 5;

	// Number of input strided array arguments:
	int64_t nin = 1;

	// Get callback arguments:
	size_t argc = 5;
	napi_value argv[ 5 ];
	status = napi_get_cb_info( env, info, &argc, argv, NULL, NULL );
	assert( status == napi_ok );

	// Check whether we were provided the correct number of arguments:
	int64_t argc64 = (int64_t)argc;
	if ( argc64 < nargs ) {
		status = napi_throw_error( env, NULL, "invalid invocation. Insufficient arguments." );
		assert( status == napi_ok );
		return;
	}
	if ( argc64 > nargs ) {
		status = napi_throw_error( env, NULL, "invalid invocation. Too many arguments." );
		assert( status == napi_ok );
		return;
	}
	// Process the provided arguments:
	uint8_t *arrays[ 2 ];
	int64_t strides[ 2 ];
	int64_t shape[ 1 ];
	int32_t types[ 2 ];

	napi_value err;
	status = stdlib_strided_napi_addon_arguments( env, argv, nargs, nin, arrays, shape, strides, types, &err );
	assert( status == napi_ok );

	// Check whether processing was successful:
	if ( err != NULL ) {
		status = napi_throw( env, err );
		assert( status == napi_ok );
		return;
	}
	// Resolve the strided array function satisfying the input array types:
	int64_t idx = stdlib_strided_function_dispatch_index_of( obj, types );

	// Check whether we were able to successfully resolve a strided array function:
	if ( idx < 0 ) {
		status = napi_throw_type_error( env, NULL, "invalid arguments. Unable to resolve a strided array function supporting the provided array argument data types." );
		assert( status == napi_ok );
		return;
	}
	// Retrieve the strided array function:
	StridedArrayFcn fcn = obj->functions[ idx ];

	// Retrieve the associated function data:
	void *clbk = obj->data[ idx ];

	// Evaluate the strided array function:
	fcn( arrays, shape, strides, clbk );

	return;
}