/*
 * © Copyright 2022 HP Development Company, L.P.
 * SPDX-License-Identifier: MIT
 */

import set from 'immutable-set';
import { EntityDefinitionJSONSchema, TransformEntityDefinitionSchemaCallback } from '../types';
import { JSONSchemaTraverser } from './JSONSchemaTraverser';

/**
 * This function helps to traverse an EntityDefinitionJSONSchema object.
 * This can be useful for creating derived schemas, like those used by OpenAPI or Ajv.
 * @param entityDefinitionSchema
 * @param callback
 */
export function transformEntityDefinitionSchema<T extends object = {}>(
	entityDefinitionSchema: Partial<EntityDefinitionJSONSchema>,
	callback: TransformEntityDefinitionSchemaCallback
) {
	let obj = {} as T;
	JSONSchemaTraverser.traverse(
		entityDefinitionSchema,
		({ schema, jsonPtr, rootSchema, parentJsonPtr, parentKeyword, parentSchema, keyIndex }) => {
			const pointerPath = jsonPtr
				.replace(/\//g, '.') // replace slashes with dots
				.replace(/(\.(\d))((\.)|$)/g, '[$2]$3') // replace `.{number}.` to `[number].
				.slice(1); // remove trailing slash
			const pointerPathParts = pointerPath.split('.');

			const result = callback({
				schema,
				jsonPtr,
				pointerPath,
				pointerPathParts,
				rootSchema,
				parentJsonPtr,
				parentKeyword,
				parentSchema,
				keyIndex
			});

			if (result && typeof result.path !== 'undefined' && result.path !== null) {
				if (result.path === '') {
					obj = result.value as T;
				} else {
					obj = set(obj, result.path, result.value, { withArrays: true });
				}
			}
		},
		{ allKeys: true }
	);

	return obj;
}